freebsd-nq/stand/lua/menu.lua
Kyle Evans 6a5a7e8a53 stand/lua: Don't descend into an empty kernels submenu
This submenu is likely going to go away in favor of kernel selection as it
is done in forth at the moment, but for the time being don't descend into it
if we have no kernels available for listing.
2018-02-16 03:14:23 +00:00

392 lines
8.1 KiB
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"}
},
-- 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();
if #kernels == 0 then
return nil;
end
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();
-- Special key behaviors
if (key == core.KEY_BACKSPACE) and (m ~= menu.welcome) then
break
elseif (key == core.KEY_ENTER) then
core.boot();
-- Should not return
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 == core.KEY_ENTER 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