sdk: fork getFiles implementation

Author: jahooma

sdk/src/client.ts

@@ -13,8 +13,8 @@ import {
   getInitialSessionState,
   SessionState,
 } from '../../common/src/types/session-state'
-import { getFiles } from '../../npm-app/src/project-files'
 import { PrintModeEvent } from '../../common/src/types/print-mode'
+import { getFiles } from './tools/read-files'
 
 type ClientToolName = 'write_file' | 'run_terminal_command'
 
@@ -215,7 +215,7 @@ export class CodebuffClient {
       const overrideResult = await override(filePath)
       return overrideResult.files
     }
-    return getFiles(filePath)
+    return getFiles(filePath, this.cwd)
   }
 
   private async handleToolCall(

sdk/src/tools/read-files.ts

@@ -0,0 +1,47 @@
+import { FILE_READ_STATUS } from '../../../common/src/constants'
+import path, { isAbsolute } from 'path'
+import fs from 'fs'
+
+export function getFiles(filePaths: string[], cwd: string) {
+  const result: Record<string, string | null> = {}
+  const MAX_FILE_SIZE = 1024 * 1024 // 1MB in bytes
+
+  for (const filePath of filePaths) {
+    if (!filePath) {
+      continue
+    }
+
+    // Convert absolute paths within project to relative paths
+    const relativePath = filePath.startsWith(cwd)
+      ? path.relative(cwd, filePath)
+      : filePath
+    const fullPath = path.join(cwd, relativePath)
+    if (isAbsolute(relativePath) || !fullPath.startsWith(cwd)) {
+      result[relativePath] = FILE_READ_STATUS.OUTSIDE_PROJECT
+      continue
+    }
+    try {
+      const stats = fs.statSync(fullPath)
+      if (stats.size > MAX_FILE_SIZE) {
+        result[relativePath] =
+          FILE_READ_STATUS.TOO_LARGE +
+          ` [${(stats.size / (1024 * 1024)).toFixed(2)}MB]`
+      } else {
+        const content = fs.readFileSync(fullPath, 'utf8')
+        result[relativePath] = content
+      }
+    } catch (error) {
+      if (
+        error &&
+        typeof error === 'object' &&
+        'code' in error &&
+        error.code === 'ENOENT'
+      ) {
+        result[relativePath] = FILE_READ_STATUS.DOES_NOT_EXIST
+      } else {
+        result[relativePath] = FILE_READ_STATUS.ERROR
+      }
+    }
+  }
+  return result
+}