#!/usr/bin/env python3
import json
import re

# 读取视觉设计项目案例数据
with open('网页未导入数据/视觉设计产业/视觉设计项目案例.json', 'r', encoding='utf-8') as f:
    visual_projects = json.load(f)

# 读取mock文件
with open('src/mocks/projectLibraryMock.js.backup_attachments', 'r', encoding='utf-8') as f:
    content = f.read()

# 创建项目名称到附件的映射
name_to_attachments = {}
for project in visual_projects:
    name = project.get("案例名称", "")
    attachments_str = project.get("附件", "")

    if name and attachments_str:
        # 分割附件字符串
        files = [f.strip() for f in attachments_str.split(',')]
        attachments_list = []

        for file in files:
            ext = file.split('.')[-1].lower()
            # 判断文件类型
            file_type = 'document'
            if ext in ['jpg', 'jpeg', 'png', 'gif']:
                file_type = 'image'
            elif ext in ['mp4', 'avi', 'mov']:
                file_type = 'video'
            elif ext == 'pdf':
                file_type = 'pdf'
            elif ext == 'psd':
                file_type = 'design'
            elif ext in ['doc', 'docx']:
                file_type = 'word'

            attachments_list.append({
                'name': file,
                'type': file_type,
                'url': f'/attachments/{file}'
            })

        name_to_attachments[name] = attachments_list

# 现在替换mock文件中每个项目的attachments字段
# 找到每个项目并替换其attachments内容
for name, attachments in name_to_attachments.items():
    # 转义名称中的特殊字符
    escaped_name = re.escape(name)

    # 查找这个项目的位置
    pattern = rf'"name": "{escaped_name}".*?"attachments": \[(.*?)\]'

    # 生成新的attachments内容
    attachments_json = json.dumps(attachments, ensure_ascii=False, indent=12)
    # 调整缩进
    attachments_json = attachments_json.replace('\n', '\n            ')
    attachments_json = attachments_json[1:-1]  # 去掉外层的[]

    # 替换attachments内容
    replacement = lambda m: m.group(0).replace(
        f'"attachments": [{m.group(1)}]',
        f'"attachments": [\n            {attachments_json}\n      ]'
    )

    content = re.sub(pattern, replacement, content, flags=re.DOTALL)

# 写回文件
with open('src/mocks/projectLibraryMock.js', 'w', encoding='utf-8') as f:
    f.write(content)

print("附件数据已成功修复！")