fix: :bug: ssh compatibility by reading from vscode.workspace.fs

2 files changed, 18 insertions, 0 deletions

diff --git a/continuedev/src/continuedev/server/ide.py b/continuedev/src/continuedev/server/ide.py
index 610a1a48..871724db 100644
--- a/continuedev/src/continuedev/server/ide.py
+++ b/continuedev/src/continuedev/server/ide.py
@@ -120,6 +120,10 @@ class TerminalContentsResponse(BaseModel):
     contents: str
 
 
+class ListDirectoryContentsResponse(BaseModel):
+    contents: List[str]
+
+
 T = TypeVar("T", bound=BaseModel)
 
 
@@ -241,6 +245,7 @@ class IdeProtocolServer(AbstractIdeProtocolServer):
             "getUserSecret",
             "runCommand",
             "getTerminalContents",
+            "listDirectoryContents",
         ]:
             self.sub_queue.post(message_type, data)
         elif message_type == "workspaceDirectory":
@@ -477,6 +482,15 @@ class IdeProtocolServer(AbstractIdeProtocolServer):
         )
         return resp.fileEdit
 
+    async def listDirectoryContents(self, directory: str) -> List[str]:
+        """List the contents of a directory"""
+        resp = await self._send_and_receive_json(
+            {"directory": directory},
+            ListDirectoryContentsResponse,
+            "listDirectoryContents",
+        )
+        return resp.contents
+
     async def applyFileSystemEdit(self, edit: FileSystemEdit) -> EditDiff:
         """Apply a file edit"""
         backward = None
diff --git a/continuedev/src/continuedev/server/ide_protocol.py b/continuedev/src/continuedev/server/ide_protocol.py
index 435c82e2..f37c1737 100644
--- a/continuedev/src/continuedev/server/ide_protocol.py
+++ b/continuedev/src/continuedev/server/ide_protocol.py
@@ -147,5 +147,9 @@ class AbstractIdeProtocolServer(ABC):
     def onFileSaved(self, filepath: str, contents: str):
         """Called when a file is saved"""
 
+    @abstractmethod
+    async def listDirectoryContents(self, directory: str) -> List[str]:
+        """List directory contents"""
+
     workspace_directory: str
     unique_id: str