import os
import pygame
from math import sin

from utils.resource_loader import import_folder
from configs.system.window_config import HITBOX_OFFSET


class AnimationHandler:

    def __init__(self):
        self.frame_index = 0
        self.animation_speed = 0.15

    def import_assets(self, sprite_type, position, name=None):
        script_dir = os.path.dirname(os.path.abspath(__file__))
        asset_path = os.path.join(
            script_dir, '../../..', 'assets', 'graphics')
        if sprite_type == 'player':

            character_path = f"{asset_path}/player"

            # Import Graphic Assets
            self.image = pygame.image.load(
                f"{character_path}/down/down_0.png").convert_alpha()
            self.rect = self.image.get_rect(topleft=position)
            self.hitbox = self.rect.inflate(HITBOX_OFFSET[sprite_type])
            self.animations = {
                'up': [], 'down': [], 'left': [], 'right': [],
                'up_idle': [], 'down_idle': [], 'left_idle': [], 'right_idle': [],
                'up_attack': [], 'down_attack': [], 'left_attack': [], 'right_attack': []
            }
            for animation in self.animations.keys():
                full_path = f"{character_path}/{animation}"
                self.animations[animation] = import_folder(full_path)

        elif sprite_type == 'enemy':

            character_path = f"{asset_path}/monsters/{name}"
            self.animations = {'idle': [], 'move': [], 'attack': []}
            main_path = f"{asset_path}/monsters/{name}"

            for animation in self.animations.keys():
                self.animations[animation] = import_folder(
                    f"{main_path}/{animation}")

    def animate(self, status, vulnerable):
        animation = self.animations[status]

        self.frame_index += self.animation_speed
        if self.frame_index >= len(animation):
            self.frame_index = 0

        # Set the image
        self.image = animation[int(self.frame_index)]
        self.rect = self.image.get_rect(center=self.hitbox.center)

        if not vulnerable:
            alpha = self.wave_value()
            self.image.set_alpha(alpha)
        else:
            self.image.set_alpha(255)

    def wave_value(self):
        value = sin(pygame.time.get_ticks())
        if value >= 0:
            return 255
        else:
            return 0