#..
    This file contains some definitions of text escape sequences, including ANSI console
    color codes.

use "lib/metaprogramming.nom"

# Text functions
action [join %strs with %glue]
    lua> ".."
        local str_bits = {}
        for i,bit in ipairs(\%strs) do str_bits[i] = stringify(bit) end
        return table.concat(str_bits, \%glue)
parse [join %strs] as: join %strs with ""

compile [capitalize %str, %str capitalized] to
    "(\(%str as lua)):gsub('%l', string.upper, 1)"

compile [%str with %patt replaced with %sub, %str s/%patt/%sub] to
    "((\(%str as lua)):gsub(\(%patt as lua), \(%sub as lua)))"
compile [%str with %patt replaced with %sub %n times, %str s/%patt/%sub/%n] to
    "((\(%str as lua)):gsub(\(%patt as lua), \(%sub as lua), \(%n as lua)))"

compile [indent %str] to "\%str:gsub('\\n','\\n'..('    '))"
compile [indent %str %n times, indent %str %n x] to "\%str:gsub('\\n','\\n'..('    '):rep(\%n))"

# Substring
# TODO: improve this syntax
compile [%str |%start|] to "\(%str as lua):sub(\(%start as lua), \(%start as lua))"
compile [%str |%start - %stop|] to "\(%str as lua):sub(\(%start as lua), \(%stop as lua))"

# Text literals
lua do> ".."
    local escapes = {
        nl="\\\\n", newline="\\\\n", tab="\\\\t", bell="\\\\a", cr="\\\\r", ["carriage return"]="\\\\r",
        backspace="\\\\b", ["form feed"]="\\\\f", formfeed="\\\\f", ["vertical tab"]="\\\\v",
    };
    local colors = {
        ["reset color"]="\\\\27[0m", bright="\\\\27[1m", dim="\\\\27[2m", underscore="\\\\27[4m",
        blink="\\\\27[5m", inverse="\\\\27[7m", hidden="\\\\27[8m",

        black="\\\\27[30m", red="\\\\27[31m", green="\\\\27[32m", yellow="\\\\27[33m", blue="\\\\27[34m",
        magenta="\\\\27[35m", cyan="\\\\27[36m", white="\\\\27[37m",

        ["on black"]="\\\\27[40m", ["on red"]="\\\\27[41m", ["on green"]="\\\\27[42m", ["on yellow"]="\\\\27[43m",
        ["on blue"]="\\\\27[44m", ["on magenta"]="\\\\27[45m", ["on cyan"]="\\\\27[46m", ["on white"]="\\\\27[47m",
    };
    for name, e in pairs(escapes) do
        local lua = "'"..e.."'";
        nomsu:define_compile_action(name, \(__line_no__), function() return {expr=str}; end, \(__src__ 1));
    end
    for name, c in pairs(colors) do
        local color = "'"..c.."'";
        local reset = "'"..colors["reset color"].."'";
        nomsu:define_compile_action(name, \(__line_no__), function() return {expr=color}; end, \(__src__ 1));
        nomsu:define_compile_action(name.." %", \(__line_no__), function(\%)
            return {expr=color..".."..nomsu:tree_to_lua(\%).expr..".."..reset};
        end, \(__src__ 1));
    end