from __future__ import annotations

from flask import Flask, jsonify, request

from .logger import setup_logger
from .task_manager import TaskManager


def create_http_app(task_manager: TaskManager | None = None) -> Flask:
    app = Flask(__name__)
    logger = setup_logger()
    tm = task_manager or TaskManager()

    @app.get('/api/health')
    def health():
        return jsonify({'ok': True})

    @app.post('/api/task/receive')
    def receive_task():
        payload = request.get_json(silent=True) or {}
        task_id = tm.create_task(payload)
        logger.info('[任务] receive task_id=%s', task_id)
        return jsonify({'ok': True, 'task_id': task_id})

    @app.post('/api/task/cancel')
    def cancel_task():
        body = request.get_json(silent=True) or {}
        task_id = str(body.get('task_id', '')).strip()
        if not task_id:
            return jsonify({'ok': False, 'error': 'task_id required'}), 400
        ok = tm.cancel_task(task_id)
        return jsonify({'ok': ok, 'task_id': task_id})

    @app.get('/api/task/status/<task_id>')
    def task_status(task_id: str):
        task = tm.get_task(task_id)
        if not task:
            return jsonify({'ok': False, 'error': 'not found'}), 404
        return jsonify({'ok': True, 'task': task})

    @app.get('/api/task/list')
    def task_list():
        limit = int(request.args.get('limit', 100))
        return jsonify({'ok': True, 'tasks': tm.list_tasks(limit=limit)})

    return app


def run_http_server(host: str, port: int, task_manager: TaskManager | None = None) -> None:
    app = create_http_app(task_manager=task_manager)
    app.run(host=host, port=port, debug=False, use_reloader=False)