From 505f5b73c5f454a24448e77809eb585e9aa6770d Mon Sep 17 00:00:00 2001
From: Simon <simobilleter@gmail.com>
Date: Thu, 31 Aug 2023 14:34:08 +0700
Subject: [PATCH] add primary aggs

---
 tubearchivist/api/src/aggs.py                 | 14 ++--
 .../home/templates/home/settings.html         | 12 ++-
 tubearchivist/static/css/style.css            |  4 +
 tubearchivist/static/stats.js                 | 77 +++++++++++++++++++
 4 files changed, 97 insertions(+), 10 deletions(-)
 create mode 100644 tubearchivist/static/stats.js

diff --git a/tubearchivist/api/src/aggs.py b/tubearchivist/api/src/aggs.py
index aac96f03..f8724df5 100644
--- a/tubearchivist/api/src/aggs.py
+++ b/tubearchivist/api/src/aggs.py
@@ -31,11 +31,13 @@ class Primary(AggBase):
     data = {
         "size": 0,
         "aggs": {
-            "video_type": {"terms": {"field": "vid_type"}},
-            "video_total": {"value_count": {"field": "media_url"}},
-            "channel_total": {"value_count": {"field": "channel_id"}},
+            "video_type": {
+                "filter": {"exists": {"field": "active"}},
+                "aggs": {"filtered": {"terms": {"field": "vid_type"}}},
+            },
+            "channel_total": {"value_count": {"field": "channel_active"}},
             "channel_sub": {"terms": {"field": "channel_subscribed"}},
-            "playlist_total": {"value_count": {"field": "playlist_id"}},
+            "playlist_total": {"value_count": {"field": "playlist_active"}},
             "playlist_sub": {"terms": {"field": "playlist_subscribed"}},
             "download": {"terms": {"field": "status"}},
         },
@@ -45,11 +47,11 @@ class Primary(AggBase):
         """make the call"""
         aggregations = self.get()
 
-        videos = {"total": aggregations["video_total"].get("value")}
+        videos = {"total": aggregations["video_type"].get("doc_count")}
         videos.update(
             {
                 i.get("key"): i.get("doc_count")
-                for i in aggregations["video_type"]["buckets"]
+                for i in aggregations["video_type"]["filtered"]["buckets"]
             }
         )
         channels = {"total": aggregations["channel_total"].get("value")}
diff --git a/tubearchivist/home/templates/home/settings.html b/tubearchivist/home/templates/home/settings.html
index d9e1860b..8d605f74 100644
--- a/tubearchivist/home/templates/home/settings.html
+++ b/tubearchivist/home/templates/home/settings.html
@@ -2,10 +2,14 @@
 {% load static %}
 {% block settings_content %}
 <div class="title-bar">
-    <h1>Dashboard Coming Soon</h1>
+    <h1>Your Archive</h1>
 </div>
-<div class="settings-group">
-    <p>Tube Archivist settings have been moved to their own pages!</p>
-    <p>You can now find them by selecting an item from the sub-menu at the top of the page.</p>
+<div>
+    <h2>Main overview</h2>
+    <div id="primaryBox" class="info-box info-box-4"></div>
 </div>
+<div>
+
+</div>
+<script type="text/javascript" src="{% static 'stats.js' %}"></script>
 {% endblock settings_content %}
diff --git a/tubearchivist/static/css/style.css b/tubearchivist/static/css/style.css
index 74b32852..df9dfadd 100644
--- a/tubearchivist/static/css/style.css
+++ b/tubearchivist/static/css/style.css
@@ -643,6 +643,10 @@ video:-webkit-full-screen {
     background-color: var(--highlight-bg);
 }
 
+.info-box-4 {
+    grid-template-columns: 1fr 1fr 1fr 1fr;
+}
+
 .info-box-3 {
     grid-template-columns: 1fr 1fr 1fr;
 }
diff --git a/tubearchivist/static/stats.js b/tubearchivist/static/stats.js
new file mode 100644
index 00000000..eecf18b6
--- /dev/null
+++ b/tubearchivist/static/stats.js
@@ -0,0 +1,77 @@
+// build stats for settings page
+
+'use strict';
+
+/* globals apiRequest */
+
+function primaryStats() {
+  let apiEndpoint = '/api/stats/primary/';
+  let responseData = apiRequest(apiEndpoint, 'GET');
+  let primaryBox = document.getElementById('primaryBox');
+  let videoTile = buildVideoTile(responseData);
+  primaryBox.appendChild(videoTile);
+  let channelTile = buildChannelTile(responseData);
+  primaryBox.appendChild(channelTile);
+  let playlistTile = buildPlaylistTile(responseData);
+  primaryBox.appendChild(playlistTile);
+  let downloadTile = buildDownloadTile(responseData);
+  primaryBox.appendChild(downloadTile);
+}
+
+function buildTile(titleText) {
+  let tile = document.createElement('div');
+  tile.classList.add('info-box-item');
+  let title = document.createElement('h3');
+  title.innerText = titleText;
+  tile.appendChild(title);
+  return tile;
+}
+
+function buildVideoTile(responseData) {
+  let tile = buildTile(`Total Videos: ${responseData.videos.total || 0}`);
+  let message = document.createElement('p');
+  message.innerHTML = `
+        videos: ${responseData.videos.videos || 0}<br>
+        shorts: ${responseData.videos.shorts || 0}<br>
+        streams: ${responseData.videos.streams || 0}<br>
+    `;
+  tile.appendChild(message);
+
+  return tile;
+}
+
+function buildChannelTile(responseData) {
+  let tile = buildTile(`Total Channels: ${responseData.channels.total || 0}`);
+  let message = document.createElement('p');
+  message.innerHTML = `subscribed: ${responseData.channels.sub_true || 0}`;
+  tile.appendChild(message);
+
+  return tile;
+}
+
+function buildPlaylistTile(responseData) {
+  let tile = buildTile(`Total Playlists: ${responseData.playlists.total || 0}`);
+  let message = document.createElement('p');
+  message.innerHTML = `subscribed: ${responseData.playlists.sub_true || 0}`;
+  tile.appendChild(message);
+
+  return tile;
+}
+
+function buildDownloadTile(responseData) {
+  let tile = buildTile('Downloads');
+  let message = document.createElement('p');
+  message.innerHTML = `
+        pending: ${responseData.downloads.pending || 0}<br>
+        ignored: ${responseData.downloads.ignore || 0}<br>
+    `;
+  tile.appendChild(message);
+
+  return tile;
+}
+
+function buildStats() {
+  primaryStats();
+}
+
+buildStats();