"""测试操作 API"""

from flask import Blueprint, jsonify, render_template, request
from flask_login import login_required, current_user
from app.models import (
    get_device_by_id,
    insert_serialnet,
    get_serialnet_stats,
    get_latest_test_state,
    get_automation_averages,
    get_automation_records,
    get_latest_wave_data,
    get_wave_records,
    clear_serialnet_records,
    insert_log,
)

bp = Blueprint("test_op", __name__)

# DG430 指令 (addr=0x01, ADDR=0x81)
COMMANDS = {
    "B0": "7F8101B03032",  # 开始测试
    "B1": "7F8101B13133",  # 测试复原
    "BA": "7F8101BA3A3C",  # 电机前进
    "BB": "7F8101BB3B3D",  # 电机后退
    "BC": "7F8101BC3C3E",  # 电机停止
}

CMD_NAMES = {
    "B0": "开始测试",
    "B1": "测试复原",
    "BA": "电机前进",
    "BB": "电机后退",
    "BC": "电机停止",
}


@bp.route("/test/<int:dnt_id>")
@login_required
def test_page(dnt_id):
    """测试操作页面"""
    device = get_device_by_id(dnt_id)
    if not device:
        return "设备不存在", 404
    return render_template("test_op.html", device=device)


@bp.route("/api/command", methods=["POST"])
@login_required
def api_command():
    """发送单次指令"""
    data = request.get_json()
    dnt_id = data.get("dnt_id")
    cmd = data.get("cmd", "").upper()

    if cmd not in COMMANDS:
        return jsonify({"ok": False, "error": f"未知指令: {cmd}"}), 400

    device = get_device_by_id(dnt_id)
    target = f"{device['serial']}" if device else f"dnt_id={dnt_id}"

    send_pkg = COMMANDS[cmd]
    cmd_name = CMD_NAMES.get(cmd, cmd)
    try:
        record_id = insert_serialnet(dnt_id, send_pkg)
        insert_log(
            current_user.id, current_user.username, "command",
            target=target,
            detail=f"{cmd_name}({cmd}) → {send_pkg}",
            result="ok",
            ip=request.remote_addr or "",
        )
        return jsonify({"ok": True, "record_id": record_id, "send_pkg": send_pkg})
    except Exception as e:
        insert_log(
            current_user.id, current_user.username, "command",
            target=target,
            detail=f"{cmd_name}({cmd}) 失败: {e}",
            result="error",
            ip=request.remote_addr or "",
        )
        return jsonify({"ok": False, "error": str(e)}), 500


@bp.route("/api/automation/start", methods=["POST"])
@login_required
def api_automation_start():
    """开始自动化测试"""
    data = request.get_json()
    dnt_id = data.get("dnt_id")
    count = int(data.get("count", 1))

    device = get_device_by_id(dnt_id)
    target = f"{device['serial']}" if device else f"dnt_id={dnt_id}"

    # 清除旧记录，然后插入第一条 0xB0
    clear_serialnet_records(dnt_id)
    record_id = insert_serialnet(dnt_id, COMMANDS["B0"])

    insert_log(
        current_user.id, current_user.username, "command",
        target=target,
        detail=f"自动化测试开始 ×{count} 次",
        result="ok",
        ip=request.remote_addr or "",
    )
    return jsonify({
        "ok": True,
        "total": count,
        "first_record_id": record_id,
    })


@bp.route("/api/automation/<int:dnt_id>/progress")
@login_required
def api_automation_progress(dnt_id):
    """获取自动化进度"""
    since = request.args.get("since", "", type=str)
    stats = get_serialnet_stats(dnt_id)
    latest = get_latest_test_state(dnt_id)
    averages = get_automation_averages(dnt_id, since if since else None)
    records = get_automation_records(dnt_id, since) if since else []
    latest_wave = get_latest_wave_data(dnt_id)
    wave_records = get_wave_records(dnt_id, since) if since else []
    return jsonify({
        "stats": stats,
        "latest": latest,
        "averages": averages,
        "records": records,
        "latest_wave": latest_wave,
        "wave_records": wave_records,
    })