刷新date值为createTime

#!/usr/bin/env python3
"""
fix_frontmatter_date.py
在 frontmatter 中根据 createTime 生成/覆盖 date 字段
跳过目录：public、.vitepress
"""

import re
from datetime import datetime
from pathlib import Path

ROOT = Path(__file__).resolve().parent
SKIP_DIRS = {"public", ".vitepress"}

# 刷新date值为createTime
FRONT_MATTER_RE = re.compile(r'^---\n(.*?)\n---\n', re.DOTALL)

# 捕获 createTime 值
CREATE_TIME_RE = re.compile(r'^createTime:\s*(.+)$', re.MULTILINE)


def should_skip(path: Path) -> bool:
    return any(part in SKIP_DIRS for part in path.parts)


def normalize_time(create_time_str: str) -> str:
    """
    把 ISO-8601 时间（可能带 Z 或 ±HH:MM）转换为 'YYYY-MM-DD HH:MM:SS'
    """
    try:
        # dateutil 更健壮，但只用标准库
        dt = datetime.fromisoformat(create_time_str.replace("Z", "+00:00"))
        return dt.strftime("%Y-%m-%d %H:%M:%S")
    except Exception:
        # 无法解析则原样返回，脚本会跳过
        return None


def process_one(md_file: Path) -> None:
    text = md_file.read_text(encoding="utf-8")

    match = FRONT_MATTER_RE.match(text)
    if not match:
        return  # 无 frontmatter

    front_raw = match.group(1)

    ct_match = CREATE_TIME_RE.search(front_raw)
    if not ct_match:
        return  # 无 createTime

    new_date_val = normalize_time(ct_match.group(1).strip())
    if new_date_val is None:
        return  # 格式异常，跳过

    # 替换或新增 date 字段
    # 1) 已存在 date: 行 -> 替换
    new_front, sub_count = re.subn(
        r'^date:.*$', f"date: {new_date_val}", front_raw, flags=re.MULTILINE
    )
    if sub_count == 0:
        # 2) 不存在 -> 在 frontmatter 末尾插入
        new_front = front_raw.rstrip() + f"\ndate: {new_date_val}\n"

    # 拼回全文
    new_text = text.replace(match.group(0), f"---\n{new_front}\n---\n", 1)

    if new_text != text:
        md_file.write_text(new_text, encoding="utf-8")
        print(f"Updated: {md_file}")


def main():
    for md in ROOT.rglob("*.md"):
        if not should_skip(md):
            process_one(md)


if __name__ == "__main__":
    main()