fix: Fix broken shutdown handler

This is a #PATCH release.

cmd/app/server.go

@@ -18,7 +18,7 @@ to handle potential shutdown requests and incoming HTTP requests.
 */
 func StartServer(client *Client, projectInfo *ProjectInfo, GitInfo git.GitData) {
 
-	s := shutdown{
+	s := shutdownService{
 		sigCh: make(chan os.Signal, 1),
 	}
 
@@ -26,7 +26,7 @@ func StartServer(client *Client, projectInfo *ProjectInfo, GitInfo git.GitData)
 	r := CreateRouter(
 		client,
 		projectInfo,
-		s,
+		&s,
 		func(a *data) error { a.projectInfo = projectInfo; return nil },
 		func(a *data) error { a.gitInfo = &GitInfo; return nil },
 		func(a *data) error { err := attachEmojis(a, fr); return err },
@@ -39,10 +39,8 @@ func StartServer(client *Client, projectInfo *ProjectInfo, GitInfo git.GitData)
 	go func() {
 		err := server.Serve(l)
 		if err != nil {
-			if errors.Is(err, http.ErrServerClosed) {
-				os.Exit(0)
-			} else {
-				fmt.Fprintf(os.Stderr, "Server did not respond: %s\n", err)
+			if !errors.Is(err, http.ErrServerClosed) {
+				fmt.Fprintf(os.Stderr, "Server crashed: %s\n", err)
 				os.Exit(1)
 			}
 		}
@@ -76,7 +74,7 @@ type data struct {
 
 type optFunc func(a *data) error
 
-func CreateRouter(gitlabClient *Client, projectInfo *ProjectInfo, s ShutdownHandler, optFuncs ...optFunc) http.Handler {
+func CreateRouter(gitlabClient *Client, projectInfo *ProjectInfo, s *shutdownService, optFuncs ...optFunc) http.Handler {
 	m := http.NewServeMux()
 
 	d := data{
@@ -188,7 +186,6 @@ func CreateRouter(gitlabClient *Client, projectInfo *ProjectInfo, s ShutdownHand
 		withPayloadValidation(methodToPayload{http.MethodPost: &DraftNotePublishRequest{}}),
 		withMethodCheck(http.MethodPost),
 	))
-
 	m.HandleFunc("/pipeline", middleware(
 		pipelineService{d, gitlabClient, git.Git{}},
 		withMethodCheck(http.MethodGet),
@@ -230,8 +227,12 @@ func CreateRouter(gitlabClient *Client, projectInfo *ProjectInfo, s ShutdownHand
 		withPayloadValidation(methodToPayload{http.MethodPost: &MergeRequestByUsernameRequest{}}),
 		withMethodCheck(http.MethodPost),
 	))
+	m.HandleFunc("/shutdown", middleware(
+		*s,
+		withPayloadValidation(methodToPayload{http.MethodPost: &ShutdownRequest{}}),
+		withMethodCheck(http.MethodPost),
+	))
 
-	m.HandleFunc("/shutdown", s.shutdownHandler)
 	m.Handle("/ping", http.HandlerFunc(pingHandler))
 
 	return LoggingServer{handler: m}

cmd/app/shutdown.go

@@ -3,9 +3,7 @@ package app
 import (
 	"context"
 	"encoding/json"
-	"errors"
 	"fmt"
-	"io"
 	"net/http"
 	"os"
 )
@@ -19,22 +17,20 @@ func (k killer) String() string {
 
 type ShutdownHandler interface {
 	WatchForShutdown(server *http.Server)
-	shutdownHandler(w http.ResponseWriter, r *http.Request)
+	ServeHTTP(w http.ResponseWriter, r *http.Request)
 }
 
-type shutdown struct {
+type shutdownService struct {
 	sigCh chan os.Signal
 }
 
-func (s shutdown) WatchForShutdown(server *http.Server) {
+func (s shutdownService) WatchForShutdown(server *http.Server) {
 	/* Handles shutdown requests */
 	<-s.sigCh
 	err := server.Shutdown(context.Background())
 	if err != nil {
 		fmt.Fprintf(os.Stderr, "Server could not shut down gracefully: %s\n", err)
 		os.Exit(1)
-	} else {
-		os.Exit(0)
 	}
 }
 
@@ -42,36 +38,20 @@ type ShutdownRequest struct {
 	Restart bool `json:"restart"`
 }
 
-/* shutdownHandler will shutdown the HTTP server and exit the process by signaling to the shutdown channel */
-func (s shutdown) shutdownHandler(w http.ResponseWriter, r *http.Request) {
-	if r.Method != http.MethodPost {
-		w.Header().Set("Allow", http.MethodPost)
-		handleError(w, errors.New("Invalid request type"), "That request type is not allowed", http.StatusMethodNotAllowed)
-		return
-	}
+/* Shuts down the HTTP server and exit the process by signaling to the shutdown channel */
+func (s shutdownService) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 
-	body, err := io.ReadAll(r.Body)
-	if err != nil {
-		handleError(w, err, "Could not read request body", http.StatusBadRequest)
-		return
-	}
-
-	var shutdownRequest ShutdownRequest
-	err = json.Unmarshal(body, &shutdownRequest)
-	if err != nil {
-		handleError(w, err, "Could not unmarshal data from request body", http.StatusBadRequest)
-		return
-	}
+	payload := r.Context().Value(payload("payload")).(*ShutdownRequest)
 
 	var text = "Shut down server"
-	if shutdownRequest.Restart {
+	if payload.Restart {
 		text = "Restarted server"
 	}
 
 	w.WriteHeader(http.StatusOK)
 	response := SuccessResponse{Message: text}
 
-	err = json.NewEncoder(w).Encode(response)
+	err := json.NewEncoder(w).Encode(response)
 	if err != nil {
 		handleError(w, err, "Could not encode response", http.StatusInternalServerError)
 	} else {