add admin teams view

admin.go

@@ -47,7 +47,7 @@ func adminLogoutHandler(w http.ResponseWriter, r *http.Request) {
 	http.Redirect(w, r, "/admin/login", http.StatusSeeOther)
 }
 
-func isAdmin(r *http.Request) bool {
+func isAdmin(w http.ResponseWriter, r *http.Request) bool {
 	cookie, err := r.Cookie("admin_session")
 	if err != nil {
 		return false
@@ -65,13 +65,65 @@ func isAdmin(r *http.Request) bool {
 	username = matches[1]
 	passwordHash = matches[2]
 	err = db.QueryRow("SELECT 1 FROM admins WHERE username=? AND PASSWORD=?", username, passwordHash).Scan(new(int))
-	return err != sql.ErrNoRows && err == nil
+	if err != sql.ErrNoRows && err == nil {
+		return true
+	}
+	http.Redirect(w, r, "/admin/login", http.StatusSeeOther)
+	return false
 }
 
 func adminHandler(w http.ResponseWriter, r *http.Request) {
-	if !isAdmin(r) {
-		http.Redirect(w, r, "/admin/login", http.StatusSeeOther)
+	if !isAdmin(w, r) {
+		http.Error(w, "Unauthorized", http.StatusUnauthorized)
 		return
 	}
-	w.Write([]byte("Welcome to the admin panel!"))
+	http.ServeFile(w, r, "templates/adminPanel.html")
+}
+
+func adminTeamsHandler(w http.ResponseWriter, r *http.Request) {
+	if !isAdmin(w, r) {
+		http.Error(w, "Unauthorized", http.StatusUnauthorized)
+		return
+	}
+	rows, err := db.Query("SELECT name, difficulty_levels.level_name, last_cipher, penalty FROM teams JOIN difficulty_levels ON teams.difficulty_level = difficulty_levels.id ORDER BY name")
+	if err != nil {
+		http.Error(w, "Database error", http.StatusInternalServerError)
+		return
+	}
+	defer rows.Close()
+	var teams []TeamTemplateS
+	for rows.Next() {
+		var team TeamTemplateS
+		if err := rows.Scan(&team.TeamName, &team.Difficulty, &team.LastCipher, &team.Penalties); err != nil {
+			http.Error(w, "Database error", http.StatusInternalServerError)
+			return
+		}
+		teams = append(teams, team)
+	}
+	if err := rows.Err(); err != nil {
+		http.Error(w, "Database error", http.StatusInternalServerError)
+		return
+	}
+	if err := AdminTeamsTemplate.Execute(w, teams); err != nil {
+		http.Error(w, "Template error", http.StatusInternalServerError)
+		return
+	}
+}
+
+func AdminStartHandler(w http.ResponseWriter, r *http.Request) {
+	if !isAdmin(w, r) {
+		http.Error(w, "Unauthorized", http.StatusUnauthorized)
+		return
+	}
+	_, err := db.Exec("UPDATE teams SET last_cipher = 1, penalty = 0")
+	if err != nil {
+		http.Error(w, "Database error", http.StatusInternalServerError)
+		return
+	}
+	_, err = db.Exec("DELETE FROM penalties")
+	if err != nil {
+		http.Error(w, "Database error", http.StatusInternalServerError)
+		return
+	}
+	http.Redirect(w, r, "/admin/", http.StatusSeeOther)
 }