from runtime import Args
from typings.zhanan.zhanan import Input, Output
import requests
import json
"""
Each file needs to export a function named `handler`. This function is the entrance to the Tool.

Parameters:
args: parameters of the entry function.
args.input - input parameters, you can get test input value by args.input.xxx.
args.logger - logger instance used to print logs, injected by runtime.

Remember to fill in input/output in Metadata, it helps LLM to recognize and use tool.

Return:
The return data of the function, which should match the declared output parameters.
"""

def handler(args: Args[Input]) -> Output:
    """
    获取宅男内容 - 基于天聚数行API

    Args:
        args: 包含输入参数和日志记录器的对象
            - args.input.key: 天聚数行API密钥

    Returns:
        Output: 包含API响应数据的字典
    """
    # 获取API密钥
    api_key = args.input.key if hasattr(args.input, 'key') else ""

    # 记录日志
    args.logger.info("开始获取宅男内容")

    if not api_key:
        args.logger.error("缺少API密钥")
        return {
            "code": 400,
            "msg": "缺少API密钥",
            "newslist": []
        }

    try:
        # 使用requests发送POST请求
        url = "https://apis.tianapi.com/zhanan/index"
        data = {'key': api_key}

        response = requests.post(url, data=data, timeout=10)
        response.raise_for_status()

        # 解析JSON响应
        result_data = response.json()

        args.logger.info("成功获取宅男内容")

        # 返回API原始响应
        return result_data

    except Exception as e:
        args.logger.error(f"获取宅男内容失败: {str(e)}")
        return {
            "code": 500,
            "msg": f"请求失败: {str(e)}",
            "newslist": []
        }