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

import json
import re
from datetime import datetime

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

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

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

    for i, match in enumerate(matches[:5], 1):  # 只取前5个问题
        q_num = match[0]
        question_text = match[1].strip()
        answer_text = match[2].strip()

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

    return questions

def update_industry_questions_directly(industry_name, position_name, visual_data):
    """直接更新指定岗位群的面试题"""

    # 查找对应的面试题内容
    interview_content = None
    for item in visual_data:
        if item.get('岗位名称') == position_name and item.get('面试题内容'):
            interview_content = item.get('面试题内容')
            break

    if not interview_content:
        return None, f"未找到{position_name}的面试题"

    # 提取问题
    questions = extract_questions(interview_content)

    if not questions:
        return None, f"未能提取{position_name}的面试题"

    # 创建带有subQuestions的结构
    result = {
        "id": "group_q1",
        "question": f"# {industry_name}面试题",
        "subQuestions": questions
    }

    return result, None

def main():
    print("=== 最终替换面试题数据 ===\n")

    # 创建备份
    timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
    backup_name = f'src/mocks/resumeInterviewMock.js.backup_{timestamp}_final'

    with open('src/mocks/resumeInterviewMock.js', 'r', encoding='utf-8') as f:
        mock_content = f.read()

    with open(backup_name, 'w', encoding='utf-8') as f:
        f.write(mock_content)
    print(f"✓ 已创建备份: {backup_name}\n")

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

    # 岗位映射
    industry_map = {
        'UI设计': 'UI设计师',
        '包装设计': '包装设计师',
        '插画设计': '插画师',
        '灯光': '影视灯光',
        '动画设计': '动画师',
        '后期特效': '特效设计师',
        '剪辑': '剪辑师',
        '品牌设计': '品牌视觉内容策划',
        '平面设计': '平面设计师',
        '三维设计': 'CG总监助理',
        '摄影/摄像': '摄影师',
        '室内设计': '室内设计师',
        '调色': '调色师',
        '新媒体运营': '新媒体运营专员',
        '音频处理': '录音师',
        '影视节目策划': '文案策划',
        '直播': '直播专员'
    }

    updated_count = 0

    for industry_name, position_name in industry_map.items():
        print(f"处理 {industry_name}...")

        # 获取新的questions数据
        new_questions, error = update_industry_questions_directly(industry_name, position_name, visual_data)

        if error:
            print(f"  ⚠️ {error}")
            continue

        # 将新的questions转换为JSON字符串
        questions_json = json.dumps([new_questions], ensure_ascii=False, indent=2)

        # 调整缩进（每行加4个空格）
        lines = questions_json.split('\n')
        adjusted_lines = []
        for line in lines:
            if line.strip():
                adjusted_lines.append('    ' + line)
            else:
                adjusted_lines.append('')
        questions_json = '\n'.join(adjusted_lines).strip()

        # 使用更简单的替换方法
        # 找到industry name，然后替换其后的整个questions数组
        pattern = rf'("name":\s*"{re.escape(industry_name)}".*?"questions":\s*)\[[\s\S]*?\](?=\s*\}})'

        replacement = rf'\1{questions_json}'

        new_mock_content = re.sub(pattern, replacement, mock_content, count=1, flags=re.MULTILINE)

        if new_mock_content != mock_content:
            mock_content = new_mock_content
            updated_count += 1
            print(f"  ✓ 成功更新 (使用岗位: {position_name}, 共{len(new_questions['subQuestions'])}个问题)")
        else:
            print(f"  ✗ 未能更新")

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

    print(f"\n✅ 更新完成！成功更新{updated_count}个岗位群")

    # 验证语法
    import subprocess
    try:
        result = subprocess.run(['node', '-c', 'src/mocks/resumeInterviewMock.js'],
                              capture_output=True, text=True)
        if result.returncode == 0:
            print("✅ 文件语法验证通过")

            # 验证一个具体的更新结果
            if updated_count > 0:
                print("\n验证UI设计的第一个问题：")
                with open('src/mocks/resumeInterviewMock.js', 'r', encoding='utf-8') as f:
                    content = f.read()
                    # 查找UI设计的第一个问题
                    match = re.search(r'"name":\s*"UI设计".*?"q1_1".*?"question":\s*"([^"]+)"', content, re.DOTALL)
                    if match:
                        print(f"  问题: {match.group(1)[:50]}...")
        else:
            print(f"❌ 语法错误:\n{result.stderr}")
            print("\n恢复备份...")
            with open(backup_name, 'r', encoding='utf-8') as f:
                backup_content = f.read()
            with open('src/mocks/resumeInterviewMock.js', 'w', encoding='utf-8') as f:
                f.write(backup_content)
            print("已恢复备份")
    except Exception as e:
        print(f"⚠️ 无法验证语法: {str(e)}")

if __name__ == '__main__':
    main()