import string
from collections import defaultdict
from typing import Any, Iterator

from ..base import BaseSolver

NOT_A_SYMBOL = "." + string.digits


class Solver(BaseSolver):
    def solve(self, input: str) -> Iterator[Any]:
        lines = input.splitlines()

        values: list[int] = []
        gears: dict[tuple[int, int], list[int]] = defaultdict(list)

        for i, line in enumerate(lines):
            j = 0
            while j < len(line):
                # skip everything until a digit is found (start of a number)
                if line[j] not in string.digits:
                    j += 1
                    continue

                # extract the range of the number and its value
                k = j + 1
                while k < len(line) and line[k] in string.digits:
                    k += 1

                value = int(line[j:k])

                # lookup around the number if there is a symbol - we go through the number
                # itself but that should not matter since it only contains digits
                found = False
                for i2 in range(max(0, i - 1), min(i + 1, len(lines) - 1) + 1):
                    for j2 in range(max(0, j - 1), min(k, len(line) - 1) + 1):
                        assert i2 >= 0 and i2 < len(lines)
                        assert j2 >= 0 and j2 < len(line)

                        if lines[i2][j2] not in NOT_A_SYMBOL:
                            found = True

                            if lines[i2][j2] == "*":
                                gears[i2, j2].append(value)

                if found:
                    values.append(value)

                # continue starting from the end of the number
                j = k

        yield sum(values)
        yield sum(v1 * v2 for v1, v2 in filter(lambda vs: len(vs) == 2, gears.values()))