#!/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 = {
        "摄影/摄像": ["摄影师", "摄影摄像助理", "影视摄像"],
        "后期特效": ["特效设计师", "特效合成师", "3D特效师"],
        "直播": ["直播专员", "直播助理", "直播运营"],
        "影视节目策划": ["文案策划", "导演"]
    }

    updated_count = 0
    not_found_positions = []

    # 为每个岗位群更新面试题
    for industry_name, position_names in industry_map.items():
        # 查找该岗位群的第一个岗位的面试题
        questions = []
        used_position = None
        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('面试题内容', ''))
                    used_position = position_name
                    break
            if questions:
                break

        if not questions:
            not_found_positions.append(industry_name)
            print(f"⚠️ 未找到面试题：{industry_name} (尝试的岗位: {', '.join(position_names)})")
            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} (使用岗位：{used_position}，共{len(questions)}个面试题)")

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

    print(f"\n✅ Mock文件更新完成！共更新{updated_count}个岗位群。")

    if not_found_positions:
        print(f"\n⚠️ 以下岗位群未找到对应的面试题数据：")
        for name in not_found_positions:
            print(f"  - {name}")

if __name__ == '__main__':
    update_mock_file()