@stupify/cli 0.0.3 → 0.0.4

package/dist/repomix-provider.js

@@ -5,18 +5,23 @@ import { pack, setLogLevel } from "repomix";
 const MAX_PACK_FILE_SIZE_BYTES = 48 * 1024;
 const MAX_PACK_TOTAL_SIZE_BYTES = 128 * 1024;
 export function emptyContextPack() {
+    const config = repomixSearchConfig();
     return {
         provider: "repomix",
         filePaths: [],
         totalCharacters: 0,
         totalTokens: 0,
         text: "",
+        config,
     };
 }
-export async function repomixContextPack(cwd, contexts, changes) {
-    const filePaths = await candidateFilePaths(cwd, contexts, changes);
+export async function repomixContextPack(cwd, contexts, changes, config = repomixSearchConfig()) {
+    const filePaths = await candidateFilePaths(cwd, contexts, changes, config);
     if (filePaths.length === 0) {
-        return emptyContextPack();
+        return {
+            ...emptyContextPack(),
+            config,
+        };
     }
     setLogLevel(-1);
     const tempDir = await mkdtemp(path.join(tmpdir(), "stupify-repomix-"));
@@ -24,7 +29,7 @@ export async function repomixContextPack(cwd, contexts, changes) {
     try {
         const result = await pack([cwd], {
             cwd,
-            input: { maxFileSize: MAX_PACK_FILE_SIZE_BYTES },
+            input: { maxFileSize: config.maxFileSizeBytes },
             output: {
                 filePath: outputPath,
                 style: "xml",
@@ -33,10 +38,10 @@ export async function repomixContextPack(cwd, contexts, changes) {
                 directoryStructure: false,
                 files: true,
                 removeComments: false,
-                removeEmptyLines: true,
-                compress: true,
+                removeEmptyLines: config.removeEmptyLines,
+                compress: config.compress,
                 topFilesLength: 0,
-                showLineNumbers: true,
+                showLineNumbers: config.showLineNumbers,
                 truncateBase64: true,
                 copyToClipboard: false,
                 includeFullDirectoryStructure: false,
@@ -54,7 +59,7 @@ export async function repomixContextPack(cwd, contexts, changes) {
                 useGitignore: true,
                 useDotIgnore: true,
                 useDefaultPatterns: true,
-                customPatterns: [],
+                customPatterns: [...config.ignorePatterns],
             },
             security: { enableSecurityCheck: false },
             tokenCount: { encoding: "o200k_base" },
@@ -65,6 +70,7 @@ export async function repomixContextPack(cwd, contexts, changes) {
             totalCharacters: result.totalCharacters,
             totalTokens: result.totalTokens,
             text: await readFile(outputPath, "utf8"),
+            config,
         };
     }
     finally {
@@ -98,23 +104,68 @@ export function entityContextsFromChanges(candidates, changes) {
             }];
     });
 }
-async function candidateFilePaths(cwd, contexts, changes) {
+async function candidateFilePaths(cwd, contexts, changes, config) {
     const byEntityId = new Map(changes.map((change) => [change.entityId, change.filePath]));
     const paths = contexts.flatMap((context) => context.filePath ?? byEntityId.get(context.entityId) ?? []);
     const safePaths = [...new Set(paths)].filter(isSafeRelativeFilePath);
     const selected = [];
     let totalBytes = 0;
     for (const filePath of safePaths) {
+        if (matchesAnyPattern(filePath, config.ignorePatterns))
+            continue;
         const bytes = await fileSize(cwd, filePath);
-        if (bytes === null || bytes > MAX_PACK_FILE_SIZE_BYTES)
+        if (bytes === null || bytes > config.maxFileSizeBytes)
             continue;
-        if (totalBytes + bytes > MAX_PACK_TOTAL_SIZE_BYTES)
+        if (totalBytes + bytes > config.maxTotalSizeBytes)
             continue;
         totalBytes += bytes;
         selected.push(filePath);
     }
     return selected;
 }
+export function repomixSearchConfig() {
+    return {
+        compress: envBoolean("STUPIFY_REPOMIX_COMPRESS", true),
+        showLineNumbers: envBoolean("STUPIFY_REPOMIX_SHOW_LINE_NUMBERS", true),
+        removeEmptyLines: envBoolean("STUPIFY_REPOMIX_REMOVE_EMPTY_LINES", true),
+        maxFileSizeBytes: envInteger("STUPIFY_REPOMIX_MAX_FILE_BYTES", MAX_PACK_FILE_SIZE_BYTES),
+        maxTotalSizeBytes: envInteger("STUPIFY_REPOMIX_MAX_TOTAL_BYTES", MAX_PACK_TOTAL_SIZE_BYTES),
+        ignorePatterns: envList("STUPIFY_REPOMIX_IGNORE_PATTERNS"),
+    };
+}
+function envBoolean(name, fallback) {
+    const value = process.env[name];
+    if (value === undefined || value === "")
+        return fallback;
+    return /^(1|true|yes|on)$/i.test(value);
+}
+function envInteger(name, fallback) {
+    const value = Number(process.env[name]);
+    return Number.isInteger(value) && value > 0 ? value : fallback;
+}
+function envList(name) {
+    return (process.env[name] ?? "")
+        .split(",")
+        .map((item) => item.trim())
+        .filter(Boolean);
+}
+function matchesAnyPattern(filePath, patterns) {
+    return patterns.some((pattern) => matchesPattern(filePath, pattern));
+}
+function matchesPattern(filePath, pattern) {
+    if (pattern === filePath)
+        return true;
+    if (!pattern.includes("*"))
+        return false;
+    const escaped = pattern
+        .split("*")
+        .map(escapeRegExp)
+        .join(".*");
+    return new RegExp(`^${escaped}$`).test(filePath);
+}
+function escapeRegExp(value) {
+    return value.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+}
 function isSafeRelativeFilePath(value) {
     if (!value || path.isAbsolute(value))
         return false;

package/dist/search-bench.d.ts

@@ -0,0 +1 @@
+export declare function runSearchBench(configPath: string): Promise<string>;