"""
    pygments.lexers.cplint
    ~~~~~~~~~~~~~~~~~~~~~~

    Lexer for the cplint language

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

from pygments.lexer import bygroups, inherit, words
from pygments.lexers import PrologLexer
from pygments.token import Operator, Keyword, Name, String, Punctuation

__all__ = ["CplintLexer"]


class CplintLexer(PrologLexer):
    """
    Lexer for cplint files, including CP-logic, Logic Programs with Annotated
    Disjunctions, Distributional Clauses syntax, ProbLog, DTProbLog.

    .. versionadded:: 2.12
    """

    name = "cplint"
    url = "https://cplint.eu"
    aliases = ["cplint"]
    filenames = ["*.ecl", "*.prolog", "*.pro", "*.pl", "*.P", "*.lpad", "*.cpl"]
    mimetypes = ["text/x-cplint"]

    tokens = {
        "root": [
            (r"map_query", Keyword),
            (
                words(
                    (
                        "gaussian",
                        "uniform_dens",
                        "dirichlet",
                        "gamma",
                        "beta",
                        "poisson",
                        "binomial",
                        "geometric",
                        "exponential",
                        "pascal",
                        "multinomial",
                        "user",
                        "val",
                        "uniform",
                        "discrete",
                        "finite",
                    )
                ),
                Name.Builtin,
            ),
            # annotations of atoms
            (r"([a-z]+)(:)", bygroups(String.Atom, Punctuation)),
            (r":(-|=)|::?|~=?|=>", Operator),
            (r"\?", Name.Builtin),
            inherit,
        ],
    }
