"""
    pygments.lexers.q
    ~~~~~~~~~~~~~~~~~

    Lexer for the Q programming language.

    :copyright: Copyright 2006-2022 by the Pygments team, see AUTHORS.
    :license: BSD, see LICENSE for details.
"""

from pygments.lexer import RegexLexer, words, include, bygroups, inherit
from pygments.token import (
    Comment,
    Keyword,
    Name,
    Number,
    Operator,
    Punctuation,
    String,
    Text,
    Whitespace,
    Literal,
    Generic,
)

__all__ = ["KLexer", "QLexer"]


class KLexer(RegexLexer):
    """
    For `K <https://code.kx.com/>`_ source code.

    .. versionadded:: 2.12
    """

    name = "K"
    aliases = ["k"]
    filenames = ["*.k"]

    tokens = {
        "whitespace": [
            # hashbang script
            (r"^#!.*", Comment.Hashbang),
            # Comments
            (r"^/\s*\n", Comment.Multiline, "comments"),
            (r"(?<!\S)/.*", Comment.Single),
            # Whitespace
            (r"\s+", Whitespace),
            # Strings
            (r"\"", String.Double, "strings"),
        ],
        "root": [
            include("whitespace"),
            include("keywords"),
            include("declarations"),
        ],
        "keywords": [
            (
                words(
                    ("abs", "acos", "asin", "atan", "avg", "bin",
                     "binr", "by", "cor", "cos", "cov", "dev",
                     "delete", "div", "do", "enlist", "exec", "exit",
                     "exp", "from", "getenv", "hopen", "if", "in",
                     "insert", "last", "like", "log", "max", "min",
                     "prd", "select", "setenv", "sin", "sqrt", "ss",
                     "sum", "tan", "update", "var", "wavg", "while",
                     "within", "wsum", "xexp"),
                    suffix=r"\b",
                ),
                Operator.Word,
            ),
        ],
        "declarations": [
            # Timing
            (r"^\\ts?", Comment.Preproc),
            (
                r"^(\\\w\s+[^/\n]*?)(/.*)",
                bygroups(Comment.Preproc, Comment.Single),
            ),
            # Generic System Commands
            (r"^\\\w.*", Comment.Preproc),
            # Prompt
            (r"^[a-zA-Z]\)", Generic.Prompt),
            # Function Names
            (
                r"([.]?[a-zA-Z][\w.]*)(\s*)([-.~=!@#$%^&*_+|,<>?/\\:']?:)(\s*)(\{)",
                bygroups(Name.Function, Whitespace, Operator, Whitespace, Punctuation),
                "functions",
            ),
            # Variable Names
            (
                r"([.]?[a-zA-Z][\w.]*)(\s*)([-.~=!@#$%^&*_+|,<>?/\\:']?:)",
                bygroups(Name.Variable, Whitespace, Operator),
            ),
            # Functions
            (r"\{", Punctuation, "functions"),
            # Parentheses
            (r"\(", Punctuation, "parentheses"),
            # Brackets
            (r"\[", Punctuation, "brackets"),
            # Errors
            (r"'`([a-zA-Z][\w.]*)?", Name.Exception),
            # File Symbols
            (r"`:([a-zA-Z/][\w./]*)?", String.Symbol),
            # Symbols
            (r"`([a-zA-Z][\w.]*)?", String.Symbol),
            # Numbers
            include("numbers"),
            # Variable Names
            (r"[a-zA-Z][\w.]*", Name),
            # Operators
            (r"[-=+*#$%@!~^&:.,<>'\\|/?_]", Operator),
            # Punctuation
            (r";", Punctuation),
        ],
        "functions": [
            include("root"),
            (r"\}", Punctuation, "#pop"),
        ],
        "parentheses": [
            include("root"),
            (r"\)", Punctuation, "#pop"),
        ],
        "brackets": [
            include("root"),
            (r"\]", Punctuation, "#pop"),
        ],
        "numbers": [
            # Binary Values
            (r"[01]+b", Number.Bin),
            # Nulls/Infinities
            (r"0[nNwW][cefghijmndzuvtp]?", Number),
            # Timestamps
            (
                (
                    r"(?:[0-9]{4}[.][0-9]{2}[.][0-9]{2}|[0-9]+)"
                    "D(?:[0-9](?:[0-9](?::[0-9]{2}"
                    "(?::[0-9]{2}(?:[.][0-9]*)?)?)?)?)?"
                ),
                Literal.Date,
            ),
            # Datetimes
            (
                (
                    r"[0-9]{4}[.][0-9]{2}"
                    "(?:m|[.][0-9]{2}(?:T(?:[0-9]{2}:[0-9]{2}"
                    "(?::[0-9]{2}(?:[.][0-9]*)?)?)?)?)"
                ),
                Literal.Date,
            ),
            # Times
            (
                (r"[0-9]{2}:[0-9]{2}(?::[0-9]{2}(?:[.][0-9]{1,3})?)?"),
                Literal.Date,
            ),
            # GUIDs
            (
                r"[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}",
                Number.Hex,
            ),
            # Byte Vectors
            (r"0x[0-9a-fA-F]+", Number.Hex),
            # Floats
            (
                r"([0-9]*[.]?[0-9]+|[0-9]+[.]?[0-9]*)[eE][+-]?[0-9]+[ef]?",
                Number.Float,
            ),
            (r"([0-9]*[.][0-9]+|[0-9]+[.][0-9]*)[ef]?", Number.Float),
            (r"[0-9]+[ef]", Number.Float),
            # Characters
            (r"[0-9]+c", Number),
            # Integers
            (r"[0-9]+[ihtuv]", Number.Integer),
            # Long Integers
            (r"[0-9]+[jnp]?", Number.Integer.Long),
        ],
        "comments": [
            (r"[^\\]+", Comment.Multiline),
            (r"^\\", Comment.Multiline, "#pop"),
            (r"\\", Comment.Multiline),
        ],
        "strings": [
            (r'[^"\\]+', String.Double),
            (r"\\.", String.Escape),
            (r'"', String.Double, "#pop"),
        ],
    }


class QLexer(KLexer):
    """
    For `Q <https://code.kx.com/>`_ source code.

    .. versionadded:: 2.12
    """

    name = "Q"
    aliases = ["q"]
    filenames = ["*.q"]

    tokens = {
        "root": [
            (
                words(
                    ("aj", "aj0", "ajf", "ajf0", "all", "and", "any",
                     "asc", "asof", "attr", "avgs", "ceiling", "cols",
                     "count", "cross", "csv", "cut", "deltas", "desc",
                     "differ", "distinct", "dsave", "each", "ej",
                     "ema", "eval", "except", "fby", "fills", "first",
                     "fkeys", "flip", "floor", "get", "group", "gtime",
                     "hclose", "hcount", "hdel", "hsym", "iasc",
                     "idesc", "ij", "ijf", "inter", "inv", "key",
                     "keys", "lj", "ljf", "load", "lower", "lsq",
                     "ltime", "ltrim", "mavg", "maxs", "mcount", "md5",
                     "mdev", "med", "meta", "mins", "mmax", "mmin",
                     "mmu", "mod", "msum", "neg", "next", "not",
                     "null", "or", "over", "parse", "peach", "pj",
                     "prds", "prior", "prev", "rand", "rank", "ratios",
                     "raze", "read0", "read1", "reciprocal", "reval",
                     "reverse", "rload", "rotate", "rsave", "rtrim",
                     "save", "scan", "scov", "sdev", "set", "show",
                     "signum", "ssr", "string", "sublist", "sums",
                     "sv", "svar", "system", "tables", "til", "trim",
                     "txf", "type", "uj", "ujf", "ungroup", "union",
                     "upper", "upsert", "value", "view", "views", "vs",
                     "where", "wj", "wj1", "ww", "xasc", "xbar",
                     "xcol", "xcols", "xdesc", "xgroup", "xkey",
                     "xlog", "xprev", "xrank"),
                    suffix=r"\b",
                ),
                Name.Builtin,
            ),
            inherit,
        ],
    }
