import string

from ._compat import PY2
from ._compat import unicode


if PY2:
    from pipenv.vendor.backports.functools_lru_cache import lru_cache
else:
    from functools import lru_cache


class TOMLChar(unicode):
    def __init__(self, c):
        super(TOMLChar, self).__init__()

        if len(self) > 1:
            raise ValueError("A TOML character must be of length 1")

    BARE = string.ascii_letters + string.digits + "-_"
    KV = "= \t"
    NUMBER = string.digits + "+-_.e"
    SPACES = " \t"
    NL = "\n\r"
    WS = SPACES + NL

    @lru_cache(maxsize=None)
    def is_bare_key_char(self):  # type: () -> bool
        """
        Whether the character is a valid bare key name or not.
        """
        return self in self.BARE

    @lru_cache(maxsize=None)
    def is_kv_sep(self):  # type: () -> bool
        """
        Whether the character is a valid key/value separator ot not.
        """
        return self in self.KV

    @lru_cache(maxsize=None)
    def is_int_float_char(self):  # type: () -> bool
        """
        Whether the character if a valid integer or float value character or not.
        """
        return self in self.NUMBER

    @lru_cache(maxsize=None)
    def is_ws(self):  # type: () -> bool
        """
        Whether the character is a whitespace character or not.
        """
        return self in self.WS

    @lru_cache(maxsize=None)
    def is_nl(self):  # type: () -> bool
        """
        Whether the character is a new line character or not.
        """
        return self in self.NL

    @lru_cache(maxsize=None)
    def is_spaces(self):  # type: () -> bool
        """
        Whether the character is a space or not
        """
        return self in self.SPACES
