from CommonServerPython import *
import demistomock as demisto
from typing import Union, List, Text


VALUE_TYPE = Union[Text, int]


class Replace(object):

    def __init__(self, value: Text, replacement: Text):
        self._value = self.get_typed_value(value)
        self.replacement = self.get_typed_value(replacement)

    def should_replace(self, value: VALUE_TYPE) -> bool:
        return self._value == value

    @staticmethod
    def get_typed_value(value: Text) -> VALUE_TYPE:
        try:
            return int(value)
        except ValueError:
            return str(value)


class RangeReplace(Replace):
    def __init__(self, start_value: Text, end_value: Text, replacement: Text):
        self._start_value = self.get_typed_value(start_value)
        self._end_value = self.get_typed_value(end_value)
        self.replacement = self.get_typed_value(replacement)

    def should_replace(self, value) -> bool:    # pylint: disable=W9014
        try:
            return self._start_value <= value <= self._end_value
        except TypeError:
            return False


def get_replace_list(map_from: List[Text], map_to: List[Text], sep: Text = '-') -> List[Replace]:
    replace_list: List[Replace] = []
    for _from, _to in zip(map_from, map_to):
        try:
            start, end = _from.split(sep)
            replace_list.append(RangeReplace(start, end, _to))
        except ValueError:
            replace_list.append(Replace(_from, _to))

    return replace_list


def replace_values(values: List[Text], replace_list: List[Replace]) -> List[VALUE_TYPE]:
    replaced_list = []
    for value in map(Replace.get_typed_value, values):
        for replace_obj in replace_list:
            if replace_obj.should_replace(value):
                value = replace_obj.replacement
                break
        replaced_list.append(value)
    return replaced_list


def main():  # pragma: no cover
    try:
        args = demisto.args()
        map_from = argToList(args['map_from'])
        map_to = argToList(args['map_to'])
        assert len(map_from) == \
            len(map_to), "the length of 'map_from' list does not match the length of 'map_to' list."
        replace_list = get_replace_list(map_from, map_to, args.get('sep', '-'))
        return_results(replace_values(argToList(args['value']), replace_list))
    except Exception as error:
        return_error(str(error), error)


if __name__ in ('__main__', '__builtin__', 'builtins'):
    main()
