Add initial application files including server implementation, API specification, configuration, and ignore files

.cursorignore

@@ -0,0 +1,13 @@
+.DS_Store
+.AppleDouble
+.LSOverride
+Icon
+._*
+.DocumentRevisions-V100
+.fseventsd
+.Spotlight-V100
+.TemporaryItems
+.Trashes
+.VolumeIcon.icns
+.com.apple.timemachine.donotpresent
+

hypercorn.toml

@@ -0,0 +1,9 @@
+# Hypercorn configuration file
+
+# Server binding options
+bind = "0.0.0.0:8000" 
+workers = 1 
+
+# Logging configuration
+accesslog = "/var/log/serverwitch/access.log"
+logconfig = "logging.toml"

openapi.yaml

@@ -0,0 +1,189 @@
+openapi: 3.1.0
+info:
+  title: Kaio Buu API
+  description: Execute commands, read files and write files on the user's server
+  version: 0.2.0
+servers:
+  - url: https://run.pyboxs.com/kaiobuu
+paths:
+  /command:
+    post:
+      operationId: post_send_command
+      summary: Execute a command on the server.
+      x-openai-isConsequential: false
+      requestBody:
+        description: Request body containing the command and session ID.
+        required: true
+        content:
+          application/json:
+            schema:
+              $ref: '#/components/schemas/CommandRequest'
+      responses:
+        '200':
+          description: Command executed successfully.
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/CommandResponse'
+        '500':
+          description: Server error.
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/ErrorResponse'
+  /read:
+    post:
+      operationId: post_read
+      summary: Read a file on the server.
+      x-openai-isConsequential: false
+      requestBody:
+        description: Request body containing the file path and session ID.
+        required: true
+        content:
+          application/json:
+            schema:
+              $ref: '#/components/schemas/ReadRequest'
+      responses:
+        '200':
+          description: File read successfully.
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/ReadResponse'
+        '500':
+          description: Server error.
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/ErrorResponse'
+  /write:
+    post:
+      operationId: post_write
+      summary: Write a file on the server.
+      x-openai-isConsequential: false
+      requestBody:
+        description: Request body containing the file path, new content and session ID.
+        required: true
+        content:
+          application/json:
+            schema:
+              $ref: '#/components/schemas/WriteRequest'
+      responses:
+        '200':
+          description: File written successfully.
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/WriteResponse'
+        '500':
+          description: Server error.
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/ErrorResponse'
+  /status:
+    get:
+      operationId: get_status
+      summary: Get the status of the server.
+      x-openai-isConsequential: false
+      responses:
+        '200':
+          description: Status of the server.
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/StatusResponse'
+components:
+  schemas:
+    CommandRequest:
+      type: object
+      required:
+        - session_id
+        - command
+      properties:
+        command:
+          type: string
+          description: Command to execute on the server.
+        session_id:
+          type: string
+          description: Session ID.
+    ReadRequest:
+      type: object
+      required:
+        - session_id
+        - path
+      properties:
+        path:
+          type: string
+          description: Absolute path of the file to be read.
+        session_id:
+          type: string
+          description: Session ID.
+    WriteRequest:
+      type: object
+      required:
+        - session_id
+        - path
+        - content
+      properties:
+        path:
+          type: string
+          description: Absolute path of the file to be written.
+        content:
+          type: string
+          description: New content of the file.
+        session_id:
+          type: string
+          description: Session ID.
+    CommandResponse:
+      type: object
+      required:
+        - return_code
+        - stdout
+        - stderr
+      properties:
+        return_code:
+          type: integer
+          description: Return code of the executed command.
+        stdout:
+          type: string
+          description: Standard output of the executed command.
+        stderr:
+          type: string
+          description: Standard error of the executed command.
+    ReadResponse:
+      type: object
+      required:
+        - content
+      properties:
+        content:
+          type: string
+          description: Content of the file.
+    WriteResponse:
+      type: object
+      required:
+        - size
+      properties:
+        size:
+          type: int
+          description: Number of characters written to the file.
+    ErrorResponse:
+      type: object
+      required:
+        - error
+      properties:
+        error:
+          type: string
+          description: Error message detailing what went wrong.
+    StatusResponse:
+      type: object
+      required:
+        - status
+        - version
+      properties:
+        status:
+          type: string
+          description: Current status of the server.
+        version:
+          type: string
+          description: Version of the server software.

poetry.toml

@@ -0,0 +1,2 @@
+[virtualenvs]
+in-project = true

requirements.txt

@@ -0,0 +1 @@
+ 

server.py

@@ -0,0 +1,103 @@
+# server.py
+import asyncio
+import uuid
+from typing import AsyncGenerator, Dict, Tuple, Any, Optional
+from dataclasses import dataclass
+from quart import Quart, websocket, request, FileResponse
+from quart_schema import QuartSchema, validate_request, validate_response
+import importlib.metadata
+import secrets
+import logging
+
+from broker import SessionBroker, SessionDoesNotExist, ClientRequest, ClientResponse, ClientError
+
+# Configuración
+TIMEOUT: int = 60
+LOG_LEVEL: int = logging.INFO
+TRUSTED_HOSTS: list[str] = ["127.0.0.1"]
+
+# Crear aplicación
+app = Quart(__name__)
+QuartSchema(app)
+app.logger.setLevel(LOG_LEVEL)
+
+broker = SessionBroker()
+
+# Definición de modelos de datos
+@dataclass
+class Status:
+    status: str
+    version: str
+
+@dataclass
+class Session:
+    session_id: str
+
+@dataclass
+class Command:
+    session_id: str
+    command: str
+
+@dataclass
+class CommandResponse:
+    return_code: int
+    stdout: str
+    stderr: str
+
+@dataclass
+class ErrorResponse:
+    error: str
+
+# Rutas API
+@app.get("/status")
+@validate_response(Status)
+async def status() -> Status:
+    return Status(status="OK", version=importlib.metadata.version('your-package-name'))
+
+@app.websocket('/session')
+async def session_handler():
+    session_id = secrets.token_hex()
+    app.logger.info(f"New session: {session_id}")
+    await websocket.send_as(Session(session_id=session_id), Session)
+
+    task = asyncio.ensure_future(_receive(session_id))
+    try:
+        async for request in broker.subscribe(session_id):
+            app.logger.info(f"Sending request {request.request_id} to client.")
+            await websocket.send_as(request, ClientRequest)
+    finally:
+        task.cancel()
+
+async def _receive(session_id: str) -> None:
+    while True:
+        response = await websocket.receive_as(ClientResponse)
+        app.logger.info(f"Received response for session {session_id}: {response}")
+        await broker.receive_response(session_id, response)
+
+@app.post('/command')
+@validate_request(Command)
+@validate_response(CommandResponse, 200)
+@validate_response(ErrorResponse, 500)
+async def command(data: Command) -> Tuple[CommandResponse | ErrorResponse, int]:
+    try:
+        response_data = await broker.send_request(
+            session_id=data.session_id, 
+            data={'action': 'command', 'command': data.command}, 
+            timeout=TIMEOUT
+        )
+        response = CommandResponse(**response_data)
+        return response, 200
+    except SessionDoesNotExist:
+        app.logger.warning(f"Invalid session ID: {data.session_id}")
+        return ErrorResponse(error='Session does not exist.'), 500
+    except ClientError as e:
+        return ErrorResponse(error=e.message), 500
+    except asyncio.TimeoutError:
+        return ErrorResponse(error='Timeout waiting for client response.'), 500
+
+# Ejecutar aplicación
+def run():
+    app.run(port=7860)
+
+if __name__ == "__main__":
+    run()