prompt-helper/dist/collectors/fileStructureCollector.js

A CLI tool to help you create and manage prompts for AI models.

"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    var desc = Object.getOwnPropertyDescriptor(m, k);
    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
      desc = { enumerable: true, get: function() { return m[k]; } };
    }
    Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
    Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
    o["default"] = v;
});
var __importStar = (this && this.__importStar) || (function () {
    var ownKeys = function(o) {
        ownKeys = Object.getOwnPropertyNames || function (o) {
            var ar = [];
            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
            return ar;
        };
        return ownKeys(o);
    };
    return function (mod) {
        if (mod && mod.__esModule) return mod;
        var result = {};
        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
        __setModuleDefault(result, mod);
        return result;
    };
})();
Object.defineProperty(exports, "__esModule", { value: true });
exports.collectFileTree = collectFileTree;
// src/collectors/fileStructureCollector.ts
const fs = __importStar(require("fs"));
const path = __importStar(require("path"));
const dependenciesCollector_1 = require("./dependenciesCollector");
const headers_1 = require("../modules/headers");
const standardizeError_1 = require("../libs/standardizeError");
const CHECK_EXTENSIONS = new Set(['.ts', '.tsx', '.js', '.jsx', '.css']);
const ALLOWED_DIRECTORIES = new Set(['public', 'src']);
/**
 * Recursively collects file metadata and structure from the provided directory.
 *
 * @param dir - The current directory being scanned.
 * @param baseDir - The root directory used to calculate relative paths.
 * @returns An array of FileNode objects representing the structure and metadata.
 */
function collectFileTree(dir, baseDir) {
    let items;
    try {
        items = fs.readdirSync(dir);
    }
    catch (err) {
        const error = (0, standardizeError_1.standardizeError)(err);
        return [
            {
                name: path.basename(dir),
                path: path.relative(baseDir, dir).replace(/\\/g, '/'),
                isDirectory: false,
                error: error.message,
            },
        ];
    }
    return items.reduce((nodes, item) => {
        // Skip promptHelper folders at root
        if (dir === baseDir && item.startsWith('promptHelper')) {
            return nodes;
        }
        const absolutePath = path.join(dir, item);
        const relativePath = path.relative(baseDir, absolutePath).replace(/\\/g, '/');
        let stat;
        try {
            stat = fs.statSync(absolutePath);
        }
        catch (err) {
            const error = (0, standardizeError_1.standardizeError)(err);
            nodes.push({
                name: item,
                path: relativePath,
                isDirectory: false,
                error: error.message,
            });
            return nodes;
        }
        if (stat.isDirectory()) {
            if (dir === baseDir && !ALLOWED_DIRECTORIES.has(item)) {
                return nodes;
            }
            nodes.push({
                name: item + '/',
                path: relativePath + '/',
                isDirectory: true,
                children: collectFileTree(absolutePath, baseDir),
            });
        }
        else {
            const ext = path.extname(item);
            if (!CHECK_EXTENSIONS.has(ext)) {
                return nodes;
            }
            const dependencies = (0, dependenciesCollector_1.getLocalDependencies)(absolutePath, baseDir);
            const hasHeader = (0, headers_1.hasCorrectHeader)(absolutePath, relativePath);
            if (!hasHeader) {
                (0, headers_1.addHeader)(absolutePath, relativePath);
            }
            const fileNode = {
                name: item,
                path: relativePath,
                isDirectory: false,
            };
            if (dependencies.length > 0) {
                fileNode.dependencies = dependencies;
            }
            if (!hasHeader) {
                fileNode.missingHeader = true;
            }
            nodes.push(fileNode);
        }
        return nodes;
    }, []);
}
//# sourceMappingURL=fileStructureCollector.js.map