from langchain_core.tools import tool
import requests

BASE_URL = "https://cxx0822.s.3q.hair/home-api/"
USERNAME = "Cxx0822"
PASSWORD = "19940822Cxx"


def get_token() -> str:
    """
    调用登录接口，返回 saToken.tokenValue
    """
    params = {
        "name": USERNAME,
        "password": PASSWORD
    }

    try:
        resp = requests.post(f"{BASE_URL}session", params=params, timeout=10)
        resp.raise_for_status()
        data = resp.json()

        # 按你给的返回结构取值
        token_value = data["saToken"]["tokenValue"]
        return token_value

    except requests.exceptions.RequestException as e:
        raise RuntimeError(f"登录失败，无法获取 token: {e}")
    except KeyError as e:
        raise RuntimeError(f"登录响应结构异常，未找到 tokenValue: {e}")


@tool
def bill_search(start_date: str, end_date: str) -> str:
    """
    查询账单列表。

    参数说明：
    - start_date (必填):
        - 开始日期，格式：YYYY-MM-DD
        - 示例：2026-06-01
    - end_date (必填):
        - 结束日期，格式：YYYY-MM-DD
        - 示例：2026-07-01
    """

    print(f"start_date: {start_date}, end_date: {end_date}")

    params = {
        "bookName": "",
        "type": "",
        "startDate": start_date,
        "endDate": end_date
    }

    token = get_token()

    headers = {
        "satoken": token
    }

    try:
        resp = requests.get(f"{BASE_URL}bill/summary", params=params, headers=headers, timeout=10)
        resp.raise_for_status()
        return resp.text
    except Exception as e:
        return f"查询账单失败: {e}"