#!/usr/bin/env python3
"""
清关文件上传 API
按照 form-data 方式提交两个字段：
 - params: JSON 字符串（包含 code、slipId、uid 等）
 - fileUpload: PDF 文件
"""

import os
import time
import json
import requests
from typing import Optional

from .api_config import API_CONFIG


def upload_clearance_file(
    code: str,
    slip_id: int,
    pdf_path: str,
    file_index: int = 0,
    uid: Optional[int] = None,
    Authorization: Optional[str] = None,
) -> str:
    """根据运单ID上传清关文件（PDF）。

    Args:
        code: 运单编号（必填）
        slip_id: 创建运单返回的ID（必填）
        pdf_path: 本地 PDF 文件路径（必填）
        file_index: 文件索引，默认 0
        uid: 文件 uid，可不传，默认使用时间戳生成
        Authorization: 授权令牌，如果提供则使用该值，否则使用 API_CONFIG 中的默认值

    Returns:
        文本：成功/失败信息。
    """
    if not code:
        return "参数错误: code 不能为空"
    if not slip_id and slip_id != 0:
        return "参数错误: slip_id 不能为空"
    if not os.path.isfile(pdf_path):
        return f"参数错误: 文件不存在 - {pdf_path}"

    url = f"{API_CONFIG['base_url']}/Exp/manager-server/attachmentNew/upload/paperless/clean"

    # 生成 uid
    real_uid = uid if isinstance(uid, int) else int(time.time() * 1000)

    params_payload = {
        "uploadType": "other",
        "code": code,
        "fileIndex": file_index,
        "fileType": "picType_1",
        "inputType": 0,
        "slipId": slip_id,
        "slipType": "slipType_chm_pdf_od",
        "fileUpload": [{"uid": real_uid}],
        "fileSizeTotal": 18425,
        "splitSuccess": 0,
        "needBackSplit": 0,
        "splitPicTotal": 0,
        "item": [],
    }

    # form-data: params 是 JSON 字符串，fileUpload 是文件
    data = {
        "params": json.dumps(params_payload, ensure_ascii=False),
    }

    files = {
        "fileUpload": (os.path.basename(pdf_path), open(pdf_path, "rb"), "application/pdf"),
    }

    # 构建 headers，优先使用传入的 Authorization
    headers = dict(API_CONFIG.get("headers", {}))
    if Authorization:
        headers['Authorization'] = Authorization
    # 移除 Content-Type（由 requests 根据 multipart 自动设置）
    headers.pop("Content-Type", None)

    try:
        resp = requests.post(url, headers=headers, data=data, files=files, timeout=120)
        # 确保文件句柄尽快关闭
        try:
            files["fileUpload"][1].close()
        except Exception:
            pass

        if resp.status_code == 200:
            return (
                f"上传清关文件成功\n"
                f"运单号：{code}\n"
                f"slipId：{slip_id}\n"
                f"uid：{real_uid}"
            )
        return f"上传失败: HTTP {resp.status_code}, 错误信息: {resp.text}"

    except requests.exceptions.RequestException as e:
        return f"网络请求失败: {str(e)}"
    except Exception as e:
        return f"上传清关文件失败: {str(e)}"


def upload_file_for_ocr(
    pdf_path: str,
    Authorization: Optional[str] = None,
) -> str:
    """上传文件图文识别
    
    只需要提供文件路径（pdf_path、Authorization），上传时文件路径不用确认，路径肯定是完整的，文件肯定是存在的，请直接调用工具接口。

    Args:
        pdf_path: 本地 PDF 文件路径（必填）
        Authorization: 授权令牌，如果提供则使用该值，否则使用 API_CONFIG 中的默认值

    Returns:
        返回 JSON 字符串，包含 status、message、chmCode（运单编号）、chmId（运单ID）。这些参数可以用于后续的 create_waybill_d_with_id 工具调用
    """
    if not os.path.isfile(pdf_path):
        return f"参数错误: 文件不存在 - {pdf_path}"

    # 从文件路径提取文件名
    file_name = os.path.basename(pdf_path)

    # 构建URL（不包含查询参数）
    url = f"{API_CONFIG['base_url']}/API/ocrDemo/pushDemo"

    # 封装参数到 params_payload
    params_payload = {
        "fileName01": file_name,
        "fileNumber": 1
    }

    # form-data: params 是 JSON 字符串，fileUpload 是文件
    data = {
        "params": json.dumps(params_payload, ensure_ascii=False),
    }

    files = {
        "file": (file_name, open(pdf_path, "rb"), "application/pdf"),
    }

    # 构建 headers，优先使用传入的 Authorization
    headers = dict(API_CONFIG.get("headers", {}))
    if Authorization:
        headers['Authorization'] = Authorization
    # 移除 Content-Type（由 requests 根据 multipart 自动设置）
    headers.pop("Content-Type", None)

    try:
        resp = requests.post(url, headers=headers, data=data, files=files, timeout=120)
        # 确保文件句柄尽快关闭
        try:
            files["file"][1].close()
        except Exception:
            pass

        if resp.status_code == 200:
            # 解析响应JSON
            result = resp.json()

            # 提取指定字段，只返回 status、message、chmCode、chmId
            filtered_result = {}
            if "status" in result:
                filtered_result["status"] = result["status"]
            if "message" in result:
                filtered_result["message"] = result["message"]
            if "chmCode" in result:
                filtered_result["chmCode"] = result["chmCode"]
            if "chmId" in result:
                filtered_result["chmId"] = result["chmId"]

            # 如果 status=9000，返回提示信息
            status = result.get("status")
            if status == 90000 or status == "90000":
                chmCode = result.get("chmCode", "")
                chmId = result.get("chmId", "")
                return f"图文识别成功，请确认是否使用该运单编号：{chmCode} ，运单ID：{chmId}"

            # 返回过滤后的JSON字符串
            return json.dumps(filtered_result, ensure_ascii=False)
        return f"上传失败: HTTP {resp.status_code}, 错误信息: {resp.text}"

    except requests.exceptions.RequestException as e:
        return f"网络请求失败: {str(e)}"
    except Exception as e:
        return f"上传文件进行OCR识别失败: {str(e)}"