chore: fix monorepo edge case (#26)

* chore: fix monorepo edge case

* Update index.test.ts

Author: khromov

src/index.test.ts

@@ -11,6 +11,15 @@ const runCLI = async (args: string = "") => {
   return execAsync(`ts-node ${cliPath} ${args}`);
 };
 
+// New helper to run CLI with specific environment variables
+const runCLIWithEnv = async (args: string = "", env: Record<string, string> = {}) => {
+  const cliPath = path.resolve(__dirname, "index.ts");
+  const envVars = Object.entries(env)
+    .map(([key, value]) => `${key}="${value}"`)
+    .join(" ");
+  return execAsync(`${envVars} ts-node ${cliPath} ${args}`);
+};
+
 describe("AI Digest CLI", () => {
   afterAll(async () => {
     // Remove the created .md files after all tests complete
@@ -319,4 +328,44 @@ describe("AI Digest CLI", () => {
       await fs.rm(tempDir2, { recursive: true, force: true });
     }
   }, 15000);
-});
+
+  // New test for working directory behavior
+  it("should respect INIT_CWD when different from process.cwd()", async () => {
+    // Create a temporary directory structure
+    const tempRootDir = await fs.mkdtemp(path.join(os.tmpdir(), "ai-digest-wd-test-"));
+    const subDir = path.join(tempRootDir, "subdir");
+    await fs.mkdir(subDir);
+    
+    // Create test files
+    await fs.writeFile(path.join(tempRootDir, "root-file.txt"), "Root file content");
+    
+    try {
+      // Run with INIT_CWD set to subdirectory but cwd unchanged
+      const env = { INIT_CWD: subDir };
+      
+      // Use the tempRootDir as input to have files to process
+      const { stdout } = await runCLIWithEnv(`--input ${tempRootDir}`, env);
+      
+      // Verify the file was created in the subdirectory (INIT_CWD)
+      const subDirOutputPath = path.join(subDir, "codebase.md");
+      const fileExists = await fs
+        .access(subDirOutputPath)
+        .then(() => true)
+        .catch(() => false);
+      
+      expect(fileExists).toBe(true);
+      
+      // Verify content includes the root file
+      const content = await fs.readFile(subDirOutputPath, "utf-8");
+      expect(content).toContain("root-file.txt");
+      expect(content).toContain("Root file content");
+      
+      // Clean up the output file
+      await fs.unlink(subDirOutputPath).catch(() => {});
+      
+    } finally {
+      // Clean up the test directories
+      await fs.rm(tempRootDir, { recursive: true, force: true });
+    }
+  }, 15000);
+});

src/index.ts

@@ -25,6 +25,16 @@ type IgnoreInstance = ReturnType<typeof ignore>;
 
 const MAX_FILE_SIZE = 10 * 1024 * 1024; // 10MB in bytes
 
+function getActualWorkingDirectory() {
+  // INIT_CWD is set by npm/npx to the directory from which the command was invoked
+  // This is more reliable than process.cwd() when running via npx
+  if (process.env.INIT_CWD) {
+    return process.env.INIT_CWD;
+  }
+  
+  return process.cwd();
+}
+
 // Simple debounce function to avoid multiple rebuilds when many files change at once
 function debounce<F extends (...args: any[]) => any>(
   func: F,
@@ -193,7 +203,7 @@ async function watchFiles(
       // Ignore the output file
       const outputAbsPath = path.isAbsolute(outputFile)
         ? outputFile
-        : path.join(process.cwd(), outputFile);
+        : path.join(getActualWorkingDirectory(), outputFile);
       if (filePath === outputAbsPath) {
         return true;
       }
@@ -453,7 +463,7 @@ async function aggregateFiles(
 
       const outputAbsPath = path.isAbsolute(outputFile)
         ? outputFile
-        : path.join(process.cwd(), outputFile);
+        : path.join(getActualWorkingDirectory(), outputFile);
 
       if (
         fullPath === outputAbsPath ||
@@ -503,7 +513,7 @@ async function aggregateFiles(
       }
     }
 
-    await fs.mkdir(path.dirname(outputFile), { recursive: true });
+    await fs.mkdir(path.dirname(path.resolve(getActualWorkingDirectory(), outputFile)), { recursive: true });
 
     // Write to a temporary file first to prevent partial writes during SIGINT
     const tempFile = `${outputFile}.temp`;
@@ -600,7 +610,7 @@ program
   .option(
     "-i, --input <directories...>",
     "Input directories (multiple allowed)",
-    [process.cwd()]
+    [getActualWorkingDirectory()] // Fix: Use getActualWorkingDirectory() instead of process.cwd()
   )
   .option("-o, --output <file>", "Output file name", "codebase.md")
   .option("--no-default-ignores", "Disable default ignore patterns")
@@ -613,9 +623,10 @@ program
   .option("--watch", "Watch for file changes and rebuild automatically")
   .action(async (options) => {
     const inputDirs = options.input.map((dir: string) => path.resolve(dir));
+    // Fix: Use getActualWorkingDirectory() instead of process.cwd()
     const outputFile = path.isAbsolute(options.output)
       ? options.output
-      : path.join(process.cwd(), options.output);
+      : path.join(getActualWorkingDirectory(), options.output);
 
     if (options.watch) {
       // Run in watch mode
@@ -641,4 +652,4 @@ program
     }
   });
 
-program.parse(process.argv);
+program.parse(process.argv);