From b07baca245e1a7d4fab02813a2d8b2f904464356 Mon Sep 17 00:00:00 2001
From: Noah Petherbridge <root@kirsle.net>
Date: Sat, 10 Jun 2023 12:54:13 -0700
Subject: [PATCH] Statistics API: also show blue v. red cameras

---
 pkg/api.go | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/pkg/api.go b/pkg/api.go
index e48fd8b..fd8ef71 100644
--- a/pkg/api.go
+++ b/pkg/api.go
@@ -29,6 +29,10 @@ func (s *Server) Statistics() http.HandlerFunc {
 		var result = struct {
 			UserCount int
 			Usernames []string
+			Cameras   struct {
+				Blue int
+				Red  int
+			}
 		}{
 			Usernames: []string{},
 		}
@@ -43,6 +47,15 @@ func (s *Server) Statistics() http.HandlerFunc {
 				}
 				result.Usernames = append(result.Usernames, sub.Username)
 				unique[sub.Username] = struct{}{}
+
+				// Count cameras by color.
+				if sub.VideoActive {
+					if sub.VideoNSFW {
+						result.Cameras.Red++
+					} else {
+						result.Cameras.Blue++
+					}
+				}
 			}
 		}