# Copyright 2012 Dan Smith <dsmith@danplanet.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import logging
import warnings

from chirp import chirp_common

LOG = logging.getLogger(__name__)
BANNED_NAME_CHARACTERS = r'%'


class InvalidValueError(Exception):

    """An invalid value was specified for a given setting"""
    pass


class InternalError(Exception):

    """A driver provided an invalid settings object structure"""
    pass


class InvalidNameError(Exception):
    """A driver provided a setting with an invalid name"""
    pass


class RadioSettingValue:

    """Base class for a single radio setting"""

    def __init__(self):
        self._current = None
        self._preinit_current = None
        self._has_changed = False
        self._validate_callback = self.null_callback
        self._mutable = True

    def initialize(self):
        """Initialize this value to the stashed value"""
        if isinstance(self._preinit_current, Exception):
            # This was communicated as an error during loading
            return
        assert self._current is None and self._preinit_current is not None
        self.set_value(self._preinit_current)
        self._preinit_current = None

    def queue_current(self, current):
        """Set this value's current on initialization"""
        self._preinit_current = current

    def null_callback(self, x):
        return x

    def set_mutable(self, mutable):
        self._mutable = mutable

    def get_mutable(self):
        return self._mutable

    def changed(self):
        """Returns True if the setting has been changed since init"""
        return self._has_changed

    def set_validate_callback(self, callback):
        self._validate_callback = callback

    def set_value(self, value):
        """Sets the current value, triggers changed"""
        if not self.get_mutable() and self.initialized:
            raise InvalidValueError("This value is not mutable")

        if self._current is not None and value != self._current:
            self._has_changed = True
        self._current = self._validate_callback(value)

    def get_value(self):
        """Gets the current value"""
        if not self.initialized:
            LOG.error('Accessing setting value before init!')
        return self._current

    def __trunc__(self):
        return int(self.get_value())

    def __int__(self):
        return int(self.get_value())

    def __float__(self):
        return float(self.get_value())

    def __str__(self):
        return str(self.get_value())

    def __floordiv__(self, value):
        return int(self.get_value()) // value

    def __div__(self, value):
        return float(self.get_value()) / value

    def __mul__(self, value):
        return self.get_value() * value

    def __gt__(self, value):
        return self.get_value() > value

    def __lt__(self, value):
        return self.get_value() < value

    def __eq__(self, value):
        return self.get_value() == value

    @property
    def initialized(self):
        return self._current is not None


class RadioSettingValueInteger(RadioSettingValue):

    """An integer setting"""

    def __init__(self, minval, maxval, current, step=1):
        RadioSettingValue.__init__(self)
        self._min = minval
        self._max = maxval
        self._step = step
        self.queue_current(current)

    def set_value(self, value):
        try:
            value = int(value)
        except Exception:
            raise InvalidValueError("An integer is required")
        if value > self._max or value < self._min:
            raise InvalidValueError("Value %i not in range %i-%i" %
                                    (value, self._min, self._max))
        RadioSettingValue.set_value(self, value)

    def get_min(self):
        """Returns the minimum allowed value"""
        return self._min

    def get_max(self):
        """Returns the maximum allowed value"""
        return self._max

    def get_step(self):
        """Returns the step increment"""
        return self._step


class RadioSettingValueFloat(RadioSettingValue):

    """A floating-point setting"""

    def __init__(self, minval, maxval, current, resolution=0.001, precision=4):
        RadioSettingValue.__init__(self)
        self._min = minval
        self._max = maxval
        self._res = resolution
        self._pre = precision
        self.queue_current(current)

    def format(self, value=None):
        """Formats the value into a string"""
        if value is None:
            value = self._current
        fmt_string = "%%.%if" % self._pre
        return fmt_string % value

    def set_value(self, value):
        try:
            value = float(value)
        except Exception:
            raise InvalidValueError("A floating point value is required")
        if value > self._max or value < self._min:
            raise InvalidValueError("Value %s not in range %s-%s" % (
                self.format(value),
                self.format(self._min), self.format(self._max)))

        # FIXME: honor resolution

        RadioSettingValue.set_value(self, value)

    def get_min(self):
        """Returns the minimum allowed value"""
        return self._min

    def get_max(self):
        """Returns the maximum allowed value"""


class RadioSettingValueBoolean(RadioSettingValue):

    """A boolean setting

    If using MemSetting, mem_vals can be set to the False, True values
    that are to be actually stored in memory instead of 0, 1
    """

    def __init__(self, current, mem_vals=(0, 1)):
        self._mem_vals = mem_vals
        RadioSettingValue.__init__(self)
        self.queue_current(current)

    def set_value(self, value):
        RadioSettingValue.set_value(self, bool(value))

    def __bool__(self):
        return bool(self.get_value())
    __nonzero__ = __bool__

    def __str__(self):
        return str(bool(self.get_value()))


class RadioSettingValueInvertedBoolean(RadioSettingValueBoolean):
    """A boolean value that is actually stored inverted in memory.

    Only really useful for MemSetting, which will invert the value when
    actually setting it in memory.
    """
    pass


class RadioSettingValueList(RadioSettingValue):

    """A list-of-strings setting"""

    def __init__(self, options, current=None, current_index=0):
        RadioSettingValue.__init__(self)
        self._options = list(options)
        if current is not None:
            # Using current_index is safer than using current because we can
            # gracefully handle out-of-range values without crashing. All new
            # code should use current_index and old code should be fixed.
            warnings.warn(
                'RadioSettingValueList should be provided with '
                'current_index instead of current (value) for '
                'safety', FutureWarning, stacklevel=2)
        self.queue_current(current if current is not None
                           else int(current_index))

    def set_value(self, value):
        if isinstance(value, int):
            try:
                value = self._options[value]
            except IndexError:
                raise InvalidValueError("Value for %s is out of range" % value)
        if value not in self._options:
            raise InvalidValueError("%s is not valid for this setting" % value)
        RadioSettingValue.set_value(self, value)

    def set_index(self, index):
        # Will raise IndexError as expected
        self.set_value(self._options[int(index)])

    def get_options(self):
        """Returns the list of valid option values"""
        return self._options

    def __trunc__(self):
        return self._options.index(self._current)

    def __int__(self):
        return self._options.index(self._current)


class RadioSettingValueString(RadioSettingValue):

    """A string setting"""

    def __init__(self, minlength, maxlength, current,
                 autopad=True, charset=chirp_common.CHARSET_ASCII,
                 mem_pad_char=' '):
        RadioSettingValue.__init__(self)
        self._minlength = minlength
        self._maxlength = maxlength
        self._charset = charset
        self._autopad = autopad
        self._mem_pad_char = mem_pad_char
        self.queue_current(current)

    @property
    def mem_pad_char(self):
        return self._mem_pad_char

    @property
    def autopad(self):
        return self._autopad

    @property
    def maxlength(self):
        return self._maxlength

    @property
    def minlength(self):
        return self._minlength

    def set_charset(self, charset):
        """Sets the set of allowed characters"""
        self._charset = charset

    def set_value(self, value):
        if len(value) < self._minlength or len(value) > self._maxlength:
            raise InvalidValueError(
                "Value %r must be between %i and %i chars" % (
                    value, self._minlength, self._maxlength))
        for char in value:
            if char not in self._charset:
                raise InvalidValueError(("Value %r contains invalid "
                                         "character `%s'") % (value, char))
        if self._autopad:
            value = value.ljust(self._maxlength)
        RadioSettingValue.set_value(self, value)

    def __str__(self):
        return self._current

    def __len__(self):
        return len(self._current)

    def __getitem__(self, i):
        return self._current[i]


class RadioSettingValueMap(RadioSettingValueList):

    """Map User Options to Radio Memory Values

    Provides User Option list for GUI, maintains state, verifies new values,
    and allows {setting,getting} by User Option OR Memory Value.  External
    conversions not needed.

    """

    def __init__(self, map_entries, mem_val=None, user_option=None):
        """Create new map

        Pass in list of 2 member tuples, typically of type (str, int),
        for each Radio Setting.  First member of each tuple is the
        User Option Name, second is the Memory Value that corresponds.
        An example is APO: ("Off", 0), ("0.5", 5), ("1.0", 10).

        """
        # Catch bugs early by testing tuple geometry
        for map_entry in map_entries:
            if not len(map_entry) == 2:
                raise InvalidValueError("map_entries must be 2 el tuples "
                                        "instead of: %s" % str(map_entry))
        user_options = [e[0] for e in map_entries]
        self._mem_vals = [e[1] for e in map_entries]
        RadioSettingValueList.__init__(self, user_options, current_index=0)
        if mem_val is not None:
            try:
                self.queue_current(self._mem_vals.index(mem_val))
            except ValueError:
                LOG.error('Memory value %r not in value list %r',
                          mem_val, self._mem_vals)
                # Set this to just out of bounds so that the invalid-current
                # handling detects it and flags it as invalid to the user.
                self.queue_current(len(self._mem_vals))
        elif user_option is not None:
            self.queue_current(user_option)
        self._has_changed = False

    def set_mem_val(self, mem_val):
        """Change setting to User Option that corresponds to 'mem_val'"""
        if mem_val in self._mem_vals:
            index = self._mem_vals.index(mem_val)
            self.set_value(self._options[index])
        else:
            raise InvalidValueError(
                "%s is not valid for this setting" % mem_val)

    def get_mem_val(self):
        """Get the mem val corresponding to the currently selected user
        option"""
        return self._mem_vals[self._options.index(self.get_value())]

    def __trunc__(self):
        """Return memory value that matches current user option"""
        index = self._options.index(self._current)
        value = self._mem_vals[index]
        return value

    def __int__(self):
        """Return memory value that matches current user option"""
        index = self._options.index(self._current)
        value = self._mem_vals[index]
        return value


def zero_indexed_seq_map(user_options):
    """RadioSettingValueMap factory method

    Radio Setting Maps commonly use a list of strings that map to a sequence
    that starts with 0.  Pass in a list of User Options and this function
    returns a list of tuples of form (str, int).

    """
    mem_vals = list(range(0, len(user_options)))
    return list(zip(user_options, mem_vals))


class RadioSettings(list):

    def __init__(self, *groups):
        list.__init__(self, groups)

    def __str__(self):
        items = [str(self[i]) for i in range(0, len(self))]
        return "\n".join(items)

    def __getitem__(self, key):
        if isinstance(key, int):
            return super().__getitem__(key)

        for element in self:
            if element.get_name() == key:
                return element

        raise KeyError(f"Invalid key {key}")

    def walk(self):
        """Iterate over all the RadioSettings in this tree

        Returns a generator.
        """
        for element in self:
            if isinstance(element, RadioSetting):
                yield element
            elif isinstance(element, RadioSettingGroup):
                for subel in element.walk():
                    yield subel
            else:
                raise ValueError('Unable to iterate %r' % element)

    def apply_to(self, memobj):
        """Walk all the settings in this tree, apply the direct-to-mem ones.

        Returns all the settings that are not MemSetting objects to be applied
        manually.
        """
        non_mem_settings = []
        for setting in self.walk():
            print("Walking %r" % setting)
            if isinstance(setting, MemSetting):
                setting.apply_to_memobj(memobj)
            else:
                non_mem_settings.append(setting)
        return non_mem_settings


class RadioSettingGroup(object):

    """A group of settings"""

    def _validate(self, element):
        # RadioSettingGroup can only contain RadioSettingGroup objects
        if not isinstance(element, RadioSettingGroup):
            raise InternalError("Incorrect type %s" % type(element))

    def __init__(self, name, shortname, *elements):
        for c in BANNED_NAME_CHARACTERS:
            if c in name:
                raise InvalidNameError(
                    'Name must not contain %r character' % c)
        self._name = name            # Setting identifier
        self._shortname = shortname  # Short human-readable name/description
        self.__doc__ = name          # Longer explanation/documentation
        self._elements = {}
        self._element_order = []
        self._frozen = False

        for i, element in enumerate(elements):
            self._validate(element)
            if (isinstance(element, RadioSettingValue) and
                    not element.initialized):
                # Late-initialize this value
                try:
                    element.initialize()
                except InvalidValueError as e:
                    LOG.error('Failed to load %s value %i: %s',
                              self._name, i, e)

            self.append(element)

    def walk(self):
        for el in self._elements.values():
            if isinstance(el, RadioSetting):
                yield el
            if isinstance(el, RadioSettingGroup):
                for subel in el.walk():
                    yield subel

    def set_frozen(self):
        self._frozen = True
        for i in self:
            if isinstance(i, RadioSettingGroup):
                i.set_frozen()

    def get_name(self):
        """Returns the group name"""
        return self._name

    def get_shortname(self):
        """Returns the short group identifier"""
        return self._shortname

    def set_shortname(self, new_name):
        self._shortname = new_name

    def set_doc(self, doc):
        """Sets the docstring for the group"""
        self.__doc__ = doc

    def __str__(self):
        string = "group '%s': {\n" % self._name
        for element in sorted(self._elements.values()):
            for line in str(element).split("\n"):
                string += "\t" + line + "\n"
        string += "}"
        return string

    # Kinda list interface

    def append(self, element):
        """Adds an element to the group"""
        if self._frozen:
            raise ValueError('Setting is frozen')
        self[element.get_name()] = element

    def __iter__(self):
        class RSGIterator:

            """Iterator for a RadioSettingGroup"""

            def __init__(self, rsg):
                self.__rsg = rsg
                self.__i = 0

            def __iter__(self):
                return self

            def next(self):
                return self.__next__()

            def __next__(self):
                """Next Iterator Interface"""
                if self.__i >= len(self.__rsg.keys()):
                    raise StopIteration()
                e = self.__rsg[self.__rsg.keys()[self.__i]]
                self.__i += 1
                return e
        return RSGIterator(self)

    # Dictionary interface

    def __len__(self):
        return len(self._elements)

    def __getitem__(self, name):
        return self._elements[name]

    def __setitem__(self, name, value):
        if name in self._element_order:
            raise KeyError("Duplicate item %s" % name)
        self._elements[name] = value
        self._element_order.append(name)

    def __delitem__(self, item):
        del self._elements[item.get_name()]
        self._element_order.remove(item.get_name())

    def __contains__(self, name):
        return name in self._elements

    def items(self):
        """Returns a key=>value set of elements, like a dict"""
        return [(name, self._elements[name]) for name in self._element_order]

    def keys(self):
        """Returns a list of string element names"""
        return self._element_order

    def values(self):
        """Returns the list of elements"""
        return [self._elements[name] for name in self._element_order]

    def __lt__(self, other):
        return self._name < other._name


class RadioSettingSubGroup(RadioSettingGroup):
    """A subgroup of related settings.

    These are to be displayed with less weight than a regular group. Used
    for repeated sections of related settings, or small groups that don't
    merit their own panel of display in the UI.
    """


class RadioSetting(RadioSettingGroup):

    """A single setting, which could be an array of items like a group"""

    def __init__(self, *args):
        super(RadioSetting, self).__init__(*args)
        self._apply_callback = None
        self._warning_text = None
        self._safe_value = None
        self._volatile = False
        if self.__doc__ == self.get_name():
            self.__doc__ = None

    @property
    def volatile(self):
        return self._volatile

    def set_volatile(self, value):
        self._volatile = value

    def set_apply_callback(self, callback, *args):
        self._apply_callback = lambda: callback(self, *args)

    def has_apply_callback(self):
        return self._apply_callback is not None

    def run_apply_callback(self):
        return self._apply_callback()

    def set_warning(self, warning_text, safe_value=None):
        """Set a warning message to be shown to the user when changed.

        This should not be over-used as it will be annoying (on purpose). This
        message will be shown to the user with an option to continue to abort.
        It should be used to warn the user of potential invalid operation or
        compatibility issues. If safe_value is set, this will only warn the
        user if the value is changed *to* something other than the safe_value.
        """
        self._warning_text = warning_text
        self._safe_value = safe_value

    def get_warning(self, value):
        if value != self._safe_value:
            return self._warning_text

    def _validate(self, value):
        # RadioSetting can only contain RadioSettingValue objects
        if not isinstance(value, RadioSettingValue):
            raise InternalError("Incorrect type")

    def changed(self):
        """Returns True if any of the elements
        in the group have been changed"""
        for element in self._elements.values():
            if element.changed():
                return True
        return False

    def __str__(self):
        return "%s:%s" % (self._name, self.value)

    def __repr__(self):
        return "[RadioSetting %s:%s]" % (self._name, self.value)

    # Magic foo.value attribute
    def __getattr__(self, name):
        if name == "value":
            if len(self) == 1:
                return self._elements[self._element_order[0]]
            else:
                return list(self._elements.values())
        elif name in ('__getstate__', '__setstate__', '__deepcopy__'):
            super().__getattr__(name)
        else:
            return self.__dict__[name]

    def __setattr__(self, name, value):
        if name == "value":
            if self._frozen:
                raise ValueError('Value is frozen')
            if len(self) == 1:
                self._elements[self._element_order[0]].set_value(value)
            else:
                raise InternalError("Setting %s is not a scalar" % self._name)
        else:
            self.__dict__[name] = value

    # List interface

    def append(self, value):
        index = len(self._element_order)
        self._elements[index] = value
        self._element_order.append(index)

    def __getitem__(self, name):
        if not isinstance(name, int):
            raise IndexError("Index `%s' is not an integer" % name)
        return self._elements[name]

    def __setitem__(self, name, value):
        if not isinstance(name, int):
            raise IndexError("Index `%s' is not an integer" % name)
        if name in self._elements:
            self._elements[name].set_value(value)
        else:
            self._elements[name] = value


class MemSetting(RadioSetting):
    """A RadioSetting that maps directly to a memory object value.

    :param path: The dot-separated path to a setting value in the object (
        i.e. "settings.dtmf.pttid")
    :param name: The human-readable name for this setting
    :param values: The RadioSettingValue (only one allowed)
    :param *kwargs: Other kwargs to be passed to RadioSetting
    """
    def __init__(self, path, name, value, **kwargs):
        self._path = path
        setting_name = path.replace('.', '_')
        super().__init__(setting_name, name, value, **kwargs)

    def apply_to_memobj(self, memobj):
        """Apply this setting value to a memory object.

        Uses the path specified at init time, operates on memobj to find and
        set the value in memory according to this setting.
        """
        if not self.value.initialized:
            LOG.warning('Unable to apply uninitialized value %s',
                        self._path)
            return
        if isinstance(self.value, RadioSettingValueInvertedBoolean):
            value = not self.value
        elif isinstance(self.value, RadioSettingValueList):
            value = int(self.value)
        elif isinstance(self.value, RadioSettingValueString):
            value = str(self.value)
            if self.value.autopad:
                value = value.rstrip().ljust(self.value.maxlength,
                                             self.value.mem_pad_char)
        else:
            value = self.value

        # After we determined if this is inverted or not, convert to the values
        # that memory wants to see.
        if isinstance(self.value, RadioSettingValueBoolean):
            value = self.value._mem_vals[int(value)]

        obj = memobj
        elements = self._path.split('.')
        for element in elements[:-1]:
            if '[' in element:
                # foo[i] syntax
                name, index = element.split('[', 1)
                obj = getattr(obj, name)
                obj = obj[int(index.replace(']', ''))]
            else:
                obj = getattr(obj, element)
        setattr(obj, elements[-1], value)
