Talon/minify.lua at main · SpartanSf/Talon · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
local nametext = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
local function mkname(n)
    local s = ""
    while n >= 52 do s, n = s .. nametext:sub(n % 52 + 1, n % 52 + 1), math.floor(n / 52) end
    return nametext:sub(n % 52 + 1, n % 52 + 1) .. s:reverse()
end

local function parseexp(tokens, upvalues, start, retval)
    -- local v = tokens[start]
    -- if v.type == "constant" or v.type == "string" or v.type == "number" then
    --     retval[start] = v

    -- end
    error("Not implemented yet")
    return retval, start
end

local function minify(tokens, upvalues, start, nextname)
    start = start or 1
    nextname = nextname or 0
    local locals = setmetatable({}, {__index = upvalues or {}})
    local state = 1
    local retval = {}
    while start <= #tokens do
        local v = tokens[start]
        if state == 1 then -- normal
            if v.type == "keyword" then
                retval[start] = v
                if v.text == "do" or v.text == "then" then
                    local t, s = minify(tokens, locals, start + 1, nextname)
                    for i = start + 1, s do retval[i] = t[i] end
                    start = s
                elseif v.text == "until" then
                    return parseexp(tokens, locals, start + 1, retval)
                elseif v.text == "function" then
                    local onn = nextname
                    if tokens[start+1].type == "name" then
                        start = start + 1
                        if locals[tokens[start].text] then retval[start] = {col = v.col, line = v.line, type = "name", text = locals[tokens[start].text]}
                        else retval[start] = tokens[start] end
                    end
                    start = start + 1
                    retval[start] = tokens[start]
                    start = start + 1
                    v = tokens[start]
                    local params = setmetatable({}, {__index = locals})
                    while v.type ~= "operator" or v.text ~= ")" do
                        if v.type == "name" then
                            params[v.text] = mkname(nextname)
                            nextname = nextname + 1
                            retval[start] = {col = v.col, line = v.line, type = "name", text = params[v.text]}
                        else retval[start] = v end
                        start = start + 1
                        v = tokens[start]
                    end
                    retval[start] = v
                    local t, s = minify(tokens, params, start + 1, nextname)
                    for i = start + 1, s do retval[i] = t[i] end
                    start = s
                    nextname = onn
                elseif v.text == "local" or v.text == "for" then
                    state = 3
                elseif v.text == "end" then
                    return retval, start
                end
            elseif v.type == "name" then
                if locals[v.text] then retval[start] = {col = v.col, line = v.line, type = "name", text = locals[v.text]}
                else retval[start] = v end
            elseif v.type == "operator" then
                retval[start] = v
                if v.text == "." or v.text == ":" then state = 2 end
            else
                retval[start] = v
            end
        elseif state == 2 then -- prefixexp
            retval[start] = v
            if not ((v.type == "operator" and (v.text == "." or v.text == ":") or v.type == "name")) then state = 1 end
        elseif state == 3 then -- local
            if v.type == "keyword" and v.text == "function" then
                v = retval[start + 1]
                locals[v.text] = mkname(nextname)
                nextname = nextname + 1
                retval[start] = {col = v.col, line = v.line, type = "name", text = locals[v.text]}
                local onn = nextname
                start = start + 2
                retval[start] = tokens[start]
                start = start + 1
                v = tokens[start]
                local params = setmetatable({}, {__index = locals})
                while v.type ~= "operator" or v.text ~= ")" do
                    if v.type == "name" then
                        params[v.text] = mkname(nextname)
                        nextname = nextname + 1
                        retval[start] = {col = v.col, line = v.line, type = "name", text = params[v.text]}
                    else retval[start] = v end
                    start = start + 1
                    v = tokens[start]
                end
                retval[start] = v
                local t, s = minify(tokens, params, start + 1, nextname)
                for i = start + 1, s do retval[i] = t[i] end
                start = s
                nextname = onn
            elseif v.type == "name" then
                locals[v.text] = mkname(nextname)
                nextname = nextname + 1
                retval[start] = {col = v.col, line = v.line, type = "name", text = locals[v.text]}
                state = 4
            else error("invalid local statement") end
        elseif state == 4 then -- local follow
            if v.type == "operator" and v.text == "," then
                state = 3
                retval[start] = v
            else
                state = 1
                start = start - 1
            end
        end
        start = start + 1
    end
    return retval, #tokens
end

return minify