from pathlib import Path
from datetime import datetime,date
from zoneinfo import ZoneInfo

import aiofiles
import base64
import os

from astrbot import logger

async def load_template(template_name: str) -> str:
    """
    异步加载模板内容（非阻塞）
    """
    plugin_dir = Path(__file__).parent.parent
    template_path = plugin_dir / "templates" / template_name

    if not template_path.exists():
        raise FileNotFoundError(f"模板文件不存在: {template_path}")

    async with aiofiles.open(template_path, "r", encoding="utf-8") as f:
        return await f.read()
    

def gold_to_string(gold_amount):
    """
    将金钱数值转换为字符串表示形式

    Args:
        gold_amount (int): 金钱数值，单位为铜币

    Returns:
        str: 格式化后的金钱字符串，例如 "1金2银3铜"
    """
    if not gold_amount:
        return "无价格"

    parts = []
    started = False  # 标记是否已经遇到第一个非零位

    bricks = gold_amount // 100000000
    gold = (gold_amount % 100000000) // 10000
    silver = (gold_amount % 10000) // 100
    copper = gold_amount % 100

    for value, unit in [(bricks, "砖"), (gold, "金"), (silver, "银"), (copper, "铜")]:
        if value != 0:
            started = True
        if started:
            parts.append(f"{value}{unit}")

    return "".join(parts)


def gold_to_parts(gold_amount):
    """将铜钱数拆成模板可渲染的货币片段"""
    try:
        amount = int(gold_amount or 0)
    except (TypeError, ValueError):
        amount = 0

    if amount <= 0:
        return []

    bricks = amount // 100000000
    gold = (amount % 100000000) // 10000
    silver = (amount % 10000) // 100
    copper = amount % 100

    parts = []
    started = False
    for key, name, value in [
        ("zhuang", "砖", bricks),
        ("jin", "金", gold),
        ("yin", "银", silver),
        ("tong", "铜", copper),
    ]:
        if value != 0:
            started = True
        if started:
            parts.append({"key": key, "name": name, "value": value})

    return parts


def week_to_num(week :str):
    week_map = {
    "一": 0,
    "二": 1,
    "三": 2,
    "四": 3,
    "五": 4,
    "六": 5,
    "日": 6,  
    }
    return week_map.get(week,None)


def compare_date_str(date_str: str) -> str:
    """
    date_str 格式：YYYY-MM-DD
    """
    d = datetime.strptime(date_str, "%Y-%m-%d").date()
    today = date.today()

    if d < today:
        return "过去"
    elif d == today:
        return "今天"
    else:
        return "将来"


def load_as_base64(icons_dir: str) -> dict[str, str]:
    """
    将指定目录下的图标文件转换为 base64 字符串。
    返回 {"文件名（不含扩展名）": "data:image/xxx;base64,xxx"} 的字典。
    """
    supported_ext = {".png", ".jpg", ".jpeg", ".gif", ".webp", ".svg", ".ico"}
    icons = {}

    for filename in os.listdir(icons_dir):
        name, ext = os.path.splitext(filename)
        if ext.lower() not in supported_ext:
            continue

        filepath = os.path.join(icons_dir, filename)
        with open(filepath, "rb") as f:
            data = base64.b64encode(f.read()).decode()

        display_name = name
        try:
            display_name = name.encode("latin-1").decode("gbk")
        except (UnicodeEncodeError, UnicodeDecodeError):
            pass

        if ext.lower() == ".svg":
            mime = "image/svg+xml"
        elif ext.lower() == ".jpg":
            mime = "image/jpeg"
        else:
            mime = f"image/{ext.lower().lstrip('.')}"

        icons[display_name] = f"data:{mime};base64,{data}"

    return icons


def format_time(ts):
    try:
        return datetime.fromtimestamp(int(ts),
            tz=ZoneInfo("Asia/Shanghai")
            ).strftime("%Y-%m-%d %H:%M:%S")
    except (TypeError, ValueError, OSError):
        return ""

def format_remaining(ts):
    try:
        seconds = max(0, int(ts) - int(datetime.now().timestamp()))
        hours, seconds = divmod(seconds, 3600)
        minutes, seconds = divmod(seconds, 60)
        return f"{hours}时{minutes:02d}分{seconds:02d}秒"
    except (TypeError, ValueError):
        return ""