Added rejection support for the rest of the candidate graph

src/repo_registry/web_ui/views.py

@@ -330,6 +330,72 @@ def reject_candidate_ability_from_form(
     )
 
 
+@router.post(
+    "/ui/repos/{repository_id}/analysis-runs/{analysis_run_id}"
+    "/candidate-capabilities/{candidate_capability_id}/reject"
+)
+def reject_candidate_capability_from_form(
+    repository_id: int,
+    analysis_run_id: int,
+    candidate_capability_id: int,
+    service: RegistryService = Depends(get_service),
+) -> RedirectResponse:
+    service.reject_candidate_capability(
+        repository_id,
+        analysis_run_id,
+        candidate_capability_id,
+        notes="Rejected from web UI",
+    )
+    return RedirectResponse(
+        f"/ui/repos/{repository_id}/analysis-runs/{analysis_run_id}",
+        status_code=303,
+    )
+
+
+@router.post(
+    "/ui/repos/{repository_id}/analysis-runs/{analysis_run_id}"
+    "/candidate-features/{candidate_feature_id}/reject"
+)
+def reject_candidate_feature_from_form(
+    repository_id: int,
+    analysis_run_id: int,
+    candidate_feature_id: int,
+    service: RegistryService = Depends(get_service),
+) -> RedirectResponse:
+    service.reject_candidate_feature(
+        repository_id,
+        analysis_run_id,
+        candidate_feature_id,
+        notes="Rejected from web UI",
+    )
+    return RedirectResponse(
+        f"/ui/repos/{repository_id}/analysis-runs/{analysis_run_id}",
+        status_code=303,
+    )
+
+
+@router.post(
+    "/ui/repos/{repository_id}/analysis-runs/{analysis_run_id}"
+    "/candidate-evidence/{candidate_evidence_id}/reject"
+)
+def reject_candidate_evidence_from_form(
+    repository_id: int,
+    analysis_run_id: int,
+    candidate_evidence_id: int,
+    service: RegistryService = Depends(get_service),
+) -> RedirectResponse:
+    service.reject_candidate_evidence(
+        repository_id,
+        analysis_run_id,
+        candidate_evidence_id,
+        notes="Rejected from web UI",
+    )
+    return RedirectResponse(
+        f"/ui/repos/{repository_id}/analysis-runs/{analysis_run_id}",
+        status_code=303,
+    )
+
+
 @router.post(
     "/ui/repos/{repository_id}/analysis-runs/{analysis_run_id}"
     "/candidate-abilities/{candidate_ability_id}/edit"
@@ -460,11 +526,11 @@ def render_candidate_capability(
     analysis_run_id: int,
 ) -> str:
     features = "".join(
-        f'<li>{escape(feature["name"])} <span class="pill">{escape(feature["type"])}</span> <span class="source">{escape(feature["location"])}</span></li>'
+        render_candidate_feature(feature, repository_id, analysis_run_id)
         for feature in capability["features"]
     )
     evidence = "".join(
-        f'<li>{escape(item["type"])} <span class="pill">{escape(item["strength"])}</span> <span class="source">{escape(item["reference"])}</span></li>'
+        render_candidate_evidence(item, repository_id, analysis_run_id)
         for item in capability["evidence"]
     )
     return f"""
@@ -472,6 +538,7 @@ def render_candidate_capability(
       <strong>{escape(capability['name'])}</strong>
       <span class="pill">{escape(capability['status'])}</span>
       <span class="pill">{capability['confidence']:.2f}</span>
+      {render_candidate_reject_form('candidate-capabilities', capability, repository_id, analysis_run_id)}
       <p class="muted">{escape(capability['description'])}</p>
       {render_candidate_edit_form('candidate-capabilities', capability, repository_id, analysis_run_id)}
       {render_sources(capability['source_refs'])}
@@ -483,6 +550,57 @@ def render_candidate_capability(
     """
 
 
+def render_candidate_feature(
+    feature: dict,
+    repository_id: int,
+    analysis_run_id: int,
+) -> str:
+    return f"""
+    <li>
+      {escape(feature["name"])}
+      <span class="pill">{escape(feature["status"])}</span>
+      <span class="pill">{escape(feature["type"])}</span>
+      <span class="source">{escape(feature["location"])}</span>
+      {render_candidate_reject_form('candidate-features', feature, repository_id, analysis_run_id)}
+    </li>
+    """
+
+
+def render_candidate_evidence(
+    evidence: dict,
+    repository_id: int,
+    analysis_run_id: int,
+) -> str:
+    return f"""
+    <li>
+      {escape(evidence["type"])}
+      <span class="pill">{escape(evidence["status"])}</span>
+      <span class="pill">{escape(evidence["strength"])}</span>
+      <span class="source">{escape(evidence["reference"])}</span>
+      {render_candidate_reject_form('candidate-evidence', evidence, repository_id, analysis_run_id)}
+    </li>
+    """
+
+
+def render_candidate_reject_form(
+    collection: str,
+    candidate: dict,
+    repository_id: int,
+    analysis_run_id: int,
+) -> str:
+    if candidate["status"] != "candidate":
+        return ""
+    action = (
+        f"/ui/repos/{repository_id}/analysis-runs/{analysis_run_id}"
+        f"/{collection}/{candidate['id']}/reject"
+    )
+    return f"""
+    <form style="display:inline" method="post" action="{action}">
+      <button class="secondary" type="submit">Reject</button>
+    </form>
+    """
+
+
 def render_ability_map(ability_map: dict) -> str:
     abilities = ability_map.get("abilities", [])
     if not abilities: