1 files changed, 26 insertions, 30 deletions

diff --git a/jb/views/common.py b/jb/views/common.py
index 7011557..d3b3e93 100644
--- a/jb/views/common.py
+++ b/jb/views/common.py
@@ -1,19 +1,18 @@
 import json
-from typing import Dict, Any
+from typing import Annotated, Any
 
 import requests
-from fastapi import Request, APIRouter, HTTPException
+from fastapi import APIRouter, Depends, HTTPException, Request
 from fastapi.responses import HTMLResponse
 from starlette.responses import RedirectResponse
 
-from jb.config import settings, JB_EVENTS_STREAM
-from jb.decorators import REDIS, HM
-from jb.flow.monitoring import emit_assignment_event, emit_mturk_notification_event
-from jb.models.definitions import AssignmentStatus
+from jb.api.auth import get_authenticated_user
+from jb.config import JB_EVENTS_STREAM, settings
+from jb.decorators import REDIS
+from jb.flow.monitoring import emit_mturk_notification_event
+from jb.models.auth import User
 from jb.models.event import MTurkEvent
 from jb.settings import BASE_HTML
-from jb.config import settings
-from jb.views.tasks import process_request
 
 common_router = APIRouter(prefix="", tags=["API"], include_in_schema=True)
 
@@ -29,32 +28,16 @@ async def work(request: Request):
     amt_hit_id = request.query_params.get("hitId", None)
     print(f"work: {amt_assignment_id=} {worker_id=} {amt_hit_id=}")
 
-    if not worker_id:
+    if (
+        not worker_id
+        or not amt_assignment_id
+        or amt_assignment_id == "ASSIGNMENT_ID_NOT_AVAILABLE"
+    ):
         return RedirectResponse(
             url=f"/preview/?{request.url.query}" if request.url.query else "/preview/",
             status_code=302,
         )
 
-    if amt_assignment_id is None or amt_assignment_id == "ASSIGNMENT_ID_NOT_AVAILABLE":
-        # Worker is previewing the HIT
-        amt_hit_type_id = "unknown"
-        if amt_hit_id:
-            hit = HM.get_from_amt_id(amt_hit_id=amt_hit_id)
-            amt_hit_type_id = hit.amt_hit_type_id
-        emit_assignment_event(
-            status=AssignmentStatus.PreviewState, amt_hit_type_id=amt_hit_type_id
-        )
-        return RedirectResponse(
-            url=f"/preview/?{request.url.query}" if request.url.query else "/preview/",
-            status_code=302,
-        )
-
-    try:
-        # The Worker has accepted the HIT
-        process_request(request)
-    except Exception:
-        raise HTTPException(status_code=500, detail="Error processing request")
-
     return HTMLResponse(BASE_HTML)
 
 
@@ -85,10 +68,23 @@ async def mturk_notifications(request: Request):
     return {"status": "ok"}
 
 
-def enqueue_mturk_notifications(msg: Dict[str, Any]) -> None:
+def enqueue_mturk_notifications(msg: dict[str, Any]) -> None:
     for evt in msg["Events"]:
         event = MTurkEvent.from_sns(evt)
         emit_mturk_notification_event(
             event_type=event.event_type, amt_hit_type_id=event.amt_hit_type_id
         )
         REDIS.xadd(JB_EVENTS_STREAM, {"data": event.model_dump_json()})
+
+
+@common_router.get(path="/work/direct/", response_class=HTMLResponse)
+async def work_direct(
+    request: Request,
+    user: Annotated[User, Depends(get_authenticated_user)],
+):
+    """
+    View for direct work (not on AMT).
+    Makes sure user is authenticated.
+    """
+    # todo: emit event
+    return HTMLResponse(BASE_HTML)