feat: clean code and 201

Signed-off-by: Jad Chahed <[email protected]>

Author: Jad31

go/admin/api.go

@@ -165,7 +165,14 @@ func (a *Admin) handleGitHubCallback(c *gin.Context) {
 
 		randomKey := random.String(32)
 		tokens[randomKey] = *token
-		c.SetCookie("ghtoken", randomKey, int(time.Hour.Seconds()), "/", "localhost", true, true)
+
+		domain := "localhost"
+
+		if a.Mode == server.ProductionMode {
+			domain = "benchmark.vitess.io"
+		}
+
+		c.SetCookie("ghtoken", randomKey, int(time.Hour.Seconds()), "/", domain, true, true)
 
 		c.Redirect(http.StatusSeeOther, "/admin/dashboard")
 	} else {
@@ -253,6 +260,10 @@ func (a *Admin) handleExecutionsAdd(c *gin.Context) {
 
 	serverAPIURL := "http://localhost:8080/api/executions/add"
 
+	if a.Mode == server.ProductionMode {
+		serverAPIURL = "https://benchmark.vitess.io/api/executions/add"
+	}
+
 	req, err := http.NewRequest("POST", serverAPIURL, bytes.NewBuffer(jsonData))
 
 	if err != nil {
@@ -275,11 +286,11 @@ func (a *Admin) handleExecutionsAdd(c *gin.Context) {
 
 	defer resp.Body.Close()
 
-	if resp.StatusCode != http.StatusOK {
+	if resp.StatusCode != http.StatusCreated {
 		slog.Error("Server API returned an error: ", resp.Status)
 		c.JSON(resp.StatusCode, gin.H{"error": "Failed to process request on server API"})
 		return
 	}
 
-	c.JSON(http.StatusOK, gin.H{"message": "Execution(s) added successfully"})
+	c.JSON(http.StatusCreated, gin.H{"message": "Execution(s) added successfully"})
 }

go/admin/templates/content.html

@@ -130,4 +130,13 @@
     document.getElementById("response").classList.add("text-red-500");
     document.getElementById("response").innerHTML = errorMessage;
 });
+
+// Display success on 201
+document.body.addEventListener("htmx:afterOnLoad", function(e) {
+  if (e.detail.xhr.status == 201) {
+    document.getElementById("response").classList.remove("text-red-500");
+    document.getElementById("response").classList.add("text-green-500");
+    document.getElementById("response").innerHTML = "Execution added successfully!";
+  }
+});
 </script>

go/server/api.go

@@ -569,8 +569,6 @@ func (s *Server) addExecutions(c *gin.Context) {
 		return
 	}
 
-	slog.Info(decryptedToken)
-
 	if req.Source == "" || req.SHA == "" || len(req.Workloads) == 0 || req.NumberOfExecutions == "" {
 		c.JSON(http.StatusBadRequest, &ErrorAPI{Error: "missing argument"})
 		return
@@ -594,9 +592,9 @@ func (s *Server) addExecutions(c *gin.Context) {
 		}
 	}
 
-	// s.appendToQueue(newElements)
+	s.appendToQueue(newElements)
 
-	c.JSON(http.StatusOK, "ok")
+	c.JSON(http.StatusCreated, "")
 }
 
 func IsUserAuthenticated(accessToken string) (bool, error) {