From 617f8ab7ef94f6a97ecfeaafa717026ce3f6ab8a Mon Sep 17 00:00:00 2001
From: WardPearce <wardpearce@pm.me>
Date: Tue, 23 Apr 2024 18:51:03 +1200
Subject: [PATCH] Add support for videoId query flags

---
 main.go | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/main.go b/main.go
index ace9222..152c11b 100644
--- a/main.go
+++ b/main.go
@@ -51,15 +51,26 @@ func main() {
 		},
 	)
 
-	app.Get("/votes/:videoId", handler)
+	// Route for /votes?videoId=:videoId
+	app.Get("/votes", handleQuery)
+
+	// Route for /votes/:videoId
+	app.Get("/votes/:videoId", handleParam)
 
 	log.Fatal(app.Listen(":3000"))
 }
 
-func handler(c *fiber.Ctx) error {
+func handleQuery(c *fiber.Ctx) error {
+	videoId := c.Query("videoId")
+	return getVotes(c, videoId)
+}
 
+func handleParam(c *fiber.Ctx) error {
 	videoId := c.Params("videoId")
+	return getVotes(c, videoId)
+}
 
+func getVotes(c *fiber.Ctx, videoId string) error {
 	match, _ := regexp.Match("^([a-zA-Z0-9_-]{11})", []byte(videoId))
 
 	if !match {