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

import json
import shutil
from datetime import datetime

def update_energy_hr_data():
    """更新侧边栏和HR访问量弹窗的能源产业HR数据"""
    
    # 读取能源产业HR数据
    with open('网页未导入数据/企业HR头像.json', 'r', encoding='utf-8') as f:
        all_hrs = json.load(f)
    
    # 筛选能源产业的HR
    energy_hrs = [hr for hr in all_hrs if hr['所属产业'] == '能源']
    print(f"找到 {len(energy_hrs)} 个能源产业HR")
    
    # 备份Sidebar组件
    sidebar_file = 'src/components/Sidebar/index.jsx'
    sidebar_backup = f"src/components/Sidebar/index.jsx.backup_{datetime.now().strftime('%Y%m%d_%H%M%S')}"
    shutil.copy(sidebar_file, sidebar_backup)
    print(f"已备份侧边栏组件到: {sidebar_backup}")
    
    # 备份HRVisitModal组件
    modal_file = 'src/components/HRVisitModal/index.jsx'
    modal_backup = f"src/components/HRVisitModal/index.jsx.backup_{datetime.now().strftime('%Y%m%d_%H%M%S')}"
    shutil.copy(modal_file, modal_backup)
    print(f"已备份HR访问弹窗组件到: {modal_backup}")
    
    # 更新Sidebar组件中的HR数据（显示前3个）
    with open(sidebar_file, 'r', encoding='utf-8') as f:
        sidebar_content = f.read()
    
    # 找到HR数据部分并替换
    sidebar_hr_data = []
    for i, hr in enumerate(energy_hrs[:3]):
        sidebar_hr_data.append(f'''{{
          id: {i + 1},
          avatar: "{hr['url']}",
          name: "{hr['企业HR名称']}",
          company: "{hr['企业HR所属公司']}"
        }}''')
    
    sidebar_hrs_str = ','.join(sidebar_hr_data)
    
    # 替换侧边栏中的HR数据
    # 找到hrAvatars定义的位置
    import re
    hr_avatars_pattern = r'const hrAvatars = \[(.*?)\];'
    hr_avatars_match = re.search(hr_avatars_pattern, sidebar_content, re.DOTALL)
    
    if hr_avatars_match:
        new_hr_avatars = f'const hrAvatars = [{sidebar_hrs_str}];'
        sidebar_content = sidebar_content[:hr_avatars_match.start()] + new_hr_avatars + sidebar_content[hr_avatars_match.end():]
        
        with open(sidebar_file, 'w', encoding='utf-8') as f:
            f.write(sidebar_content)
        print("已更新侧边栏HR数据")
    
    # 更新HRVisitModal组件中的HR数据（显示全部14个）
    with open(modal_file, 'r', encoding='utf-8') as f:
        modal_content = f.read()
    
    # 构建HR数据数组
    modal_hr_data = []
    for i, hr in enumerate(energy_hrs):
        modal_hr_data.append(f'''    {{
      id: {i + 1},
      avatar: "{hr['url']}",
      name: "{hr['企业HR名称']}",
      company: "{hr['企业HR所属公司']}",
      position: "HR",
      visitCount: {25 - i}
    }}''')
    
    modal_hrs_str = ',\n'.join(modal_hr_data)
    
    # 替换弹窗中的HR数据
    hr_data_pattern = r'const hrData = \[(.*?)\];'
    hr_data_match = re.search(hr_data_pattern, modal_content, re.DOTALL)
    
    if hr_data_match:
        new_hr_data = f'const hrData = [\n{modal_hrs_str}\n  ];'
        modal_content = modal_content[:hr_data_match.start()] + new_hr_data + modal_content[hr_data_match.end():]
        
        with open(modal_file, 'w', encoding='utf-8') as f:
            f.write(modal_content)
        print("已更新HR访问弹窗数据")
    
    # 显示更新结果
    print("\n能源产业HR列表:")
    for i, hr in enumerate(energy_hrs, 1):
        print(f"{i}. {hr['企业HR名称']} - {hr['企业HR所属公司']}")

if __name__ == "__main__":
    update_energy_hr_data()