import requests
import time
import json
import os
import random
import re

# -------------------------- 尝试导入通知模块 --------------------------
try:
    from notify import send
except ImportError:
    def send(title, content):
        print(f"\n[通知] {title}\n{content}")
        print("\n 未找到 notify.py 模块，取消推送")

# -------------------------- 核心逻辑 --------------------------
class V2exSigner:
    def __init__(self, cookie, index):
        self.cookie = cookie
        self.index = index
        self.headers = {
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/122.0.0.0 Safari/537.36",
            "Referer": "https://www.v2ex.com/",
            "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8",
            "Cookie": self.cookie
        }
        self.session = requests.Session()
        self.session.headers.update(self.headers)

    def sign(self):
        print(f"\n==== 账号 {self.index} 开始签到 ====")
        try:
            # 1. 访问签到页面获取 once 参数
            res = self.session.get("https://www.v2ex.com/mission/daily", timeout=20)
            if "登出" not in res.text:
                return {"status": "失败", "msg": f"账号 {self.index}: Cookie 已失效或未登录"}

            if "每日登录奖励已领取" in res.text:
                return {"status": "成功", "msg": f"账号 {self.index}: 今日已签过"}

            # 正则匹配签到链接
            pattern = r"location\.href = \'(.*?once=.*?)\';"
            urls = re.findall(pattern, res.text)
            
            if not urls:
                return {"status": "失败", "msg": f"账号 {self.index}: 未找到签到链接，可能已签到或页面结构变化"}

            # 2. 执行签到请求
            sign_url = "https://www.v2ex.com" + urls[0]
            self.session.headers.update({"Referer": "https://www.v2ex.com/mission/daily"})
            self.session.get(sign_url, timeout=20)

            # 3. 验证签到结果
            res = self.session.get("https://www.v2ex.com/mission/daily", timeout=20)
            if "每日登录奖励已领取" in res.text:
                # 获取余额和奖励信息
                balance_res = self.session.get("https://www.v2ex.com/balance", timeout=20)
                # 尝试获取今日金币
                reward = re.findall(r'<td class="d"><span class="gray">(.*?)</span></td>', balance_res.text)
                reward_info = reward[0] if reward else "成功获取"
                
                # 获取总余额
                total = re.findall(r'<td class="d" style="text-align: right;">(\d+\.\d+)</td>', balance_res.text)
                total_info = total[0] if total else "未知"

                return {
                    "status": "成功",
                    "msg": f"账号 {self.index}: 签到成功！奖励: {reward_info}，余额: {total_info}"
                }
            else:
                return {"status": "失败", "msg": f"账号 {self.index}: 签到请求已发，但页面仍显示未领取"}

        except Exception as e:
            return {"status": "异常", "msg": f"账号 {self.index}: 异常 -> {str(e)}"}

def main():
    # 环境变量：V2EX_COOKIE
    cookie_env = os.getenv("V2EX_COOKIE")
    
    if not cookie_env:
        print(" 未找到环境变量 V2EX_COOKIE，请在青龙后台设置")
        return

    # 分割多个 Cookie
    cookies = re.split(r'[&\n]', cookie_env)
    cookies = [c.strip() for c in cookies if c.strip()]
    
    print(f"共发现 {len(cookies)} 个 V2EX 账号")

    # 随机签到模式 (保持一致风格)
    random_signin = os.getenv("RANDOM_SIGNIN", "false").lower() == "true"
    if random_signin:
        delay = random.randint(1, 1800)
        print(f"随机签到已启用，随机延迟时间窗口: 30 分钟")
        print(f"当前任务将延迟 {delay} 秒后执行...")
        time.sleep(delay)
    else:
        print("随机签到: 禁用")

    all_results = []
    for i, ck in enumerate(cookies, 1):
        signer = V2exSigner(ck, i)
        res = signer.sign()
        print(res["msg"])
        all_results.append(res["msg"])

    # 统一通知
    if all_results:
        notify_content = "\n".join(all_results)
        send("V2EX 签到通知", notify_content)

if __name__ == "__main__":
    main()