doodle3d-firmware/src/util/utils.lua

--[[--
	TODO: finish documentation
	The unavoidable collection of utility functions.
]]

local M = {}


function string:split(div)
	local div, pos, arr = div or ':', 0, {}
	for st,sp in function() return self:find(div, pos, true) end do
		table.insert(arr, self:sub(pos, st - 1))
		pos = sp + 1
	end
	table.insert(arr, self:sub(pos))
	return arr
end

function M.fileSize(file)
	local current = file:seek()
	local size = file:seek('end')
	file:seek('set', current)
	return size
end

function M.toboolean(s)
	if not s then return false end
	
	local b = type(s) == 'string' and s:lower() or s
	local textTrue = (b == '1' or b == 't' or b == 'true')
	local boolTrue = (type(b) == 'boolean' and b == true)
	local numTrue = (type(b) == 'number' and b > 0)
	return textTrue or boolTrue or numTrue
end

function M.dump(o)
	if type(o) == 'table' then
		local s = '{ '
		for k,v in pairs(o) do
			if type(k) ~= 'number' then k = '"'..k..'"' end
			s = s .. '['..k..'] = ' .. M.dump(v) .. ','
		end
		return s .. '} '
	else
		return tostring(o)
	end
end

--- Returns the name of a section in a UCI config.
-- This name is necessary to be able to refer to the corresponding section and
-- the UCI library does not provide a way to look it up.
-- @tparam string config Name of the UCI config to search through.
-- @tparam string type UCI type of the section to find.
-- @treturn string Name of the section matching the parameters, or nil if it could not be found.
function M.getUciSectionName(config, type)
	local uci = require('uci').cursor()
	local sname = nil
	uci:foreach(config, type, function(s) sname = s['.name'] end)
	return sname
end

function M.exists(file)
	if not file or type(file) ~= 'string' or file:len() == 0 then
		return nil, "file must be a non-empty string"
	end
	
	local r = io.open(file, 'r') -- ignore returned message
	if r then r:close() end
	return r ~= nil
end

--creates and returns true if not exists, returns false it does, nil+msg on error
function M.create(file)
	local r,m = M.exists(file)
	
	if r == nil then
		return r,m
	elseif r == true then
		return true
	end
	
	r,m = io.open(file, 'a') -- append mode is probably safer in case the file does exist after all
	if not r then return r,m end
	
	r:close()
	return true
end

--FIXME: somehow protect this function from running arbitrary commands
function M.symlink(from, to)
	if from == nil or from == '' or to == nil or to == '' then return -1 end
	local x = 'ln -s ' .. from .. ' ' .. to
	return os.execute(x)
end

function M.readFile(filePath)
	local f, msg, nr = io.open(filePath, 'r')
	if not f then return nil,msg,nr end
	
	local res = f:read('*all')
	f:close()
	
	return res
end

-- TODO: this function has been duplicated from rest/api/api_system.lua
function M.captureCommandOutput(cmd)
	local f = assert(io.popen(cmd, 'r'))
	local output = assert(f:read('*all'))
	f:close()
	return output;
end

return M