-- bl library -- Main lua-side functionality of bllua, -- provided through the global table 'bl.' local _bllua_ts = ts bl = bl or {} -- Config local tsMaxArgs = 16 local tsArgsLocal = '%a,%b,%c,%d,%e,%f,%g,%h,%i,%j,%k,%l,%m,%n,%o,%p' local tsArgsGlobal = '$_bllua_hook_arg1,$_bllua_hook_arg2,$_bllua_hook_arg3,$_bllua_hook_arg4,' .. '$_bllua_hook_arg5,$_bllua_hook_arg6,$_bllua_hook_arg7,$_bllua_hook_arg8,' .. '$_bllua_hook_arg9,$_bllua_hook_arg10,$_bllua_hook_arg11,$_bllua_hook_arg12,' .. '$_bllua_hook_arg13,$_bllua_hook_arg14,$_bllua_hook_arg15,$_bllua_hook_arg16' -- Misc local function ipairsNilable(t) local maxk = 0 for k, _ in pairs(t) do if k > maxk then maxk = k end end local i = 0 return function() i = i + 1 if i > maxk then return nil else return i, t[i] end end end -- Validation local function isValidFuncName(name) return type(name) == 'string' and name:find('^[a-zA-Z_][a-zA-Z0-9_]*$') end local function isValidFuncNameNs(name) return type(name) == 'string' and ( name:find('^[a-zA-Z0-9_]+$') or name:find('^[a-zA-Z0-9_]+::[a-zA-Z0-9_]+$')) end local function isValidFuncNameNsArgn(name) return type(name) == 'string' and ( name:find('^[a-zA-Z0-9_]+$') or name:find('^[a-zA-Z0-9_]+%.[a-zA-Z0-9_]+$') or name:find('^[a-zA-Z0-9_]+::[a-zA-Z0-9_]+$') or name:find('^[a-zA-Z0-9_]+:[0-9]+$') or name:find('^[a-zA-Z0-9_]+::[a-zA-Z0-9_]+:[0-9]+$')) end -- Whether a var can be converted into a TS vector local function isTsVector(val) if type(val) ~= 'table' then return false end if #val ~= 3 and #val ~= 2 then return false end if val.__is_vector then return true end for _, v in ipairs(val) do if type(v) ~= 'number' then return false end end return true end -- Use strings for object types instead of integer bitmasks like in TS local tsTypesByName = { ['all'] = -1, ['static'] = 1, ['environment'] = 2, ['terrain'] = 4, ['water'] = 16, ['trigger'] = 32, ['marker'] = 64, ['gamebase'] = 1024, ['shapebase'] = 2048, ['camera'] = 4096, ['staticshape'] = 8192, ['player'] = 16384, ['item'] = 32768, ['vehicle'] = 65536, ['vehicleblocker'] = 131072, ['projectile'] = 262144, ['explosion'] = 524288, ['corpse'] = 1048576, ['debris'] = 4194304, ['physicalzone'] = 8388608, ['staticts'] = 16777216, ['brick'] = 33554432, ['brickalways'] = 67108864, ['staticrendered'] = 134217728, ['damagableitem'] = 268435456, } local tsTypesByNum = {} for k, v in pairs(tsTypesByName) do tsTypesByNum[v] = k end -- Type conversion from Lua to TS local toTsObject -- Convert a string from TS into a boolean -- Note: Nonempty nonnumeric strings evaluate to 1, unlike in TS local function tsBool(v) return v ~= '' and v ~= '0' end -- Convert a Lua var into a TS string, or error if not possible local function valToTs(val) if val == nil then -- nil -> '' return '' elseif type(val) == 'boolean' then -- bool -> 0 or 1 return val and '1' or '0' elseif type(val) == 'number' then -- number return tostring(val) elseif type(val) == 'string' then -- string return val elseif type(val) == 'table' then if val._tsObjectId then -- object -> object id return tostring(val._tsObjectId) elseif isTsVector(val) then -- vector - > 3 numbers return table.concat(val, ' ') elseif #val == 2 and isTsVector(val[1]) and isTsVector(val[2]) then -- box - > 6 numbers return table.concat(val[1], ' ') .. ' ' .. table.concat(val[2], ' ') else error('valToTs: cannot pass Lua tables to TorqueScript.', 3) end else error('valToTs: could not convert value to TorqueScript: ' .. tostring(val), 3) end end -- Type conversion from TS to Lua local fromTsForceTypes = { ['boolean'] = function(val) return tsBool(val) end, ['object'] = function(val) return toTsObject(val) end, -- wrap because toTsObject not defined yet ['string'] = function(val) return val end, } local function forceValFromTs(val, typ) local func = fromTsForceTypes[typ] if not func then error('valFromTs: invalid force type \'' .. typ .. '\'', 4) end return func(val) end local function vectorFromTs(val) local xS, yS, zS = val:match('^(%-?[0-9%.e]+) (%-?[0-9%.e]+) (%-?[0-9%.e]+)$') if xS then return vector { tonumber(xS), tonumber(yS), tonumber(zS) } else return nil end end local function boxFromTs(val) local x1S, y1S, z1S, x2S, y2S, z2S = val:match( '^(%-?[0-9%.e]+) (%-?[0-9%.e]+) (%-?[0-9%.e]+) ' .. '(%-?[0-9%.e]+) (%-?[0-9%.e]+) (%-?[0-9%.e]+)$') if x1S then return { vector { tonumber(x1S), tonumber(y1S), tonumber(z1S) }, vector { tonumber(x2S), tonumber(y2S), tonumber(z2S) } } else return nil end end local function multinumericFromTs(val) local tsNumPat = '%-?[0-9]+%.?[0-9]*e?[0-9]*' if val:find('^' .. tsNumPat) then local nums = {} for _, part in ipairs(val:split(' ')) do if part:find('^' .. tsNumPat .. '$') then table.insert(nums, tonumber(part)) else return nil end end if #nums == 2 or #nums == 3 then return vector(nums) elseif #nums == 6 then -- box return { vector { nums[1], nums[2], nums[3] }, vector { nums[4], nums[5], nums[6] } } elseif #nums == 7 then -- axis return nil --return matrix(nums) else return nil end end end bl._forceType = bl._forceType or {} -- todo: ensure name and name2 are already lowercase local function valFromTs(val, name, name2) if type(val) ~= 'string' then error('valFromTs: expected string, got ' .. type(val), 3) end if name then name = name:lower() if bl._forceType[name] then return forceValFromTs(val, bl._forceType[name]) end end if name2 then name2 = name2:lower() if bl._forceType[name2] then return forceValFromTs(val, bl._forceType[name2]) end end -- '' -> nil if val == '' then return nil end -- number local num = tonumber(val) if num then return num end -- vector, box, or axis->matrix local vec = multinumericFromTs(val) if vec then return vec end -- net string if val:sub(1, 1) == '\001' then return _bllua_ts.call('getTaggedString', val) end -- string return val end local function arglistFromTs(name, argsS) local args = {} for i, arg in ipairsNilable(argsS) do args[i] = valFromTs(arg, name .. ':' .. i) end return args end local function arglistToTs(args) local argsS = {} for i, v in ipairsNilable(args) do table.insert(argsS, valToTs(v)) end return argsS end local function classFromForceTypeStr(name) local class, rest = name:match('^([a-zA-Z0-9_]+)(::.+)$') if not class then class, rest = name:match('^([a-zA-Z0-9_]+)(%..+)$') end return class, rest end local setForceType setForceType = function(ftname, typ) if typ ~= nil and not fromTsForceTypes[typ] then error('bl.type: invalid type \'' .. typ .. '\'', 2) end if not isValidFuncNameNsArgn(ftname) then error('bl.type: invalid function or variable name \'' .. ftname .. '\'', 2) end ftname = ftname:lower() bl._forceType[ftname] = typ -- apply to child classes if present local cname, rest = classFromForceTypeStr(ftname) if cname then local meta = bl._objectUserMetas[cname] if meta then for chcname, _ in pairs(meta._children) do setForceType(chcname .. rest, typ) end end end end bl.type = setForceType -- Type conversion TS->Lua backwards, convert back to string local function numFromTsTostring(val) -- todo: as-good-as-possible scientific notation for numbers return tostring(val) end local function valFromTsTostring(val) if type(val) == 'string' then return val elseif type(val) == 'number' then return numFromTsTostring(val) elseif type(val) == 'table' then if val.__is_vector then local strs = {} for i = 1, #val do strs[i] = numFromTsTostring(val[i]) end return table.concat(strs, ' ') elseif val.__is_matrix then -- todo: matrix back to axis-angle string error('bl.string: matrix not yet supported', 3) end end error('bl.string: cannot convert \'' .. type(val) .. '\'', 3) end -- Getting info from TS about functions and objects local function isTsObject(t) return type(t) == 'table' and t._tsObjectId ~= nil end local function tsIsObject(name) return tsBool(_bllua_ts.call('isObject', name)) end local function tsIsFunction(name) return tsBool(_bllua_ts.call('isFunction', name)) end local function tsIsFunctionNs(ns, name) return tsBool(_bllua_ts.call('isFunction', ns, name)) end local function tsIsFunctionNsname(nsname) local ns, name = nsname:match('^([^:]+)::([^:]+)$') if ns then return tsIsFunctionNs(ns, name) else return tsIsFunction(nsname) end end -- sanity check to make sure objects that don't isObject are always marked as ._deleted -- can be removed later local function assertTsObjectExists(obj) local is = tsIsObject(obj._tsObjectId) if not is then print('Warning: TS object :exists or isobject from lua but no longer exists') end return is end function bl.isObject(obj) if type(obj) == 'number' then -- object id return tsIsObject(tostring(obj)) elseif type(obj) == 'string' then -- object name return tsIsObject(obj) elseif isTsObject(obj) then -- lua object if obj._deleted then return false else return assertTsObjectExists(obj) end else error('bl.isObject: argument #1: expected torque object, number, or string', 2) end end function bl.isFunction(name) return tsIsFunctionNsname(name) end -- Torque object pseudo-class local tsClassMeta = { __tostring = function(t) return 'torqueClass:' .. t._name .. (t._inherit and (':' .. t._inherit._name) or '') end, } bl._objectUserMetas = bl._objectUserMetas or {} function bl.class(cname, inhname) if not (type(cname) == 'string' and isValidFuncName(cname)) then error('bl.class: argument #1: invalid class name', 2) end if not (inhname == nil or (type(inhname) == 'string' and isValidFuncName(inhname))) then error('bl.class: argument #2: inherit name must be a string or nil', 2) end cname = cname:lower() local met = bl._objectUserMetas[cname] or { _name = cname, _inherit = nil, _children = {}, } bl._objectUserMetas[cname] = met setmetatable(met, tsClassMeta) if inhname then inhname = inhname:lower() local inh = bl._objectUserMetas[inhname] if not inh then error('bl.class: argument #2: \'' .. inhname .. '\' is not the ' .. 'name of an existing class', 2) end inh._children[cname] = true local inhI = met._inherit if inhI and inhI ~= inh then error('bl.class: argument #2: class already exists and ' .. 'inherits a different parent.', 2) end met._inherit = inh -- apply inherited method and field types for ftname, typ in pairs(bl._forceType) do local cname2, rest = classFromForceTypeStr(ftname) if cname2 == inhname then setForceType(cname .. rest, typ) end end end end local function objectInheritedMetas(name) local inh = bl._objectUserMetas[name:lower()] return function() local inhP = inh if inhP == nil then return nil end inh = inh._inherit return inhP end end local tsObjectMeta = { -- __index: Called when accessing fields that don't exist in the object itself -- Return torque member function or value __index = function(t, name) if rawget(t, '_deleted') then error('ts object index: object no longer exists', 2) end if type(name) ~= 'string' and type(name) ~= 'number' then error('ts object index: index must be a string or number', 2) end if getmetatable(t)[name] then return getmetatable(t)[name] elseif type(name) == 'number' then if not tsIsFunctionNs(rawget(t, '_tsNamespace'), 'getObject') then error('ts object __index: index is number, but object does not have ' .. 'getObject method', 2) end return toTsObject(_bllua_ts.callobj(t._tsObjectId, 'getObject', tostring(name))) else for inh in objectInheritedMetas(rawget(t, '_tsClassName')) do if inh[name] then return inh[name] end end if tsIsFunctionNs(rawget(t, '_tsNamespace'), name) or tsIsFunctionNs(rawget(t, '_tsName'), name) then return function(t2, ...) if t2 == nil or type(t2) ~= 'table' or not t2._tsObjectId then error('ts object method: be sure to use :func() not .func()', 2) end local argsS = arglistToTs({ ... }) local res = _bllua_ts.callobj(t2._tsObjectId, name, unpack(argsS)) return valFromTs(res, t2._tsName and t2._tsName .. '::' .. name, t2._tsNamespace .. '::' .. name) end else local res = _bllua_ts.getfield(rawget(t, '_tsObjectId'), name) return valFromTs(res, rawget(t, '_tsName') and rawget(t, '_tsName') .. '.' .. name, rawget(t, '_tsNamespace') .. '.' .. name) end end end, -- __newindex: Called when setting fields on the object -- Set lua data -- Use :set() to set Torque data __newindex = function(t, name, val) if rawget(t, '_deleted') then error('ts object newindex: object no longer exists', 2) end if type(name) ~= 'string' then error('ts object newindex: index must be a string', 2) end rawset(t, name, val) -- create strong reference since it's now storing lua data bl._objectsStrong[rawget(t, '_tsObjectId')] = t end, -- object:set(fieldName, value) -- Use to set torque data set = function(t, name, val) if t == nil or type(t) ~= 'table' or not t._tsObjectId then error('ts object method: be sure to use :func() not .func()', 2) end if t._deleted then error('ts object method: object no longer exists', 2) end if type(name) ~= 'string' then error('ts object :set(): index must be a string', 2) end _bllua_ts.setfield(t._tsObjectId, name, valToTs(val)) end, -- __tostring: Called when printing -- Display a nice info string __tostring = function(t) return 'torque:' .. t._tsNamespace .. ':' .. t._tsObjectId .. (t._tsName ~= '' and ('(' .. t._tsName .. ')') or '') .. (t._deleted and '(deleted)' or '') end, -- #object -- If the object has a getCount method, return its count __len = function(t) if t._deleted then error('ts object __len: object no longer exists', 2) end if not tsIsFunctionNs(t._tsNamespace, 'getCount') then error('ts object __len: object has no getCount method', 2) end return tonumber(_bllua_ts.callobj(t._tsObjectId, 'getCount')) end, -- object:members() -- Return an iterator for Torque objects with the getCount and getObject methods -- for index, object in group:members() do ... end members = function(t) if t == nil then error('ts object method: be sure to use :func() not .func()', 2) end if t._deleted then error('ts object method: object no longer exists', 2) end if not ( tsIsFunctionNs(t._tsNamespace, 'getCount') and tsIsFunctionNs(t._tsNamespace, 'getObject')) then error('ts object :members() - ' .. 'Object does not have getCount and getObject methods', 2) end local count = tonumber(_bllua_ts.callobj(t._tsObjectId, 'getCount')) local idx = 0 return function() if idx < count then local obj = toTsObject(_bllua_ts.callobj(t._tsObjectId, 'getObject', tostring(idx))) idx = idx + 1 --return idx-1, obj return obj else return nil end end end, -- Wrap some Torque functions for performance and error checking getName = function(t) if t == nil then error('ts object method: be sure to use :func() not .func()', 2) end if t._deleted then error('ts object method: object no longer exists', 2) end return t._tsName end, getId = function(t) if t == nil then error('ts object method: be sure to use :func() not .func()', 2) end if t._deleted then error('ts object method: object no longer exists', 2) end return tonumber(t._tsObjectId) end, getType = function(t) if t == nil then error('ts object method: be sure to use :func() not .func()', 2) end if t._deleted then error('ts object method: object no longer exists', 2) end return tsTypesByNum[_bllua_ts.callobj(t._tsObjectId, 'getType')] end, ---- Schedule method for objects --schedule = function(t, time, cb, ...) -- if type(t)~='table' or not t._tsObjectId then -- error('ts object method: be sure to use :func() not .func()', 2) end -- if t._deleted then -- error('ts object method: object no longer exists', 2) end -- if type(time)~='number' then -- error('ts object schedule: argument #2: time must be a number', 2) end -- if type(cb)~='function' then -- error('ts object schedule: argument #3: callback must be a function', 2) end -- local args = {...} -- bl.schedule(time, function() -- if tsBool(__bllua_ts.call('isObject', t._tsObjectId)) then -- pcall(cb, unpack(args)) -- end -- end) --end, exists = function(t) if t == nil then error('ts object method: be sure to use :func() not .func()', 2) end if t._deleted then return false else return assertTsObjectExists(t) end end, } -- Weak-values table for caching Torque object references -- Objects in this table can be garbage collected if there are no other refs to them if not bl._objectsWeak then bl._objectsWeak = {} setmetatable(bl._objectsWeak, { __mode = 'v' }) end -- Strong table for preserving Torque object references containing lua data -- If an object in this table, it will remain here and in the Weak table until deleted if not bl._objectsStrong then bl._objectsStrong = {} end -- Hook object deletion to clean up its lua data -- idS is expected to be the object ID number, NOT the object name function _bllua_objectDeleted(idS) local obj = bl._objectsWeak[idS] if obj then obj._deleted = true bl._objectsStrong[idS] = nil bl._objectsWeak[idS] = nil bl._objectsWeak[obj._tsName:lower()] = nil end end -- Return a Torque object for the object ID string, or create one if none exists toTsObject = function(idiS) if type(idiS) ~= 'string' then error('toTsObject: input must be a string', 2) end idiS = idiS:lower() if bl._objectsWeak[idiS] then return bl._objectsWeak[idiS] end if not tsIsObject(idiS) then --error('toTsObject: object \''..idiS..'\' does not exist', 2) end return nil end local className = _bllua_ts.callobj(idiS, 'getClassName') local obj = { _tsObjectId = _bllua_ts.callobj(idiS, 'getId'), _tsName = _bllua_ts.callobj(idiS, 'getName'), _tsNamespace = className, _tsClassName = className:lower(), } setmetatable(obj, tsObjectMeta) bl._objectsWeak[obj._tsObjectId] = obj bl._objectsWeak[obj._tsName:lower()] = obj return obj end -- Allow bl['namespaced::function']() local function safeNamespaceName(name) return tostring(name:gsub(':', '_')) end bl._cachedNamespaceCalls = {} local function tsNamespacedCallTfname(name) local tfname = bl._cachedNamespaceCalls[name] if not tfname then tfname = '_bllua_nscall_' .. safeNamespaceName(name) local tfcode = 'function ' .. tfname .. '(' .. tsArgsLocal .. '){' .. name .. '(' .. tsArgsLocal .. ');}' _bllua_ts.eval(tfcode) bl._cachedNamespaceCalls[name] = tfname end return tfname end local function tsCallGen(name) return function(...) local argsS = arglistToTs({ ... }) local res = _bllua_ts.call(name, unpack(argsS)) return valFromTs(res, name) end end -- Metatable for the global bl library -- Allows accessing Torque objects, variables, and functions by indexing it local tsMeta = { -- __index: Called when accessing fields that don't exist in the table itself -- Allow indexing by object id: bl[1234] -- by object name: bl.mainMenuGui -- by function name: bl.quit() -- by variable name: bl.iAmAdmin __index = function(t, name) if getmetatable(t)[name] then return getmetatable(t)[name] elseif type(name) == 'string' and bl._objectUserMetas[name:lower()] then return bl._objectUserMetas[name:lower()] else if type(name) == 'number' then return toTsObject(tostring(name)) elseif name:find('::') then local ns, rest = name:match('^([^:]+)::(.+)$') if not ns then error('ts index: invalid name \'' .. name .. '\'', 2) end if not rest:find('::') and tsIsFunctionNs(ns, rest) then return tsCallGen(tsNamespacedCallTfname(name)) else local res = _bllua_ts.getvar(name) return valFromTs(res, name) end elseif tsIsFunction(name) then return tsCallGen(name) elseif tsIsObject(name) then return toTsObject(name) else local res = _bllua_ts.getvar(name) return valFromTs(res, name) end end end, } -- bl.set(name, value) -- Used to set global variables function bl.set(name, val) _bllua_ts.setvar(name, valToTs(val)) end -- Utility functions function bl.call(func, ...) local args = { ... } local argsS = arglistToTs(args) return _bllua_ts.call(func, unpack(argsS)) end function bl.eval(code) local res = _bllua_ts.eval(code) return valFromTs(res) end function bl.exec(file) local res = _bllua_ts.call('exec', file) return valFromTs(res) end function bl.array(name, ...) local rest = { ... } return name .. table.concat(rest, '_') end function bl.boolean(val) return val ~= nil and val ~= false and --val~='' and --val~='0' and val ~= 0 end function bl.object(id) if type(id) == 'table' and id._tsObjectId then return id elseif type(id) == 'string' or type(id) == 'number' then return toTsObject(tostring(id)) else error('bl.object: id must be a ts object, number, or string', 2) end end function bl.string(val) return valFromTsTostring(val) end -- Lua calling from TS local luaLookup luaLookup = function(tbl, name, set, val) if name:find('%.') then local first, rest = name:match('^([^%.:]+)%.(.+)$') if not isValidFuncName(first) then error('luaLookup: invalid name \'' .. tostring(first) .. '\'', 3) end if tbl[first] == nil then if set then tbl[first] = {} else return nil end end return luaLookup(tbl[first], rest, set, val) elseif name:find(':') then local first, rest = name:match('^([^%.:]+):(.*)$') if rest:find('[%.:]') then error('luacall: cannot have : or . after :', 3) end if not isValidFuncName(first) then error('luacall: invalid name \'' .. tostring(first) .. '\'', 3) end if not isValidFuncName(rest) then error('luacall: invalid method name \'' .. tostring(first) .. '\'', 3) end if not tbl[first] then error('luacall: no object named \'' .. rest .. '\'', 3) end if tbl[first][rest] == nil then error('luacall: no method named \'' .. rest .. '\'', 3) end return function(...) tbl[first][rest](tbl[first], ...) end else if set then tbl[name] = val else return tbl[name] end end end -- Todo: similar deep access for luaget and luaset function _bllua_call(fname, ...) local args = arglistFromTs(fname:lower(), { ... }) -- todo: separate lua from ts func names? local func = luaLookup(_G, fname) if not func then error('luacall: no global in lua named \'' .. fname .. '\'', 2) end local res = func(unpack(args)) return valToTs(res) end function _bllua_getvar(vname) return valToTs(luaLookup(_G, vname)) end function _bllua_setvar(vname, valS) return valToTs(luaLookup(_G, vname, true, valFromTs(valS, vname))) -- todo: separate lua from ts var names? end function _bllua_eval(code) return loadstring(code)() end function _bllua_exec(fn) return dofile(fn, 2) end -- bl.schedule: Use TS's schedule function to schedule lua calls -- bl.schedule(time, function, args...) bl._scheduleTable = bl._scheduleTable or {} bl._scheduleNextId = bl._scheduleNextId or 1 local function cancelTsSched(sched) if not (sched and sched.handle) then error('schedule:cancel() - invalid object', 2) end _bllua_ts.call('cancel', sched.handle) bl._scheduleTable[id] = nil end function bl.schedule(time, cb, ...) local id = bl._scheduleNextId bl._scheduleNextId = bl._scheduleNextId + 1 local args = { ... } local handle = tonumber(_bllua_ts.call('schedule', time, 0, '_bllua_luacall', '_bllua_schedule_callback', id)) local sch = { callback = cb, args = args, handle = handle, cancel = cancelTsSched, } bl._scheduleTable[id] = sch return sch end function _bllua_schedule_callback(idS) local id = tonumber(idS) or error('_bllua_schedule_callback: invalid id: ' .. tostring(idS)) local sch = bl._scheduleTable[id] if not sch then error('_bllua_schedule_callback: no schedule with id ' .. id) end bl._scheduleTable[id] = nil sch.callback(unpack(sch.args)) end -- serverCmd and clientCmd bl._cmds = bl._cmds or {} function _bllua_process_cmd(cmdS, ...) local cmd = cmdS:lower() local args = arglistFromTs(cmd, { ... }) local func = bl._cmds[cmd] if not func then error('_bllua_process_cmd: no cmd named \'' .. cmd .. '\'') end func(unpack(args)) --pcall(func, unpack(args)) end local function addCmd(cmd, func) if not isValidFuncName(cmd) then error('addCmd: invalid function name \'' .. tostring(cmd) .. '\'') end bl._cmds[cmd] = func _bllua_ts.eval('function ' .. cmd .. '(' .. tsArgsLocal .. '){' .. '_bllua_luacall(_bllua_process_cmd,"' .. cmd .. '",' .. tsArgsLocal .. ');}') end function bl.addServerCmd(name, func) name = name:lower() addCmd('servercmd' .. name, func) bl._forceType['servercmd' .. name .. ':1'] = 'object' end function bl.addClientCmd(name, func) name = name:lower() addCmd('clientcmd' .. name, func) end -- commandToServer and commandToClient function bl.commandToServer(cmd, ...) _bllua_ts.call('commandToServer', _bllua_ts.call('addTaggedString', cmd), unpack(arglistToTs({ ... }))) end function bl.commandToClient(client, cmd, ...) _bllua_ts.call('commandToClient', valToTs(client), _bllua_ts.call('addTaggedString', cmd), unpack(arglistToTs({ ... }))) end function bl.commandToAll(cmd, ...) _bllua_ts.call('commandToAll', _bllua_ts.call('addTaggedString', cmd), unpack(arglistToTs({ ... }))) end -- Hooks (using TS packages) local function isPackageActive(pkg) local numpkgs = tonumber(_bllua_ts.call('getNumActivePackages')) for i = 0, numpkgs - 1 do local apkg = _bllua_ts.call('getActivePackage', tostring(i)) if apkg == pkg then return true end end return false end local function activatePackage(pkg) if not isPackageActive(pkg) then _bllua_ts.call('activatePackage', pkg) end end local function deactivatePackage(pkg) if isPackageActive(pkg) then _bllua_ts.call('deactivatePackage', pkg) end end bl._hooks = bl._hooks or {} function _bllua_process_hook_before(pkgS, nameS, ...) local args = arglistFromTs(nameS, { ... }) local func = bl._hooks[pkgS] and bl._hooks[pkgS][nameS] and bl._hooks[pkgS][nameS].before if not func then error('_bllua_process_hook_before: no hook for ' .. pkgs .. ':' .. nameS) end _bllua_ts.setvar('_bllua_hook_abort', '0') func(args) --pcall(func, args) if args._return then _bllua_ts.setvar('_bllua_hook_abort', '1') _bllua_ts.setvar('_bllua_hook_return', valToTs(args._return)) end for i = 1, tsMaxArgs do _bllua_ts.setvar('_bllua_hook_arg' .. i, valToTs(args[i])) end end function _bllua_process_hook_after(pkgS, nameS, resultS, ...) nameS = nameS:lower() local args = arglistFromTs(nameS, { ... }) args._return = valFromTs(resultS, nameS) local func = bl._hooks[pkgS] and bl._hooks[pkgS][nameS] and bl._hooks[pkgS][nameS].after if not func then error('_bllua_process_hook_after: no hook for ' .. pkgs .. ':' .. nameS) end func(args) --pcall(func, args) return valToTs(args._return) end local function updateHook(pkg, name, hk) local beforeCode = hk.before and ('_bllua_luacall("_bllua_process_hook_before","' .. pkg .. '","' .. name .. '",' .. tsArgsLocal .. ');') or '' local arglist = (hk.before and tsArgsGlobal or tsArgsLocal) local parentCode = tsIsFunctionNsname(name) and -- only call parent if it exists (hk.before and 'if($_bllua_hook_abort)return $_bllua_hook_return;else ' or '') .. ((hk.after and '%result=' or 'return ') .. 'parent::' .. name:match('[^:]+$') .. '(' .. arglist .. ');') or '' local afterCode = hk.after and ('return _bllua_luacall("_bllua_process_hook_after","' .. pkg .. '","' .. name .. '",%result,' .. arglist .. ');') or '' local code = 'package ' .. pkg .. '{' .. 'function ' .. name .. '(' .. tsArgsLocal .. '){' .. beforeCode .. parentCode .. afterCode .. '}' .. '};' print('bl.hook eval output: [[' .. code .. ']]') _bllua_ts.eval(code) end function bl.hook(pkg, name, time, func) if not isValidFuncName(pkg) then error('bl.hook: argument #1: invalid package name \'' .. tostring(pkg) .. '\'', 2) end if not isValidFuncNameNs(name) then error('bl.hook: argument #2: invalid function name \'' .. tostring(name) .. '\'', 2) end if time ~= 'before' and time ~= 'after' then error('bl.hook: argument #3: time must be \'before\' or \'after\'', 2) end if type(func) ~= 'function' then error('bl.hook: argument #4: expected a function', 2) end bl._hooks[pkg] = bl._hooks[pkg] or {} bl._hooks[pkg][name] = bl._hooks[pkg][name] or {} bl._hooks[pkg][name][time] = func updateHook(pkg, name, bl._hooks[pkg][name]) activatePackage(pkg) end local function tableEmpty(t) return next(t) ~= nil end function bl.unhook(pkg, name, time) if not isValidFuncName(pkg) then error('bl.unhook: argument #1: invalid package name \'' .. tostring(pkg) .. '\'', 2) end if name and not isValidFuncNameNs(name) then error('bl.unhook: argument #2: invalid function name \'' .. tostring(name) .. '\'', 2) end if time and time ~= 'before' and time ~= 'after' then error('bl.unhook: argument #3: time must be \'before\' or \'after\'', 2) end if not name then if bl._hooks[pkg] then for name, hk in pairs(bl._hooks[pkg]) do updateHook(pkg, name, {}) end bl._hooks[pkg] = nil else --error('bl.unhook: no hooks registered under package name \''.. -- pkg..'\'', 2) end deactivatePackage(pkg) else if bl._hooks[pkg][name] then if not time then bl._hooks[pkg][name] = nil if table.empty(bl._hooks[pkg]) then bl._hooks[pkg] = nil deactivatePackage(pkg) end updateHook(pkg, name, {}) else if time ~= 'before' and time ~= 'after' then error('bl.unhook: argument #3: time must be nil, \'before\', or \'after\'', 2) end bl._hooks[pkg][name][time] = nil if tableEmpty(bl._hooks[pkg][name]) then bl._hooks[pkg][name] = nil end if tableEmpty(bl._hooks[pkg]) then bl._hooks[pkg] = nil updateHook(pkg, name, {}) deactivatePackage(pkg) else updateHook(pkg, name, bl._hooks[pkg][name]) end end else --error('bl.unhook: no hooks registered for function \''..name.. -- '\' under package name \''..pkg..'\'', 2) end end end -- Container search/raycast local function vecToTs(v) if not isTsVector(v) then error('vecToTs: argument is not a vector', 3) end return table.concat(v, ' ') end local function maskToTs(mask) if type(mask) == 'string' then local val = tsTypesByName[mask:lower()] if not val then error('maskToTs: invalid mask \'' .. mask .. '\'', 3) end return tostring(val) elseif type(mask) == 'table' then local tval = 0 local seen = {} for i, v in ipairs(mask) do if not seen[v] then local val = tsTypesByName[v:lower()] if not val then error('maskToTs: invalid mask \'' .. v .. '\' at index ' .. i .. ' in mask list', 3) end tval = tval + val seen[v] = true end end return tostring(tval) else error('maskToTs: mask must be a string or table', 3) end end local function objToTs(obj) if type(obj) == 'number' or type(obj) == 'string' then return tostring(obj) elseif type(obj) == 'table' and obj._tsObjectId then return tostring(obj._tsObjectId) else error('objToTs: invalid object \'' .. tostring(obj) .. '\'', 3) end end function bl.raycast(start, stop, mask, ignores) local startS = vecToTs(start) local stopS = vecToTs(start) local maskS = maskToTs(mask) local ignoresS = {} for _, v in ipairsNilable(ignores) do table.insert(ignoresS, objToTs(v)) end local retS = _bllua_ts.call('containerRaycast', startS, stopS, maskS, unpack(ignoresS)) if retS == '0' then return nil else local hitS, pxS, pyS, pzS, nxS, nyS, nzS = retS:match('^([0-9]+) ' .. '(%-?[0-9%.e]+) (%-?[0-9%.e]+) (%-?[0-9%.e]+) ' .. '(%-?[0-9%.e]+) (%-?[0-9%.e]+) (%-?[0-9%.e]+)$') local hit = toTsObject(hitS) local pos = vector { tonumber(pxS), tonumber(pyS), tonumber(pzS) } local norm = vector { tonumber(nxS), tonumber(nyS), tonumber(nzS) } return hit, pos, norm end end local function tsContainerSearchIterator() local retS = _bllua_ts.call('containerSearchNext') if retS == '0' then return nil else return toTsObject(retS) end end function bl.boxSearch(pos, size, mask) local posS = vecToTs(pos) local sizeS = vecToTs(size) local maskS = maskToTs(mask) _bllua_ts.call('initContainerBoxSearch', posS, sizeS, maskS) return tsContainerSearchIterator end function bl.radiusSearch(pos, radius, mask) local posS = vecToTs(pos) if type(radius) ~= 'number' then error('bl.radiusSearch: argument #2: radius must be a number', 2) end local radiusS = tostring(radius) local maskS = maskToTs(mask) _bllua_ts.call('initContainerRadiusSearch', posS, radiusS, maskS) return tsContainerSearchIterator end -- Print/Talk/Echo local maxTsArgLen = 8192 local function valsToString(vals) local strs = {} for i, v in ipairsNilable(vals) do local tstr = table.tostring(v) if #tstr > maxTsArgLen then tstr = tostring(v) end strs[i] = tstr end return table.concat(strs, ' ') end bl.echo = function(...) local str = valsToString({ ... }) _bllua_ts.call('echo', str) end print = bl.echo bl.talk = function(...) local str = valsToString({ ... }) _bllua_ts.call('echo', str) _bllua_ts.call('talk', str) end -- bl.new and bl.datablock local function createTsObj(keyword, class, name, inherit, props) local propsT = {} if props then for k, v in pairs(props) do if not isValidFuncName(k) then error('bl.new/bl.datablock: invalid property name \'' .. k .. '\'') end table.insert(propsT, k .. '="' .. valToTs(v) .. '";') end end local objS = _bllua_ts.eval( 'return ' .. keyword .. ' ' .. class .. '(' .. (name or '') .. (inherit and (':' .. inherit) or '') .. '){' .. table.concat(propsT) .. '};') local obj = toTsObject(objS) if not obj then error('bl.new/bl.datablock: failed to create object', 3) end return obj end local function parseTsDecl(decl) local class, name, inherit if decl:find(' ') then -- class ... local cl, rest = decl:match('^([^ ]+) ([^ ]+)$') class = cl if rest:find(':') then -- class name:inherit name, inherit = rest:match('^([^:]*):([^:]+)$') if not name then class = nil end -- error if name == '' then name = nil end -- class :inherit else name = rest end else -- class class = decl end if not ( isValidFuncName(class) and (name == nil or isValidFuncName(name)) and (inherit == nil or isValidFuncName(inherit))) then error('bl.new/bl.datablock: invalid decl \'' .. decl .. '\'\n' .. 'must be of the format: \'className\', \'className name\', ' .. '\'className :inherit\', or \'className name:inherit\'', 3) end return class, name, inherit end function bl.new(decl, props) local class, name, inherit = parseTsDecl(decl) return createTsObj('new', class, name, inherit, props) end function bl.datablock(decl, props) local class, name, inherit = parseTsDecl(decl) if not name then error('bl.datablock: must specify a name', 2) end return createTsObj('datablock', class, name, inherit, props) end setmetatable(bl, tsMeta) print(' Executed libbl.lua')