# library imports
import pygame
from pygame.math import Vector2 as vector
from os import walk


class Player(pygame.sprite.Sprite):

	def __init__(self,pos, groups, path, collision_sprites):
		super().__init__(groups)

		self.import_assets(path)
		self.frame_index = 0 
		self.status = 'down_idle'

		# sprite
		self.image = self.animations[self.status][self.frame_index]
		self.rect = self.image.get_rect(center = pos)

		# movement
		self.pos = vector(self.rect.center)
		self.direction = vector()
		self.speed = 200

		# collisions
		self.hitbox = self.rect.inflate(0,-self.rect.height / 2)
		self.collision_sprites = collision_sprites

		self.attacking = False

	def get_status(self):
		# idle
		if self.direction.x == 0 and self.direction.y == 0:
			self.status = self.status.split('_')[0] + '_idle'
		# attacking
		if self.attacking:
			self.status = self.status.split('_')[0] + '_attack'

	def import_assets(self, path):
		self.animations = {}

		for index,folder in enumerate(walk(path)):
			if index == 0:
				for name in folder[1]:
					self.animations[name] = []

			else:
				for file_name in sorted(folder[2], key = lambda string: int(string.split('.')[0])):
					path = folder[0] + '/' + file_name
					surf = pygame.image.load(path).convert_alpha()
					key = folder[0].split('/')[3]
					self.animations[key].append(surf)

	def input(self):
		keys = pygame.key.get_pressed()
		
		if not self.attacking:
			# horizontal input 
			if keys[pygame.K_RIGHT]:
				self.direction.x = 1
				self.status = 'right'
			elif keys[pygame.K_LEFT]:
				self.direction.x = -1
				self.status = 'left'
			else:
				self.direction.x = 0

			# vertical input 
			if keys[pygame.K_UP]:
				self.direction.y = -1
				self.status = 'up'
			elif keys[pygame.K_DOWN]:
				self.direction.y = 1
				self.status = 'down'
			else:
				self.direction.y = 0

			if keys[pygame.K_SPACE]:
				self.attacking = True
				self.direction = vector()
				self.frame_index = 0

	def move(self,dt):

		# normalize movement
		if self.direction.magnitude() != 0:
			self.direction = self.direction.normalize()

		# vertical movement + hitbox
		self.pos.y += self.direction.y * self.speed * dt
		self.hitbox.centery = round(self.pos.y)
		self.rect.centery = self.hitbox.centery
		self.collision('vertical')


		# horizontal movement + hitbox
		self.pos.x += self.direction.x * self.speed * dt
		self.hitbox.centerx = round(self.pos.x)
		self.rect.centerx = self.hitbox.centerx
		self.collision('horizontal')

	def animate(self,dt):
		current_animation = self.animations[self.status]

		self.frame_index += 10 * dt
		if self.frame_index >= len(current_animation):
			self.frame_index = 0
			if self.attacking:
				self.attacking = False

		self.image = current_animation[int(self.frame_index)]

	def collision(self,direction):

		for sprite in self.collision_sprites.sprites():
			if sprite.hitbox.colliderect(self.hitbox):
				# if hasattr(sprite,'name') and sprite.name == '':
				# 	pygame.quit
				# 	sys.exit

				if direction == 'horizontal':
					if self.direction.x > 0: #moving right
						self.hitbox.right = sprite.hitbox.left
						self.rect.centerx = self.hitbox.centerx
						self.pos.x = self.hitbox.centerx
					if self.direction.x < 0: #moving right
						self.hitbox.left = sprite.hitbox.right
						self.rect.centerx = self.hitbox.centerx
						self.pos.x = self.hitbox.centerx	
				else:		
					if self.direction.y > 0: # moving down
						self.hitbox.bottom = sprite.hitbox.top
						self.rect.centery = self.hitbox.centery
						self.pos.y = self.hitbox.centery
					if self.direction.y < 0: # moving up
						self.hitbox.top = sprite.hitbox.bottom
						self.rect.centery = self.hitbox.centery
						self.pos.y = self.hitbox.centery


	def update(self,dt):
		self.input()
		self.get_status()
		self.move(dt)
		self.animate(dt)