#!/usr/bin/env python3
"""
每日身体 AI 建议推送（卡片版 + 周期追踪）
每天推 3 次，每次挑不同的建议 + 吃药提醒 + 周期状态。
用法: python3 body_push.py [0|1|2]  — 第几次推送（默认自动判断）
"""

import json
import os
import re
import sys
import glob
import html
import urllib.request
from datetime import date, timedelta

WEBHOOK_URL = "https://open.feishu.cn/open-apis/bot/v2/hook/5dd37bb4-0e98-4880-9b47-98c8c7823347"
BODY_DIR = os.path.expanduser("~/Desktop/12身体重启")
CYCLE_FILE = os.path.join(BODY_DIR, "cycle.json")
ITEMS_PER_PUSH = 3

HEADER_COLORS = ["blue", "turquoise", "violet"]
STATUS_COLOR = {"good": "green", "warn": "orange", "bad": "red", "neutral": "blue"}


# ─── 周期追踪 ───

def load_cycle():
    if os.path.exists(CYCLE_FILE):
        with open(CYCLE_FILE) as f:
            return json.load(f)
    return None


def cycle_status(cycle):
    if not cycle:
        return None

    start = date.fromisoformat(cycle["cycle_start"])
    today = date.today()
    day = (today - start).days + 1
    active = cycle.get("active_days", 21)
    brk = cycle.get("break_days", 7)
    total = active + brk

    if day > total:
        day = ((day - 1) % total) + 1

    break_start = start + timedelta(days=active)
    break_end = start + timedelta(days=total - 1)

    if day <= 7:
        pill = f"💊 优思明第{day}天/21天，记得吃药！"
        phase = "📍 服药期第一周（激素稳定期）"
        body_tip = "身体状态平稳，精力恢复中，适合正常饮食和运动"
    elif day <= 14:
        pill = f"💊 优思明第{day}天/21天，记得吃药！"
        phase = "📍 服药期第二周（黄金期）"
        body_tip = "状态最好的一周！运动表现好，蛋白质吸收效率高，适合加强度"
    elif day <= active:
        days_to_break = active - day
        pill = f"💊 优思明第{day}天/21天，记得吃药！"
        phase = f"📍 服药期第三周（{break_start.month}/{break_start.day}进入停药期）"
        if days_to_break <= 3:
            body_tip = f"⚠️ 还有{days_to_break}天停药，可能已有PMS：情绪波动、腹胀、水肿、食欲增加。体重涨1-2kg是水分，不是胖了！"
        else:
            body_tip = "可能出现轻微水钠潴留，体重微涨属正常。注意补钾（香蕉/土豆），不要因为体重焦虑而节食"
    else:
        break_day = day - active
        next_start = break_end + timedelta(days=1)
        pill = f"💊 停药期第{break_day}天/{brk}天（不用吃优思明）"
        phase = f"📍 停药期（{next_start.month}/{next_start.day}开新一轮）"
        if break_day <= 2:
            body_tip = "撤退性出血通常停药2-4天后出现。可能疲劳、情绪低，运动散步瑜伽即可，允许多吃100-200kcal"
        elif break_day <= 5:
            body_tip = "🩸 出血期，铁流失增加——多吃红肉/菠菜/配维C。运动量可以降低，不要硬撑"
        else:
            body_tip = f"出血接近尾声，{next_start.month}/{next_start.day}开始新一轮服药，记得准备好药"

    return {"day": day, "pill": pill, "phase": phase, "body_tip": body_tip}


# ─── 数据提取 ───

def find_latest_dashboard():
    pattern = os.path.join(BODY_DIR, "W*_dashboard.html")
    files = glob.glob(pattern)
    if not files:
        main = os.path.join(BODY_DIR, "dashboard.html")
        return main if os.path.exists(main) else None
    files.sort(key=os.path.getmtime)
    return files[-1]


def find_latest_weekly():
    pattern = os.path.join(BODY_DIR, "W*_0*.md")
    files = glob.glob(pattern)
    if not files:
        return None
    files.sort(key=os.path.getmtime)
    return files[-1]


def strip_html(text):
    text = html.unescape(text)
    return re.sub(r'<[^>]+>', '', text).strip()


def extract_summary_metrics(filepath):
    """从 dashboard 提取顶部汇总指标"""
    if not filepath:
        return {}
    with open(filepath) as f:
        content = f.read()

    metrics = {}
    for card in re.finditer(
        r'<div class="summary-card\s+(\w+)">\s*'
        r'<div class="label">(.+?)</div>\s*'
        r'<div class="value"[^>]*>(.+?)</div>\s*'
        r'<div class="unit">(.+?)</div>\s*'
        r'<div class="target">(.+?)</div>',
        content, re.DOTALL
    ):
        status, label, value, unit, target = [strip_html(g) for g in card.groups()]
        metrics[label] = {"value": value, "unit": unit, "target": target, "status": status}

    return metrics


def collect_all_advice(filepath):
    with open(filepath) as f:
        content = f.read()

    items = []
    for box in re.finditer(
        r'<div class="insight-box">\s*<h3>(.+?)</h3>\s*<ul>(.*?)</ul>\s*</div>',
        content, re.DOTALL
    ):
        for li in re.finditer(r'<li>(.*?)</li>', box.group(2), re.DOTALL):
            raw = li.group(1)
            if 'tag green' in raw:
                continue
            text = strip_html(raw)
            color = "red" if 'tag red' in raw else "yellow"
            items.append({"text": text, "color": color})

    return items


def extract_exercise_summary(filepath):
    with open(filepath) as f:
        content = f.read()

    today = date.today()
    year = today.year
    cutoff = today - timedelta(days=7)
    exercise_days = 0
    total_active = 0

    for section in re.split(r'(?=^## 周[一二三四五六日]\s+\d+/\d+)', content, flags=re.MULTILINE):
        header = re.match(r'## 周[一二三四五六日]\s+(\d+)/(\d+)', section)
        if not header:
            continue
        try:
            d = date(year, int(header.group(1)), int(header.group(2)))
        except ValueError:
            continue
        if d < cutoff or d > today:
            continue
        m = re.search(r'含(?:运动|活动)(\d+)', section)
        if m:
            exercise_days += 1
            total_active += int(m.group(1))

    goal = "3-4次/周"
    if exercise_days >= 3:
        return f"本周运动{exercise_days}次，活动消耗{total_active}kcal，继续保持！"
    elif exercise_days >= 1:
        return f"本周运动{exercise_days}次（目标{goal}），还差{3 - exercise_days}次，今天动一动？"
    else:
        return f"本周还没运动（目标{goal}），找20分钟动起来！"


def pick_items(all_items, push_index):
    if not all_items:
        return []
    n = len(all_items)
    day_offset = date.today().toordinal()
    start = (day_offset * 3 + push_index) * ITEMS_PER_PUSH % n
    return [all_items[(start + i) % n] for i in range(ITEMS_PER_PUSH)]


# ─── 卡片构建 ───

def build_card(cycle_info, metrics, diet_items, exercise_line, push_index):
    today = date.today()
    label = ["早安", "午间", "晚间"][push_index]
    elements = []

    # 周期状态
    if cycle_info:
        elements.append({
            "tag": "markdown",
            "content": f"{cycle_info['pill']}\n{cycle_info['phase']}\n{cycle_info['body_tip']}"
        })
        elements.append({"tag": "hr"})

    # 汇总指标（三列）
    keys_order = ["日均摄入", "日均蛋白质", "日均赤字"]
    columns = []
    for key in keys_order:
        m = metrics.get(key)
        if not m:
            continue
        color = STATUS_COLOR.get(m["status"], "blue")
        columns.append({
            "tag": "column",
            "width": "weighted",
            "weight": 1,
            "vertical_align": "center",
            "elements": [{
                "tag": "markdown",
                "content": f"**{key}**\n<font color='{color}'>{m['value']}</font> {m['unit']}\n{m['target']}"
            }]
        })
    if columns:
        elements.append({
            "tag": "column_set",
            "flex_mode": "bisect",
            "background_style": "grey",
            "columns": columns,
        })
        elements.append({"tag": "hr"})

    # 饮食建议
    if diet_items:
        elements.append({"tag": "markdown", "content": "**📋 今日建议**"})
        for item in diet_items:
            icon = "🔴" if item["color"] == "red" else "🟡"
            text = item["text"]
            parts = text.split(" ", 1)
            if len(parts) == 2 and len(parts[0]) <= 15:
                md = f"{icon} **{parts[0]}** — {parts[1]}"
            else:
                md = f"{icon} {text}"
            elements.append({"tag": "markdown", "content": md})
        elements.append({"tag": "hr"})

    # 运动
    if exercise_line:
        elements.append({"tag": "markdown", "content": f"🏃 {exercise_line}"})

    # 累计赤字
    cum = metrics.get("累计赤字")
    if cum:
        elements.append({
            "tag": "markdown",
            "content": f"📊 累计赤字 **{cum['value']}** {cum['unit']} | {cum['target']}"
        })

    return {
        "msg_type": "interactive",
        "card": {
            "header": {
                "template": HEADER_COLORS[push_index],
                "title": {"tag": "plain_text", "content": f"🤖 {label}提醒 · {today.month}/{today.day}"}
            },
            "elements": elements,
        }
    }


# ─── 发送 ───

def send_webhook(card_payload):
    payload = json.dumps(card_payload).encode()
    req = urllib.request.Request(
        WEBHOOK_URL, data=payload,
        headers={"Content-Type": "application/json"},
        method="POST",
    )
    try:
        with urllib.request.urlopen(req, timeout=10) as resp:
            result = json.loads(resp.read())
            print(f"[推送结果] {result}")
            return result
    except Exception as e:
        print(f"[推送失败] {e}")
        return None


def auto_push_index():
    from datetime import datetime
    h = datetime.now().hour
    if h < 11:
        return 0
    elif h < 16:
        return 1
    else:
        return 2


def main():
    push_index = int(sys.argv[1]) if len(sys.argv) > 1 else auto_push_index()

    cycle_info = cycle_status(load_cycle())

    dashboard = find_latest_dashboard()
    all_advice = collect_all_advice(dashboard) if dashboard else []
    metrics = extract_summary_metrics(dashboard)

    weekly = find_latest_weekly()
    exercise_line = extract_exercise_summary(weekly) if weekly else None

    diet_items = pick_items(all_advice, push_index)
    card = build_card(cycle_info, metrics, diet_items, exercise_line, push_index)

    # 日志
    for el in card["card"]["elements"]:
        if el.get("tag") == "markdown":
            print(el["content"])
    print()

    send_webhook(card)


if __name__ == "__main__":
    main()