mirror of
https://github.com/reconurge/flowsint.git
synced 2026-05-06 03:38:31 -05:00
36 lines
982 B
Python
36 lines
982 B
Python
from fastapi import FastAPI, HTTPException, Depends
|
|
from pydantic import BaseModel
|
|
|
|
from app.core.celery import celery_app
|
|
from app.scanners.registry import ScannerRegistry
|
|
from app.core.auth import get_current_user
|
|
|
|
app = FastAPI()
|
|
|
|
class ScanRequest(BaseModel):
|
|
value: str
|
|
sketch_id: str
|
|
scanner: str
|
|
|
|
@app.post("/scan")
|
|
async def scan(
|
|
item: ScanRequest,
|
|
user=Depends(get_current_user)
|
|
):
|
|
try:
|
|
if not ScannerRegistry.scanner_exists(name=item.scanner):
|
|
raise HTTPException(status_code=400, detail="Scanner not found")
|
|
user_id = user.get("sub")
|
|
task = celery_app.send_task("run_scan", args=[item.scanner, item.value, item.sketch_id])
|
|
print(user_id)
|
|
return {"id": task.id}
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
print(e)
|
|
raise HTTPException(status_code=500, detail=f"Server error: {str(e)}")
|
|
|
|
@app.get("/health")
|
|
async def health():
|
|
return {"status": "ok"}
|