mirror of
https://github.com/smittix/intercept.git
synced 2026-07-07 09:08:12 -07:00
James Smith
96172ca593
First-time run of ruff-format via pre-commit hook normalises quote style, trailing commas, and whitespace across 188 Python files. No logic changes. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
133 lines
3.8 KiB
Python
133 lines
3.8 KiB
Python
"""
|
|
TSCM Case Management Routes
|
|
|
|
Handles /cases/* endpoints.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
|
|
from flask import jsonify, request
|
|
|
|
from routes.tscm import tscm_bp
|
|
|
|
logger = logging.getLogger("intercept.tscm")
|
|
|
|
|
|
@tscm_bp.route("/cases", methods=["GET"])
|
|
def list_cases():
|
|
"""List all TSCM cases."""
|
|
from utils.database import get_all_tscm_cases
|
|
|
|
status = request.args.get("status")
|
|
limit = request.args.get("limit", 50, type=int)
|
|
|
|
cases = get_all_tscm_cases(status=status, limit=limit)
|
|
|
|
return jsonify({"status": "success", "count": len(cases), "cases": cases})
|
|
|
|
|
|
@tscm_bp.route("/cases", methods=["POST"])
|
|
def create_case():
|
|
"""Create a new TSCM case."""
|
|
from utils.database import create_tscm_case
|
|
|
|
data = request.get_json() or {}
|
|
|
|
name = data.get("name")
|
|
if not name:
|
|
return jsonify({"status": "error", "message": "name is required"}), 400
|
|
|
|
case_id = create_tscm_case(
|
|
name=name,
|
|
description=data.get("description"),
|
|
location=data.get("location"),
|
|
priority=data.get("priority", "normal"),
|
|
created_by=data.get("created_by"),
|
|
metadata=data.get("metadata"),
|
|
)
|
|
|
|
return jsonify({"status": "success", "message": "Case created", "case_id": case_id})
|
|
|
|
|
|
@tscm_bp.route("/cases/<int:case_id>", methods=["GET"])
|
|
def get_case(case_id: int):
|
|
"""Get a TSCM case with all linked sweeps, threats, and notes."""
|
|
from utils.database import get_tscm_case
|
|
|
|
case = get_tscm_case(case_id)
|
|
if not case:
|
|
return jsonify({"status": "error", "message": "Case not found"}), 404
|
|
|
|
return jsonify({"status": "success", "case": case})
|
|
|
|
|
|
@tscm_bp.route("/cases/<int:case_id>", methods=["PUT"])
|
|
def update_case(case_id: int):
|
|
"""Update a TSCM case."""
|
|
from utils.database import update_tscm_case
|
|
|
|
data = request.get_json() or {}
|
|
|
|
success = update_tscm_case(
|
|
case_id=case_id,
|
|
status=data.get("status"),
|
|
priority=data.get("priority"),
|
|
assigned_to=data.get("assigned_to"),
|
|
notes=data.get("notes"),
|
|
)
|
|
|
|
if not success:
|
|
return jsonify({"status": "error", "message": "Case not found"}), 404
|
|
|
|
return jsonify({"status": "success", "message": "Case updated"})
|
|
|
|
|
|
@tscm_bp.route("/cases/<int:case_id>/sweeps/<int:sweep_id>", methods=["POST"])
|
|
def link_sweep_to_case(case_id: int, sweep_id: int):
|
|
"""Link a sweep to a case."""
|
|
from utils.database import add_sweep_to_case
|
|
|
|
success = add_sweep_to_case(case_id, sweep_id)
|
|
|
|
return jsonify(
|
|
{
|
|
"status": "success" if success else "error",
|
|
"message": "Sweep linked to case" if success else "Already linked or not found",
|
|
}
|
|
)
|
|
|
|
|
|
@tscm_bp.route("/cases/<int:case_id>/threats/<int:threat_id>", methods=["POST"])
|
|
def link_threat_to_case(case_id: int, threat_id: int):
|
|
"""Link a threat to a case."""
|
|
from utils.database import add_threat_to_case
|
|
|
|
success = add_threat_to_case(case_id, threat_id)
|
|
|
|
return jsonify(
|
|
{
|
|
"status": "success" if success else "error",
|
|
"message": "Threat linked to case" if success else "Already linked or not found",
|
|
}
|
|
)
|
|
|
|
|
|
@tscm_bp.route("/cases/<int:case_id>/notes", methods=["POST"])
|
|
def add_note_to_case(case_id: int):
|
|
"""Add a note to a case."""
|
|
from utils.database import add_case_note
|
|
|
|
data = request.get_json() or {}
|
|
|
|
content = data.get("content")
|
|
if not content:
|
|
return jsonify({"status": "error", "message": "content is required"}), 400
|
|
|
|
note_id = add_case_note(
|
|
case_id=case_id, content=content, note_type=data.get("note_type", "general"), created_by=data.get("created_by")
|
|
)
|
|
|
|
return jsonify({"status": "success", "message": "Note added", "note_id": note_id})
|