from Snake import Snake
from Vec import Vec2 
from Box import Box

class Walls:
    """Contains the walls of a game.
    Supports iterating over the walls and checking if there is a wall at a specific coordinate.
    Useful as a wrapper to later increase performance of these operations."""

    _walls = []

    def __init__(self, walls):
        self._walls = walls

    def fromString(wallString):
        walls = []
        y = 0
        for line in wallString.split('\n'):
            x = 0
            for char in line:
                if char == "#":
                    walls.append(Vec2(x, y))
                x += 1
            y += 1
        return Walls(walls)

    def walls(self):
        return self._walls
        
    def wallAt(self, pos):
        return (pos in self._walls)

    def width(self):
        return max(self._walls, key=lambda p: p.x).x + 1

    def height(self):
        return max(self._walls, key=lambda p: p.y).y + 1


class Game:
    """Class responsible for the main logic of a game.
       For a game, this will probably be a singleton."""
    
    snake = None
    boxes = []
    walls = None

    def __init__(self):
        self.snake = Snake([Vec2(6, 6), Vec2(6, 7), Vec2(6,8), Vec2(7,8)], self)
        _box1 = Box(Vec2(11, 4))
        _box2 = Box(Vec2(3, 10))
        self.boxes = [_box1, _box2]
        self.walls = Walls.fromString("""###############
#             #
#  f          #
#             #
#          b  #
#             #
#     h       #
#     t       #
#     tt      #
#             #
#  b          #
#      ###    #
#       f     #
###############""")

    def width(self): return self.walls.width()
    def height(self): return self.walls.height()

    def tick(self):
        self.snake.move()

    def isLost(self):
        return self.snake.hasCollided

    def enter(self, pos: Vec2, inDir: Vec2) -> bool:
        boxAt = next(filter(lambda box: box.pos == pos, self.boxes), None)
        if self.walls.wallAt(pos):
            return False
        elif boxAt != None:
            if self.enter(pos + inDir, inDir):
                boxAt.pos = pos + inDir
                return True
            else:
                return False
        else:
            return True