ai_enemy.py

import pygame
from .enemy_bullet import EnemyBullet  # For relative import in a package setup


class EnemyShip:
  def __init__(self, x, y):
    self.x, self.y = x, y
    self.speed = 3
    self.color = (255, 0, 0) #red AI-enemies/ships
    self.rect = pygame.Rect(self.x, self.y, 40, 40) #entire Hitbox
    self.bullets = []



 def move_towards_player(self, player_x, player_y):
    dx, dy = player_x - self.x, player_y - self.y
    distance = math.sqrt(dx ** 2 + dy ** 2)
    if distance > 1e-6   #avoid 0/ error
         dx, dy = dx / distance, dy / distance
         self.x += dx * self.speed
         self.y += dy * self.speed
         self.rect.topleft = (self.x, self.y)

 def fire(self, player_x, player_y) 
    bullet = EnemyBullet(self.x + 20, self.y + 20, target_x, target_y) #spawn mid-center rect
    self.bullets.append(bullet)

 def update_bullets(self, screen):
        for bullet in self.bullets:
            bullet.update()
            bullet.draw(screen)
        # Remove offscreen bullets (optional)
        self.bullets = [b for b in self.bullets if 0 <= b.x <= screen.get_width() and 0 <= b.y <= screen.get_height()]

  def draw(self, screen):
        pygame.draw.rect(screen, self.color, self.rect)
        self.update_bullets(screen)