#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import json
import re

def escape_js_string(text):
    """
    正确转义JavaScript字符串
    """
    if not text:
        return ""

    # 转义特殊字符
    text = text.replace('\\', '\\\\')  # 反斜杠
    text = text.replace('"', '\\"')    # 双引号
    text = text.replace('\n', '\\n')   # 换行符
    text = text.replace('\r', '\\r')   # 回车符
    text = text.replace('\t', '\\t')   # 制表符

    return text

def extract_sections_from_content(content):
    """
    从项目案例内容中提取overview、process、keyPoints等部分
    """
    # 提取一、项目概述部分
    overview_match = re.search(r'# 一、项目概述\n\n(.*?)(?=\n# 二、)', content, re.DOTALL)
    overview = overview_match.group(1).strip() if overview_match else ""

    # 提取二、项目整体流程介绍部分
    process_match = re.search(r'# 二、项目整体流程介绍\n\n(.*?)(?=\n# 三、)', content, re.DOTALL)
    process = process_match.group(1).strip() if process_match else ""

    # 提取三、项目案例关键技术点部分
    keypoints_match = re.search(r'# 三、项目案例关键技术点\n\n(.*)', content, re.DOTALL)
    keypoints = keypoints_match.group(1).strip() if keypoints_match else ""

    return overview, process, keypoints

def fix_project_details_mock_data():
    """
    修复projectLibraryMock.js文件中的项目详情数据格式问题
    """

    # 读取能源项目案例数据
    with open("网页未导入数据/能源产业/能源项目案例.json", 'r', encoding='utf-8') as f:
        energy_projects = json.load(f)

    # 转换项目详情数据
    converted_details = []

    for idx, project in enumerate(energy_projects, 1):
        # 提取项目案例内容的各个部分
        content = project.get("项目案例内容", "")
        overview, process, keypoints = extract_sections_from_content(content)

        # 处理岗位信息
        resume_names = project.get("对应个人简历名称", "").split(",")
        positions = []
        for name in resume_names:
            name = name.strip()
            if name:
                positions.append({
                    "level": "技术骨干岗",
                    "position": name
                })

        if not positions:
            positions = [{"level": "技术骨干岗", "position": "能源工程师"}]

        # 提取对应单元名称
        vertical_units = project.get("对应单元名称（垂直能力课）", "").split(",")
        unit = vertical_units[0].strip() if vertical_units else "能源技术基础"

        # 安全处理文本内容
        safe_overview = escape_js_string(overview) if overview else "本项目聚焦能源产业核心技术领域，通过系统化的技术实施和管理优化，实现高效、安全、可持续的能源系统运营。"
        safe_process = escape_js_string(process) if process else "### 流程一：前期准备与需求分析\\n\\n1. 进行全面的技术需求调研和可行性分析\\n2. 制定详细的项目实施方案和时间计划\\n\\n### 流程二：系统设计与优化\\n\\n1. 完成核心技术方案设计\\n2. 进行系统集成和性能优化\\n\\n### 流程三：实施与验收\\n\\n1. 按计划实施项目各项内容\\n2. 完成系统测试和项目验收"
        safe_keypoints = escape_js_string(keypoints) if keypoints else "### （一）核心技术应用\\n\\n1. 应用先进的能源技术和管理方法\\n2. 实现系统的高效稳定运行\\n\\n### （二）安全管理体系\\n\\n1. 建立完善的安全管理制度\\n2. 确保项目实施过程的安全可控\\n\\n### （三）持续优化改进\\n\\n1. 建立持续改进机制\\n2. 不断提升系统性能和效益"

        converted_detail = {
            "id": idx,
            "name": project["案例名称"],
            "positions": positions,
            "unit": unit,
            "overview": safe_overview,
            "process": safe_process,
            "keyPoints": safe_keypoints
        }

        converted_details.append(converted_detail)

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

    # 生成新的项目详情数据JavaScript格式
    details_js = "[\n"
    for detail in converted_details:
        # 处理positions数组
        positions_str = ""
        for pos in detail['positions']:
            positions_str += f'''      {{
        "level": "{pos['level']}",
        "position": "{pos['position']}"
      }},
'''
        positions_str = positions_str.rstrip(',\n') if positions_str else ""

        details_js += f'''  {{
    "id": {detail['id']},
    "name": "{detail['name']}",
    "positions": [
{positions_str}
    ],
    "unit": "{detail['unit']}",
    "overview": "{detail['overview']}",
    "process": "{detail['process']}",
    "keyPoints": "{detail['keyPoints']}"
  }},
'''

    details_js = details_js.rstrip(',\n') + "\n  ]"

    # 替换项目详情数组，查找getMockProjectDetail函数中的projects数组
    pattern = r'(export const getMockProjectDetail = \(id\) => \{\s*// 直接根据ID返回对应项目的详情\s*const projects = )\[[\s\S]*?\];'
    replacement = f'\\g<1>{details_js};'

    updated_content = re.sub(pattern, replacement, content)

    # 写入更新后的内容
    with open("src/mocks/projectLibraryMock.js", 'w', encoding='utf-8') as f:
        f.write(updated_content)

    print(f"✅ 修复完成，已正确转义 {len(converted_details)} 个能源项目详情的文本内容")

if __name__ == "__main__":
    fix_project_details_mock_data()