from typing import Dict
from typing import List
from typing import Optional

from ytdl_sub.script.types.array import Array
from ytdl_sub.script.types.map import Map
from ytdl_sub.script.types.resolvable import Hashable
from ytdl_sub.script.types.resolvable import Resolvable
from ytdl_sub.script.types.resolvable import String
from ytdl_sub.utils.exceptions import StringFormattingException


class MapFunctions:
    @staticmethod
    def map(*key_values: Array) -> Map:
        output: Dict[Resolvable, Resolvable] = {}

        for key_value in key_values:
            if len(key_value.value) != 2:
                raise StringFormattingException(
                    "%map must take Arrays containing pairs of keys and values"
                )

            output[key_value.value[0]] = key_value.value[1]

        return Map(output)

    @staticmethod
    def get(mapping: Map, key: Hashable, default: Optional[Resolvable] = None) -> Resolvable:
        if key not in mapping.value:
            if default is not None:
                return default
            raise StringFormattingException("key not found")
        return mapping.value[key]