utils.lua

--[[
	Copyright (c) 2019 igor725, scaledteam
	released under The MIT license http://opensource.org/licenses/MIT
]]

ffi = require('ffi')
C = ffi.C
ffi.cdef[[
	typedef struct {
		uint8_t r;
		uint8_t g;
		uint8_t b;
	} color;
	typedef struct {
		float x;
		float y;
		float z;
	} vector;
	typedef struct {
		float yaw;
		float pitch;
	} angle;

	size_t fread(void*, size_t, size_t, void*);
	size_t fwrite(const void*, size_t, size_t, void*);
	int    ferror(void*);
	void   free(void*);
]]

local ext = (jit.os == 'Windows'and'dll')or'so'
package.cpath = ('./bin/%s/?.%s;'):format(jit.arch, ext)
package.path = './lua/?.lua;./?.lua;./misc/?.lua'

function checkEnv(ev, val)
	local evar = os.getenv(ev)
	if evar then
		return evar:lower():find(val:lower())
	else
		return false
	end
end

local terms = {'xterm', 'screen', 'linux', 'cygwin', 'vt100'}

local function isColoredTerm()
	for i = 1, #terms do
		if checkEnv('TERM', terms[i])then
			return true
		end
	end
end

enableConsoleColors = checkEnv('ConEmuANSI', 'on')or isColoredTerm()

function packTo(file, fmt, ...)
	if select('#', ...) < 1 then return false end
	local data = struct.pack(fmt, ...)
	return file:write(data)
end

function unpackFrom(file, fmt)
	local sz = struct.size(fmt)
	local data = file:read(sz)
	return struct.unpack(fmt, data)
end

function writeString(f, v)
	assert(#v < 255, 'String too long')
	f:write(string.char(#v), v)
end

function readString(f)
	return f:read(f:read(1):byte())
end

function newColor(r, g, b)
	r, g, b = r or 255, g or 255, b or 255
	return ffi.new('color', r, g, b)
end

function newAngle(y, p)
	return ffi.new('angle', y, p)
end

function newVector(x, y, z)
	return ffi.new('vector', x, y, z)
end

function distance(x1, y1, z1, x2, y2, z2)
	return math.sqrt( (x2 - x1) ^ 2 + (y2 - y1) ^ 2 + (z2 - z1) ^ 2 )
end

function toAngle(x, y)
	if y == 0 then
		if x < 0 then
			return 180
		elseif x > 0 then
			return 0
		else
			return 0
		end
	else
		angle = math.atan(y / x) / math.pi * 180

		if x < 0 then
			x = -x
			if y < 0 then
				angle = angle - 180
			else
				angle = angle + 180
			end
		end

		return angle
	end
end

floor = math.floor
ceil = math.ceil

local colorReplace

if enableConsoleColors then
	local rt = {
		['&0'] = '30',
		['&1'] = '34',
		['&2'] = '32',
		['&3'] = '36',
		['&4'] = '31',
		['&5'] = '35',
		['&6'] = '33',
		['&7'] = '1;30',
		['&8'] = '1;30',
		['&9'] = '1;34',
		['&a'] = '1;32',
		['&b'] = '1;36',
		['&c'] = '1;31',
		['&d'] = '1;35',
		['&e'] = '1;33',
		['&f'] = '0'
	}
	colorReplace = function(s)
		s = s:lower()
		return string.format('\27[%sm', rt[s])
	end
else
	colorReplace = function()
		return ''
	end
end

function mc2ansi(str)
	local pattern = '(&%x)'
	if str:find(pattern)then
		str = str:gsub(pattern, colorReplace)
	end
	return str
end

function printf(...)
	local str = string.format(...)
	log.info(str)
	return str
end

function parseSizeStr(str)
	local sz, suff = '', ''

	for i = 1, #str do
		local b = str:byte(i, i)
		if b >= 48 and b <= 57 then
			sz = sz .. string.char(b)
		elseif b == 46 then
			sz = sz .. '.'
		elseif (b >= 65 and b <= 90)or (b >= 97 and b <= 122)then
			suff = suff .. string.char(b)
		end
	end

	if sz then
		sz = tonumber(sz)
		suff = suff:lower()

		if sz < 0 then
			return 0
		end
		if suff == 'b'or suff == ''then
			return sz
		elseif suff == 'kb'or suff == 'k'then
			return sz * 1000
		elseif suff == 'mb'or suff == 'm'then
			return sz * 1e6
		elseif suff == 'gb'or suff == 'g'then
			return sz * 1e9
		end
	end
	return -1
end

function toDHMS(sec)
	local d = sec / 86400
	local h = (sec % 86400) / 3600
	local m = (sec / 60) % 60
	local s = sec % 60
	return d, h, m, s
end

function getn(t)
	local c = 0
	for _ in pairs(t)do c = c + 1 end
	return c
end

function getCurrentOnline(world)
	local count = 0
	for i = 0, 127 do
		local ent = entities[i]
		if isPlayer(ent) and (not world or getWorld(world) == getWorld(ent))then
			count = count + 1
		end
	end
	return count
end

local function woSpaces(...) -- It works faster than string.match
	local idx = 1
	local stStart, stEnd

	while true do
		local b = select(idx, ...)
		if not stStart then
			if b ~= 32 and b ~= 0 then
				stStart = idx
				idx = select('#', ...)
			else
				idx = idx + 1
			end
		else
			if b ~= 32 and b ~= 0 then
				stEnd = idx
				break
			end
			idx = idx - 1
			if idx == 1 then
				stEnd = 1
				break
			end
		end
	end

	return stStart, stEnd
end

function trimStr(str)
	return str:sub(woSpaces(str:byte(1, -1)))
end

function randomStr(len)
	local str = ffi.new('char[?]', len + 1)
	for i = 0, len - 1 do
		if math.random(0, 100) > 30 then
			str[i] = math.random(48, 57)
		else
			str[i] = math.random(97, 122)
		end
	end
	return ffi.string(str)
end

function getAddr(void)
	return tonumber(ffi.cast('uintptr_t', void))
end

function table.hasValue(tbl, ...)
	for i=1, #tbl do
		local idx = 1
		while true do
			local tv = select(idx, ...)
			if tv == nil then
				break
			end
			if tv == tbl[i]then
				return true
			end
			idx = idx + 1
		end
	end
	return false
end

function string.split(self, sep)
	local sep, fields = sep or ':', {}
	local pattern = string.format('([^%s]+)', sep)
	self:gsub(pattern, function(c) fields[#fields + 1] = c end)
	return fields
end

function string.startsWith(self, ...)
	local idx = 1
	while true do
		local str = select(idx, ...)
		if str == nil then
			break
		end
		if self:sub(1, #str) == str then
			return true, idx
		end
		idx = idx + 1
	end
	return false
end

if jit.os == 'Windows'then
	ffi.cdef[[
		typedef struct {
		  uint32_t dwLowDateTime;
		  uint32_t dwHighDateTime;
		} filetime;

		int MoveFileExA(
			const char *lpExistingFileName,
			const char *lpNewFileName,
			unsigned long dwFlags
		);

		typedef struct {
			unsigned int dwFileAttributes;
			filetime  ftCreationTime;
			filetime  ftLastAccessTime;
			filetime  ftLastWriteTime;
			uint32_t  nFileSizeHigh;
			uint32_t  nFileSizeLow;
			uint32_t  dwReserved0;
			uint32_t  dwReserved1;
			char     cFileName[260];
			char     cAlternateFileName[14];
		} WIN32_FIND_DATA;

		void  GetSystemTimeAsFileTime(filetime*);
		void* FindFirstFileA(const char*, WIN32_FIND_DATA*);
		bool  CreateDirectoryA(const char*, uint32_t);
		bool  FindNextFileA(void*, WIN32_FIND_DATA*);
		bool  FindClose(void*);
		void  Sleep(uint32_t);
	]]

	INVALID_HANDLE = ffi.cast('void*', -1)

	local ft = ffi.new('filetime')
	function gettime()
		C.GetSystemTimeAsFileTime(ft)
		local wtime = ft.dwLowDateTime / 1.0e7 + ft.dwHighDateTime * 429.4967296
		return wtime - 11644473600
	end

	function sleep(ms)
		C.Sleep(ms)
	end

	function scanDir(path, ext)
		local fdata = ffi.new('WIN32_FIND_DATA')
		local file

		local function getName()
			local name = ffi.string(fdata.cFileName)
			if #name > 0 then
				return name
			end
		end

		local function isFile()
			return bit.band(fdata.dwFileAttributes, 16) == 0
		end

		if not ext then
			ext = '*'
		end

		return function()
			if not file then
				file = C.FindFirstFileA(path .. '\\*.' .. ext, fdata)
				if file == INVALID_HANDLE then
					return
				end
				return getName(), isFile()
			end

			if C.FindNextFileA(file, fdata)then
				return getName(), isFile()
			else
				C.FindClose(file)
			end
		end
	end

	function createDirectory(path)
		return C.CreateDirectoryA(path, 0)
	end

	function os.rename(oldfile, newfile)
		local ret = C.MoveFileExA(oldfile, newfile, 1)
		if ret == 0 then
			return nil, C.GetLastError()
		else
			return true
		end
	end
else
	ffi.cdef[[
		struct dirent {
			unsigned long  d_ino;
			unsigned long  d_off;
			unsigned short d_reclen;
			unsigned char  d_type;
			char           d_name[256];
		};

		struct timeval {
			long tv_sec;
			long tv_usec;
		};

		typedef unsigned short mode_t;

		void  gettimeofday(struct timeval*, void*);
		int   mkdir(const char*, mode_t);
		void  usleep(unsigned int);
		void* opendir(const char*);
		void  closedir(void*);
		struct dirent* readdir(void*);
		mode_t umask(mode_t);
	]]

	local function scanNext(dir, ext)
		while true do
			local _ent = C.readdir(dir)
			if _ent == nil then break end
			local name = ffi.string(_ent.d_name)
			if name:sub(-#ext) == ext then
				return name, _ent.d_type == 8
			end
		end
		C.closedir(dir)
	end

	function sleep(ms)
		C.usleep(ms * 1000)
	end

	local t = ffi.new('struct timeval')
	function gettime()
		C.gettimeofday(t, nil)
		local tm = tonumber(t.tv_sec) + 1e-6 * tonumber(t.tv_usec)
		return tm
	end

	function createDirectory(path, chmod)
		chmod = tonumber(chmod or 755, 8)
		return C.mkdir(path, chmod) == 0
	end

	function scanDir(dir, ext)
		local _dir = C.opendir(dir)
		if _dir == nil then return end

		return function()
			return scanNext(_dir, ext)
		end
	end
end

function dirForEach(dir, ext, func)
	for name, isFile in scanDir(dir, ext)do
		if isFile then
			func(name, dir .. '/' .. name)
		end
	end
end

function makeNormalCube(p1, p2)
	local px1, py1, pz1 = unpack(p1)
	local px2, py2, pz2 = unpack(p2)

	if p1[1] - p2[1] < 0 then
		px1 = p2[1] + 1
		px2 = p1[1]
	else
		px1 = p1[1] + 1
	end
	if p1[2] - p2[2] < 0 then
		py1 = p2[2] + 1
		py2 = p1[2]
	else
		py1 = p1[2] + 1
	end
	if p1[3] - p2[3] < 0 then
		pz1 = p2[3] + 1
		pz2 = p1[3]
	else
		pz1 = p1[3] + 1
	end

	return px1, py1, pz1, px2, py2, pz2
end

function watchThreads(threads)
	while #threads > 0 do
		local thread = threads[#threads]
		if thread then
			if thread.status == 'error'then
				log.error(thread[-1])
				table.remove(threads, #threads)
			elseif thread.status == 'done'then
				table.remove(threads, #threads)
			end
		else
			sleep(40)
		end
	end
end

lanes = require('other.lanes').configure{
	with_timers = false
}
struct = require('struct')
require('hooks')
require('log')
require('lng')
local cats = {'data', 'hash', 'network', 'objs', 'other'}
for i = 1, #cats do
	dirForEach('lua/' .. cats[i], 'lua', function(_, path)
		assert(loadfile(path))()
	end)
end

local dirs = {'worlds', 'saves', 'playerdata'}
for i = 1, #dirs do
	createDirectory(dirs[i])
end