import pygame
import os
import random

SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
LOGO_SCALING = 0.1


class DVDLogo(pygame.sprite.Sprite):
    def __init__(self, window_size):
        super().__init__()

        self.base_image = pygame.image.load(SCRIPT_DIR + '/resources/dvd.png')
        self.base_image.convert_alpha()
        self.base_image = pygame.transform.scale(self.base_image, (self.base_image.get_width() * LOGO_SCALING,
                                                                   self.base_image.get_height() * LOGO_SCALING))
        self.image = self.base_image.copy()

        color_surface = pygame.Surface(self.image.get_size(), pygame.SRCALPHA)
        color_surface.fill((64, 64, 255))
        self.image.blit(color_surface, (0, 0), special_flags=pygame.BLEND_RGBA_MULT)

        self.rect = self.image.get_rect()
        self.x_speed = 4
        self.y_speed = 4

        self.max_x = window_size[0] - self.rect.width
        self.max_y = window_size[1] - self.rect.height

    def random_color(self):
        color_surface = pygame.Surface(self.image.get_size(), pygame.SRCALPHA)
        new_color = (random.randint(0, 255),
                     random.randint(0, 255),
                     random.randint(0, 255))
        color_surface.fill(new_color)
        self.image = self.base_image.copy()
        self.image.blit(color_surface, (0, 0), special_flags=pygame.BLEND_RGBA_MULT)

    def update(self):
        self.rect.x += self.x_speed
        self.rect.y += self.y_speed

        if self.rect.x < 0 or self.rect.x > self.max_x:
            self.x_speed *= -1
            self.random_color()
        if self.rect.y < 0 or self.rect.y > self.max_y:
            self.y_speed *= -1
            self.random_color()

        self.rect.x = max(0, min(self.rect.x, self.max_x))
        self.rect.y = max(0, min(self.rect.y, self.max_y))


class Surface(pygame.Surface):
    def __init__(self, window_size):
        super().__init__(window_size, pygame.SRCALPHA)
        self.running = True
        self.quit = False
        self.next_surface = ''

        pygame.mouse.set_visible(False)

        dvd_logo = DVDLogo(window_size)
        self.all_sprites = pygame.sprite.Group(dvd_logo)

    def update(self):
        self.fill(pygame.colordict.THECOLORS.get('black'))

        for event in pygame.event.get():
            match event.type:
                case pygame.MOUSEBUTTONDOWN:
                    self.running = False
                    self.quit = True

        self.all_sprites.update()
        self.all_sprites.draw(self)