[locations] Modularize additional column handling

Additional columns are now added through a hook,
moving specialized code where it belongs.

1 files changed, 76 insertions, 0 deletions

diff --git a/modules-available/baseconfig/hooks/locations-column.inc.php b/modules-available/baseconfig/hooks/locations-column.inc.php
new file mode 100644
index 00000000..ca30d56e
--- /dev/null
+++ b/modules-available/baseconfig/hooks/locations-column.inc.php
@@ -0,0 +1,76 @@
+<?php
+
+if (!User::hasPermission('.baseconfig.view'))
+	return null;
+
+class BaseconfigLocationColumn extends AbstractLocationColumn
+{
+
+	private $byLoc;
+	private $byMachine;
+
+	public function __construct(array $allowedLocationIds)
+	{
+		if (in_array(0, $allowedLocationIds)) {
+			$extra = 'OR m.locationid IS NULL';
+		} else {
+			$extra = '';
+		}
+		// Count overridden config vars
+		$this->byLoc = Database::queryKeyValueList("SELECT locationid, Count(*) AS cnt
+				FROM `setting_location`
+				WHERE locationid IN (:allowedLocationIds)
+				GROUP BY locationid", compact('allowedLocationIds'));
+		// Confusing because the count might be inaccurate within a branch
+		//$this->propagateFields($locationList, '', 'overriddenVars', 'overriddenClass');
+		// Count machines with overriden var(s)
+		$this->byMachine = Database::queryKeyValueList("SELECT IFNULL(m.locationid, 0), Count(DISTINCT sm.machineuuid) AS cnt
+				FROM setting_machine sm
+				INNER JOIN machine m USING (machineuuid)
+				WHERE (m.locationid IN (:allowedLocationIds) $extra)
+				GROUP BY m.locationid", compact('allowedLocationIds'));
+		// There WHERE statement drops clients without location - but this cannot be displayed by the locations
+		// table anyways as of now - maybe implement some day? Or just encourage everyone to have a root location
+	}
+
+	public function getColumnHtml(int $locationId): string
+	{
+		$ret = '';
+		if ($this->byLoc[$locationId] ?? 0) {
+			$title = htmlspecialchars(Dictionary::translateFileModule('baseconfig', 'module', 'overriden-vars-for-location'));
+			$ret .= <<<EOF
+			<span class="badge" title="{$title}">
+				<span class="glyphicon glyphicon-home"></span> {$this->byLoc[$locationId]}
+			</span>
+EOF;
+		}
+		if ($this->byMachine[$locationId] ?? 0) {
+			$title = htmlspecialchars(Dictionary::translateFileModule('baseconfig', 'module', 'overriden-vars-machines'));
+			$ret .= <<<EOF
+				<span class="badge" title="{$title}">
+				<span class="glyphicon glyphicon-tasks"></span> {$this->byMachine[$locationId]}
+			</span>
+EOF;
+		}
+		return $ret;
+	}
+
+	public function getEditUrl(int $locationId): string
+	{
+		if (!User::hasPermission('.baseconfig.edit', $locationId))
+			return '';
+		return '?do=baseconfig&module=locations&locationid=' . $locationId;
+	}
+
+	public function header(): string
+	{
+		return Dictionary::translateFileModule('baseconfig', 'module', 'location-column-header');
+	}
+
+	public function priority(): int
+	{
+		return 1000;
+	}
+}
+
+return new BaseconfigLocationColumn($allowedLocationIds);
\ No newline at end of file