# core/jx3jiaoyihang.py
from astrbot.api import logger
from urllib.parse import quote
from datetime import datetime

from .api_data import api_data_get, api_data_post


def fetch_all_pages(base_url, initial_params, max_pages=None):
    """
    获取所有分页数据
    
    Args:
        base_url: API基础URL
        initial_params: 初始请求参数
        max_pages: 最大页数限制（可选）
    
    Returns:
        list: 所有页面的数据列表
    """
    all_data = []
    current_page = 1
    
    while True:
        # 设置当前页码
        params = initial_params.copy()
        params["page"] = str(current_page)
        
        # 请求数据
        data = api_data_get(base_url, params)
        
        if not data or "list" not in data:
            break
            
        # 添加当前页数据到总列表
        all_data.extend(data["list"])
        
        # 检查是否还有更多页面
        total_pages = data.get("pages", 1)
        if current_page >= total_pages:
            break
            
        # 检查是否达到最大页数限制
        if max_pages and current_page >= max_pages:
            break
            
        current_page += 1
        
    return all_data


def merge_item_data_by_itemid(price_data, item_list_data):
    """
    根据ItemId和id字段将物品名称信息合并到价格数据中
    
    Args:
        price_data: 第一组数据，包含价格信息的字典
        item_list_data: 第二组数据，可以是字典（包含"list"键）或直接是物品列表
    
    Returns:
        dict: 合并后的数据，包含价格和名称信息
    """
    # 提取价格数据中的物品信息
    price_items = price_data.get("data", {})
    
    # 处理不同类型的item_list_data输入
    if isinstance(item_list_data, dict) and "list" in item_list_data:
        # 如果是字典且包含"list"键
        item_list = item_list_data.get("list", [])
    elif isinstance(item_list_data, list):
        # 如果直接是列表
        item_list = item_list_data
    else:
        # 其他情况，使用空列表
        item_list = []
    
    # 创建一个字典用于快速查找物品名称，使用id作为键
    item_name_map = {}
    for item in item_list:
        item_id = item.get("id")
        item_name = item.get("Name")
        if item_id and item_name:
            item_name_map[item_id] = item_name
    
    # 创建一个新的结果字典
    merged_data = {"code": price_data.get("code", 0), "msg": price_data.get("msg", ""), "data": {}}
    
    # 遍历价格数据，添加名称信息
    for item_id, price_info in price_items.items():
        # 复制价格信息
        merged_item = price_info.copy()
        
        # 使用ItemId字段查找对应的名称
        item_id_to_match = merged_item.get("ItemId")
        if item_id_to_match and item_id_to_match in item_name_map:
            merged_item["Name"] = item_name_map[item_id_to_match]
        else:
            merged_item["Name"] = "未知物品"
        
        # 添加到结果中
        merged_data["data"][item_id] = merged_item
    
    return merged_data


#交易行数据查询函数
def jx3_data_jiaoyihang(inserver="眉间雪", inname="武技殊影图"):
    """
    获取剑三交易行某区某物品价格数据
    
    Args:
        inserver: 第一组数据，服务器名称
        inname: 物品名称
    
    Returns:
        list: 合并后的数据，包含价格和名称信息
    """
    # 第一步：获取所有物品列表数据（处理分页）
    custom_url = f"https://node.jx3box.com/item_merged/name/{quote(inname)}"
    initial_params = {
        "client": "std",
        "strict": "0",
        "per": "50"  # 每页数量，可以根据需要调整
    }
    
    # 获取所有页面的物品数据
    all_items = fetch_all_pages(custom_url, initial_params)
    
    if not all_items:
        #logger.error(f"未找到物品: {inname}")
        return "未找到改物品"
    
    # 提取所有ID
    ids = [item.get("id") for item in all_items if item.get("id")]
    
    if not ids:
        #logger.error(f"物品 {inname} 没有有效的ID")
        return "未找到改物品"
    
    # 返回格式化结果
    stringids = ",".join(ids)
    logger.info(f"搜索到物品: {inname}, 共找到 {len(ids)} 个物品\nIDs: {stringids}")
    
    # 第二步：获取价格数据
    price_url = "https://next2.jx3box.com/api/item-price/list"
    price_params = {
        "server": inserver,
        "itemIds": stringids
    }
    
    price_data = api_data_get(price_url, price_params)
    
    if not price_data or "data" not in price_data:
        return "无交易行数据"
    
    if not price_data["data"] or (isinstance(price_data["data"], dict) and not price_data["data"]) or (isinstance(price_data["data"], list) and not price_data["data"]):
        return "无交易行数据"
    
    # 第三步：合并数据
    merged_data = merge_item_data_by_itemid(price_data, {"list": all_items})
    
    # 处理合并后的数据
    result_items = []
    for item_id, item_info in merged_data.get("data", {}).items():
        # 格式化价格（假设价格是以铜钱为单位，转换为金）
        lowest_price = item_info.get("LowestPrice", 0)
        bricks = lowest_price // 100000000 if lowest_price else 0
        gold = (lowest_price % 100000000) // 10000 if lowest_price else 0
        silver = (lowest_price % 10000) // 100 if lowest_price else 0
        copper = lowest_price % 100 if lowest_price else 0
        
        price_str = f"{bricks}砖{gold}金{silver}银{copper}铜" if lowest_price else "无价格"
        
        result_items.append({
            "name": item_info.get("Name", "未知物品"),
            "price": price_str,
            "avg_price": item_info.get("AvgPrice", 0),
            "sample_size": item_info.get("SampleSize", 0)
        })
    
    # 按价格排序
    result_items.sort(key=lambda x: x["avg_price"])
    
    return result_items
  

def jx3_data_wujia(inname="秃盒"):
    """
    获取剑三外观物品价格数据
    
    Args:
        inname: 物品名称
    
    Returns:
        dict: 包含价格和名称信息的字典
    """
    # 初始化返回数据结构
    datas = {
        "code": 0,
        "msg": "未获取数据",
        "data": {}
    }
    
    try:
        # 1. 获取物品ID和名称
        idname = api_data_get(inname)
        if not idname:
            datas.update({"code": 201, "msg": "未找到该外观信息"})
            return datas
            
        # 更新基础信息
        datas.update({
            "code": 1,
            "msg": "获取外观名称ID完成",
            "data": {
                "showName": idname.get("showName", "未知物品"),
                "searchId": idname.get("searchId", "未知ID"),
                "Name": inname
            }
        })
        
        # 2. 查询爱剑三获取外观数据
        aj3_data = get_aj3_goods_data(datas["data"]["showName"])
        if aj3_data:
            datas["data"].update(aj3_data)
            datas.update({"code": 2, "msg": "获取外观数据完成"})
        
        # 3. 查询万宝楼数据（公示和在售）
        wbl_data = get_wbl_data(idname['searchId'])
        if wbl_data:
            datas["data"].update(wbl_data)
            datas.update({"code": 200, "msg": "获取万宝楼数据完成"})
            
    except Exception as e:
        logger.error(f"物价查询出错: {e}")
        datas.update({"code": 500, "msg": f"查询过程中出错: {e}"})
    
    return datas


def get_aj3_goods_data(goods_name):
    """获取爱剑三商品数据"""
    try:
        aj3_url = "https://www.aijx3.cn/api2/aijx3-wj/goods/getGoodsDetail"
        aj3_params = {"goodsName": goods_name}
        
        dataaj3 = api_data_post(aj3_url, aj3_params, "data")
        if not dataaj3:
            return None
            
        imgs = dataaj3.get("imgs", [])
        return {
            "goodsDesc": dataaj3.get("goodsDesc", "无描述"),
            "publishTime": dataaj3.get("publishTime", "无数据"),
            "priceNum": dataaj3.get("priceNum", 0),
            "imgs": imgs[0] if imgs else "",
            "goodsId": dataaj3.get("goodsId", "无数据")
        }
    except Exception as e:
        logger.error(f"获取爱剑三数据出错: {e}")
        return None


def get_wbl_data(search_id):
    """获取万宝楼数据（公示和在售）"""
    try:
        wbl_url = "https://www.aijx3.cn/api2/aijx3-wblwg/record/queryByCondition"
        base_params = {
            "accoSeq": "",
            "orderMode": 1,
            "orderBy": "price_num",
            "searchId": [search_id],
            "current": 1,
            "size": 10
        }
        
        # 获取公示数据
        gs_params = base_params.copy()
        gs_params["tradeStatus"] = "3"
        datawblgs = api_data_post(wbl_url, gs_params, "data")
        
        # 获取在售数据
        zs_params = base_params.copy()
        zs_params["tradeStatus"] = "5"
        datawblzs = api_data_post(wbl_url, zs_params, "data")
        
        return {
            "wblgs": process_wbl_records(datawblgs.get("records", [])),
            "wblzs": process_wbl_records(datawblzs.get("records", []))
        }
    except Exception as e:
        logger.error(f"获取万宝楼数据出错: {e}")
        return None


def process_wbl_records(records):
    """处理万宝楼记录数据"""
    processed = []
    for record in records:
        try:
            # 转换时间戳为可读格式
            timestamp = record.get("replyTime", 0)
            dt = datetime.fromtimestamp(timestamp / 1000) if timestamp else datetime.now()
            
            processed.append({
                "priceNum": record.get("priceNum", 0),
                "belongQf2": record.get("belongQf2", "无数据"),
                "replyTime": dt.strftime("%Y-%m-%d %H:%M:%S"),
                "discountRate": record.get("discountRate", 0.0),
            })
        except Exception as e:
            logger.error(f"处理万宝楼记录出错: {e}")
            continue
    
    return processed