from .base import Plant
from ..bullet import Bullet
from ...config import CELL_W, CELL_H

class Peashooter(Plant):
    def __init__(self, col, row):
        super().__init__(col, row, "peashooter")
        self.hp = 200
        self.max_hp = 200
        self.shoot_interval = 1.5
        self.cost = 100

    def shoot(self, game_state):
        # Check if any zombie in the same row and to the right
        has_target = False
        for z in game_state.zombies:
            if z.row == self.row and z.x > self.x:
                has_target = True
                break
        
        if has_target:
            # Spawn bullet
            bx = self.x + 40
            by = self.y + 20
            damages = [20,22,24,26,28,30,33,36,39,42]
            b = Bullet(bx, by, self.row, 360, 0, damages[self.level], "pea")
            game_state.bullets.append(b)