From dc96f75429b7abfe166e879d926961f9ba3399dd Mon Sep 17 00:00:00 2001
From: Michael Tremer <michael.tremer@ipfire.org>
Date: Mon, 3 Feb 2020 12:12:19 +0000
Subject: [PATCH] fireinfo: Show total amount of profiles in database

Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>
---
 Makefile.am                       |  1 +
 src/templates/base.html           | 10 +++++++++-
 src/templates/fireinfo/admin.html | 19 +++++++++++++++++++
 src/web/__init__.py               |  5 ++++-
 src/web/fireinfo.py               | 13 +++++++++++++
 5 files changed, 46 insertions(+), 2 deletions(-)
 create mode 100644 src/templates/fireinfo/admin.html

diff --git a/Makefile.am b/Makefile.am
index a5f6e265..f9657875 100644
--- a/Makefile.am
+++ b/Makefile.am
@@ -197,6 +197,7 @@ templates_download_DATA = \
 templates_downloaddir = $(templatesdir)/download
 
 templates_fireinfo_DATA = \
+	src/templates/fireinfo/admin.html \
 	src/templates/fireinfo/driver.html \
 	src/templates/fireinfo/index.html \
 	src/templates/fireinfo/i-use-1.png \
diff --git a/src/templates/base.html b/src/templates/base.html
index 5b0ad327..3a56f88c 100644
--- a/src/templates/base.html
+++ b/src/templates/base.html
@@ -142,6 +142,14 @@
 
 						<div class="collapse navbar-collapse" id="navbar">
 							<ul class="navbar-nav ml-auto">
+								{% if current_user and current_user.is_staff() %}
+									<li class="nav-item">
+										<a class="nav-link {% if request.path.startswith("/admin") %}active{% end %}" href="/admin">
+											{{ _("Admin") }}
+										</a>
+									</li>
+								{% end %}
+
 								<li class="nav-item">
 									<a class="nav-link {% if request.path.startswith("/vendors") %}active{% end %}" href="/vendors">
 										{{ _("Vendors") }}
@@ -361,7 +369,7 @@
 						<div class="container">
 							<div class="row flex-lg-row-reverse">
 								<div class="col-12 col-lg-6 text-center text-lg-right">
-									{% if not current_user and hostname in ("blog.ipfire.org", "nopaste.ipfire.org", "wiki.ipfire.org") %}
+									{% if not current_user and hostname in ("blog.ipfire.org", "fireinfo.ipfire.org", "nopaste.ipfire.org", "wiki.ipfire.org") %}
 										<a href="/login">{{ _("Login") }}</a>
 									{% elif current_user %}
 										<p class="mb-0">
diff --git a/src/templates/fireinfo/admin.html b/src/templates/fireinfo/admin.html
new file mode 100644
index 00000000..66269ba8
--- /dev/null
+++ b/src/templates/fireinfo/admin.html
@@ -0,0 +1,19 @@
+{% extends "../base.html" %}
+
+{% block title %}{{ _("Admin") }}{% end block %}
+
+{% block container %}
+	<section>
+		<div class="container">
+			<div class="row">
+				<div class="col col-lg-8">
+					<h1 class="display-2">{{ "{:,d}".format(count) }}</h1>
+
+					<p>
+						{{ _("Total amount of systems reporting back to Fireinfo") }}
+					</p>
+				</div>
+			</div>
+		</div>
+	</section>
+{% end block %}
diff --git a/src/web/__init__.py b/src/web/__init__.py
index 0119d9a2..4821c438 100644
--- a/src/web/__init__.py
+++ b/src/web/__init__.py
@@ -198,6 +198,9 @@ class Application(tornado.web.Application):
 		self.add_handlers(r"fireinfo(\.dev)?\.ipfire\.org", [
 			(r"/", fireinfo.IndexHandler),
 
+			# Admin
+			(r"/admin", fireinfo.AdminIndexHandler),
+
 			# Vendors
 			(r"/vendors", fireinfo.VendorsHandler),
 			(r"/vendors/(pci|usb)/([0-9a-f]{4})", fireinfo.VendorHandler),
@@ -215,7 +218,7 @@ class Application(tornado.web.Application):
 
 			# Send profiles
 			(r"/send/([a-z0-9]+)", fireinfo.ProfileSendHandler),
-		])
+		] + authentication_handlers)
 
 		# i-use.ipfire.org
 		self.add_handlers(r"i-use(\.dev)?\.ipfire\.org", [
diff --git a/src/web/fireinfo.py b/src/web/fireinfo.py
index 47bc9258..a6a9858a 100644
--- a/src/web/fireinfo.py
+++ b/src/web/fireinfo.py
@@ -286,3 +286,16 @@ class DeviceAndGroupsTableModule(ui_modules.UIModule):
 
 		return self.render_string("fireinfo/modules/table-devices-and-groups.html",
 				groups=groups)
+
+
+class AdminIndexHandler(BaseHandler):
+	@tornado.web.authenticated
+	def prepare(self):
+		if not self.current_user.is_staff():
+			raise tornado.web.HTTPError(401)
+
+	@tornado.web.authenticated
+	def get(self):
+		count = self.backend.fireinfo.get_profile_count()
+
+		self.render("fireinfo/admin.html", count=count)
-- 
2.39.2