setup project

Author: BaseMax

app/api/execute.py

@@ -0,0 +1,13 @@
+import redis.asyncio as redis
+from fastapi import APIRouter
+from app.core.config import settings
+from app.schemas.execution import ExecutionRequest
+
+router = APIRouter(prefix="/execute")
+r = redis.from_url(settings.REDIS_URL)
+
+
+@router.post("")
+async def enqueue_exec(req: ExecutionRequest):
+    await r.lpush("exec_queue", req.model_dump_json())
+    return {"status": "queued"}

app/api/snippets.py

@@ -0,0 +1,31 @@
+from fastapi import APIRouter, Depends
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy import select
+from app.db.session import SessionLocal
+from app.db.models import Snippet, SnippetVersion
+from app.schemas.snippet import SnippetCreate, SnippetOut
+
+router = APIRouter(prefix="/snippets")
+
+
+async def get_db():
+    async with SessionLocal() as s:
+        yield s
+
+
+@router.post("", response_model=SnippetOut)
+async def create_snippet(data: SnippetCreate, db: AsyncSession = Depends(get_db)):
+    snippet = Snippet(title=data.title)
+    db.add(snippet)
+    await db.flush()
+
+    version = SnippetVersion(snippet_id=snippet.id, code=data.code, version=1)
+    db.add(version)
+    await db.commit()
+    return snippet
+
+
+@router.get("/{snippet_id}", response_model=SnippetOut)
+async def get_snippet(snippet_id: int, db: AsyncSession = Depends(get_db)):
+    q = await db.execute(select(Snippet).where(Snippet.id == snippet_id))
+    return q.scalar_one()

app/core/config.py

@@ -0,0 +1,12 @@
+from pydantic import BaseModel
+
+
+class Settings(BaseModel):
+    DB_URL: str = "mysql+aiomysql://app:app@mariadb:3306/code"
+    REDIS_URL: str = "redis://redis:6379/0"
+    SANDBOX_CONTAINER: str = "code-sandbox"
+    EXEC_TIMEOUT: int = 5
+    MAX_OUTPUT: int = 20000
+
+
+settings = Settings()

app/db/models.py

@@ -0,0 +1,34 @@
+from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
+from sqlalchemy import Text, String, Integer, DateTime, ForeignKey
+from datetime import datetime
+
+
+class Base(DeclarativeBase):
+    pass
+
+
+class Snippet(Base):
+    __tablename__ = "snippets"
+
+    id: Mapped[int] = mapped_column(primary_key=True)
+    title: Mapped[str] = mapped_column(String(200))
+    created_at: Mapped[datetime] = mapped_column(default=datetime.utcnow)
+
+
+class SnippetVersion(Base):
+    __tablename__ = "snippet_versions"
+
+    id: Mapped[int] = mapped_column(primary_key=True)
+    snippet_id: Mapped[int] = mapped_column(ForeignKey("snippets.id"))
+    code: Mapped[str] = mapped_column(Text)
+    version: Mapped[int]
+
+
+class Execution(Base):
+    __tablename__ = "executions"
+
+    id: Mapped[int] = mapped_column(primary_key=True)
+    snippet_version_id: Mapped[int] = mapped_column(ForeignKey("snippet_versions.id"))
+    output: Mapped[str] = mapped_column(Text)
+    status: Mapped[str] = mapped_column(String(50))
+    created_at: Mapped[datetime] = mapped_column(default=datetime.utcnow)

app/db/session.py

@@ -0,0 +1,5 @@
+from sqlalchemy.ext.asyncio import create_async_engine, async_sessionmaker
+from app.core.config import settings
+
+engine = create_async_engine(settings.DB_URL, echo=False)
+SessionLocal = async_sessionmaker(engine, expire_on_commit=False)

app/docker-compose.yml

@@ -0,0 +1,34 @@
+services:
+  mariadb:
+    image: mariadb:11
+    environment:
+      MYSQL_ROOT_PASSWORD: root
+      MYSQL_DATABASE: code
+      MYSQL_USER: app
+      MYSQL_PASSWORD: app
+    ports:
+      - "3306:3306"
+
+  redis:
+    image: redis:7
+    ports:
+      - "6379:6379"
+
+  sandbox:
+    build: ./sandbox
+    container_name: code-sandbox
+    networks: [default]
+
+  api:
+    build: .
+    command: uvicorn app.main:app --host 0.0.0.0 --port 8000
+    volumes: [".:/app"]
+    ports:
+      - "8000:8000"
+    depends_on: [mariadb, redis, sandbox]
+
+  worker:
+    build: .
+    command: python -m app.worker.worker
+    volumes: [".:/app"]
+    depends_on: [redis, mariadb, sandbox]

app/main.py

@@ -0,0 +1,7 @@
+from fastapi import FastAPI
+from app.api import snippets, execute
+
+app = FastAPI(title="Code Snippet Execution Platform")
+
+app.include_router(snippets.router)
+app.include_router(execute.router)

app/schemas/execution.py

@@ -0,0 +1,14 @@
+from pydantic import BaseModel
+
+
+class ExecutionRequest(BaseModel):
+    snippet_version_id: int
+
+
+class ExecutionOut(BaseModel):
+    id: int
+    status: str
+    output: str
+
+    class Config:
+        from_attributes = True

app/schemas/snippet.py

@@ -0,0 +1,14 @@
+from pydantic import BaseModel
+
+
+class SnippetCreate(BaseModel):
+    title: str
+    code: str
+
+
+class SnippetOut(BaseModel):
+    id: int
+    title: str
+
+    class Config:
+        from_attributes = True

app/services/executor.py

@@ -0,0 +1,18 @@
+import docker
+from app.core.config import settings
+
+client = docker.from_env()
+
+
+def run_code(code: str) -> str:
+    container = client.containers.get(settings.SANDBOX_CONTAINER)
+
+    exec_id = client.api.exec_create(
+        container.id,
+        cmd=["python", "/runner.py"],
+        stdin=True,
+        tty=False,
+    )
+
+    output = client.api.exec_start(exec_id, stdin=True, socket=False, demux=False, stream=False, detach=False, tty=False, input=code.encode())
+    return output.decode(errors="ignore")[: settings.MAX_OUTPUT]