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

import json
import re

def load_visual_design_data():
    """加载视觉设计岗位简历数据"""
    with open('网页未导入数据/视觉设计产业/视觉设计岗位简历.json', 'r', encoding='utf-8') as f:
        return json.load(f)

def extract_questions_from_content(content):
    """从面试题内容中提取问题和答案"""
    questions = []

    # 提取所有问题
    pattern = r'(\d+)\.\s*问题[：:]?\s*(.*?)(?:\n\s*)?(?:参考回答[：:]?)(.*?)(?=\d+\.\s*问题|$)'
    matches = re.findall(pattern, content, re.DOTALL)

    for match in matches[:5]:  # 限制为前5个问题
        q_num = match[0]
        question_text = match[1].strip()
        answer_text = match[2].strip()

        questions.append({
            "id": f"q{q_num}",
            "question": question_text,
            "answer": answer_text
        })

    return questions

def update_mock_file():
    """更新mock文件中的面试题数据"""

    # 加载视觉设计数据
    visual_data = load_visual_design_data()

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

    # 创建岗位群到岗位名称的映射
    industry_map = {
        "UI设计": ["UI设计师"],
        "包装设计": ["包装设计师", "包装设计师助理"],
        "插画设计": ["插画师", "插画设计师", "插画师助理"],
        "摄影": ["摄影师", "摄影师助理", "摄影助理"],
        "角色原画": ["角色原画师", "原画师助理"],
        "动画设计": ["动画师", "二维动画师", "动画制作助理"],
        "平面": ["平面设计师", "广告设计助理"],
        "品牌": ["品牌视觉内容策划", "品牌创意文案"],
        "灯光": ["影视灯光", "影视摄像", "摄影美术指导助理"],
        "AIGC": ["AI绘画师", "CG总监助理", "数字创意师"]
    }

    updated_count = 0

    # 为每个岗位群更新面试题
    for industry_name, position_names in industry_map.items():
        # 查找该岗位群的第一个岗位的面试题
        questions = []
        for position_name in position_names:
            for item in visual_data:
                if item.get('岗位名称') == position_name and item.get('面试题内容'):
                    questions = extract_questions_from_content(item.get('面试题内容', ''))
                    break
            if questions:
                break

        if not questions:
            print(f"⚠️ 未找到面试题内容：{industry_name}")
            continue

        # 查找并替换该岗位群的questions
        # 使用更宽松的正则表达式
        pattern = rf'("name":\s*"{re.escape(industry_name)}".*?"questions":\s*\[)(.*?)(\]\s*\}})'

        def replace_func(match):
            prefix = match.group(1)
            suffix = match.group(3)

            # 构建新的questions数组
            questions_str = "\n"
            for i, q in enumerate(questions):
                if i > 0:
                    questions_str += ",\n"

                # 转义特殊字符
                question_escaped = q['question'].replace('\\', '\\\\').replace('"', '\\"')
                answer_escaped = q['answer'].replace('\\', '\\\\').replace('"', '\\"')

                questions_str += f'''      {{
        "id": "{q['id']}",
        "question": "{question_escaped}",
        "answer": "{answer_escaped}"
      }}'''

            questions_str += "\n    "

            return prefix + questions_str + suffix

        # 执行替换
        new_content = re.sub(pattern, replace_func, mock_content, count=1, flags=re.DOTALL)

        if new_content != mock_content:
            mock_content = new_content
            updated_count += 1
            print(f"✓ 已更新岗位群：{industry_name} (共{len(questions)}个面试题)")
        else:
            # 尝试第二种格式
            pattern2 = rf'("name":\s*"{re.escape(industry_name)}".*?"questions":\s*\[[\s\S]*?\])'
            if re.search(pattern2, mock_content, re.DOTALL):
                print(f"⚠️ 找到岗位群但未能更新：{industry_name}")
            else:
                print(f"✗ 未找到岗位群：{industry_name}")

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

    print(f"\n✅ Mock文件更新完成！共更新{updated_count}个岗位群。")
    print("请刷新页面查看效果。")

if __name__ == '__main__':
    update_mock_file()