baselib.lua

--=========== Copyright © 2019, Planimeter, All rights reserved. ===========--
--
-- Purpose: Extends the base library
--
--==========================================================================--

require( "engine.shared.concommand" )

function accessor( class, member, verb, key, default )
	if ( type( class ) ~= "table" ) then
		typerror( 1, "table", class )
	end

	class[ "set" .. string.capitalize( member ) ] = function( self, value )
		self[ key or member ] = value
	end

	class[ ( verb or "get" ) .. string.capitalize( member ) ] = function( self )
		return self[ key or member ] or default
	end
end

if ( rawdofile == nil ) then
	rawdofile = dofile

	function dofile( filename )
		local f = assert( love.filesystem.load( filename ) )
		return f()
	end
end

if ( rawprint == nil and rawtype == nil ) then
	rawprint           = print
	rawtype            = type

	local rawtype      = rawtype
	local tonumber     = tonumber
	local assert       = assert
	local getmetatable = getmetatable
	local rawget       = rawget

	function print( ... )
		rawprint( ... )

		if ( _CLIENT and g_Console ) then
			g_Console.print( ... )
		end
	end

	function type( object )
		local mt = getmetatable( object )
		if ( mt ~= nil and rawget( mt, "__type" ) ~= nil ) then
			return rawget( mt, "__type" )
		end

		return rawtype( object )
	end
end

function toboolean( v )
	local n = tonumber( v )
	return n ~= nil and n ~= 0
end

function typeof( object, class )
	if ( type( object ) == class ) then
		return true
	end

	if ( rawtype( object ) == "table" ) then
		local base = getbaseclass( object )
		while ( base ~= nil ) do
			if ( base.__type == class ) then
				return true
			end

			base = getbaseclass( base )
		end
	end

	return rawtype( object ) == class
end

function typerror( narg, tname, value )
	local info = debug.getinfo( 2, "n" )
	error( "bad argument #" .. narg .. " " ..
	       "to '" .. info.name ..
	       "' (" .. tname .. " expected, " ..
	       "got " .. type( value ) .. ")", 3 )
end

concommand( "lua_dofile", "Loads and runs the given file",
	function( self, player, command, argString, argTable )
		if ( argTable[ 1 ] == nil ) then
			print( "lua_dofile <filename>" )
			return
		end

		local status, ret = pcall( love.filesystem.load, argString )
		if ( status == true ) then
			status, ret = pcall( ret )
			if ( status == false ) then
				print( ret )
			end
		else
			print( ret )
		end
	end,

	nil,

	function( argS )
		local autocomplete = {}
		local dir = string.stripfilename( argS )
		local files = love.filesystem.getDirectoryItems( dir )
		for _, v in ipairs( files ) do
			local info = love.filesystem.getInfo( dir .. v )
			local isDirectory = info and info.type == "directory" or false
			if ( isDirectory or string.fileextension( v ) == "lua" ) then
				local filename = ( dir ~= "" and dir or "" ) .. v
				local cmd = "lua_dofile " .. filename
				if ( string.find( cmd, "lua_dofile " .. argS, 1, true ) ) then
					table.insert( autocomplete, cmd )
				end
			end
		end

		table.sort( autocomplete )

		return autocomplete
	end
)

concommand( "lua_dostring", "Loads and runs the given string",
	function( self, player, command, argString, argTable )
		if ( argTable[ 1 ] == nil ) then
			print( "lua_dostring <string>" )
			return
		end

		local f, err = loadstring( argString )
		if ( f ) then
			local status, err = pcall( f )
			if ( status == false ) then
				print( err )
			end
		else
			print( err )
		end
	end
)

concommand( "lua_watch", "Watches an expression",
	function( self, player, command, argString, argTable )
		if ( argTable[ 1 ] == nil ) then
			print( "lua_watch <expression>" )
			return
		end

		if ( _G.g_Watch ) then
			_G.g_Watch:remove()
		end

		if ( argTable[ 1 ] == "nil" ) then
			return
		end

		-- Initialize watch expression
		local watch = gui.watch( _G.g_Viewport )
		watch:setExpression( argString )
		_G.g_Watch = watch
	end
)