from typing import Any
import random
import pygame
from pygame.sprite import Sprite
from bullet import Bullet3, Bullet4, Bullet5


class Enemy(Sprite):  # 敌人的基类
    def __init__(self, ai_game):
        super().__init__()
        # 获取屏幕设置
        self.screen = ai_game.screen
        self.screen_rect = ai_game.screen.get_rect()
        self.settings = ai_game.settings
        # 自身速度
        self.speed_x = 2.0
        self.speed_y = 2.0

    def update(self, *args: Any, **kwargs: Any) -> None:
        self.moving()
        self.fire()

    def moving(self):
        self.rect.y += self.speed_y
        self.rect.x += self.speed_x
        if self.rect.y > self.settings.screen_height or \
                self.rect.left > self.screen_rect.width or \
                self.rect.right < 0:
            self.kill()

    def fire(self):
        pass


class Rock(Enemy):
    def __init__(self, ai_game):
        super().__init__(ai_game)  # 继承父类的类属性
        # 导入图片
        self.image_original = random.choice(self.settings.rocks_images).convert()  # 从所有图片中随机取一个，并convert
        self.image_original.set_colorkey(self.settings.BLACK)  # 将白色背景设置为透明
        self.image = self.image_original
        self.rect = self.image.get_rect()
        self.radius = self.rect.height / 2
        # pygame.draw.circle(self.image, self.settings.RED, self.rect.center, self.radius)
        # 固定出现位置
        self.rect.x = random.randrange(0, self.settings.screen_width - self.rect.width)
        self.rect.y = 0 - self.rect.height
        # 速度
        self.speed_x = random.randrange(-8, 8)
        self.speed_y = random.randrange(6, 16)
        # 方向检测
        self.direction = 1
        # 转动
        self.degree = 3
        self.total_degree = 0

    def moving(self):
        self.rotate()
        self.rect.y += self.speed_y
        self.rect.x += self.speed_x * self.direction
        if self.rect.right > self.screen_rect.width or self.rect.left < 0:
            self.direction *= -1
        if self.rect.y > self.settings.screen_height:
            self.kill()

    def rotate(self):
        self.total_degree += self.degree
        self.total_degree %= 360
        self.image = pygame.transform.rotate(self.image_original, self.total_degree)
        center = self.rect.center
        self.rect = self.image.get_rect(center=center)


class Alien1(Enemy):  # 外星飞船
    def __init__(self, ai_game):
        super().__init__(ai_game)  # 继承父类的类属性
        # 导入图片
        self.image = random.choice(self.settings.alien1_image).convert()
        self.image.set_colorkey(self.settings.BLACK)
        self.rect = self.image.get_rect()
        # 进入的固定位置
        self.rect.x = random.randrange(0, self.settings.screen_width - self.rect.width)
        self.rect.y = 0 - self.rect.height
        # 半径
        self.radius = self.rect.height / 2
        # 速度
        self.speed_x = 0
        self.speed_y = random.randrange(2, 4)
        # 开火时间
        self.target = ai_game.ship
        self.auto_fire_time = 0
        self.fire_rate = random.randrange(45, 55)

    def update(self, *args: Any, **kwargs: Any) -> None:
        self.moving()
        self.fire()

    def fire(self):
        if self.target.alive:
            if self.auto_fire_time == 0:
                bullet = Bullet4(self)
                # bullet.rect.center = self.rect.center
                self.settings.enemy_bullets.add(bullet)
            self.auto_fire_time += 1
            if self.auto_fire_time > self.fire_rate:
                self.auto_fire_time = 0


class Alien2(Enemy):  # ufo飞船
    def __init__(self, ai_game):
        super().__init__(ai_game)  # 继承父类的类属性
        # 导入图片
        self.rotate = False
        self.original = None
        self.image = random.choice(self.settings.alien2_image).convert()
        self.image.set_colorkey(self.settings.BLACK)
        self.rect = self.image.get_rect()
        # 进入的固定位置
        self.rect.x = random.randrange(0, self.settings.screen_width - self.rect.width)
        self.rect.y = 0 - self.rect.height
        # 半径
        self.radius = self.rect.height / 2
        # 速度
        self.speed_x = random.randrange(-1, 1)
        self.speed_y = random.randrange(5, 10)
        # 方向检测
        self.direction = 1
        # 目标为飞船
        self.target = ai_game.ship
        # 外星飞船转向
        self.angle = 0
        self.angle_total = 0
        # 向量
        self.vect = pygame.Vector2
        # 开火时间
        self.auto_fire_time = 0

    def update(self, *args: Any, **kwargs: Any) -> None:
        self.moving()
        self.rotate_to_ship()
        self.fire()

    def rotate_to_ship(self):
        self.get_ship_position()  # 返回一个角度，和飞船的相对角度
        if not self.rotate:
            self.original = self.image
            self.rotate = True
        if self.rotate:
            center = self.rect.center
            self.image = pygame.transform.rotate(self.original, -self.angle - 270)
            self.rect = self.image.get_rect(center=center)

    def get_ship_position(self):
        pos2 = self.vect(self.target.rect.centerx, self.target.rect.centery)  # 飞船的坐标
        pos1 = self.vect(self.rect.centerx, self.rect.centery)  # 自身的坐标
        pos0 = self.vect(0, 0)
        self.angle = pos0.angle_to(pos2 - pos1)
        # print(self.angle)

    def fire(self):
        if self.target.hp >= 0:
            if self.auto_fire_time == 0:
                bullet = Bullet3(self)
                bullet.rect.center = self.rect.center
                self.settings.enemy_bullets.add(bullet)
            self.auto_fire_time += 1
            if self.auto_fire_time > 10:
                self.auto_fire_time = 0


class Alien3(Enemy):  # Boss
    def __init__(self, ai_game):
        super().__init__(ai_game)  # 继承父类的类属性
        # 导入图片
        self.images = self.settings.alienBoss_images  # 列表
        self.frame = 0
        self.image = self.images[self.frame]
        self.image = pygame.transform.scale2x(self.image)
        self.rect = self.image.get_rect()
        self.last_frame = 14

        self.last_time = pygame.time.get_ticks()
        self.now_time = pygame.time.get_ticks()
        self.now_time = None
        self.frame_rate = 0
        self.temp = 0
        # 进入的固定位置
        self.rect.midbottom = ai_game.screen_rect.midtop
        self.rect.y = 0 - self.rect.height * 2
        self.radius = self.rect.width / 2
        # 速度
        self.direction = 1
        self.speed_x = 0.0
        self.speed_x_flag = 0
        self.speed_y = 4.0
        # 开火
        self.vect = pygame.Vector2
        self.auto_fire_time = 0
        self.fire_flag1 = 30
        self.auto_fire_time2 = 0
        self.fire_flag2 = 1
        self.auto_fire_time3 = 0
        self.fire_flag3 = 30
        self.target = ai_game.ship
        self.angle = None
        # 生命值
        self.hp = 200
        self.alive = 1

    def update(self, *args: Any, **kwargs: Any) -> None:
        self.change_image()
        self.moving()
        self.rotate_to_ship()
        self.fire()

    def change_image(self):
        if self.frame <= self.last_frame:
            self.image = self.images[self.frame]
            self.image = pygame.transform.scale2x(self.image)
            self.frame += 1
        else:
            self.frame = 0

    def moving(self):
        self.rect.y += self.speed_y
        self.rect.x += self.speed_x * self.direction
        if self.hp < 100 and self.speed_x_flag == 1:
            self.speed_x *= 2
            self.speed_x_flag = 2
        if self.rect.bottom == 200 and self.speed_x_flag == 0:
            while True:
                if self.speed_x != 0:
                    break
                self.speed_x = 3
            self.speed_y = 0
            self.speed_x_flag = 1
        if self.rect.right > self.screen_rect.width or self.rect.left < 0:
            self.direction *= -1
        if self.rect.y > self.settings.screen_height:
            self.kill()

    def fire(self):
        if self.target.alive:
            if self.auto_fire_time == 0 and self.hp > 150:  # 直线子弹和旋转子弹初级
                bullet3 = Bullet3(self)
                bullet3.rect.center = self.rect.center
                bullet4 = Bullet4(self)
                self.settings.enemy_bullets.add(bullet3, bullet4)
            elif self.auto_fire_time == 0 and self.hp > 85:  # 直线子弹和旋转子弹中级
                bullet3 = Bullet3(self)
                bullet3.rect.center = self.rect.center
                bullet4 = Bullet4(self)
                self.settings.enemy_bullets.add(bullet3, bullet4)
            self.auto_fire_time += 1
            if self.auto_fire_time > self.fire_flag1:
                self.auto_fire_time = 0
            if self.hp <= 100:
                self.fire_flag1 = 20

            if self.auto_fire_time2 == 0 and self.hp <= 80:  # 旋转子弹高级
                bullet3 = Bullet3(self)
                bullet3.rect.center = self.rect.center
                self.settings.enemy_bullets.add(bullet3)
            self.auto_fire_time2 += 1
            if self.auto_fire_time2 > self.fire_flag2:
                self.auto_fire_time2 = 0

            if self.auto_fire_time3 == 0 and self.hp <= 40:  # 旋转子弹高级和超级光波
                bullet5 = Bullet5(self)
                self.settings.boss_bullets.add(bullet5)
            self.auto_fire_time3 += 1
            if self.auto_fire_time3 > self.fire_flag3:
                self.auto_fire_time3 = 0

    def rotate_to_ship(self):
        self.get_ship_position()  # 返回一个角度，和飞船的相对角度

    def get_ship_position(self):
        pos2 = self.vect(self.target.rect.centerx, self.target.rect.centery)  # 飞船的坐标
        pos1 = self.vect(self.rect.centerx, self.rect.centery)  # 自身的坐标
        pos0 = self.vect(0, 0)
        self.angle = pos0.angle_to(pos2 - pos1)