2 -- SPDX-License-Identifier: BSD-2-Clause-FreeBSD
4 -- Copyright (c) 2015 Pedro Souza <pedrosouza@freebsd.org>
5 -- Copyright (c) 2018 Kyle Evans <kevans@FreeBSD.org>
6 -- All rights reserved.
8 -- Redistribution and use in source and binary forms, with or without
9 -- modification, are permitted provided that the following conditions
11 -- 1. Redistributions of source code must retain the above copyright
12 -- notice, this list of conditions and the following disclaimer.
13 -- 2. Redistributions in binary form must reproduce the above copyright
14 -- notice, this list of conditions and the following disclaimer in the
15 -- documentation and/or other materials provided with the distribution.
17 -- THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18 -- ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 -- IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 -- ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21 -- FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 -- DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 -- OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 -- HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 -- LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 -- OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 local color = require("color")
33 local config = require("config")
34 local core = require("core")
35 local screen = require("screen")
42 local function menuEntryName(drawing_menu, entry)
43 local name_handler = drawer.menu_name_handlers[entry.entry_type]
45 if name_handler ~= nil then
46 return name_handler(drawing_menu, entry)
48 if type(entry.name) == "function" then
54 local function getBranddef(brand)
59 local branddef = drawer.branddefs[brand]
62 if branddef == nil then
63 try_include('brand-' .. brand)
64 branddef = drawer.branddefs[brand]
70 local function getLogodef(logo)
75 local logodef = drawer.logodefs[logo]
78 if logodef == nil then
79 try_include('logo-' .. logo)
80 logodef = drawer.logodefs[logo]
87 " ______ ____ _____ _____ ",
88 " | ____| | _ \\ / ____| __ \\ ",
89 " | |___ _ __ ___ ___ | |_) | (___ | | | |",
90 " | ___| '__/ _ \\/ _ \\| _ < \\___ \\| | | |",
91 " | | | | | __/ __/| |_) |____) | |__| |",
92 " | | | | | | || | | |",
93 " |_| |_| \\___|\\___||____/|_____/|_____/ "
98 drawer.default_brand = 'fbsd'
100 drawer.menu_name_handlers = {
101 -- Menu name handlers should take the menu being drawn and entry being
102 -- drawn as parameters, and return the name of the item.
103 -- This is designed so that everything, including menu separators, may
104 -- have their names derived differently. The default action for entry
105 -- types not specified here is to use entry.name directly.
106 [core.MENU_SEPARATOR] = function(_, entry)
107 if entry.name ~= nil then
108 if type(entry.name) == "function" then
115 [core.MENU_CAROUSEL_ENTRY] = function(_, entry)
116 local carid = entry.carousel_id
117 local caridx = config.getCarouselIndex(carid)
118 local choices = entry.items
119 if type(choices) == "function" then
122 if #choices < caridx then
125 return entry.name(caridx, choices[caridx], choices)
129 drawer.brand_position = {x = 2, y = 1}
130 drawer.logo_position = {x = 46, y = 4}
131 drawer.menu_position = {x = 5, y = 10}
132 drawer.frame_size = {w = 42, h = 13}
133 drawer.default_shift = {x = 0, y = 0}
134 drawer.shift = drawer.default_shift
137 -- Indexed by valid values for loader_brand in loader.conf(5). Valid
138 -- keys are: graphic (table depicting graphic)
140 graphic = fbsd_brand,
147 function drawer.addBrand(name, def)
148 drawer.branddefs[name] = def
151 function drawer.addLogo(name, def)
152 drawer.logodefs[name] = def
156 -- Indexed by valid values for loader_logo in loader.conf(5). Valid keys
157 -- are: requires_color (boolean), graphic (table depicting graphic), and
158 -- shift (table containing x and y).
160 graphic = fbsd_brand,
163 graphic = fbsd_brand,
167 shift = {x = 17, y = 0},
171 drawer.frame_styles = {
172 -- Indexed by valid values for loader_menu_frame in loader.conf(5).
173 -- All of the keys appearing below must be set for any menu frame style
174 -- added to drawer.frame_styles.
187 bottom_left = "\xC0",
189 bottom_right = "\xD9",
195 bottom_left = "\xC8",
197 bottom_right = "\xBC",
201 function drawer.drawscreen(menu_opts)
202 -- drawlogo() must go first.
203 -- it determines the positions of other elements
207 return drawer.drawmenu(menu_opts)
210 function drawer.drawmenu(menudef)
211 local x = drawer.menu_position.x
212 local y = drawer.menu_position.y
214 x = x + drawer.shift.x
215 y = y + drawer.shift.y
217 -- print the menu and build the alias table
218 local alias_table = {}
220 local menu_entries = menudef.entries
221 local effective_line_num = 0
222 if type(menu_entries) == "function" then
223 menu_entries = menu_entries()
225 for _, e in ipairs(menu_entries) do
226 -- Allow menu items to be conditionally visible by specifying
227 -- a visible function.
228 if e.visible ~= nil and not e.visible() then
231 effective_line_num = effective_line_num + 1
232 if e.entry_type ~= core.MENU_SEPARATOR then
233 entry_num = entry_num + 1
234 screen.setcursor(x, y + effective_line_num)
236 printc(entry_num .. ". " .. menuEntryName(menudef, e))
238 -- fill the alias table
239 alias_table[tostring(entry_num)] = e
240 if e.alias ~= nil then
241 for _, a in ipairs(e.alias) do
246 screen.setcursor(x, y + effective_line_num)
247 printc(menuEntryName(menudef, e))
254 function drawer.drawbox()
255 local x = drawer.menu_position.x - 3
256 local y = drawer.menu_position.y - 1
257 local w = drawer.frame_size.w
258 local h = drawer.frame_size.h
260 local framestyle = loader.getenv("loader_menu_frame") or "double"
261 local framespec = drawer.frame_styles[framestyle]
262 -- If we don't have a framespec for the current frame style, just don't
264 if framespec == nil then
268 local hl = framespec.horizontal
269 local vl = framespec.vertical
271 local tl = framespec.top_left
272 local bl = framespec.bottom_left
273 local tr = framespec.top_right
274 local br = framespec.bottom_right
276 x = x + drawer.shift.x
277 y = y + drawer.shift.y
279 screen.setcursor(x, y); printc(tl)
280 screen.setcursor(x, y + h); printc(bl)
281 screen.setcursor(x + w, y); printc(tr)
282 screen.setcursor(x + w, y + h); printc(br)
284 screen.setcursor(x + 1, y)
289 screen.setcursor(x + 1, y + h)
295 screen.setcursor(x, y + i)
297 screen.setcursor(x + w, y + i)
301 local menu_header = loader.getenv("loader_menu_title") or
303 local menu_header_align = loader.getenv("loader_menu_title_align")
306 if menu_header_align ~= nil then
307 menu_header_align = menu_header_align:lower()
308 if menu_header_align == "left" then
309 -- Just inside the left border on top
310 menu_header_x = x + 1
311 elseif menu_header_align == "right" then
312 -- Just inside the right border on top
313 menu_header_x = x + w - #menu_header
316 if menu_header_x == nil then
317 menu_header_x = x + (w / 2) - (#menu_header / 2)
319 screen.setcursor(menu_header_x, y)
323 function drawer.draw(x, y, logo)
325 screen.setcursor(x, y + i - 1)
330 function drawer.drawbrand()
331 local x = tonumber(loader.getenv("loader_brand_x")) or
332 drawer.brand_position.x
333 local y = tonumber(loader.getenv("loader_brand_y")) or
334 drawer.brand_position.y
336 local branddef = getBranddef(loader.getenv("loader_brand"))
338 if branddef == nil then
339 branddef = getBranddef(drawer.default_brand)
342 local graphic = branddef.graphic
344 x = x + drawer.shift.x
345 y = y + drawer.shift.y
346 drawer.draw(x, y, graphic)
349 function drawer.drawlogo()
350 local x = tonumber(loader.getenv("loader_logo_x")) or
351 drawer.logo_position.x
352 local y = tonumber(loader.getenv("loader_logo_y")) or
353 drawer.logo_position.y
355 local logo = loader.getenv("loader_logo")
356 local colored = color.isEnabled()
358 local logodef = getLogodef(logo)
360 if logodef == nil or logodef.graphic == nil or
361 (not colored and logodef.requires_color) then
362 -- Choose a sensible default
364 logodef = getLogodef("orb")
366 logodef = getLogodef("orbbw")
370 if logodef ~= nil and logodef.graphic == none then
371 drawer.shift = logodef.shift
373 drawer.shift = drawer.default_shift
376 x = x + drawer.shift.x
377 y = y + drawer.shift.y
379 if logodef ~= nil and logodef.shift ~= nil then
380 x = x + logodef.shift.x
381 y = y + logodef.shift.y
384 drawer.draw(x, y, logodef.graphic)