feat: add method to retrieve open buffer information

bigcodegen · bigcodegen · commit 364c86add19d · 2025-02-09T16:00:07.000-06:00
diff --git a/src/neovim.ts b/src/neovim.ts
@@ -7,6 +7,13 @@ interface NeovimStatus {
   fileName: string;
 }
 
+interface BufferInfo {
+  number: number;
+  name: string;
+  isListed: boolean;
+  isLoaded: boolean;
+}
+
 export class NeovimManager {
   private static instance: NeovimManager;
 
@@ -124,4 +131,32 @@ export class NeovimManager {
       return 'Error editing lines';
     }
   }
+
+  public async getOpenBuffers(): Promise<BufferInfo[]> {
+    try {
+      const nvim = await this.connect();
+      const buffers = await nvim.buffers;
+      const bufferInfos: BufferInfo[] = [];
+
+      for (const buffer of buffers) {
+        const [isLoaded, isListedOption] = await Promise.all([
+          buffer.loaded,
+          buffer.getOption('buflisted')
+        ]);
+        const isListed = Boolean(isListedOption);
+
+        bufferInfos.push({
+          number: buffer.id,
+          name: await buffer.name,
+          isListed,
+          isLoaded
+        });
+      }
+
+      return bufferInfos;
+    } catch (error) {
+      console.error('Error getting open buffers:', error);
+      return [];
+    }
+  }
 }
