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

import json
import re

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 convert_energy_project_details():
    """
    将能源项目案例数据转换为项目详情数据格式
    """

    # 读取能源项目案例数据
    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 "能源技术基础"

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

        converted_details.append(converted_detail)

    print(f"✅ 转换完成，共转换 {len(converted_details)} 个能源项目详情")

    return converted_details

def update_project_details_mock_data(converted_details):
    """
    更新projectLibraryMock.js文件中的项目详情数据
    """

    # 读取现有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 ""

        # 处理多行文本，确保正确转义
        overview = detail['overview'].replace('"', '\\"').replace('\n', '\\n')
        process = detail['process'].replace('"', '\\"').replace('\n', '\\n')
        keypoints = detail['keyPoints'].replace('"', '\\"').replace('\n', '\\n')

        details_js += f'''  {{
    "id": {detail['id']},
    "name": "{detail['name']}",
    "positions": [
{positions_str}
    ],
    "unit": "{detail['unit']}",
    "overview": "{overview}",
    "process": "{process}",
    "keyPoints": "{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"✅ projectLibraryMock.js 项目详情数据已更新")

if __name__ == "__main__":
    converted_details = convert_energy_project_details()
    update_project_details_mock_data(converted_details)