from .base import Plant

class Reshaper(Plant):
    def __init__(self, col, row):
        super().__init__(col, row, "reshaper")
        self.hp = 200
        self.max_hp = 200
        self.shoot_interval = 0
        self.cost = 50
        self.life_timer = 0

    def update(self, dt, game_state):
        # Instant effect then die
        self.life_timer += dt
        if self.life_timer > 3.0:
            self.active = False
            
            # Find target plant at this location (excluding self)
            target = None
            for p in game_state.plants:
                if p.col == self.col and p.row == self.row and p != self and p.active:
                    target = p
                    break
            
            if target:
                # Refund sun
                # We need to know the cost of the target. 
                # Since we don't have a global config easily accessible here, 
                # we rely on the plant instance having a 'cost' attribute.
                base_refund = getattr(target, 'cost', 0)
                
                # 计算植物玩家数量
                plant_player_count = sum(1 for role in game_state.roles.values() if role == 'plant')
                
                # 双人植物玩家时返还减半
                refund_amount = base_refund // 2 if plant_player_count >= 2 else base_refund
                
                if game_state.player_states:
                    for username in game_state.player_states:
                        game_state.player_states[username]['sun'] += refund_amount
                        if target.type in game_state.player_states[username]['cooldowns']:
                            del game_state.player_states[username]['cooldowns'][target.type]
                else:
                    game_state.sun += refund_amount
                    # Reset cooldown
                    if target.type in game_state.plant_cooldowns:
                        del game_state.plant_cooldowns[target.type]
                
                # Remove target
                target.active = False