import httpx
import json
import os
import re
import sentry_sdk
import random
import time
import yaml
import logging

# --- Logging Setup ---
if os.environ.get("MHYY_LOGLEVEL", "").upper() == "DEBUG":
    loglevel = logging.DEBUG
elif os.environ.get("MHYY_LOGLEVEL", "").upper() == "WARNING":
    loglevel = logging.WARNING
elif os.environ.get("MHYY_LOGLEVEL", "").upper() == "ERROR":
    loglevel = logging.ERROR
else:
    loglevel = logging.INFO

logging.basicConfig(
    level=loglevel,
    format="%(asctime)s [%(levelname)s]: %(message)s",
    datefmt="%Y-%m-%d %H:%M:%S",
)

logger = logging.getLogger()


# --- Config Reading Function ---
def ReadConf(variable_name, default_value=None):
    """
    Reads YAML configuration from environment variable or config.yml.
    Assumes the variable_name contains the full YAML content.
    """
    env_value = os.environ.get(variable_name)

    if env_value:
        try:
            # Attempt to load from environment variable (assuming it's YAML)
            config_data = yaml.load(env_value, Loader=yaml.FullLoader)
            logger.debug("Configuration loaded from environment variable.")
            return config_data
        except yaml.YAMLError as e:
            logger.error(
                f"Failed to parse YAML from environment variable '{variable_name}': {e}"
            )
            return default_value  # Return default or None if env parsing fails
        except Exception as e:
            logger.error(
                f"An unexpected error occurred reading environment variable '{variable_name}': {e}"
            )
            return default_value

    # If not found or failed in environment, try to read from config.yml
    try:
        with open("config.yml", "r", encoding="utf-8") as config_file:
            config_data = yaml.load(config_file, Loader=yaml.FullLoader)
            logger.debug("Configuration loaded from config.yml file.")
            return config_data
    except FileNotFoundError:
        logger.warning("config.yml not found.")
        return default_value
    except yaml.YAMLError as e:
        logger.error(f"Failed to parse YAML from config.yml: {e}")
        return default_value
    except Exception as e:
        logger.error(f"An unexpected error occurred reading config.yml: {e}")
        return default_value


# --- Sentry Setup ---
sentry_sdk.init(
    "https://425d7b4536f94c9fa540fe34dd6609a2@o361988.ingest.sentry.io/6352584",
    traces_sample_rate=1.0,
)

# --- Load Configuration ---
full_config = ReadConf("MHYY_CONFIG", {})  # Read the entire config
accounts_conf = full_config.get("accounts")
notification_settings = full_config.get(
    "notifications", {}
)  # Get notification settings, default to empty dict

if not accounts_conf:
    logger.error(
        "请正确配置环境变量 MHYY_CONFIG 或者 config.yml 并包含 'accounts' 部分后再运行本脚本！"
    )
    os._exit(0)
logger.info(f"检测到 {len(accounts_conf)} 个账号，正在进行任务……")


def send_notifications(message: str, settings: dict):
    """Sends message to configured notification services."""
    if not message or not settings:
        logger.debug("No message to send or no notification settings configured.")
        return

    logger.info("Attempting to send notifications...")

    # ServerChan (SCT)
    sct_conf = settings.get("serverchan", {})
    sct_key = sct_conf.get("key")
    if sct_key:
        sct_url = f"https://sctapi.ftqq.com/{sct_key}.send"
        try:
            payload = {"title": "MHYY-AutoCheckin 状态推送", "desp": message}
            response = httpx.get(sct_url, params=payload, timeout=10)
            response.raise_for_status()
            logger.info("ServerChan notification sent successfully.")
        except httpx.HTTPStatusError as e:
            logger.error(
                f"ServerChan HTTP error occurred: {e.response.status_code} - {e.response.text}"
            )
        except httpx.RequestError as e:
            logger.error(f"An error occurred while requesting ServerChan: {e}")
        except Exception as e:
            logger.error(
                f"An unexpected error occurred sending ServerChan notification: {e}"
            )
    else:
        logger.debug("ServerChan not configured.")

    # DingTalk
    dingtalk_conf = settings.get("dingtalk", {})
    dingtalk_webhook_url = dingtalk_conf.get("webhook_url")
    if dingtalk_webhook_url:
        try:
            payload = {"msgtype": "text", "text": {"content": message}}
            response = httpx.post(dingtalk_webhook_url, json=payload, timeout=10)
            response.raise_for_status()
            result = response.json()
            if result.get("errcode") == 0:
                logger.info("DingTalk notification sent successfully.")
            else:
                logger.error(
                    f"DingTalk error: {result.get('errcode')} - {result.get('errmsg')}"
                )
        except httpx.HTTPStatusError as e:
            logger.error(
                f"DingTalk HTTP error occurred: {e.response.status_code} - {e.response.text}"
            )
        except httpx.RequestError as e:
            logger.error(f"An error occurred while requesting DingTalk: {e}")
        except Exception as e:
            logger.error(
                f"An unexpected error occurred sending DingTalk notification: {e}"
            )
    else:
        logger.debug("DingTalk not configured.")

    telegram_conf = settings.get("telegram", {})
    telegram_bot_token = telegram_conf.get("bot_token")
    telegram_chat_id = telegram_conf.get("chat_id")
    if telegram_bot_token and telegram_chat_id:
        telegram_url = f"https://api.telegram.org/bot{telegram_bot_token}/sendMessage"
        try:
            # Telegram text message parameters
            params = {
                "chat_id": telegram_chat_id,
                "text": message,
                # Optional: parse_mode can be 'MarkdownV2', 'HTML', or None
                # For simplicity, sending as plain text. Be careful with special characters if using Markdown/HTML.
                # "parse_mode": "HTML"
            }
            response = httpx.get(telegram_url, params=params, timeout=10)
            response.raise_for_status()  # Raise an exception for bad status codes
            result = response.json()
            if result.get("ok"):
                logger.info("Telegram notification sent successfully.")
            else:
                logger.error(
                    f"Telegram error: {result.get('error_code')} - {result.get('description')}"
                )
        except httpx.HTTPStatusError as e:
            logger.error(
                f"Telegram HTTP error occurred: {e.response.status_code} - {e.response.text}"
            )
        except httpx.RequestError as e:
            logger.error(f"An error occurred while requesting Telegram: {e}")
        except Exception as e:
            logger.error(
                f"An unexpected error occurred sending Telegram notification: {e}"
            )
    else:
        logger.debug("Telegram not configured.")


class RunError(Exception):
    pass


if __name__ == "__main__":
    if not os.environ.get("MHYY_DEBUG", False):
        wait_time = random.randint(10, 11)  # Random Sleep to Avoid Ban
        logger.info(
            f"为了避免同一时间签到人数太多导致被官方怀疑，开始休眠 {wait_time} 秒"
        )
        time.sleep(wait_time)

    version = "5.0.0"  # Default version
    try:
        ver_info = httpx.get(
            "https://hyp-api.mihoyo.com/hyp/hyp-connect/api/getGameBranches?game_ids[]=1Z8W5NHUQb&launcher_id=jGHBHlcOq1",
            timeout=60,
            verify=False,
        ).text
        version = json.loads(ver_info)["data"]["game_branches"][0]["main"]["tag"]
        logger.info(f"从官方API获取到云·原神最新版本号：{version}")
    except Exception as e:
        logger.warning(f"获取版本号失败，使用默认版本：{version}. Error: {e}")

    for config in accounts_conf:
        notification_msg = "【MHYY】签到状态推送\n\n"  # Message container for the current account

        # 各种API的URL
        NotificationURL = "https://api-cloudgame.mihoyo.com/hk4e_cg_cn/gamer/api/listNotifications?status=NotificationStatusUnread&type=NotificationTypePopup&is_sort=true"
        WalletURL = "https://api-cloudgame.mihoyo.com/hk4e_cg_cn/wallet/wallet/get"
        AnnouncementURL = (
            "https://api-cloudgame.mihoyo.com/hk4e_cg_cn/gamer/api/getAnnouncementInfo"
        )

        # Validate account config entry
        if not isinstance(config, dict) or "token" not in config:
            error_msg = f"跳过无效的账号配置条目: {config}"
            logger.error(error_msg)
            notification_msg += error_msg + "\n"
            send_notifications(
                notification_msg, notification_settings
            )  # Notify about invalid config
            continue  # Skip this entry

        try:
            token = config["token"]
            client_type = config.get("type", 5)
            sysver = config.get("sysver", "14.0") 
            deviceid = config["deviceid"]
            devicename = config.get("devicename", "iPhone 13")
            devicemodel = config.get("devicemodel", "iPhone13,3")
            appid = config.get("appid", "1953439978")

            # Construct headers
            headers = {
                "x-rpc-combo_token": token,
                "x-rpc-client_type": str(client_type),
                "x-rpc-app_version": str(version),
                "x-rpc-sys_version": str(sysver),
                "x-rpc-channel": "cyydmihoyo",
                "x-rpc-device_id": deviceid,
                "x-rpc-device_name": devicename,
                "x-rpc-device_model": devicemodel,
                "x-rpc-vendor_id": "1",
                "x-rpc-cg_game_biz": "hk4e_cn",
                "x-rpc-op_biz": "clgm_cn",
                "x-rpc-language": "zh-cn",
                "Host": "api-cloudgame.mihoyo.com",
                "Connection": "Keep-Alive",
                "Accept-Encoding": "gzip",
                "User-Agent": f"Mozilla/5.0 (iPhone; CPU iPhone OS {sysver} like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Mobile/15E148",
            }

            bbsid_match = re.search(r"oi=(\d+)", token)
            bbsid = bbsid_match.group(1) if bbsid_match else "N/A"

            region = config.get("region", "cn")
            if region == "os":
                headers["x-rpc-channel"] = "mihoyo"
                headers["x-rpc-cg_game_biz"] = "hk4e_global"
                headers["x-rpc-op_biz"] = "clgm_global"
                headers["x-rpc-cg_game_id"] = "9000254"
                headers["x-rpc-app_id"] = "600493"
                headers["User-Agent"] = "okhttp/4.10.0"
                headers["Host"] = "sg-cg-api.hoyoverse.com"
                NotificationURL = "https://sg-cg-api.hoyoverse.com/hk4e_global/cg/gamer/api/listNotifications?status=NotificationStatusUnread&type=NotificationTypePopup&is_sort=true"
                WalletURL = (
                    "https://sg-cg-api.hoyoverse.com/hk4e_global/cg/wallet/wallet/get"
                )
                AnnouncementURL = "https://sg-cg-api.hoyoverse.com/hk4e_global/cg/gamer/api/getAnnouncementInfo"

            logger.info(
                f"--- 正在进行第 {accounts_conf.index(config) + 1} 个账号 (BBSID: {bbsid})，服务器为{'CN' if region != 'os' else 'GLOBAL'} ---"
            )
            notification_msg += (
                f"☁️ 云原神签到结果 ({'CN' if region != 'os' else 'GLOBAL'}):\n"
            )
            notification_msg += (
                f"账号 {accounts_conf.index(config) + 1} (BBSID: {bbsid})\n\n"
            )

            try:
                wallet_res = httpx.get(
                    WalletURL, headers=headers, timeout=30, verify=False
                )
                wallet_res.raise_for_status()
                wallet_data = wallet_res.json()
                logger.debug(f"Wallet response: {wallet_data}")

                if wallet_data.get("retcode") == -100:
                    error_msg = f"当前登录已过期，请重新登陆！返回为：{wallet_data.get('message', 'Unknown error')}"
                    logger.error(error_msg)
                    notification_msg += error_msg + "\n"
                elif wallet_data.get("retcode") == 0 and wallet_data.get("data"):
                    free_time = wallet_data["data"]["free_time"]["free_time"]
                    play_card_msg = wallet_data["data"]["play_card"]["short_msg"]
                    coin_num = wallet_data["data"]["coin"]["coin_num"]
                    coin_minutes = int(coin_num) / 10 if coin_num is not None else 0
                    wallet_status = f"✅ 钱包：免费时长 {free_time} 分钟，畅玩卡状态为 {play_card_msg}，拥有原点 {coin_num} 点 ({coin_minutes:.0f}分钟)\n"
                    logger.info(wallet_status.strip())
                    notification_msg += wallet_status
                else:
                    error_msg = f"获取钱包信息失败: {wallet_data.get('retcode')} - {wallet_data.get('message', 'Unknown error')}"
                    logger.error(error_msg)
                    notification_msg += error_msg + "\n"

            except httpx.HTTPStatusError as e:
                error_msg = f"获取钱包信息HTTP错误: {e.response.status_code} - {e.response.text}"
                logger.error(error_msg)
                notification_msg += error_msg + "\n"
            except httpx.RequestError as e:
                error_msg = f"请求钱包信息失败: {e}"
                logger.error(error_msg)
                notification_msg += error_msg + "\n"
            except Exception as e:
                error_msg = f"解析钱包信息出错: {e}"
                logger.error(error_msg)
                notification_msg += error_msg + "\n"

            # --- Check Sign-in Status ---
            try:
                announcement_res = httpx.get(
                    AnnouncementURL, headers=headers, timeout=30, verify=False
                )
                announcement_res.raise_for_status()
                # logger.debug(f'Announcement response: {announcement_res.text}') # Too verbose usually

                notification_res = httpx.get(
                    NotificationURL, headers=headers, timeout=30, verify=False
                )
                notification_res.raise_for_status()
                notification_data = notification_res.json()
                logger.debug(f"Notification response: {notification_data}")

                sign_in_status = "❓ 未知签到状态"  # Default status

                if notification_data.get("retcode") == 0 and notification_data.get(
                    "data"
                ):
                    notification_list = notification_data["data"].get("list", [])

                    if not notification_list:
                        sign_in_status = "✅ 今天似乎已经签到过了！(通知列表为空)"
                        logger.info(sign_in_status)
                        notification_msg += sign_in_status + "\n"
                    else:
                        # Look for a notification indicating sign-in reward or limit reached
                        # The logic here was a bit fragile, let's try to be more robust
                        # Look for specific message patterns if possible, or just check the presence of notifications

                        last_notification_msg = notification_list[0].get(
                            "msg"
                        )  # Assume the first one is the latest? Or the last one? Original code used [-1]... let's stick to that for now.
                        if len(notification_list) > 0:
                            last_notification_msg = notification_list[-1].get("msg")

                        try:
                            # Attempt to parse the 'msg' field which is often a JSON string itself
                            msg_payload = json.loads(last_notification_msg)
                            logger.debug(
                                f"Parsed last notification msg payload: {msg_payload}"
                            )

                            if msg_payload.get("msg") == "每日登录奖励":
                                sign_in_status = f"✅ 获取签到情况成功！{msg_payload.get('msg')}：获得 {msg_payload.get('free_time')} 分钟"
                                logger.info(sign_in_status)
                                notification_msg += sign_in_status + "\n"
                            elif msg_payload.get("over_num", 0) > 0:
                                sign_in_status = f"✅ 获取签到情况成功！免费时长已达上限，未能获得 {msg_payload.get('free_time')} 分钟 (超出 {msg_payload.get('over_num')} 分钟)"
                                logger.info(sign_in_status)
                                notification_msg += sign_in_status + "\n"
                            else:
                                # Catch-all for other notification types or unexpected payloads
                                sign_in_status = f"❓ 获取到其他通知，可能已经签到或状态未知: {last_notification_msg}"
                                logger.info(sign_in_status)
                                notification_msg += sign_in_status + "\n"

                        except json.JSONDecodeError:
                            # 'msg' is not a JSON string
                            sign_in_status = f"❓ 获取到非标准通知，可能已经签到或状态未知: {last_notification_msg}"
                            logger.info(sign_in_status)
                            notification_msg += sign_in_status + "\n"
                        except Exception as e:
                            # Other errors during parsing msg
                            sign_in_status = f"❌ 解析通知详情时出错: {e}. Raw msg: {last_notification_msg}"
                            logger.error(sign_in_status)
                            notification_msg += sign_in_status + "\n"

                elif notification_data.get("retcode") != 0:
                    error_msg = f"获取通知列表失败: {notification_data.get('retcode')} - {notification_data.get('message', 'Unknown error')}"
                    logger.error(error_msg)
                    notification_msg += error_msg + "\n"

            except httpx.HTTPStatusError as e:
                error_msg = f"获取通知列表HTTP错误: {e.response.status_code} - {e.response.text}"
                logger.error(error_msg)
                notification_msg += error_msg + "\n"
            except httpx.RequestError as e:
                error_msg = f"请求通知列表失败: {e}"
                logger.error(error_msg)
                notification_msg += error_msg + "\n"
            except Exception as e:
                error_msg = f"检查签到状态时出错: {e}"
                logger.error(error_msg)
                notification_msg += error_msg + "\n"

        except KeyError as e:
            # This catches missing required keys in account config
            error_msg = f"账号配置缺少必需的键: {e}"
            logger.error(error_msg)
            notification_msg += f"❌ 账号配置错误: {error_msg}\n"
        except Exception as e:
            # Catch any other unexpected errors during account processing
            error_msg = f"处理账号时发生未知错误: {e}"
            logger.error(error_msg)
            notification_msg += f"❌ 账号处理错误: {error_msg}\n"

        if accounts_conf.index(config) < len(accounts_conf) - 1:
            notification_msg += "\n---\n\n"

        send_notifications(notification_msg, notification_settings)

    logger.info("所有任务已经执行完毕！")