Initial CRUD API for workspaces

This adds a simple and unimplemented REST API for workspaces. Workspaces
will be the base for all other resources in terms of REST resource
mapping, so these go first.

These are initially left entirely unimplemented as
#600 needs to merge

Signed-off-by: Juan Antonio Osorio <[email protected]>

Author: JAORMX

src/codegate/api/__init__.py

@@ -0,0 +1,40 @@
+from fastapi import APIRouter
+from fastapi.routing import APIRoute
+
+from codegate.api import v1_models
+
+v1 = APIRouter()
+
+
+def uniq_name(route: APIRoute):
+    return f"v1_{route.name}"
+
+
+@v1.get("/workspaces", tags=["Workspaces"], generate_unique_id_function=uniq_name)
+async def list_workspaces() -> v1_models.ListWorkspacesResponse:
+    """List all workspaces."""
+    raise NotImplementedError
+
+@v1.get("/workspaces/active", tags=["Workspaces"], generate_unique_id_function=uniq_name)
+async def list_active_workspaces() -> v1_models.ListActiveWorkspacesResponse:
+    """List all active workspaces.
+
+    In it's current form, this function will only return one workspace. That is,
+    the globally active workspace."""
+    raise NotImplementedError
+
+@v1.post("/workspaces", tags=["Workspaces"], generate_unique_id_function=uniq_name, status_code=201)
+async def create_workspace(request: v1_models.CreateWorkspaceRequest):
+    """Create a new workspace."""
+    raise NotImplementedError
+
+@v1.get("/workspaces/{workspace_name}", tags=["Workspaces"], generate_unique_id_function=uniq_name)
+async def get_workspace(workspace_name: str) -> v1_models.Workspace:
+    """Get a workspace by name."""
+    raise NotImplementedError
+
+@v1.delete("/workspaces/{workspace_name}", tags=["Workspaces"],
+           generate_unique_id_function=uniq_name, status_code=204)
+async def delete_workspace(workspace_name: str):
+    """Delete a workspace by name."""
+    raise NotImplementedError

src/codegate/api/v1.py

@@ -0,0 +1,20 @@
+from typing import Any
+
+import pydantic
+
+
+class Workspace(pydantic.BaseModel):
+    name: str
+
+class ActiveWorkspace(Workspace):
+    # TODO: use a more specific type for last_updated
+    last_updated: Any
+
+class ListWorkspacesResponse(pydantic.BaseModel):
+    workspaces: list[Workspace]
+
+class ListActiveWorkspacesResponse(pydantic.BaseModel):
+    workspaces: list[ActiveWorkspace]
+
+class CreateWorkspaceRequest(pydantic.BaseModel):
+    name: str

src/codegate/api/v1_models.py

@@ -127,7 +127,7 @@ def extract_snippets(message: str) -> List[CodeSnippet]:
         lang_map = {
             "typescript": "javascript"
         }
-        lang = lang_map.get(lang, lang)        
+        lang = lang_map.get(lang, lang)
         snippets.append(CodeSnippet(filepath=filename, code=content, language=lang))
 
     return snippets

src/codegate/pipeline/extract_snippets/extract_snippets.py

@@ -1,4 +1,3 @@
-import json
 from typing import AsyncIterator, Optional, Union
 
 import structlog

src/codegate/providers/ollama/completion_handler.py

@@ -7,6 +7,7 @@
 from starlette.middleware.errors import ServerErrorMiddleware
 
 from codegate import __description__, __version__
+from codegate.api.v1 import v1
 from codegate.dashboard.dashboard import dashboard_router
 from codegate.pipeline.factory import PipelineFactory
 from codegate.providers.anthropic.provider import AnthropicProvider
@@ -97,4 +98,7 @@ async def health_check():
     app.include_router(system_router)
     app.include_router(dashboard_router)
 
+    # CodeGate API
+    app.include_router(v1, prefix="/api/v1", tags=["CodeGate API"])
+
     return app