-- mod-version:2 -- lite-xl 2.0

-- Syntax highlighting for the Rivet programming language.
-- by StunxFS :)

local syntax = require "core.syntax"

syntax.add {
    name = "Rivet",
    files = {"%.ri$"},
    comment = "//",
    block_comment = {"/*", "*/"},
    patterns = {
        {pattern = "//.-\n", type = "comment"},
        {pattern = {"/%*", "%*/"}, type = "comment"},
        {pattern = {'[cbr]?"', '"', "\\"}, type = "string"},
        {pattern = {"[b]?'", "'", '\\' }, type = "string"},
        {pattern = "0b[01_]+", type = "number"},
        {pattern = "0o[0-7_]+", type = "number"},
        {pattern = "0x[%x_]+", type = "number"},
        {pattern = "%d[%d_]*%.[%d_]*[eE][-+]?%d+", type = "number"},
        {pattern = "%d[%d_]*%.[%d_]*", type = "number"},
        {pattern = "%d[%d_]*", type = "number"},
        {pattern = "-?%.?%d+", type = "number"},
        {pattern = "[%+%-=/%*%^%%<>!~|&%.%?]", type = "operator"},
        {pattern = "[%a_][%w_]*::", type = "keyword2"},
        { -- Uppercase constants of at least 2 chars in len
            pattern = "_?%u[%u_][%u%d_]*%f[%s%+%*%-%.%)%]}%?%^%%=/<>~|&;:,!]",
            type = "number"
        },
        {pattern = "[A-Z][%w_]*", type = "keyword2"}, -- types
        {pattern = "[%a_][%w_]*%f[(]", type = "function"},
        {pattern = "[%a_][%w_]*!%f[(]", type = "keyword2"},
        {pattern = "[%a_][%w_]*", type = "symbol"},
        {pattern = {"#%[", "%]"}, type = "keyword"},
        {pattern = "%$%s?[%a_][%w_]*", type = "keyword2"},
        {pattern = "%@%s?[%a_][%w_]*", type = "keyword2"},
    },
    symbols = {
        ["extern"] = "keyword",
        ["using"] = "keyword",

        ["pub"] = "keyword",
        ["as"] = "keyword",

        ["pkg"] = "keyword",
        ["mod"] = "keyword",
        ["const"] = "keyword",
        ["static"] = "keyword",
        ["trait"] = "keyword",
        ["struct"] = "keyword",
        ["union"] = "keyword",
        ["type"] = "keyword",
        ["errtype"] = "keyword",
        ["enum"] = "keyword",
        ["fn"] = "keyword",
        ["test"] = "keyword",
        ["extend"] = "keyword",

        -- comptime `if` and `match` expr
        ["$if"] = "keyword",
        ["$elif"] = "keyword",
        ["$else"] = "keyword",
        ["$match"] = "keyword",

        ["if"] = "keyword",
        ["elif"] = "keyword",
        ["else"] = "keyword",
        ["match"] = "keyword",
        ["while"] = "keyword",
        ["for"] = "keyword",

        ["break"] = "keyword",
        ["continue"] = "keyword",
        ["return"] = "keyword",
        ["raise"] = "keyword",

        ["let"] = "keyword",
        ["mut"] = "keyword",
        ["unsafe"] = "keyword",
        ["goto"] = "keyword",
        ["orelse"] = "keyword",
        ["catch"] = "keyword",
        ["or"] = "keyword",
        ["and"] = "keyword",
        ["is"] = "keyword",
        ["in"] = "keyword",

        -- types
        ["void"] = "keyword2",
        ["no_return"] = "keyword2",
        ["bool"] = "keyword2",
        ["i8"] = "keyword2",
        ["i16"] = "keyword2",
        ["i32"] = "keyword2",
        ["i64"] = "keyword2",
        ["u8"] = "keyword2",
        ["u16"] = "keyword2",
        ["u32"] = "keyword2",
        ["u64"] = "keyword2",
        ["f32"] = "keyword2",
        ["f64"] = "keyword2",
        ["rune"] = "keyword2",
        ["isize"] = "keyword2",
        ["usize"] = "keyword2",
        ["str"] = "keyword2",
        ["Self"] = "keyword2",

        -- literals
        ["super"] = "literal",
        ["self"] = "literal",
        ["true"] = "literal",
        ["false"] = "literal",
        ["none"] = "literal"
    }
}