freebsd-dev/stand/lua/menu.lua

--
-- Copyright (c) 2015 Pedro Souza <pedrosouza@freebsd.org>
-- All rights reserved.
--
-- Redistribution and use in source and binary forms, with or without
-- modification, are permitted provided that the following conditions
-- are met:
-- 1. Redistributions of source code must retain the above copyright
--    notice, this list of conditions and the following disclaimer.
-- 2. Redistributions in binary form must reproduce the above copyright
--    notice, this list of conditions and the following disclaimer in the
--    documentation and/or other materials provided with the distribution.
--
-- THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
-- ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
-- IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
-- ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
-- FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
-- DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
-- OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
-- HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
-- LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
-- OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
-- SUCH DAMAGE.
--
-- $FreeBSD$
--


local menu = {};

local core = require("core");
local color = require("color");
local config = require("config");
local screen = require("screen");
local drawer = require("drawer");

local OnOff;
local skip;
local run;
local autoboot;

--loader menu tree:
--rooted at menu.welcome
--submenu declarations:
local kernel_options;
local boot_options;
local welcome;

menu.kernel_options = {
	-- this table is dynamically appended to when accessed
	-- return to welcome menu
	{
		entry_type = "return",
		name = function()
			return "Back to main menu"..color.highlight(" [Backspace]");
		end,
		alias = {"\08"}
	}
};

menu.boot_options = {
	-- return to welcome menu
	{
		entry_type = "return",
		name = function()
			return "Back to main menu"..color.highlight(" [Backspace]");
		end,
		alias = {"\08"}
	},

	-- load defaults
	{
		entry_type = "entry",
		name = function()
			return "Load System "..color.highlight("D").."efaults";
		end,
		func = function()
			core.setDefaults()
		end,
		alias = {"d", "D"}
	},

	{
		entry_type = "separator",
		name = function()
			return "";
		end
	},

	{
		entry_type = "separator",
		name = function()
			return "Boot Options:";
		end
	},

	-- acpi
	{
		entry_type = "entry",
		name = function()
			return OnOff(color.highlight("A").."CPI       :", core.acpi);
		end,
		func = function()
			core.setACPI();
		end,
		alias = {"a", "A"}
	},
	-- safe mode
	{
		entry_type = "entry",
		name = function()
			return OnOff("Safe "..color.highlight("M").."ode  :", core.sm);
		end,
		func = function()
			core.setSafeMode();
		end,
		alias = {"m", "M"}
	},
	-- single user
	{
		entry_type = "entry",
		name = function()
			return OnOff(color.highlight("S").."ingle user:", core.su);
		end,
		func = function()
			core.setSingleUser();
		end,
		alias = {"s", "S"}
	},
	-- verbose boot
	{
		entry_type = "entry",
		name = function()
			return OnOff(color.highlight("V").."erbose    :", core.verbose);
		end,
		func = function()
			core.setVerbose();
		end,
		alias = {"v", "V"}
	},
};

menu.welcome = {
	-- boot multi user
	{
		entry_type = "entry",
		name = function()
			return color.highlight("B").."oot Multi user "..color.highlight("[Enter]");
		end,
		func = function()
			core.setSingleUser(false);
			core.boot();
		end,
		alias = {"b", "B", "\013"}
	},

	-- boot single user
	{
		entry_type = "entry",
		name = function()
			return "Boot "..color.highlight("S").."ingle user";
		end,
		func = function()
			core.setSingleUser(true);
			core.boot();
		end,
		alias = {"s", "S"}
	},

	-- escape to interpreter
	{
		entry_type = "return",
		name = function()
			return color.highlight("Esc").."ape to lua interpreter";
		end,
		alias = {"\027"}
	},

	-- reboot
	{
		entry_type = "entry",
		name = function()
			return color.highlight("R").."eboot";
		end,
		func = function()
			loader.perform("reboot");
		end,
		alias = {"r", "R"}
	},


	{
		entry_type = "separator",
		name = function()
			return "";
		end
	},

	{
		entry_type = "separator",
		name = function()
			return "Options:";
		end
	},

	-- kernel options
	{
		entry_type = "submenu",
		name = function()
			local kernels = core.kernelList();
			if #kernels == 0 then
				return "Kernels (not available)";
			end
			return color.highlight("K").."ernels";
		end,
		submenu = function()

			-- dynamically build the kernel menu:
			local kernels = core.kernelList();
			for k, v in ipairs(kernels) do
				menu.kernel_options[#menu.kernel_options + 1] = {
					entry_type = "entry",
					name = function()
						return v;
					end,
					func = function()
						config.reload(v);
					end,
					alias = {} -- automatically enumerated
				}
			end

			return menu.kernel_options;
		end,
		alias = {"k", "K"}
	},

	-- boot options
	{
		entry_type = "submenu",
		name = function()
			return "Boot "..color.highlight("O").."ptions";
		end,
		submenu = function()
			return menu.boot_options;
		end,
		alias = {"o", "O"}
	}

};

function menu.run(m)

	if (menu.skip()) then
		core.autoboot();
		return false;
	end

	if (m == nil) then
		m = menu.welcome;
	end

	-- redraw screen
	screen.clear();
	screen.defcursor();
	local alias_table = drawer.drawscreen(m);

--	menu.autoboot();

	cont = true;
	while cont do
		local key = io.getchar();

		-- Exit on backspace
		if (key == 127) and (m ~= menu.welcome) then
			break
		end

		key = string.char(key)
		-- check to see if key is an alias
		local sel_entry = nil;
		for k, v in pairs(alias_table) do
			if (key == k) then
				sel_entry = v;
			end
		end

		-- if we have an alias do the assigned action:
		if(sel_entry ~= nil) then
			if (sel_entry.entry_type == "entry") then
				-- run function
				sel_entry.func();
			elseif (sel_entry.entry_type == "submenu") then
				-- recurse
				cont = menu.run(sel_entry.submenu());
			elseif (sel_entry.entry_type == "return") then
				-- break recurse
				cont = false;
			end
			-- if we got an alias key the screen is out of date:
			screen.clear();
			screen.defcursor();
			alias_table = drawer.drawscreen(m);
		end
	end

	if (m == menu.welcome) then
		screen.defcursor();
		print("Exiting menu!");
		return false;
	end

	return true;
end

function menu.skip()
	if core.bootserial() then
		return true;
	end
	local c = string.lower(loader.getenv("console") or "");
	if (c:match("^efi[ ;]") or c:match("[ ;]efi[ ;]")) ~= nil then
		return true;
	end

	c = string.lower(loader.getenv("beastie_disable") or "");
	print("beastie_disable", c);
	return c == "yes";
end

function menu.autoboot()
	if menu.already_autoboot == true then
		return;
	end
	menu.already_autoboot = true;

	local ab = loader.getenv("autoboot_delay");
	if ab == "NO" or ab == "no" then
		core.boot();
	end
	ab = tonumber(ab) or 10;

	local x = loader.getenv("loader_menu_timeout_x") or 5;
	local y = loader.getenv("loader_menu_timeout_y") or 22;

	local endtime = loader.time() + ab;
	local time;

	repeat
		time = endtime - loader.time();
		screen.setcursor(x, y);
		print("Autoboot in "..time.." seconds, hit [Enter] to boot"
			      .." or any other key to stop     ");
		screen.defcursor();
		if io.ischar() then
			local ch = io.getchar();
			if ch == 13 then
				break;
			else
				-- prevent autoboot when escaping to interpreter
				loader.setenv("autoboot_delay", "NO");
				-- erase autoboot msg
				screen.setcursor(0, y);
				print("                                        "
					      .."                                        ");
				screen.defcursor();
				return;
			end
		end

		loader.delay(50000);
	until time <= 0
	core.boot();

end

function OnOff(str, b)
	if (b) then
		return str .. color.escapef(color.GREEN).."On"..color.escapef(color.WHITE);
	else
		return str .. color.escapef(color.RED).."off"..color.escapef(color.WHITE);
	end
end

return menu