from survival import esper, GameMap
from survival.components.movement_component import MovementComponent
from survival.components.moving_component import MovingComponent
from survival.components.position_component import PositionComponent
from survival.components.sprite_component import SpriteComponent


class MovementSystem(esper.Processor):
    def __init__(self, game_map: GameMap):
        self.map = game_map

    def process(self, dt):
        for ent, (mov, pos, moving, sprite) in self.world.get_components(MovementComponent, PositionComponent,
                                                                         MovingComponent,
                                                                         SpriteComponent):
            # cost = self.map.get_cost(moving.target)
            # pos.position[0] += moving.direction_vector[0] * mov.speed * dt / 100 / cost
            # pos.position[1] += moving.direction_vector[1] * mov.speed * dt / 100 / cost
            #
            # if abs(moving.target[0] * 32 - pos.position[0]) < 1 * mov.speed and abs(
            #         pos.position[1] - moving.target[1] * 32) < 1 * mov.speed:
            #     pos.position = [moving.target[0] * 32, moving.target[1] * 32]
            #     self.world.remove_component(ent, MovingComponent)
            pos.position = [moving.target[0] * 32, moving.target[1] * 32]
            self.world.remove_component(ent, MovingComponent)