From c2a6dfb649107ce3ad64162e2bb2c3c7275650fb Mon Sep 17 00:00:00 2001
From: Simon Rettberg
Date: Tue, 12 Feb 2019 14:55:12 +0100
Subject: [serversetup*] PXELinux and iPXE side-by-side

---
 .../serversetup-bwlp-ipxe/inc/ipxemenu.inc.php     | 142 +++++++++++++++++++++
 1 file changed, 142 insertions(+)
 create mode 100644 modules-available/serversetup-bwlp-ipxe/inc/ipxemenu.inc.php

(limited to 'modules-available/serversetup-bwlp-ipxe/inc/ipxemenu.inc.php')

diff --git a/modules-available/serversetup-bwlp-ipxe/inc/ipxemenu.inc.php b/modules-available/serversetup-bwlp-ipxe/inc/ipxemenu.inc.php
new file mode 100644
index 00000000..5c1a87d5
--- /dev/null
+++ b/modules-available/serversetup-bwlp-ipxe/inc/ipxemenu.inc.php
@@ -0,0 +1,142 @@
+<?php
+
+class IPxeMenu
+{
+
+	protected $menuid;
+	protected $timeoutMs;
+	protected $title;
+	protected $defaultEntryId;
+	/**
+	 * @var MenuEntry[]
+	 */
+	protected $items = [];
+
+	public function __construct($menu)
+	{
+		if (!is_array($menu)) {
+			$menu = Database::queryFirst("SELECT menuid, timeoutms, title, defaultentryid FROM serversetup_menu
+				WHERE menuid = :menuid LIMIT 1", ['menuid' => $menu]);
+			if (!is_array($menu)) {
+				$menu = ['menuid' => 'foo', 'title' => 'Invalid Menu ID: ' . (int)$menu];
+			}
+		}
+		$this->menuid = (int)$menu['menuid'];
+		$this->timeoutMs = (int)$menu['timeoutms'];
+		$this->title = $menu['title'];
+		$this->defaultEntryId = $menu['defaultentryid'];
+		$res = Database::simpleQuery("SELECT e.menuentryid, e.entryid, e.hotkey, e.title, e.hidden, e.sortval, e.md5pass,
+			b.data AS bootentry
+			FROM serversetup_menuentry e
+			LEFT JOIN serversetup_bootentry b USING (entryid)
+			WHERE e.menuid = :menuid
+			ORDER BY e.sortval ASC, e.title ASC", ['menuid' => $menu['menuid']]);
+		while ($row = $res->fetch(PDO::FETCH_ASSOC)) {
+			$this->items[] = new MenuEntry($row);
+		}
+	}
+
+	public function getMenuDefinition($targetVar, $mode, $slxExtensions)
+	{
+		$str = "menu -- {$this->title}\n";
+		foreach ($this->items as $item) {
+			$str .= $item->getMenuItemScript("m_{$this->menuid}", $this->defaultEntryId, $mode, $slxExtensions);
+		}
+		if ($this->defaultEntryId === null) {
+			$defaultLabel = "mx_{$this->menuid}_poweroff";
+		} else {
+			$defaultLabel = "m_{$this->menuid}_{$this->defaultEntryId}";
+		}
+		$str .= "choose";
+		if ($this->timeoutMs > 0) {
+			$str .= " --timeout {$this->timeoutMs}";
+		}
+		$str .= " $targetVar || goto $defaultLabel || goto fail\n";
+		if ($this->defaultEntryId === null) {
+			$str .= "goto skip_{$defaultLabel}\n"
+				. ":{$defaultLabel}\n"
+				. "poweroff || goto fail\n"
+				. ":skip_{$defaultLabel}\n";
+		}
+		return $str;
+	}
+
+	public function getItemsCode($mode)
+	{
+		$str = '';
+		foreach ($this->items as $item) {
+			$str .= $item->getBootEntryScript("m_{$this->menuid}", 'fail', $mode);
+			$str .= "goto slx_menu\n";
+		}
+		return $str;
+	}
+
+	/*
+	 *
+	 */
+
+	public static function forLocation($locationId)
+	{
+		$chain = null;
+		if (Module::isAvailable('location')) {
+			$chain = Location::getLocationRootChain($locationId);
+		}
+		if (!empty($chain)) {
+			$res = Database::simpleQuery("SELECT m.menuid, m.timeoutms, m.title, IFNULL(ml.defaultentryid, m.defaultentryid) AS defaultentryid, ml.locationid
+			FROM serversetup_menu m
+			INNER JOIN serversetup_menu_location ml USING (menuid)
+			WHERE ml.locationid IN (:chain)", ['chain' => $chain]);
+			if ($res->rowCount() > 0) {
+				// Make the location id key, preserving order (closest location is first)
+				$chain = array_flip($chain);
+				while ($row = $res->fetch(PDO::FETCH_ASSOC)) {
+					// Overwrite the value (numeric ascending values, useless) with menu array of according location
+					$chain[(int)$row['locationid']] = $row;
+				}
+				// Use first one that was found
+				foreach ($chain as $menu) {
+					if (is_array($menu)) {
+						return new IPxeMenu($menu);
+					}
+				}
+				// Should never end up here, but we'd just fall through and use the default
+			}
+		}
+		// We're here, no specific menu, use default
+		$menu = Database::queryFirst("SELECT menuid, timeoutms, title, defaultentryid
+			FROM serversetup_menu
+			ORDER BY isdefault DESC LIMIT 1");
+		if ($menu === false) {
+			return new EmptyIPxeMenu;
+		}
+		return new IPxeMenu($menu);
+	}
+
+	public static function forClient($ip, $uuid)
+	{
+		$locationId = 0;
+		if (Module::isAvailable('location')) {
+			$locationId = Location::getFromIpAndUuid($ip, $uuid);
+		}
+		return self::forLocation($locationId);
+	}
+
+}
+
+class EmptyIPxeMenu extends IPxeMenu
+{
+
+	/** @noinspection PhpMissingParentConstructorInspection */
+	public function __construct()
+	{
+		$this->title = 'No menu defined';
+		$this->menuid = -1;
+		$this->items[] = new MenuEntry([
+			'title' => 'Please create a menu in Server-Setup first'
+		]);
+		$this->items[] = new MenuEntry([
+			'title' => 'Bitte erstellen Sie zunächst ein Menü'
+		]);
+	}
+
+}
\ No newline at end of file
-- 
cgit v1.2.3-55-g7522