Merge pull request #16 from coder-inbox/code-execution

implement code execution using rapid api judge0 api

Author: wiseaidev

.env.example

@@ -21,3 +21,6 @@ DETA_PROJECT_KEY=
 
 # OpenAI
 OPENAI_API_KEY=
+
+# RAPIDAPI JUDGE0 API Key
+RAPIDAPI_KEY=

poetry.lock

@@ -32,6 +32,7 @@ deta = "^1.1.0"
 nylas = "^5.14.1"
 openai = "^0.28.1"
 apscheduler = "^3.10.4"
+httpx = "^0.25.0"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^7.2.0"

pyproject.toml

@@ -16,6 +16,8 @@ fastapi==0.103.2
 frozenlist==1.3.3
 h11==0.14.0
 httptools==0.6.0
+httpx==0.25.0
+httpcore==0.18.0
 idna==3.4
 motor==3.1.2
 multidict==6.0.4

requirements.txt

@@ -29,6 +29,7 @@ class Settings(BaseSettings):
         DETA_PROJECT_KEY (str) : A Deta project key.
         NYLAS_SYSTEM_TOKEN (str) : A Nylas access token for sending email as system.
         OPENAI_API_KEY (str) : An openai api key for generating emails.
+        RAPIDAPI_KEY (str): Rapid api key
 
     Example:
         >>> MONGODB_HOST=svc-123456789.svc.MONGODB.com
@@ -40,6 +41,7 @@ class Settings(BaseSettings):
         >>> DETA_PROJECT_KEY=12312dSDJHJSBA
         >>> NYLAS_SYSTEM_TOKEN=12312dSDJHJSBA
         >>> OPENAI_API_KEY=12312dSDJHJSBA
+        >>> RAPIDAPI_KEY=12312dSDJHJSBA
     """
 
     MONGODB_HOST: str = os.getenv("MONGODB_HOST")  # type: ignore
@@ -55,6 +57,7 @@ class Settings(BaseSettings):
     CLIENT_URI: str = os.getenv("CLIENT_URI")  # type: ignore
     NYLAS_SYSTEM_TOKEN: str = os.getenv("NYLAS_SYSTEM_TOKEN")  # type: ignore
     OPENAI_API_KEY: str = os.getenv("OPENAI_API_KEY")  # type: ignore
+    RAPIDAPI_KEY: str = os.getenv("RAPIDAPI_KEY")  # type: ignore
 
     class Config:  # pylint: disable=R0903
         """

src/config.py

@@ -1,9 +1,12 @@
 """Nylas router module."""
 
+import asyncio
 from fastapi import (
     APIRouter,
     Depends,
+    HTTPException,
 )
+import httpx
 from odmantic.session import (
     AIOSession,
 )
@@ -29,6 +32,15 @@
 
 router = APIRouter(prefix="/api/v1")
 
+JUDGE0_API_URL = "https://judge0-ce.p.rapidapi.com/submissions"
+
+headers = {
+    "X-RapidAPI-Key": settings().RAPIDAPI_KEY,
+    "Content-Type": "application/json",
+}
+
+submission_status_dict = {}
+
 
 @router.post(
     "/nylas/generate-auth-url",
@@ -334,3 +346,77 @@ async def search_emails(
         item.as_json(enforce_read_only=False) for item in threads_with_messages
     ]
     return res_json
+
+
+@router.post(
+    "/nylas/execute-code",
+    response_model=None,
+    status_code=200,
+    name="nylas:execute-code",
+)
+async def execute_code(
+    request_body: nylas_schemas.CodeExecutionSchema,
+    current_user: users_schemas.UserObjectSchema = Depends(
+        dependencies.get_current_user
+    ),
+) -> Any:
+    try:
+        payload = {
+            "source_code": request_body.code,
+            "language_id": int(request_body.language_id),
+            "stdin": "",
+            "expected_output": "",
+            "cpu_time_limit": 2,
+            "cpu_extra_time": 0.5,
+            "wall_time_limit": 5,
+            "memory_limit": 512000,
+        }
+
+        async with httpx.AsyncClient() as client:
+            response = await client.post(
+                JUDGE0_API_URL,
+                json=payload,
+                headers=headers,
+            )
+
+        if response.status_code == 201:
+            submission_token = response.json()["token"]
+            submission_status_dict[submission_token] = "Running"
+            timeout = 60
+            start_time = asyncio.get_event_loop().time()
+            result = None
+            while submission_status_dict.get(submission_token) == "Running":
+                await asyncio.sleep(1)
+
+                if asyncio.get_event_loop().time() - start_time > timeout:
+                    print("Timeout")
+                    submission_status_dict[submission_token] = "Timeout"
+                    break
+
+                async with httpx.AsyncClient() as client:
+                    new_response = await client.get(
+                        f"{JUDGE0_API_URL}/{submission_token}",
+                        headers=headers,
+                    )
+
+                if new_response.status_code == 200:
+                    print(new_response.json())
+                    submission_stdout = new_response.json()["stdout"]
+                    if submission_stdout:
+                        print("Finished")
+                        submission_status_dict[submission_token] = "Finished"
+                        result = new_response.json()
+                        break
+                else:
+                    return HTTPException(
+                        status_code=500, detail="Failed to retrieve result"
+                    )
+            return result
+        else:
+            return HTTPException(
+                status_code=500, detail="Code execution failed"
+            )
+
+    except Exception as e:
+        print(e)
+        return HTTPException(status_code=500, detail=str(e))

src/nylas/router.py

@@ -84,3 +84,8 @@ class ReplyEmailSchema(BaseModel):
         ..., description="thread id", example="6gq33fhrgpts8wmb2sl98jdvo"
     )
     body: str = Field(..., description="email body", example="Hello there!")
+
+
+class CodeExecutionSchema(BaseModel):
+    code: str = Field(..., description="code", example="print('hi')")
+    language_id: str = Field(..., description="language id", example="71")