#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
生成符合前端期望的扁平化面试题数据
前端组件期望的数据结构：
questions: [
  { id: "q1", question: "问题内容", answer: "答案内容" },
  { id: "q2", question: "问题内容", answer: "答案内容" }
]
"""

import json
import re
import sys
from datetime import datetime

def load_health_resume_data():
    """加载大健康岗位简历数据"""
    try:
        with open('网页未导入数据/大健康产业/大健康岗位简历.json', 'r', encoding='utf-8') as f:
            return json.load(f)
    except Exception as e:
        print(f"Error loading health resume data: {e}")
        return None

def parse_interview_content_to_flat_array(content):
    """解析面试题内容，转换为扁平的问答数组"""
    if not content:
        return []

    questions = []

    # 按大标题分割（# 一、二、三等）
    if content.startswith('# '):
        content = '\n' + content
    sections = re.split(r'\n# ([一二三四五六七八九十]+、[^#\n]+)', content)

    if len(sections) < 2:
        return []

    question_counter = 1
    for i in range(1, len(sections), 2):
        if i + 1 < len(sections):
            section_title = sections[i].strip()
            section_content = sections[i + 1].strip()

            # 按问题编号分割 (1. 2. 3. 等)
            question_parts = re.split(r'\n\s*(\d+\.)\s+', section_content)

            for j in range(1, len(question_parts), 2):
                if j + 1 < len(question_parts):
                    question_block = question_parts[j + 1].strip()

                    # 提取问题和答案
                    lines = question_block.split('\n')
                    question_text = ""
                    answer_text = ""
                    in_answer = False

                    for line in lines:
                        line = line.strip()
                        if line.startswith('示例答案：'):
                            in_answer = True
                            continue

                        if not in_answer and line and not line.startswith('示例答案：'):
                            if question_text:
                                question_text += " "
                            question_text += line
                        elif in_answer and line:
                            if answer_text:
                                answer_text += " "
                            answer_text += line

                    if question_text:
                        questions.append({
                            "id": f"q{question_counter}",
                            "question": question_text,
                            "answer": answer_text
                        })
                        question_counter += 1

    return questions

def update_interview_questions():
    """更新面试题为扁平化数据"""
    try:
        # 加载大健康数据
        health_data = load_health_resume_data()
        if not health_data:
            print("Failed to load health resume data")
            return False

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

        # 创建岗位到面试题的映射
        position_to_questions = {}
        for item in health_data:
            position_name = item.get('岗位名称', '')
            interview_content = item.get('面试题内容', '')

            if position_name and interview_content:
                questions = parse_interview_content_to_flat_array(interview_content)
                position_to_questions[position_name] = questions

        print(f"解析了 {len(position_to_questions)} 个岗位的面试题")

        # 更新每个岗位的面试题
        updated_content = content
        update_count = 0

        for position_name, questions in position_to_questions.items():
            if not questions:
                continue

            # 将questions数组转换为JavaScript格式的字符串
            questions_js_parts = []
            for q in questions:
                q_text = q['question'].replace('"', '\\"')
                a_text = q['answer'].replace('"', '\\"')
                question_js = '''      {
        "id": "%s",
        "question": "%s",
        "answer": "%s"
      }''' % (q['id'], q_text, a_text)
                questions_js_parts.append(question_js)

            questions_js = '''[
%s
    ]''' % ',\n'.join(questions_js_parts)

            # 查找该岗位并更新面试题
            # 先尝试替换现有的questions字段
            pattern = rf'"title": "{re.escape(position_name)}",[^}}]+?"questions": \[[^\]]*?\]'

            def replace_questions(match):
                matched_text = match.group(0)
                return re.sub(r'"questions": \[[^\]]*?\]', f'"questions": {questions_js}', matched_text, flags=re.DOTALL)

            new_content = re.sub(pattern, replace_questions, updated_content, flags=re.DOTALL)

            if new_content != updated_content:
                updated_content = new_content
                update_count += 1
                print(f"✅ 更新 {position_name} 的面试题 ({len(questions)} 个问题)")
            else:
                # 如果没有questions字段，则添加
                pattern2 = rf'"title": "{re.escape(position_name)}",[^}}]+?"requirements": \[[^\]]*?\]'

                def add_questions(match):
                    matched_text = match.group(0)
                    return matched_text + ',\n    "questions": ' + questions_js

                new_content2 = re.sub(pattern2, add_questions, updated_content, flags=re.DOTALL)

                if new_content2 != updated_content:
                    updated_content = new_content2
                    update_count += 1
                    print(f"✅ 添加 {position_name} 的面试题 ({len(questions)} 个问题)")

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

        print(f"\n🎉 成功更新 {update_count} 个岗位的扁平化面试题数据！")
        return True

    except Exception as e:
        print(f"Error updating interview questions: {e}")
        import traceback
        traceback.print_exc()
        return False

def main():
    """主函数"""
    print("开始生成符合前端期望的扁平化面试题数据...")

    success = update_interview_questions()

    if success:
        print("扁平化面试题数据更新完成！")
    else:
        print("面试题数据更新失败！")

    return success

if __name__ == "__main__":
    main()