From a0c590944ff82d3584a532cd7400689297735757 Mon Sep 17 00:00:00 2001
From: Castor Regex <regex@wit.com>
Date: Wed, 22 Oct 2025 05:31:24 -0500
Subject: feat: add support for .proto files

---
 packages/core/src/services/fileSystemService.ts | 14 ++++++++++++++
 packages/core/src/utils/fileUtils.ts            | 15 ++++++++++++++-
 2 files changed, 28 insertions(+), 1 deletion(-)

(limited to 'packages/core/src')

diff --git a/packages/core/src/services/fileSystemService.ts b/packages/core/src/services/fileSystemService.ts
index e2f30cf4..3e4cd30f 100644
--- a/packages/core/src/services/fileSystemService.ts
+++ b/packages/core/src/services/fileSystemService.ts
@@ -5,6 +5,7 @@
  */
 
 import fs from 'fs/promises';
+import protobuf from 'protobufjs';
 
 /**
  * Interface for file system operations that may be delegated to different implementations
@@ -18,6 +19,14 @@ export interface FileSystemService {
    */
   readTextFile(filePath: string): Promise<string>;
 
+  /**
+   * Read and parse a proto file
+   *
+   * @param filePath - The path to the file to read
+   * @returns The file content as a JSON string
+   */
+  readProtoFile(filePath: string): Promise<string>;
+
   /**
    * Write text content to a file
    *
@@ -35,6 +44,11 @@ export class StandardFileSystemService implements FileSystemService {
     return fs.readFile(filePath, 'utf-8');
   }
 
+  async readProtoFile(filePath: string): Promise<string> {
+    const root = await protobuf.load(filePath);
+    return JSON.stringify(root.toJSON(), null, 2);
+  }
+
   async writeTextFile(filePath: string, content: string): Promise<void> {
     await fs.writeFile(filePath, content, 'utf-8');
   }
diff --git a/packages/core/src/utils/fileUtils.ts b/packages/core/src/utils/fileUtils.ts
index f0b491ed..009a4d72 100644
--- a/packages/core/src/utils/fileUtils.ts
+++ b/packages/core/src/utils/fileUtils.ts
@@ -121,7 +121,9 @@ export async function isBinaryFile(filePath: string): Promise<boolean> {
  */
 export async function detectFileType(
   filePath: string,
-): Promise<'text' | 'image' | 'pdf' | 'audio' | 'video' | 'binary' | 'svg'> {
+): Promise<
+  'text' | 'image' | 'pdf' | 'audio' | 'video' | 'binary' | 'svg' | 'proto'
+> {
   const ext = path.extname(filePath).toLowerCase();
 
   // The mimetype for various TypeScript extensions (ts, mts, cts, tsx) can be
@@ -135,6 +137,10 @@ export async function detectFileType(
     return 'svg';
   }
 
+  if (ext === '.proto') {
+    return 'proto';
+  }
+
   const lookedUpMimeType = mime.lookup(filePath); // Returns false if not found, or the mime type string
   if (lookedUpMimeType) {
     if (lookedUpMimeType.startsWith('image/')) {
@@ -280,6 +286,13 @@ export async function processSingleFileContent(
           returnDisplay: `Read SVG as text: ${relativePathForDisplay}`,
         };
       }
+      case 'proto': {
+        const content = await fileSystemService.readProtoFile(filePath);
+        return {
+          llmContent: content,
+          returnDisplay: `Read proto file: ${relativePathForDisplay}`,
+        };
+      }
       case 'text': {
         const content = await fileSystemService.readTextFile(filePath);
         const lines = content.split('\n');
-- 
cgit v1.2.3