#!/usr/bin/env python3
"""
跳过目录：public、.vitepress
"""

import re
from pathlib import Path

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

# vuepress 图片卡片转vitepress
IMAGE_CARD_RE = re.compile(
    r'<ImageCard\b(.*?)\s*/>',
    re.IGNORECASE | re.DOTALL
)

# 属性正则：捕获 key="value" 或 key='value'
ATTR_RE = re.compile(
    r'(\w+)\s*=\s*(["\'])(.*?)\2',
    re.IGNORECASE
)

# 属性映射表
ATTR_MAP = {
    'image': '- img',
    'title': '  name',
    'description': '  desc',
    'author': '  author',
    # date -> 丢弃
}


def should_skip(path: Path) -> bool:
    """判断路径是否包含需要跳过的目录"""
    return any(part in SKIP_DIRS for part in path.parts)


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

    def replacer(match: re.Match) -> str:
        raw_attrs = match.group(1)
        props = {}
        for k, _, v in ATTR_RE.findall(raw_attrs):
            mapped = ATTR_MAP.get(k.lower())
            if mapped:          # 忽略未映射或 date
                props[mapped] = v

        yaml_lines = []
        for k, v in props.items():
            yaml_lines.append(f'{k}: {v}')

        # 合成块
        block = (
            '::: imgCard\n'
            '```yaml\n' +
            '\n'.join(yaml_lines) + '\n' +
            '```\n' +
            ':::\n'
        )
        return block

    new_text, count = IMAGE_CARD_RE.subn(replacer, text)
    if count:
        md_file.write_text(new_text, encoding='utf-8')
        print(f'Updated: {md_file}  ({count} tag(s))')


def main():
    for md in ROOT.rglob('*.md'):
        if not should_skip(md):
            convert_one(md)


if __name__ == '__main__':
    main()