1 files changed, 81 insertions, 0 deletions
diff --git a/src/uglify.lua b/src/uglify.lua
new file mode 100644
index 0000000..143f67f
--- /dev/null
+++ b/src/uglify.lua
@@ -0,0 +1,81 @@
+local u = {}
+local nonames = {"if", "for", "end", "do", "local", "then", "else", "elseif", "return", "goto", "function", "nil", "false", "true", "repeat", "return", "break", "and", "or", "not", "in", "repeat", "until", "while", "continue"}
+
+u.uglify = function(str)
+	assert(str ~= nil, "Cannot uglify a nil string")
+	local avalchars = {}
+	local capture_chars = {"%", "(", "[", "\13"}
+	local skipchars = {}
+
+	for k, v in pairs(capture_chars) do
+		skipchars[string.byte(v, 1)] = true
+	end
+
+	for k = 1, 128 do
+		if not skipchars[k] then
+			if string.find(str, string.char(k)) then
+				avalchars[k] = false
+			else
+				avalchars[k] = true
+			end
+		end
+	end
+
+	for k, v in pairs(skipchars) do
+		avalchars[k] = false
+	end
+
+	local counts = {}
+	for k, v in ipairs(nonames) do
+		counts[v] = 0
+	end
+	for k,_ in pairs(counts) do
+		for i,j in string.gmatch(str,k) do
+			counts[k] = counts[k] + 1
+		end
+	end
+
+	local prettifycode = [[
+	local function p(s) local r = {%s} for k,v in pairs(r) do s = s:gsub(v[2],v[1]) end return s end]]
+	local replacementtbl = {}
+	local cursor = 1 --Start at 1 because 0 is the null character
+
+	for k, v in ipairs(nonames) do
+		if counts[v] ~= 0 then
+			while not avalchars[cursor] do
+				cursor = cursor + 1
+			end
+
+			replacementtbl[v] = cursor
+			avalchars[cursor] = false
+		end
+	end
+
+	assert(cursor < 128, "Unable to uglify file, not enough unused characters!")
+	local replacementstr = {}
+
+	for k, v in pairs(replacementtbl) do
+		local trs = string.format("{%q,%q}", k, string.char(v))
+		replacementstr[#replacementstr + 1] = trs
+	end
+
+	local frepstr = table.concat(replacementstr, ",")
+	local pcd = string.format(prettifycode, frepstr)
+
+	for k, v in pairs(replacementtbl) do
+		str = str:gsub(k, string.char(v))
+	end
+
+	local numdeepcom = 1
+
+	local uglified = table.concat({pcd, "\n", "return assert(loadstring(p([", string.rep("=", numdeepcom), "[", str, "]", string.rep("=", numdeepcom), "])))()"})
+
+	if #uglified > #str then
+		print("WARNING: Uglification hurt rather than helped! Put more code in one file!")
+	end
+
+	return uglified
+end
+
+--prettifycode = string.format(prettifycode,)
+return u