From bca689b6ca813aed03bb83b1620ed90ca4eeac37 Mon Sep 17 00:00:00 2001
From: Scott Francis <scott@planetscott.ca>
Date: Mon, 18 Nov 2024 14:58:16 -0500
Subject: [PATCH] Show currently playing stream on dashboard

---
 debug.go                         |  4 ++
 release.go                       |  4 ++
 server.go                        | 68 +++++++++++++++++++++++++-------
 web/static/js/script.js          | 25 ++++++++++++
 web/templates/base.html          |  4 +-
 web/templates/home.html          |  6 +--
 web/templates/stream_counts.html |  2 -
 web/templates/stream_info.html   | 19 +++++++++
 8 files changed, 111 insertions(+), 21 deletions(-)
 create mode 100644 web/static/js/script.js
 delete mode 100644 web/templates/stream_counts.html
 create mode 100644 web/templates/stream_info.html

diff --git a/debug.go b/debug.go
index ef9dc2e..309609a 100644
--- a/debug.go
+++ b/debug.go
@@ -7,3 +7,7 @@ import "github.com/gin-gonic/gin"
 func SetGinMode() {
 	gin.SetMode(gin.DebugMode)
 }
+
+func IsDebugMode() bool {
+	return true
+}
diff --git a/release.go b/release.go
index 19384c0..3bd37fa 100644
--- a/release.go
+++ b/release.go
@@ -7,3 +7,7 @@ import "github.com/gin-gonic/gin"
 func SetGinMode() {
 	gin.SetMode(gin.ReleaseMode)
 }
+
+func IsDebugMode() bool {
+	return false
+}
diff --git a/server.go b/server.go
index 0ab854d..f0ba286 100644
--- a/server.go
+++ b/server.go
@@ -316,12 +316,6 @@ func (s *Server) streamTracker(c *gin.Context) {
 	}
 }
 
-func (s *Server) getActiveStreamCount() int {
-	s.lock.Lock()
-	defer s.lock.Unlock()
-	return len(s.streams)
-}
-
 func (s *Server) getActiveStreams() []*streamInfo {
 	s.lock.Lock()
 	defer s.lock.Unlock()
@@ -368,22 +362,68 @@ func (s *Server) debug() gin.HandlerFunc {
 	}
 }
 
-func (s *Server) getStreamCountData() gin.H {
-	return gin.H{
-		"ActiveStreams": s.getActiveStreamCount(),
-		"TotalStreams":  atomic.LoadInt64(&s.totalStreams),
+func (s *Server) headContent() template.HTML {
+	if IsDebugMode() {
+		return template.HTML(`<script src="/static/js/htmx.min.js?v=debug"></script>
+		<script src="https://cdn.tailwindcss.com"></script>
+    <script>
+        tailwind.config = {
+            darkMode: 'class',
+            theme: {
+                extend: {
+                    colors: {
+                        dark: {
+                            bg: '#1a202c',
+                            text: '#e2e8f0',
+                        },
+                    },
+                },
+            },
+        }
+    </script>
+		<style type="text/tailwindcss">
+			@layer utilities {
+					.toggle-checkbox:checked {
+							@apply: right-0 border-green-400;
+							right: 0;
+							border-color: #68D391;
+					}
+					.toggle-checkbox:checked + .toggle-label {
+							@apply: bg-green-400;
+							background-color: #68D391;
+					}
+			}
+		</style>`)
+	} else {
+		return template.HTML(fmt.Sprintf(`<script src="/static/js/htmx.min.js?v=%[1]s"></script>
+		<link href="/static/css/output.css?v=%[1]s" rel="stylesheet">`, s.version))
 	}
 }
 
 func (s *Server) homePage() gin.HandlerFunc {
+	streamInfoData := s.getStreamInfoData()
+	streamInfoData["HeadContent"] = s.headContent()
+	if IsDebugMode() {
+		streamInfoData["Version"] = "debug"
+	} else {
+		streamInfoData["Version"] = s.version
+	}
 	return func(c *gin.Context) {
-		c.HTML(http.StatusOK, "base.html", s.getStreamCountData())
+		c.HTML(http.StatusOK, "base.html", streamInfoData)
+	}
+}
+
+func (s *Server) getStreamInfoData() gin.H {
+	return gin.H{
+		"ActiveStreams": s.getActiveStreams(),
+		"TotalStreams":  atomic.LoadInt64(&s.totalStreams),
+		"Now":           time.Now(),
 	}
 }
 
-func (s *Server) getStreamCounts() gin.HandlerFunc {
+func (s *Server) getStreamInfo() gin.HandlerFunc {
 	return func(c *gin.Context) {
-		c.HTML(http.StatusOK, "stream_counts.html", s.getStreamCountData())
+		c.HTML(http.StatusOK, "stream_info.html", s.getStreamInfoData())
 	}
 }
 
@@ -400,7 +440,7 @@ func (s *Server) Start(provider *Provider) chan error {
 	s.router.GET(fmt.Sprintf("%s:channelId", channelURIPrefix), s.streamChannel())
 	s.router.PUT("/refresh", s.refresh())
 	s.router.GET("/debug", s.debug())
-	s.router.GET("/stream-counts", s.getStreamCounts())
+	s.router.GET("/stream-info", s.getStreamInfo())
 	s.router.StaticFS("/static", static.AssetFile())
 
 	s.server = &http.Server{
diff --git a/web/static/js/script.js b/web/static/js/script.js
new file mode 100644
index 0000000..5d6c400
--- /dev/null
+++ b/web/static/js/script.js
@@ -0,0 +1,25 @@
+function updateDurations() {
+  const entries = document.querySelectorAll('[data-start-time]');
+  const now = Math.floor(Date.now() / 1000);
+
+  entries.forEach(entry => {
+      const startTime = parseInt(entry.dataset.startTime);
+      const durationSeconds = now - startTime;
+
+      const hours = Math.floor(durationSeconds / 3600);
+      const minutes = Math.floor((durationSeconds % 3600) / 60);
+      const seconds = durationSeconds % 60;
+
+      let durationStr = '';
+      if (hours > 0) durationStr += `${hours}h `;
+      if (minutes > 0) durationStr += `${minutes}m `;
+      durationStr += `${seconds}s`;
+
+      const startTimeStr = new Date(startTime * 1000).toLocaleTimeString();
+      entry.title = `Started: ${startTimeStr} (Duration: ${durationStr})`;
+  });
+}
+
+// Update durations immediately and every second
+updateDurations();
+setInterval(updateDurations, 1000);
diff --git a/web/templates/base.html b/web/templates/base.html
index 8dae9e9..dbce18c 100644
--- a/web/templates/base.html
+++ b/web/templates/base.html
@@ -4,8 +4,7 @@
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <title>ProxyTV</title>
-    <script src="/static/js/htmx.min.js"></script>
-    <link href="/static/css/output.css" rel="stylesheet">
+    {{ .HeadContent }}
 </head>
 <body class="bg-gray-100 dark:bg-dark-bg text-gray-900 dark:text-dark-text">
     <div class="container mx-auto p-4">
@@ -40,5 +39,6 @@
             }
         });
     </script>
+    <script src="/static/js/script.js?v={{ .Version }}"></script>
 </body>
 </html>
\ No newline at end of file
diff --git a/web/templates/home.html b/web/templates/home.html
index 2f71326..1e7081c 100644
--- a/web/templates/home.html
+++ b/web/templates/home.html
@@ -2,9 +2,9 @@
 <h1 class="text-3xl font-bold mb-4 dark:text-dark-text">ProxyTV Dashboard</h1>
 <div class="grid grid-cols-1 md:grid-cols-2 gap-4">
     <div class="bg-white dark:bg-gray-800 p-4 rounded shadow">
-        <h2 class="text-xl font-semibold mb-2 dark:text-dark-text">Server Status</h2>
-        <div hx-get="/stream-counts" hx-trigger="every 5s">
-            {{ template "stream_counts.html" . }}
+        <h2 class="text-xl font-semibold dark:text-dark-text">Server Status</h2>
+        <div hx-get="/stream-info" hx-trigger="every 5s">
+            {{ template "stream_info.html" . }}
         </div>
     </div>
     <div class="bg-white dark:bg-gray-800 p-4 rounded shadow">
diff --git a/web/templates/stream_counts.html b/web/templates/stream_counts.html
deleted file mode 100644
index 6f375e8..0000000
--- a/web/templates/stream_counts.html
+++ /dev/null
@@ -1,2 +0,0 @@
-<p class="dark:text-dark-text">Active Streams: <span id="active-streams">{{ .ActiveStreams }}</span></p>
-<p class="dark:text-dark-text">Total Streams: <span id="total-streams">{{ .TotalStreams }}</span></p>
diff --git a/web/templates/stream_info.html b/web/templates/stream_info.html
new file mode 100644
index 0000000..441d245
--- /dev/null
+++ b/web/templates/stream_info.html
@@ -0,0 +1,19 @@
+<div class="p-4">
+    <h3 class="text-lg font-semibold mb-5">Active Streams: {{len .ActiveStreams}} / Total Streams: {{.TotalStreams}}</h3>
+
+    <div class="flex flex-col gap-3 mt-5">
+        {{range .ActiveStreams}}
+        <div class="flex items-center p-3 bg-gray-50 dark:bg-dark-bg rounded-lg transition-colors hover:bg-gray-100" data-start-time="{{.StartTime.Unix}}" title="Started: {{.StartTime.Format "15:04:05"}}">
+            {{if .LogoURL}}
+            <img src="{{.LogoURL}}" alt="{{.Name}}" class="w-10 h-10 object-contain mr-4">
+            {{end}}
+            <div class="flex gap-x-6">
+                <div class="font-bold">{{.Name}}</div>
+                <div class="text-sm text-gray-500 content-center">{{.ClientIP}}</div>
+            </div>
+        </div>
+        {{else}}
+        <div class="text-center font-bold text-gray-500 py-5">No active streams</div>
+        {{end}}
+    </div>
+</div>