modlib/text.lua

-- Localize globals
local math, modlib, pairs, setmetatable, string, table = math, modlib, pairs, setmetatable, string, table

-- Set environment
local _ENV = {}
setfenv(1, _ENV)

function upper_first(text) return text:sub(1, 1):upper() .. text:sub(2) end

function lower_first(text) return text:sub(1, 1):lower() .. text:sub(2) end

function starts_with(text, prefix) return text:sub(1, #prefix) == prefix end

function ends_with(text, suffix) return text:sub(-#suffix) == suffix end

function contains(str, substr, plain)
	return not not str:find(substr, 1, plain == nil and true or plain)
end

function trim(text, to_remove)
	local j = 1
	for i = 1, string.len(text) do
		if text:sub(i, i) ~= to_remove then
			j = i
			break
		end
	end
	local k = 1
	for i = string.len(text), j, -1 do
		if text:sub(i, i) ~= to_remove then
			k = i
			break
		end
	end
	return text:sub(j, k)
end

function trim_begin(text, to_remove)
	local j = 1
	for i = 1, string.len(text) do
		if text:sub(i, i) ~= to_remove then
			j = i
			break
		end
	end
	return text:sub(j)
end

trim_left = trim_begin

function trim_end(text, to_remove)
	local k = 1
	for i = string.len(text), 1, -1 do
		if text:sub(i, i) ~= to_remove then
			k = i
			break
		end
	end
	return text:sub(1, k)
end

trim_right = trim_end

function trim_spacing(text)
	return text:match"^%s*(.-)%s*$"
end

local inputstream_metatable = {
	__index = {
		read = function(self, count)
			local cursor = self.cursor + 1
			self.cursor = self.cursor + count
			local text = self.text:sub(cursor, self.cursor)
			return text ~= "" and text or nil
		end,
		seek = function(self) return self.cursor end
	}
}
--> inputstream "handle"; only allows reading characters (given a count), seeking does not accept any arguments
function inputstream(text)
	return setmetatable({text = text, cursor = 0}, inputstream_metatable)
end

function hexdump(text)
	local dump = {}
	for index = 1, text:len() do
		dump[index] = ("%02X"):format(text:byte(index))
	end
	return table.concat(dump)
end

function split(text, delimiter, limit, is_regex)
	limit = limit or math.huge
	local no_regex = not is_regex
	local parts = {}
	local occurences = 1
	local last_index = 1
	local index = string.find(text, delimiter, 1, no_regex)
	while index and occurences < limit do
		table.insert(parts, string.sub(text, last_index, index - 1))
		last_index = index + string.len(delimiter)
		index = string.find(text, delimiter, index + string.len(delimiter), no_regex)
		occurences = occurences + 1
	end
	table.insert(parts, string.sub(text, last_index))
	return parts
end

function split_without_limit(text, delimiter, is_regex) return split(text, delimiter, nil, is_regex) end

split_unlimited = split_without_limit

function split_lines(text, limit) return modlib.text.split(text, "\r?\n", limit, true) end

function lines(text) return text:gmatch"[^\r\n]*" end

local zero = string.byte"0"
local nine = string.byte"9"
local letter_a = string.byte"A"
local letter_f = string.byte"F"

function is_hexadecimal(byte)
	return byte >= zero and byte <= nine or byte >= letter_a and byte <= letter_f
end

magic_chars = {
	"%",
	"(",
	")",
	".",
	"+",
	"-",
	"*",
	"?",
	"[",
	"^",
	"$"
}
local magic_charset = {}
for _, magic_char in pairs(magic_chars) do table.insert(magic_charset, "%" .. magic_char) end
magic_charset = "[" .. table.concat(magic_charset) .. "]"

function escape_magic_chars(text) return text:gsub("(" .. magic_charset .. ")", "%%%1") end

local keywords = modlib.table.set{"and", "break", "do", "else", "elseif", "end", "false", "for", "function", "if", "in", "local", "nil", "not", "or", "repeat", "return", "then", "true", "until", "while"}
keywords["goto"] = true -- Lua 5.2 (LuaJIT) support

function is_keyword(text)
	return keywords[text]
end

function is_identifier(text)
	return (not keywords[text]) and text:match"^[A-Za-z_][A-Za-z%d_]*$"
end

-- Export environment
return _ENV