Adding /internal/airdrop.artifacts.upload-url endpoint to the mock DevRev server.

Author: dusano

mock_devrev_server.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, HTTPException, Header, UploadFile, File, Request
+from fastapi import FastAPI, HTTPException, Header, Request
 from pydantic import BaseModel
 from typing import Optional, List, Dict
 import uuid
@@ -22,6 +22,11 @@ class ArtifactPrepareResponse(BaseModel):
 class ExternalWorkerResponse(BaseModel):
     state: str
 
+class AirdropArtifactResponse(BaseModel):
+    artifact_id: str
+    upload_url: str
+    form_data: List[FormDataField]
+
 @app.post("/upload/{artifact_id}")
 async def upload_artifact(
     artifact_id: str,
@@ -102,6 +107,36 @@ async def get_snap_ins(request: Request):
     print("Received /internal/snap-ins.get GET request")
     return {"status": "success"}
 
+@app.get("/internal/airdrop.artifacts.upload-url", response_model=AirdropArtifactResponse)
+async def airdrop_artifacts_upload_url(
+    file_type: str,
+    file_name: str,
+    request_id: Optional[str] = None,
+    authorization: Optional[str] = Header(None)
+):
+    # Generate a unique artifact ID
+    artifact_id = str(uuid.uuid4())
+    
+    # Create a mock S3-like URL for the upload
+    upload_url = f"http://localhost:8003/upload/{artifact_id}"
+    
+    # Create form data fields that would typically be required for S3 upload
+    form_data = [
+        FormDataField(key="key", value=f"airdrop-artifacts/{artifact_id}/{file_name}"),
+        FormDataField(key="Content-Type", value=file_type),
+        FormDataField(key="x-amz-meta-artifact-id", value=artifact_id),
+    ]
+    
+    # Add request_id to form data if provided
+    if request_id:
+        form_data.append(FormDataField(key="x-amz-meta-request-id", value=request_id))
+    
+    return AirdropArtifactResponse(
+        artifact_id=artifact_id,
+        upload_url=upload_url,
+        form_data=form_data
+    )
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="localhost", port=8003)