routes restructoring

cmd/internal/routes/user.go

@@ -0,0 +1,78 @@
+package routes
+
+import (
+	"encoding/json"
+	"net/http"
+	"time"
+
+	"github.com/siestaw/laterna/server/cmd/internal/config"
+	"github.com/siestaw/laterna/server/cmd/internal/db"
+	"github.com/siestaw/laterna/server/cmd/internal/logger"
+	"github.com/siestaw/laterna/server/cmd/internal/models"
+	"github.com/siestaw/laterna/server/cmd/utils"
+)
+
+func RegisterUserRoutes(mux *http.ServeMux) {
+	mux.HandleFunc("GET /api/v1/id/{ID}", getCurrent)
+	mux.HandleFunc("PUT /api/v1/id/{ID}", setCurrent)
+}
+
+func getCurrent(w http.ResponseWriter, r *http.Request) {
+	idStr := r.PathValue("ID")
+	id, err := utils.IDtoInt(idStr)
+	if err != nil {
+		utils.HTTPErrorHandling(w, r, http.StatusBadRequest, err.Error())
+		return
+	}
+
+	state, err := db.ViewColor(id)
+	if err != nil {
+		utils.HTTPErrorHandling(w, r, http.StatusBadRequest, "Lamp not found")
+		return
+	}
+
+	w.Header().Set("Content-Type", "application/json")
+	json.NewEncoder(w).Encode(state)
+	defer r.Body.Close()
+}
+
+func setCurrent(w http.ResponseWriter, r *http.Request) {
+	idStr := r.PathValue("ID")
+	id, err := utils.IDtoInt(idStr)
+	if err != nil {
+		utils.HTTPErrorHandling(w, r, http.StatusBadRequest, err.Error())
+		return
+	}
+	currentState, err := db.ViewColor(id)
+	if err != nil {
+		utils.HTTPErrorHandling(w, r, http.StatusInternalServerError, "Failed to get current lamp color")
+		return
+	}
+
+	if time.Since(currentState.UpdatedAt).Seconds() < config.AppConfig.HTTP.Cooldown {
+		utils.HTTPErrorHandling(w, r, http.StatusTooManyRequests, "Slow down!")
+		return
+	}
+
+	var req models.LampUpdateRequest
+	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
+		logger.HTTPLogger.Print(err)
+		utils.HTTPErrorHandling(w, r, http.StatusBadRequest, "Invalid JSON")
+		return
+	}
+
+	err = db.SetColor(id, req.Color)
+	if err != nil {
+		logger.HTTPLogger.Printf("Could not update lamp %d: %s", id, err)
+		utils.HTTPErrorHandling(w, r, http.StatusInternalServerError, err.Error())
+		return
+	}
+
+	updatedState := currentState
+	updatedState.Color = req.Color
+	updatedState.UpdatedAt = time.Now()
+
+	w.Header().Set("Content-Type", "application/json")
+	json.NewEncoder(w).Encode(updatedState)
+	defer r.Body.Close()
+}