import sys
import os
import time
import sdl2
import sdl2.ext
from PIL import Image

class GameWindow:
    """Classe che gestisce la finestra di gioco e il rendering grafico."""
    def __init__(self, width, height, cell_size, title="Mice!", key_callback=None):
        self.cell_size = cell_size
        self.width = width * cell_size
        self.height = height * cell_size
        self.actual_screen_size = (640, 480)
        if self.width > self.actual_screen_size[0] or self.height > self.actual_screen_size[1]:
            self.target_size = self.actual_screen_size
        else:
            self.target_size = (self.width, self.height)
        self.w_start_offset = (self.target_size[0] - self.width) // 2
        self.h_start_offset = (self.target_size[1] - self.height) // 2
        self.w_offset = self.w_start_offset
        self.h_offset = self.h_start_offset
        self.max_w_offset = self.target_size[0] - self.width
        self.max_h_offset = self.target_size[1] - self.height
        print(f"Screen size: {self.width}x{self.height}")
        self.delay = 30
        sdl2.ext.init(joystick=True)
        self.load_joystick()
        self.window = sdl2.ext.Window(title=title, size=self.target_size,)# flags=sdl2.SDL_WINDOW_FULLSCREEN)
        self.window.show()
        self.renderer = sdl2.ext.Renderer(self.window, flags=sdl2.SDL_RENDERER_ACCELERATED)
        self.factory = sdl2.ext.SpriteFactory(renderer=self.renderer)
        self.fonts = self.generate_fonts("assets/AmaticSC-Regular.ttf")
        self.running = True
        self.key_callback = key_callback

    def load_joystick(self):
        sdl2.SDL_Init(sdl2.SDL_INIT_JOYSTICK)
        sdl2.SDL_JoystickOpen(0)
        
    def generate_fonts(self,font_file):
        fonts = {}
        for i in range(10, 70, 1):
            fonts.update({i: sdl2.ext.FontManager(font_path=font_file, size=i)})
        return fonts
    
    def load_image(self, path, transparent_color=None):
        image_path = os.path.join(os.path.dirname(__file__), "..", "assets", path)
        image = Image.open(image_path)
        if transparent_color:
            image = image.convert("RGBA")
            datas = image.getdata()
            new_data = []
            for item in datas:
                if item[:3] == transparent_color:
                    new_data.append((255, 255, 255, 0))
                else:
                    new_data.append(item)
            image.putdata(new_data)
        # Ridimensiona l'immagine in base a cell_size
        scale = self.cell_size // 20
        image = image.resize((image.width * scale, image.height * scale), Image.NEAREST)
        return self.factory.from_surface(sdl2.ext.pillow_to_surface(image))
    
    def draw_text(self, text, font, position, color):
        sprite = self.factory.from_text(text, color=color, fontmanager=font)
        if position == "center":
            sprite.position = (self.size[0] // 2 - sprite.size[0] // 2, self.size[1] // 2 - sprite.size[1] // 2)
        else:
            sprite.position = position
        self.renderer.copy(sprite, dstrect=sprite.position)

    def draw_image(self, x, y, sprite, tag, anchor="nw"):
        sprite.position = (x+self.w_offset, y+self.h_offset)
        self.renderer.copy(sprite, dstrect=sprite.position)
        
    def draw_rectangle(self, x, y, width, height, tag, outline="red"):
        x, y = x + self.w_offset, y + self.h_offset
        self.renderer.draw_rect((x, y, width, height), color=sdl2.ext.Color(255, 0, 0))
        
    def draw_pointer(self, x, y):
        x=x+self.w_offset
        y=y+self.h_offset
        for i in range(3):
            self.renderer.draw_rect((x + i,y+i, self.cell_size-2*i, self.cell_size-2*i), color=sdl2.ext.Color(255, 0, 0))
        
    def delete_tag(self, tag):
        pass
    
    def get_image_size(self, image):
        return image.size
    
    def update_status(self, text):
        self.renderer.fill((10, 3, 100, 40), sdl2.ext.Color(255, 255, 255))
        self.draw_text(text, self.fonts[30], (20, 5), sdl2.ext.Color(0, 0, 0))
    
    def new_cycle(self, delay, callback):
        pass
    
    def full_screen(self,flag):
        sdl2.SDL_SetWindowFullscreen(self.window.window, flag)
        
    def mainloop(self, **kwargs):
        while self.running:
            self.renderer.clear()
            if "bg_update" in kwargs:
                kwargs["bg_update"]()
            kwargs["update"]()
            events = sdl2.ext.get_events()
            for event in events:
                if event.type == sdl2.SDL_QUIT:
                    self.running = False
                elif event.type == sdl2.SDL_KEYDOWN and self.key_callback:
                    key = sdl2.SDL_GetKeyName(event.key.keysym.sym).decode('utf-8')
                    self.key_callback(key)
                elif event.type == sdl2.SDL_MOUSEMOTION:
                    self.scroll_view((event.motion.x//self.cell_size, event.motion.y//self.cell_size))
                elif event.type == sdl2.SDL_JOYBUTTONDOWN:
                    key = sdl2.SDL_JoystickGetButton(event.jbutton.which, event.jbutton.button)
                    self.key_callback(key)
                    
            # Disegna qui gli sprite
            self.renderer.present()
            sdl2.SDL_Delay(self.delay)
            
    def close(self):
        self.running = False
        sdl2.ext.quit()

    def scroll_view(self, pointer):
        x, y = pointer
        x = x//2
        y = y//2
        x = -x * self.cell_size
        y = -y * self.cell_size
        
        if not x <= self.max_w_offset + self.cell_size:
            self.w_offset = x
        else:
            self.w_offset = self.max_w_offset 
        
        
        if not y < self.max_h_offset:
            self.h_offset = y