"""
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
    })