summaryrefslogtreecommitdiff
path: root/server/continuedev/headless
diff options
context:
space:
mode:
authorNate Sesti <33237525+sestinj@users.noreply.github.com>2023-10-09 18:37:27 -0700
committerGitHub <noreply@github.com>2023-10-09 18:37:27 -0700
commitf09150617ed2454f3074bcf93f53aae5ae637d40 (patch)
tree5cfe614a64d921dfe58b049f426d67a8b832c71f /server/continuedev/headless
parent985304a213f620cdff3f8f65f74ed7e3b79be29d (diff)
downloadsncontinue-f09150617ed2454f3074bcf93f53aae5ae637d40.tar.gz
sncontinue-f09150617ed2454f3074bcf93f53aae5ae637d40.tar.bz2
sncontinue-f09150617ed2454f3074bcf93f53aae5ae637d40.zip
Preview (#541)
* Strong typing (#533) * refactor: :recycle: get rid of continuedev.src.continuedev structure * refactor: :recycle: switching back to server folder * feat: :sparkles: make config.py imports shorter * feat: :bookmark: publish as pre-release vscode extension * refactor: :recycle: refactor and add more completion params to ui * build: :building_construction: download from preview S3 * fix: :bug: fix paths * fix: :green_heart: package:pre-release * ci: :green_heart: more time for tests * fix: :green_heart: fix build scripts * fix: :bug: fix import in run.py * fix: :bookmark: update version to try again * ci: 💚 Update package.json version [skip ci] * refactor: :fire: don't check for old extensions version * fix: :bug: small bug fixes * fix: :bug: fix config.py import paths * ci: 💚 Update package.json version [skip ci] * ci: :green_heart: platform-specific builds test #1 * feat: :green_heart: ship with binary * fix: :green_heart: fix copy statement to include.exe for windows * fix: :green_heart: cd extension before packaging * chore: :loud_sound: count tokens generated * fix: :green_heart: remove npm_config_arch * fix: :green_heart: publish as pre-release! * chore: :bookmark: update version * perf: :green_heart: hardcode distro paths * fix: :bug: fix yaml syntax error * chore: :bookmark: update version * fix: :green_heart: update permissions and version * feat: :bug: kill old server if needed * feat: :lipstick: update marketplace icon for pre-release * ci: 💚 Update package.json version [skip ci] * feat: :sparkles: auto-reload for config.py * feat: :wrench: update default config.py imports * feat: :sparkles: codelens in config.py * feat: :sparkles: select model param count from UI * ci: 💚 Update package.json version [skip ci] * feat: :sparkles: more model options, ollama error handling * perf: :zap: don't show server loading immediately * fix: :bug: fixing small UI details * ci: 💚 Update package.json version [skip ci] * feat: :rocket: headers param on LLM class * fix: :bug: fix headers for openai.;y * feat: :sparkles: highlight code on cmd+shift+L * ci: 💚 Update package.json version [skip ci] * feat: :lipstick: sticky top bar in gui.tsx * fix: :loud_sound: websocket logging and horizontal scrollbar * ci: 💚 Update package.json version [skip ci] * feat: :sparkles: allow AzureOpenAI Service through GGML * ci: 💚 Update package.json version [skip ci] * fix: :bug: fix automigration * ci: 💚 Update package.json version [skip ci] * ci: :green_heart: upload binaries in ci, download apple silicon * chore: :fire: remove notes * fix: :green_heart: use curl to download binary * fix: :green_heart: set permissions on apple silicon binary * fix: :green_heart: testing * fix: :green_heart: cleanup file * fix: :green_heart: fix preview.yaml * fix: :green_heart: only upload once per binary * fix: :green_heart: install rosetta * ci: :green_heart: download binary after tests * ci: 💚 Update package.json version [skip ci] * ci: :green_heart: prepare ci for merge to main --------- Co-authored-by: GitHub Action <action@github.com>
Diffstat (limited to 'server/continuedev/headless')
-rw-r--r--server/continuedev/headless/__init__.py20
-rw-r--r--server/continuedev/headless/headless_ide.py181
2 files changed, 201 insertions, 0 deletions
diff --git a/server/continuedev/headless/__init__.py b/server/continuedev/headless/__init__.py
new file mode 100644
index 00000000..2ecdcce6
--- /dev/null
+++ b/server/continuedev/headless/__init__.py
@@ -0,0 +1,20 @@
+from typing import Optional, Union
+
+import typer
+
+from ..core.config import ContinueConfig
+from ..server.session_manager import Session, session_manager
+from .headless_ide import LocalIdeProtocol
+
+app = typer.Typer()
+
+
+async def start_headless_session(
+ config: Optional[Union[str, ContinueConfig]] = None
+) -> Session:
+ if config is not None:
+ if isinstance(config, str):
+ config: ContinueConfig = ContinueConfig.from_filepath(config)
+
+ ide = LocalIdeProtocol()
+ return await session_manager.new_session(ide, config=config)
diff --git a/server/continuedev/headless/headless_ide.py b/server/continuedev/headless/headless_ide.py
new file mode 100644
index 00000000..088da2c9
--- /dev/null
+++ b/server/continuedev/headless/headless_ide.py
@@ -0,0 +1,181 @@
+import os
+import subprocess
+import uuid
+from typing import Any, Callable, Coroutine, List, Optional
+
+from dotenv import load_dotenv
+from fastapi import WebSocket
+
+from ..models.filesystem import (
+ FileSystem,
+ RangeInFile,
+ RangeInFileWithContents,
+ RealFileSystem,
+)
+from ..models.filesystem_edit import EditDiff, FileEdit, FileSystemEdit
+from ..server.ide_protocol import AbstractIdeProtocolServer
+
+load_dotenv()
+
+
+def get_mac_address():
+ mac_num = hex(uuid.getnode()).replace("0x", "").upper()
+ mac = "-".join(mac_num[i : i + 2] for i in range(0, 11, 2))
+ return mac
+
+
+class LocalIdeProtocol(AbstractIdeProtocolServer):
+ websocket: WebSocket = None
+ session_id: Optional[str]
+ workspace_directory: str = os.getcwd()
+ unique_id: str = get_mac_address()
+
+ filesystem: FileSystem = RealFileSystem()
+
+ async def handle_json(self, data: Any):
+ """Handle a json message"""
+ pass
+
+ def showSuggestion(self, file_edit: FileEdit):
+ """Show a suggestion to the user"""
+ pass
+
+ async def setFileOpen(self, filepath: str, open: bool = True):
+ """Set whether a file is open"""
+ pass
+
+ async def showMessage(self, message: str):
+ """Show a message to the user"""
+ print(message)
+
+ async def showVirtualFile(self, name: str, contents: str):
+ """Show a virtual file"""
+ pass
+
+ async def setSuggestionsLocked(self, filepath: str, locked: bool = True):
+ """Set whether suggestions are locked"""
+ pass
+
+ async def getSessionId(self):
+ """Get a new session ID"""
+ pass
+
+ async def showSuggestionsAndWait(self, suggestions: List[FileEdit]) -> bool:
+ """Show suggestions to the user and wait for a response"""
+ pass
+
+ def onAcceptRejectSuggestion(self, accepted: bool):
+ """Called when the user accepts or rejects a suggestion"""
+ pass
+
+ def onFileSystemUpdate(self, update: FileSystemEdit):
+ """Called when a file system update is received"""
+ pass
+
+ def onCloseGUI(self, session_id: str):
+ """Called when a GUI is closed"""
+ pass
+
+ def onOpenGUIRequest(self):
+ """Called when a GUI is requested to be opened"""
+ pass
+
+ async def getOpenFiles(self) -> List[str]:
+ """Get a list of open files"""
+ pass
+
+ async def getVisibleFiles(self) -> List[str]:
+ """Get a list of visible files"""
+ pass
+
+ async def getHighlightedCode(self) -> List[RangeInFile]:
+ """Get a list of highlighted code"""
+ pass
+
+ async def readFile(self, filepath: str) -> str:
+ """Read a file"""
+ return self.filesystem.read(filepath)
+
+ async def readRangeInFile(self, range_in_file: RangeInFile) -> str:
+ """Read a range in a file"""
+ return self.filesystem.read_range_in_file(range_in_file)
+
+ async def editFile(self, edit: FileEdit):
+ """Edit a file"""
+ self.filesystem.apply_file_edit(edit)
+
+ async def applyFileSystemEdit(self, edit: FileSystemEdit) -> EditDiff:
+ """Apply a file edit"""
+ return self.filesystem.apply_edit(edit)
+
+ async def saveFile(self, filepath: str):
+ """Save a file"""
+ pass
+
+ async def getUserSecret(self, key: str):
+ """Get a user secret"""
+ return os.environ.get(key)
+
+ async def highlightCode(self, range_in_file: RangeInFile, color: str):
+ """Highlight code"""
+ pass
+
+ async def runCommand(self, command: str) -> str:
+ """Run a command using subprocess (don't pass, actually implement)"""
+ return subprocess.check_output(command, shell=True).decode("utf-8")
+
+ def onHighlightedCodeUpdate(self, range_in_files: List[RangeInFileWithContents]):
+ """Called when highlighted code is updated"""
+ pass
+
+ def onDeleteAtIndex(self, index: int):
+ """Called when a step is deleted at a given index"""
+ pass
+
+ async def showDiff(self, filepath: str, replacement: str, step_index: int):
+ """Show a diff"""
+ pass
+
+ def subscribeToFilesCreated(self, callback: Callable[[List[str]], None]):
+ """Subscribe to files created event"""
+ pass
+
+ def subscribeToFilesDeleted(self, callback: Callable[[List[str]], None]):
+ """Subscribe to files deleted event"""
+ pass
+
+ def subscribeToFilesRenamed(self, callback: Callable[[List[str], List[str]], None]):
+ """Subscribe to files renamed event"""
+ pass
+
+ def subscribeToFileSaved(self, callback: Callable[[str, str], None]):
+ """Subscribe to file saved event"""
+ pass
+
+ def onFilesCreated(self, filepaths: List[str]):
+ """Called when files are created"""
+ pass
+
+ def onFilesDeleted(self, filepaths: List[str]):
+ """Called when files are deleted"""
+ pass
+
+ def onFilesRenamed(self, old_filepaths: List[str], new_filepaths: List[str]):
+ """Called when files are renamed"""
+ pass
+
+ def onFileSaved(self, filepath: str, contents: str):
+ """Called when a file is saved"""
+ pass
+
+ async def fileExists(self, filepath: str) -> Coroutine[Any, Any, str]:
+ """Check if a file exists"""
+ return self.filesystem.exists(filepath)
+
+ async def getTerminalContents(self) -> Coroutine[Any, Any, str]:
+ return ""
+
+ async def listDirectoryContents(
+ self, directory: str, recursive: bool = False
+ ) -> List[str]:
+ return self.filesystem.list_directory_contents(directory, recursive=recursive)