Update setup.lua
This commit is contained in:
CadenCoaster
GitHub
280
setup.lua
280
setup.lua
@@ -1,274 +1,3 @@
|
|||||||
local function installMBS(...)
|
|
||||||
local arg = {...}
|
|
||||||
local root_dir = settings.get("mbs.install_path", ".mbs")
|
|
||||||
local rom_dir = "rom/.mbs"
|
|
||||||
local install_dir = fs.exists(root_dir) and root_dir or rom_dir
|
|
||||||
local repo_url = "https://raw.githubusercontent.com/SquidDev-CC/mbs/master/"
|
|
||||||
|
|
||||||
--- Write a string with the given colour to the terminal
|
|
||||||
local function write_coloured(colour, text)
|
|
||||||
local old = term.getTextColour()
|
|
||||||
term.setTextColour(colour)
|
|
||||||
io.write(text)
|
|
||||||
term.setTextColour(old)
|
|
||||||
end
|
|
||||||
|
|
||||||
--- Print usage for this program
|
|
||||||
local commands = { "install", "modules", "module", "download" }
|
|
||||||
local function print_usage()
|
|
||||||
local name = fs.getName(shell.getRunningProgram()):gsub("%.lua$", "")
|
|
||||||
write_coloured(colours.cyan, name .. " modules ") io.write("Print the status of all modules\n")
|
|
||||||
write_coloured(colours.cyan, name .. " module ") io.write("Print information about a given module\n")
|
|
||||||
write_coloured(colours.cyan, name .. " install ") io.write("Download all modules and create a startup file\n")
|
|
||||||
write_coloured(colours.cyan, name .. " download ") io.write("Download all modules WITHOUT creating a startup file\n")
|
|
||||||
end
|
|
||||||
|
|
||||||
--- Attempt to load a module from the given path, returning the module or false
|
|
||||||
-- and an error message.
|
|
||||||
local function load_module(path)
|
|
||||||
if fs.isDir(path) then return false, "Invalid module (is directory)" end
|
|
||||||
|
|
||||||
local fn, err = loadfile(path, _ENV)
|
|
||||||
if not fn then return false, "Invalid module (" .. err .. ")" end
|
|
||||||
|
|
||||||
local ok, res = pcall(fn)
|
|
||||||
if not ok then return false, "Invalid module (" .. res .. ")" end
|
|
||||||
|
|
||||||
if type(res) ~= "table" or type(res.description) ~= "string" or type(res.enabled) ~= "function" then
|
|
||||||
return false, "Malformed module"
|
|
||||||
end
|
|
||||||
|
|
||||||
return res
|
|
||||||
end
|
|
||||||
|
|
||||||
--- Setup all modules
|
|
||||||
local function setup_module(module)
|
|
||||||
for _, setting in ipairs(module.settings) do
|
|
||||||
if settings.define then
|
|
||||||
settings.define(setting.name, {
|
|
||||||
description = setting.description,
|
|
||||||
type = setting.type,
|
|
||||||
default = setting.default,
|
|
||||||
})
|
|
||||||
elseif settings.get(setting.name) == nil then
|
|
||||||
settings.set(setting.name, setting.default)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
--- Download a set of files
|
|
||||||
local function download_files(files)
|
|
||||||
if #files == 0 then return end
|
|
||||||
|
|
||||||
local urls = {}
|
|
||||||
for _, file in ipairs(files) do
|
|
||||||
local url = repo_url .. file
|
|
||||||
http.request(url)
|
|
||||||
urls[url] = file
|
|
||||||
end
|
|
||||||
|
|
||||||
while true do
|
|
||||||
local event, url, arg1 = os.pullEvent()
|
|
||||||
if event == "http_success" and urls[url] then
|
|
||||||
local handle = fs.open(fs.combine(root_dir, urls[url]), "w")
|
|
||||||
handle.write(arg1.readAll())
|
|
||||||
handle.close()
|
|
||||||
arg1.close()
|
|
||||||
|
|
||||||
urls[url] = nil
|
|
||||||
if next(urls) == nil then return end
|
|
||||||
elseif event == "http_failure" and urls[url] then
|
|
||||||
error("Could not download " .. urls[url], 0)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
--- read completion helper, completes text using the given options
|
|
||||||
local function complete_multi(text, options, add_spaces)
|
|
||||||
local results = {}
|
|
||||||
for n = 1, #options do
|
|
||||||
local option = options[n]
|
|
||||||
if #option + (add_spaces and 1 or 0) > #text and option:sub(1, #text) == text then
|
|
||||||
local result = option:sub(#text + 1)
|
|
||||||
if add_spaces then
|
|
||||||
results[#results + 1] = result .. " "
|
|
||||||
else
|
|
||||||
results[#results + 1] = result
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
return results
|
|
||||||
end
|
|
||||||
|
|
||||||
--- Append an object to a list if it is not already contained within
|
|
||||||
local function add_unique(list, x)
|
|
||||||
for i = 1, #list do if list[i] == x then return end end
|
|
||||||
list[#list + 1] = x
|
|
||||||
end
|
|
||||||
|
|
||||||
local function load_all_modules()
|
|
||||||
-- Load all modules and update them.
|
|
||||||
local module_dir = fs.combine(root_dir, "modules")
|
|
||||||
local modules = fs.isDir(module_dir) and fs.list(module_dir) or {}
|
|
||||||
|
|
||||||
-- Add the default modules if not already there.
|
|
||||||
for _, module in ipairs { "lua.lua", "pager.lua", "readline.lua", "shell.lua" } do
|
|
||||||
add_unique(modules, module)
|
|
||||||
end
|
|
||||||
|
|
||||||
local files = {}
|
|
||||||
for i = 1, #modules do files[i] = "modules/" .. modules[i] end
|
|
||||||
download_files(files)
|
|
||||||
|
|
||||||
-- Scan for dependencies in enabled modules, downloading them as well
|
|
||||||
local deps = {}
|
|
||||||
for i = 1, #files do
|
|
||||||
local module = load_module(fs.combine(root_dir, files[i]))
|
|
||||||
if module then
|
|
||||||
setup_module(module)
|
|
||||||
if module.enabled() then
|
|
||||||
for _, dep in ipairs(module.dependencies) do deps[#deps + 1] = dep end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
download_files(deps)
|
|
||||||
end
|
|
||||||
|
|
||||||
if arg.n == 0 then
|
|
||||||
printError("Expected some command")
|
|
||||||
print_usage()
|
|
||||||
error()
|
|
||||||
elseif arg[1] == "download" then
|
|
||||||
load_all_modules()
|
|
||||||
elseif arg[1] == "install" then
|
|
||||||
load_all_modules()
|
|
||||||
|
|
||||||
-- Move the existing startup file. We have to read the whole thing,
|
|
||||||
-- as otherwise we'd end up copying inside ourselves.
|
|
||||||
if fs.exists("startup") and not fs.isDir("startup") then
|
|
||||||
write_coloured(colours.cyan, "Moving your existing startup file to startup/30_startup.lua.\n")
|
|
||||||
|
|
||||||
local handle = fs.open("startup", "r")
|
|
||||||
local contents = handle.readAll()
|
|
||||||
handle.close()
|
|
||||||
fs.delete("startup")
|
|
||||||
|
|
||||||
handle = fs.open("startup/30_startup.lua", "w")
|
|
||||||
handle.write(contents)
|
|
||||||
handle.close()
|
|
||||||
end
|
|
||||||
|
|
||||||
-- Also move the startup.lua file afterwards
|
|
||||||
if fs.exists("startup.lua") and not fs.isDir("startup.lua") then
|
|
||||||
write_coloured(colours.cyan, "Moving your existing startup.lua file to startup/31_startup.lua.\n")
|
|
||||||
fs.move("startup.lua", "startup/31_startup.lua")
|
|
||||||
end
|
|
||||||
|
|
||||||
if fs.exists("startup/99_mbs.lua") then
|
|
||||||
write_coloured(colours.cyan, "Deleting the old startup/99_mbs.lua file. We now run before other startup files.\n")
|
|
||||||
fs.delete("startup/99_mbs.lua")
|
|
||||||
end
|
|
||||||
|
|
||||||
-- We'll run at the first possible position to ensure
|
|
||||||
local handle = fs.open("startup/00_mbs.lua", "w")
|
|
||||||
local current = shell.getRunningProgram()
|
|
||||||
handle.writeLine(("assert(loadfile(%q, _ENV))('startup', %q)"):format(current, current))
|
|
||||||
handle.close()
|
|
||||||
|
|
||||||
write_coloured(colours.green, "Installed! ")
|
|
||||||
io.write("Please reboot to apply changes.\n")
|
|
||||||
elseif arg[1] == "startup" then
|
|
||||||
-- Gather a list of all modules
|
|
||||||
local module_dir = fs.combine(install_dir, "modules")
|
|
||||||
local files = fs.isDir(module_dir) and fs.list(module_dir) or {}
|
|
||||||
|
|
||||||
-- Load those modules and determine which are enabled.
|
|
||||||
local enabled = {}
|
|
||||||
local module_names = {}
|
|
||||||
for _, file in ipairs(files) do
|
|
||||||
local module = load_module(fs.combine(module_dir, file))
|
|
||||||
if module then
|
|
||||||
setup_module(module)
|
|
||||||
module_names[#module_names + 1] = file:gsub("%.lua$", "")
|
|
||||||
if module.enabled() then enabled[#enabled + 1] = module end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
local current = arg[2] or shell.getRunningProgram()
|
|
||||||
shell.setCompletionFunction(current, function(_, index, text, previous)
|
|
||||||
if index == 1 then
|
|
||||||
return complete_multi(text, commands, true)
|
|
||||||
elseif index == 2 and previous[#previous] == "module" then
|
|
||||||
return complete_multi(text, module_names, false)
|
|
||||||
end
|
|
||||||
end)
|
|
||||||
|
|
||||||
-- Setup those modules
|
|
||||||
for _, module in ipairs(enabled) do
|
|
||||||
if type(module.setup) == "function" then module.setup(install_dir) end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- And run the startup hook if needed
|
|
||||||
for _, module in ipairs(enabled) do
|
|
||||||
if type(module.startup) == "function" then module.startup(install_dir) end
|
|
||||||
end
|
|
||||||
|
|
||||||
elseif arg[1] == "modules" then
|
|
||||||
local module_dir = fs.combine(install_dir, "modules")
|
|
||||||
local files = fs.isDir(module_dir) and fs.list(module_dir) or {}
|
|
||||||
local found_any = false
|
|
||||||
|
|
||||||
for _, file in ipairs(files) do
|
|
||||||
local res, err = load_module(fs.combine(module_dir, file))
|
|
||||||
write_coloured(colours.cyan, file:gsub("%.lua$", "") .. " ")
|
|
||||||
if res then
|
|
||||||
write(res.description)
|
|
||||||
if res.enabled() then
|
|
||||||
write_coloured(colours.green, " (enabled)")
|
|
||||||
else
|
|
||||||
write_coloured(colours.red, " (disabled)")
|
|
||||||
end
|
|
||||||
found_any = true
|
|
||||||
else
|
|
||||||
write_coloured(colours.red, err)
|
|
||||||
end
|
|
||||||
|
|
||||||
io.write("\n")
|
|
||||||
end
|
|
||||||
|
|
||||||
if not found_any then error("No modules found. Maybe try running the `install` command?", 0) end
|
|
||||||
elseif arg[1] == "module" then
|
|
||||||
if not arg[2] then error("Expected module name", 0) end
|
|
||||||
local module, err = load_module(fs.combine(install_dir, fs.combine("modules", arg[2] .. ".lua")))
|
|
||||||
if not module then error(err, 0) end
|
|
||||||
|
|
||||||
io.write(module.description)
|
|
||||||
if module.enabled() then
|
|
||||||
write_coloured(colours.green, " (enabled)")
|
|
||||||
else
|
|
||||||
write_coloured(colours.red, " (disabled)")
|
|
||||||
end
|
|
||||||
io.write("\n\n")
|
|
||||||
|
|
||||||
for _, setting in ipairs(module.settings) do
|
|
||||||
local value = settings.get(setting.name)
|
|
||||||
write_coloured(colours.cyan, setting.name)
|
|
||||||
io.write(" " .. setting.description .. " (")
|
|
||||||
write_coloured(colours.yellow, textutils.serialise(value))
|
|
||||||
if value ~= setting.default then
|
|
||||||
io.write(", default is \n")
|
|
||||||
write_coloured(colours.yellow, textutils.serialise(setting.default))
|
|
||||||
end
|
|
||||||
|
|
||||||
io.write(")\n")
|
|
||||||
end
|
|
||||||
else
|
|
||||||
printError("Unknown command")
|
|
||||||
print_usage()
|
|
||||||
error()
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
local function saveFile(url,file)
|
local function saveFile(url,file)
|
||||||
fs.makeDir(fs.getDir(file))
|
fs.makeDir(fs.getDir(file))
|
||||||
local file = fs.open(file,"w")
|
local file = fs.open(file,"w")
|
||||||
@@ -324,10 +53,11 @@ if netmount then
|
|||||||
print("done setting up netmountcc!")
|
print("done setting up netmountcc!")
|
||||||
end
|
end
|
||||||
|
|
||||||
if mbs and not netmount then
|
if mbs then
|
||||||
installMBS("install")
|
if not fs.exists("/rom/programs/mbs.lua") then
|
||||||
elseif mbs and netmount then
|
saveFile("https://raw.githubusercontent.com/SquidDev-CC/mbs/master/mbs.lua", "/mbs.lua")
|
||||||
installMBS("install")
|
end
|
||||||
|
shell.run("mbs install")
|
||||||
end
|
end
|
||||||
|
|
||||||
os.reboot()
|
os.reboot()
|
||||||
|
|||||||
Reference in New Issue
Block a user