# from maix import app, key, uart, pinmap, time
# import hashlib
# import hmac
# import ujson

# # 配置 UART2（用于 HTTP 上传）
# pinmap.set_pin_function("A29", "UART2_RX")
# pinmap.set_pin_function("A28", "UART2_TX")
# http_serial = uart.UART("/dev/ttyS2", 115200, uart.BITS.BITS_8,
#                         uart.PARITY.PARITY_NONE, uart.STOP.STOP_1)

# # 按键初始化
# key_triggered = False
# def on_key_event(key_id, state):
#     global key_triggered
#     if state == key.State.KEY_PRESSED:
#         key_triggered = True
# key_obj = key.Key(on_key_event)

# # Token生成
# def generate_token(device_id):
#     SALT = "shootMessageFire"
#     SALT2 = "shoot"
#     return "Arrow_" + hmac.new((SALT + device_id).encode(), SALT2.encode(), hashlib.sha256).hexdigest()

# # 发送AT命令
# http_instance_id = -1
# def send_cmd(cmd_str, expect_create_id=False):
#     global http_instance_id
#     print("[AT指令] =>", cmd_str)
#     http_serial.write((cmd_str + "\r\n").encode())
#     buffer = b""
#     start = time.ticks_ms()
#     while time.ticks_ms() - start < 3000:
#         data = http_serial.read(128)
#         if data:
#             buffer += data
#             try:
#                 decoded = buffer.decode()
#                 print("返回:", decoded.strip())
#                 if expect_create_id and "+MHTTPCREATE:" in decoded:
#                     http_instance_id = int(decoded.split(":")[1].split("\r")[0].strip())
#                 if "OK" in decoded: return True
#                 if "+CME ERROR" in decoded or "ERROR" in decoded: return False
#             except:
#                 print("解码异常")
#         time.sleep_ms(10)
#     return False

# def create_http_instance(url):
#     return send_cmd(f'AT+MHTTPCREATE="{url}"', True) and http_instance_id != -1

# def send_data(instance_id, token, api_path, json_data):
#     # 设置Header（使用统一的AT+MHTTPCFG="header"）
#     send_cmd(f'AT+MHTTPCFG="header",{instance_id},"Content-Type: application/json"')
#     send_cmd(f'AT+MHTTPCFG="header",{instance_id},"Authorization: {token}"')
#     send_cmd(f'AT+MHTTPCFG="header",{instance_id},"DeviceId: {device_id}"')
    
#     # 发送Body数据
#     json_str = ujson.dumps(json_data)
#     send_cmd(f'AT+MHTTPCONTENT={instance_id},0,0,"{json_str}"')
#     send_cmd(f'AT+MHTTPREQUEST={instance_id},2,0,"{api_path}"')

# def read_response(timeout_ms=5000):
#     start = time.ticks_ms()
#     while time.ticks_ms() - start < timeout_ms:
#         data = http_serial.read(128)
#         if data:
#             try:
#                 print("响应:", data.decode("utf-8").strip())
#             except:
#                 print("响应(raw):", data)
#         time.sleep_ms(50)

# # 参数配置
# device_id = "wZhC7kAZ" #需要根据实际硬件ID来测试
# url = "http://ws.shelingxingqiu.com"
# api_path = "/home/shoot/device_fire/arrow/fire"
# token = generate_token(device_id)
# print("生成Token:", token)

# # 主循环：仅监听按键并上传模拟数据
# while not app.need_exit():
#     if key_triggered:
#         key_triggered = False
#         print("按键按下，准备上传...")

#         # 模拟数据（可替换为实际测量值）
#         timestamp = int(time.time() * 1000)
#         json_data = {
#             "id": timestamp,
#             "DeviceId": device_id,
#             "x": 12.34,
#             "y": -5.67,
#             "rag": 90.0,
#             "time": timestamp,
#             "dst": 234.5,
#             "battery": 80,
#             "errorCode": 0
#         }

#         if http_instance_id == -1 and not create_http_instance(url):
#             print("创建 HTTP 实例失败")
#         elif http_instance_id != -1:
#             send_data(http_instance_id, token, api_path, json_data)
#             read_response()
#     else:
#         time.sleep_ms(100)
from maix import app, uart, pinmap, time
import hashlib
import hmac
import re
import ujson

# ========== 配置 ==========
# UART2 for HTTP
pinmap.set_pin_function("A29", "UART2_RX")
pinmap.set_pin_function("A28", "UART2_TX")
http_serial = uart.UART("/dev/ttyS2", 115200, uart.BITS.BITS_8,
                        uart.PARITY.PARITY_NONE, uart.STOP.STOP_1)

# 设备参数
device_id = "wZhC7kAZ"
url = "http://ws.shelingxingqiu.com"
api_path = "/home/shoot/device_fire/arrow/fire"

# ========== 工具函数 ==========
def generate_token(device_id):
    SALT = "shootMessageFire"
    SALT2 = "shoot"
    return "Arrow_" + hmac.new((SALT + device_id).encode(), SALT2.encode(), hashlib.sha256).hexdigest()

def send_cmd(cmd_str, timeout_ms=3000):
    """发送 AT 指令并返回完整响应；超时返回已收到的内容。"""
    print("[AT] =>", cmd_str)
    http_serial.write((cmd_str + "\r\n").encode())
    buffer = b""
    start = time.ticks_ms()
    while time.ticks_diff(time.ticks_ms(), start) < timeout_ms:
        data = http_serial.read(128)
        if data:
            buffer += data
            try:
                decoded = buffer.decode("utf-8", "ignore")
                if "OK" in decoded or "+CME ERROR" in decoded or "ERROR" in decoded:
                    print("[AT] <=", decoded.strip())
                    return decoded
            except:
                pass
        time.sleep_ms(10)
    decoded = buffer.decode("utf-8", "ignore")
    print("[AT] !! timeout", timeout_ms, "ms, response:", decoded.strip() or "<empty>")
    return decoded


def response_ok(response):
    return "OK" in response and "ERROR" not in response


def wait_modem_ready():
    """等待模组响应，并确认 PDP 上下文已经获得 IP。"""
    for attempt in range(15):
        if response_ok(send_cmd("AT", 1000)):
            break
        print("[4G] 等待模组启动", attempt + 1, "/15")
        time.sleep_ms(1000)
    else:
        print("[4G] UART2 无 AT 响应，请检查模组供电、A28/A29 接线和串口占用")
        return False

    send_cmd("ATE0", 1000)
    cpin = send_cmd("AT+CPIN?", 3000)
    if "READY" not in cpin:
        print("[4G] SIM 卡未就绪:", cpin.strip())
        return False

    addr = send_cmd("AT+CGPADDR=1", 3000)
    match = re.search(r'\+CGPADDR:\s*1,"([^\"]+)"', addr)
    if match and match.group(1) != "0.0.0.0":
        print("[4G] PDP ready, IP:", match.group(1))
        return True

    send_cmd("AT+MIPCALL=1,1", 15000)
    for _ in range(20):
        addr = send_cmd("AT+CGPADDR=1", 3000)
        match = re.search(r'\+CGPADDR:\s*1,"([^\"]+)"', addr)
        if match and match.group(1) != "0.0.0.0":
            print("[4G] PDP ready, IP:", match.group(1))
            return True
        time.sleep_ms(1000)

    print("[4G] PDP 未获得 IP，请检查 SIM 流量、信号和 APN")
    return False


def clear_http_instances():
    for instance_id in range(6):
        send_cmd(f"AT+MHTTPDEL={instance_id}", 1200)

def create_http_instance(url):
    cmd = f'AT+MHTTPCREATE="{url}"'
    response = send_cmd(cmd, 8000)
    match = re.search(r"\+MHTTPCREATE:\s*(\d+)", response)
    if not response_ok(response) or not match:
        print("❌ 创建 HTTP 实例失败，模组响应:", response.strip() or "<empty>")
        return None
    return int(match.group(1))

def send_http_request(url, api_path, token, device_id, json_data):
    # 1. 创建 HTTP 实例
    instance_id = create_http_instance(url)
    if instance_id is None:
        return False

    # 2. 设置 Headers
    commands = (
        f'AT+MHTTPCFG="header",{instance_id},"Content-Type: application/json"',
        f'AT+MHTTPCFG="header",{instance_id},"Authorization: {token}"',
        f'AT+MHTTPCFG="header",{instance_id},"DeviceId: {device_id}"',
    )
    for command in commands:
        if not response_ok(send_cmd(command)):
            print("❌ HTTP Header 配置失败")
            send_cmd(f"AT+MHTTPDEL={instance_id}", 2000)
            return False

    # 3. 发送 Body
    json_str = ujson.dumps(json_data)
    at_json = json_str.replace("\\", "\\\\").replace('"', '\\"')
    if not response_ok(send_cmd(f'AT+MHTTPCONTENT={instance_id},0,0,"{at_json}"', 8000)):
        print("❌ HTTP Body 配置失败")
        send_cmd(f"AT+MHTTPDEL={instance_id}", 2000)
        return False

    # 4. 发起 POST 请求
    if response_ok(send_cmd(f'AT+MHTTPREQUEST={instance_id},2,0,"{api_path}"', 15000)):
        print("✅ HTTP 请求已发送")
        return True
    else:
        print("❌ 发送请求失败")
        return False

def read_response(timeout_ms=5000):
    print("⏳ 等待响应...")
    start = time.ticks_ms()
    while time.ticks_ms() - start < timeout_ms:
        data = http_serial.read(128)
        if data:
            try:
                print("📡 响应:", data.decode("utf-8").strip())
            except:
                print("📡 响应(raw):", data)
        time.sleep_ms(100)

# ========== 主程序：直接上传 ==========
print("🚀 启动直接上传流程...")

token = generate_token(device_id)
print("🔑 Token 已生成:", token[:12] + "...")

# 构造模拟数据
timestamp = int(time.time() * 1000)
json_data = {
    "id": timestamp,
    "DeviceId": device_id,
    "x": 12.34,
    "y": -5.67,
    "rag": 90.0,
    "time": timestamp,
    "dst": 234.5,
    "battery": 80,
    "errorCode": 0
}

# 执行上传
upload_ok = False
if not wait_modem_ready():
    print("💥 4G 模组未就绪")
else:
    clear_http_instances()
    upload_ok = send_http_request(url, api_path, token, device_id, json_data)

if upload_ok:
    read_response()
else:
    print("💥 上传流程失败")

print("🔚 程序结束")