import pygame
from Environment import Environment
from Package import Package

TILE_SIZE = 50

class Forklift:
    def __init__(self, window):
        self.window = window
        self.image = pygame.image.load('resources/images.png').convert_alpha()
        self.x = 50
        self.y = 350
        self.environment = Environment(self.window)
        self.hasPackage = False
        self.package = Package(self.window)

    def drawForklift(self):
        self.environment.drawEnviroment()
        self.package.drawPackage()
        self.window.blit(self.image, (self.x, self.y))
        pygame.display.flip()

    def moveForkliftRight(self):
        self.x += TILE_SIZE
        self.drawForklift()

    def moveForkliftLeft(self):
        self.x -= TILE_SIZE
        self.drawForklift()

    def moveForkliftUp(self):
        self.y -= TILE_SIZE
        self.drawForklift()

    def moveForkliftDown(self):
        self.y += TILE_SIZE
        self.drawForklift()