From fb20eb77d0328d1fcf8b4b1f22223c446c5f9cdf Mon Sep 17 00:00:00 2001
From: Christoph Reiter <reiter.christoph@gmail.com>
Date: Mon, 1 Mar 2021 10:28:28 +0100
Subject: [PATCH] Add a traffic stats page

hacky iframe include..
---
 app/templates/navbar.html |  3 +++
 app/templates/stats.html  | 18 ++++++++++++++++++
 app/web.py                |  5 +++++
 tests/test_main.py        |  2 +-
 4 files changed, 27 insertions(+), 1 deletion(-)
 create mode 100644 app/templates/stats.html

diff --git a/app/templates/navbar.html b/app/templates/navbar.html
index 314c64c..5dbbe77 100644
--- a/app/templates/navbar.html
+++ b/app/templates/navbar.html
@@ -7,6 +7,9 @@
 <li class="nav-item">
     <a class="nav-link {{ 'active' if is_endpoint('outofdate') else '' }}" href="{{ url_for('outofdate') }}">Outdated Packages</a>
 </li>
+<li class="nav-item">
+    <a class="nav-link {{ 'active' if is_endpoint('stats') else '' }}" href="{{ url_for('stats') }}">Traffic Stats</a>
+</li>
 <hr>
 <li class="nav-item">
     <a class="nav-link {{ 'active' if is_endpoint('repos') else '' }}" href="{{ url_for('repos') }}">Repos</a>
diff --git a/app/templates/stats.html b/app/templates/stats.html
new file mode 100644
index 0000000..38a8c0f
--- /dev/null
+++ b/app/templates/stats.html
@@ -0,0 +1,18 @@
+{% extends "layout.html" %}
+{% block title %}Traffic Stats{% endblock %}
+{% block inner_content %}
+
+<style>
+    iframe {
+        width: 100%;
+        height: calc(100vh - 8rem);
+    }
+</style>
+
+<div class="card">
+  <div class="overflow-auto">
+    <iframe src="https://repo.msys2.org/stats/?i=enp2s0"></iframe>
+  </div>
+</div>
+
+{% endblock %}
diff --git a/app/web.py b/app/web.py
index 36e70f1..9343bab 100644
--- a/app/web.py
+++ b/app/web.py
@@ -147,6 +147,11 @@ async def repos(request: Request, response: Response) -> Response:
     return templates.TemplateResponse("repos.html", {"request": request, "repos": get_repositories()}, headers=dict(response.headers))
 
 
+@router.get('/stats', dependencies=[Depends(Etag(get_etag))])
+async def stats(request: Request, response: Response) -> Response:
+    return templates.TemplateResponse("stats.html", {"request": request}, headers=dict(response.headers))
+
+
 @router.get('/', dependencies=[Depends(Etag(get_etag))])
 async def index(request: Request, response: Response) -> Response:
     return RedirectResponse(request.url_for('queue'), headers=dict(response.headers))
diff --git a/tests/test_main.py b/tests/test_main.py
index 67c9c5e..1912847 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -25,7 +25,7 @@ def client():
 @pytest.mark.parametrize("endpoint", [
     '', 'repos', 'base', 'group', 'updates', 'outofdate', 'queue', 'new',
     'search', 'base/foo', 'group/foo', 'package/foo',
-    'package',
+    'package', 'stats',
 ])
 def test_main_endpoints(client, endpoint):
     r = client.get('/' + endpoint)