from playwright.sync_api import sync_playwright
from jinja2 import Environment, FileSystemLoader
import utils
import json
import os


class ImageGenerator:
    def __init__(self):
        self.env = Environment(loader=FileSystemLoader('templates'))
        # 加载英雄数据
        with open('heroes.json', 'r', encoding='utf-8') as f:
            self.heroes_data = json.load(f)
        
    def get_hero_image_url(self, hero_id):
        """根据英雄id获取图片URL"""
        hero_info = self.heroes_data.get(str(hero_id))
        if hero_info:
            return f"https://cdn.cloudflare.steamstatic.com{hero_info['img']}"
        return None

    def generate_match_report(self, match_data):
        """
        根据match_data生成比赛报告图片
        match_data应该是dota.py中serialize_match方法返回的格式
        """
        # 处理数据，标记最高经济和最高伤害的玩家
        radiant_players = []
        dire_players = []
        
        # 分离天辉和夜魇玩家
        for player in match_data['players']:
            player_data = {
                'name': player['personaname'] if player['personaname'] else '-',
                'is_friend': bool(player.get('nickname', '')),
                'hero': player['hero'],
                'level': player['level'],
                'kills': player['kills'],
                'deaths': player['deaths'],
                'assists': player['assists'],
                'total_gold': player['total_gold'],
                'hero_damage': player['hero_damage'],
                'hero_img': self.get_hero_image_url(player['hero_id']),
                'highest_gold': False,
                'highest_damage': False,
                'rank': player.get('rank'),
                'rank_tier': player.get('rank_tier'),
            }
            
            # 如果是好友，使用昵称
            if player.get('nickname'):
                player_data['name'] = player['nickname']
                
            # 转换英雄名称为中文
            player_data['hero'] = utils.get_hero_chinese_name(player['hero'])
                
            if player['is_radiant']:
                radiant_players.append(player_data)
            else:
                dire_players.append(player_data)
        
        # 找出天辉最高经济和伤害
        if radiant_players:
            max_gold_player = max(radiant_players, key=lambda p: int(p['total_gold']))
            max_damage_player = max(radiant_players, key=lambda p: int(p['hero_damage']))
            
            for player in radiant_players:
                if player == max_gold_player:
                    player['highest_gold'] = True
                if player == max_damage_player:
                    player['highest_damage'] = True
        
        # 找出夜魇最高经济和伤害
        if dire_players:
            max_gold_player = max(dire_players, key=lambda p: int(p['total_gold']))
            max_damage_player = max(dire_players, key=lambda p: int(p['hero_damage']))
            
            for player in dire_players:
                if player == max_gold_player:
                    player['highest_gold'] = True
                if player == max_damage_player:
                    player['highest_damage'] = True
        
        
        # 短数字
        for player in radiant_players:
            player['total_gold'] = utils.shorten_digits(player['total_gold'])
            player['hero_damage'] = utils.shorten_digits(player['hero_damage'])
        for player in dire_players:
            player['total_gold'] = utils.shorten_digits(player['total_gold'])
            player['hero_damage'] = utils.shorten_digits(player['hero_damage'])

        end_time = match_data.get('end_time', '')
        # '2025-03-05T01:04:30.000+08:00' get 01:04:30
        end_time = end_time.split('T')[1].split('.')[0]

        # 准备模板数据
        template_data = {
            'radiant_players': radiant_players,
            'dire_players': dire_players,
            'radiant_score': match_data.get('radiant_score', 0),
            'dire_score': match_data.get('dire_score', 0),
            'duration': match_data.get('duration', 0),
            'radiant_win': match_data.get('radiant_win', False),
            'start_time': match_data.get('start_time', ''),
            'end_time': end_time,
        }
        
        # 渲染模板
        template = self.env.get_template('match_report.html')
        html_content = template.render(**template_data)
        
        # 使用Playwright生成图片
        with sync_playwright() as playwright:
            browser = playwright.chromium.launch()
            page = browser.new_page()
            page.set_content(html_content)
            page.set_viewport_size({"width": 800, "height": 800})
            
            # 等待内容完全加载
            page.wait_for_timeout(1000)
            
            # 调整截图高度以适应内容
            body_height = page.evaluate('document.body.scrollHeight')
            body_width = page.evaluate('document.body.offsetWidth')
            page.set_viewport_size({"width": body_width, "height": body_height})
            
            # 截图
            image_path = "match_report_%s.png" % match_data.get('match_id')
            page.screenshot(path=image_path, full_page=True)
            browser.close()
            image_url = utils.upload_image(image_path, image_path)
            os.remove(image_path)
            return image_url
        return None