import requests

import datetime

from ruamel.yaml import YAML

import sys

import hashlib  # 用于计算内容MD5，精准比对内容

  
  

# -------------------------- 1. 核心配置（替换为你的实际信息） --------------------------

# -------------------------- 新手必改：3处核心配置 --------------------------

MOMENTS_API = "你的实际API域名/api/memo/list"  # ❗ 必须替换为你的实际API地址（如https://api.example.com/moments）

JWT_TOKEN = "eyJhbGc58_4kLQfo"               # ❗ 替换为你的实际令牌（若API不需要令牌，可改为空字符串：JWT_TOKEN = ""）

ESSAY_YML_PATH = "/opt/blog-source/source/_data/essay.yml"  # ✅ 这个路径通常不用改，确认存在即可

  
  

def get_content_md5(content):

    """计算内容的MD5哈希（确保内容完全一致才返回相同值）"""

    # 去除内容前后空格，避免因空格差异误判

    clean_content = content.strip()

    # 计算MD5（需编码为UTF-8）

    md5_obj = hashlib.md5(clean_content.encode("utf-8"))

    return md5_obj.hexdigest()  # 返回32位MD5字符串

  
  

def convert_to_timestamp(iso_time):

    """将API的ISO时间（如2025-09-17T04:48:14.768737624+08:00）转换为毫秒级时间戳（统一精度）"""

    try:

        # 解析带时区的ISO时间

        dt = datetime.datetime.fromisoformat(iso_time)

        # 转换为毫秒级时间戳（乘以1000，取整避免小数位差异）

        timestamp_ms = int(dt.timestamp() * 1000)

        return timestamp_ms

    except Exception as e:

        print(f"⚠️  时间格式转换失败：{iso_time}，错误：{str(e)}")

        return None

  
  

def fetch_moments_data():

    """拉取API数据：适配{"code":0,"data":{"list":[...]}}结构"""

    headers = {

        "x-api-token": JWT_TOKEN,

        "Content-Type": "application/json",

        "User-Agent": "Mozilla/5.0 (Linux; Android 10; SM-G975F) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Mobile Safari/537.36"

    }

    payload = {"page": 1, "size": 100, "showType": 1}

  

    try:

        response = requests.post(MOMENTS_API, headers=headers, json=payload, timeout=10)

        response.raise_for_status()

        api_result = response.json()

  

        if api_result.get("code") != 0:

            print(f"❌ API错误：code={api_result.get('code')}，msg={api_result.get('msg')}")

            sys.exit(1)

        moments_list = api_result.get("data", {}).get("list", [])

        total = api_result.get("data", {}).get("total", 0)

        print(f"✅ API请求成功：共{total}条动态（本次返回{len(moments_list)}条）")

        return moments_list

  

    except Exception as e:

        print(f"❌ 拉取动态失败：{str(e)}")

        sys.exit(1)

  
  

def load_existing_essay():

    """加载现有essay.yml，提取“时间戳+内容MD5”作为去重key"""

    yaml = YAML()

    yaml.preserve_quotes = True

    yaml.indent(mapping=2, sequence=4, offset=2)

  

    try:

        with open(ESSAY_YML_PATH, "r", encoding="utf-8") as f:

            essay_config = yaml.load(f)

            if not isinstance(essay_config, list) or len(essay_config) == 0:

                print(f"❌ essay.yml格式错误：需为列表结构")

                sys.exit(1)

        existing_essay_list = essay_config[0].get("essay_list", [])

        existing_keys = set()  # 存储去重key："时间戳_ms_内容MD5"

  

        for old_item in existing_essay_list:

            # 提取现有动态的“时间戳”和“内容”（首次执行时可能没有timestamp，跳过）

            old_timestamp = old_item.get("timestamp_ms")

            old_content = old_item.get("content", "").strip()

  

            if old_timestamp and old_content:

                # 生成去重key：时间戳+内容MD5

                old_content_md5 = get_content_md5(old_content)

                old_key = f"{old_timestamp}_{old_content_md5}"

                existing_keys.add(old_key)

  

        print(f"✅ 加载现有essay.yml：共{len(existing_essay_list)}条短文，已提取{len(existing_keys)}个去重key")

        return essay_config, existing_essay_list, existing_keys

  

    except FileNotFoundError:

        print(f"❌ 未找到essay.yml：{ESSAY_YML_PATH}")

        sys.exit(1)

    except Exception as e:

        print(f"❌ 读取essay.yml错误：{str(e)}")

        sys.exit(1)

  
  

def deduplicate_and_merge(moments_list, existing_essay_list, existing_keys):

    """去重逻辑：时间戳（毫秒级）+ 内容MD5 完全一致才视为重复"""

    new_essay_items = []

  

    for moment in moments_list:

        # 1. 提取API字段

        content = moment.get("content", "").strip()

        created_at = moment.get("createdAt", "")  # ISO时间

        location = moment.get("location", "")

        nickname = moment.get("user", {}).get("nickname", "mzxi")

  

        # 跳过无效动态（无内容或无时间）

        if not content or not created_at:

            print(f"ℹ️ 跳过无效动态：content={content[:20]}...，createdAt={created_at}")

            continue

  

        # 2. 转换时间戳（毫秒级，统一精度）

        timestamp_ms = convert_to_timestamp(created_at)

        if not timestamp_ms:

            continue  # 时间转换失败，跳过

  

        # 3. 生成去重key：时间戳+内容MD5

        content_md5 = get_content_md5(content)

        new_key = f"{timestamp_ms}_{content_md5}"

  

        # 4. 去重判断：key不在现有集合中，视为新动态

        if new_key in existing_keys:

            print(f"ℹ️ 重复动态：时间戳={timestamp_ms}，内容MD5={content_md5[:8]}...，已跳过")

            continue

  

        # 5. 处理显示日期（YYYY/MM/DD）

        try:

            dt = datetime.datetime.fromisoformat(created_at)

            show_date = dt.strftime("%Y/%m/%d")

        except Exception as e:

            show_date = "未知日期"

            print(f"⚠️  日期格式处理失败：{created_at}，错误：{str(e)}")

  

        # 6. 构造新动态（含timestamp_ms，用于下次去重）

        new_item = {

            "timestamp_ms": timestamp_ms,  # 存储毫秒级时间戳（去重用）

            "content": content.replace("\n", "<br>"),  # 换行转HTML

            "date": show_date,             # 页面显示日期

            "from": nickname,              # 作者昵称

            "address": location            # 位置（无则为空）

        }

        new_essay_items.append(new_item)

        existing_keys.add(new_key)  # 新增key到集合，避免同批次重复

  

    # 7. 合并：新动态放前面（最新动态显示在顶部）

    merged_essay_list = new_essay_items + existing_essay_list

    print(f"ℹ️ 去重完成：新增{len(new_essay_items)}条动态，合并后共{len(merged_essay_list)}条")

    return new_essay_items, merged_essay_list

  
  

def save_essay_yml(essay_config, merged_essay_list):

    """保存更新后的essay.yml，保留注释"""

    yaml = YAML()

    yaml.preserve_quotes = True

    yaml.indent(mapping=2, sequence=4, offset=2)

  

    try:

        essay_config[0]["essay_list"] = merged_essay_list

        with open(ESSAY_YML_PATH, "w", encoding="utf-8") as f:

            yaml.dump(essay_config, f)

        print(f"✅ 保存essay.yml成功：{ESSAY_YML_PATH}")

        return True

    except Exception as e:

        print(f"❌ 保存essay.yml错误：{str(e)}（需权限：chmod 644 {ESSAY_YML_PATH}）")

        sys.exit(1)

  
  

if __name__ == "__main__":

    print("=" * 60)

    start_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")

    print(f"📅 同步开始：{start_time}")

  

    # 1. 拉取API数据

    moments_data = fetch_moments_data()

    if not moments_data:

        print(f"ℹ️ 无动态数据，流程结束")

        print(f"📅 同步结束：{datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")

        print("=" * 60)

        sys.exit(0)

  

    # 2. 加载现有essay.yml及去重key

    essay_config, existing_essay, existing_keys = load_existing_essay()

  

    # 3. 去重合并

    new_items, merged_list = deduplicate_and_merge(moments_data, existing_essay, existing_keys)

  

    # 4. 处理结果

    if len(new_items) == 0:

        print(f"ℹ️ 无新动态（已去重）")

        print(f"📅 同步结束：{datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")

        print("=" * 60)

        sys.exit(0)  # 无新动态，返回0

    else:

        save_essay_yml(essay_config, merged_list)

        print(f"📅 同步结束：{datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")

        print("=" * 60)

        sys.exit(2)  # 有新动态，返回2