from collections import deque
from maze import MazeBuilder

def build_path(previous, start, finish):
    if finish not in previous:
        return []
    path = []
    current = finish
    while current != start:
        path.append(current)
        current = previous[current]
    path.append(start)
    path.reverse()
    return path

class BfsStrategy:
    def solve(self, maze):
        start = maze.start
        finish = maze.finish
        queue = deque([start])
        previous = {start: None}
        visited_count = 0

        while queue:
            current = queue.popleft()
            visited_count += 1

            if current == finish:
                break

            for next_cell in maze.neighbors(current[0], current[1]):
                if next_cell not in previous:
                    previous[next_cell] = current
                    queue.append(next_cell)

        path = build_path(previous, start, finish)
        return {
            "name": "BFS",
            "path": path,
            "visited": visited_count,
            "length": len(path)
        }
class DfsStrategy:
    def solve(self, maze):
        start = maze.start
        finish = maze.finish
        stack = [start]
        previous = {start: None}
        visited_count = 0

        while stack:
            current = stack.pop()
            visited_count += 1

            if current == finish:
                break

            for next_cell in maze.neighbors(current[0], current[1]):
                if next_cell not in previous:
                    previous[next_cell] = current
                    stack.append(next_cell)

        path = build_path(previous, start, finish)
        return {
            "name": "DFS",
            "path": path,
            "visited": visited_count,
            "length": len(path)
        }

class MazeSolver:
    def __init__(self, strategy):
        self.strategy = strategy

    def solve(self, maze):
        return self.strategy.solve(maze)

if __name__ == "__main__":
    maze = MazeBuilder().from_file("raskatovia/docs/data/task2/maps/simple.txt").build()
    strategies = [BfsStrategy(), DfsStrategy()]

    for strategy in strategies:
        solver = MazeSolver(strategy)
        result = solver.solve(maze)
        print("algorithm:", result["name"])
        print("visited:", result["visited"])
        print("length:", result["length"])
        print(maze.draw(result["path"]))
        print()