from flask import request, jsonify
from app.utils import make_response
from pathlib import Path
import uuid
import threading
import signal
import os
import json

# 全局状态（模块内部）
task_running = {}
GLOBAL_ASR_SERVICE = None
GLOBAL_DIAR_SERVICE = None
ASR_MODEL_LOADED = False
DIAR_MODEL_LOADED = False
ASR_MODEL_LOCK = threading.Lock()
DIAR_MODEL_LOCK = threading.Lock()

def register_asr_routes(app):
    """【模块唯一入口】只注册路由，不创建app"""

    @app.route('/api/recognize', methods=['GET'])
    def recognize():
        task_id = str(uuid.uuid4())
        task_running[task_id] = True
        try:
            path = request.args.get('path', '')
            if not path:
                return jsonify(make_response(status="error", message="缺少path参数")), 400

            server_dir = Path(__file__).parent.parent.parent.absolute()
            os.chdir(server_dir)

            def timeout_handler(signum, frame):
                task_running[task_id] = False
                raise TimeoutError(f"超时 {app.config['TASK_TIMEOUT']}s")

            use_alarm = False
            try:
                signal.signal(signal.SIGALRM, timeout_handler) # pyright: ignore
                signal.alarm(app.config['TASK_TIMEOUT']) # pyright: ignore
                use_alarm = True
            except:
                pass

            try:
                from app.asr.core import main
                result = main(path)

                # 检查 main() 的返回值，如果是错误信息，返回 400
                if result and isinstance(result, str):
                    task_running[task_id] = False
                    return jsonify(make_response(status="error", message=result)), 400

            finally:
                if use_alarm:
                    signal.alarm(0) # pyright: ignore
                task_running[task_id] = False

            return jsonify(make_response(message="推理完成", data={"task_id": task_id, "path": path}))

        except Exception as e:
            task_running[task_id] = False
            return jsonify(make_response(status="error", message=str(e))), 500

    @app.route('/api/result', methods=['GET'])
    def result():
        path = request.args.get('path', '')
        if not path:
            return jsonify(make_response(status="error", message="缺少path")), 400
        result_file = Path(app.config['OUTPUT_DIR']) / "SpeechRecognition" / f"{Path(path).stem}_result.json"
        if not result_file.exists():
            return jsonify(make_response(status="error", message="结果不存在")), 404
        with open(result_file, 'r', encoding='utf-8') as f:
            data = json.load(f)
        return jsonify(make_response(data=data))