feat: add flags for metrics and health endpoint

cmd/flags.go

@@ -12,4 +12,6 @@ const (
 	FlagGasLimit      = "gas-limit"
 	FlagNoTLS         = "no-tls"
 	FlagTLSSkipVerify = "tls-skip-verify"
+	FlagMetrics       = "metrics"
+	FlagHealth        = "health"
 )

cmd/root.go

@@ -3,6 +3,7 @@ package cmd
 import (
 	"errors"
 	"fmt"
+	"okp4/cosmos-faucet/internal/server"
 	"os"
 
 	"okp4/cosmos-faucet/pkg"
@@ -20,6 +21,7 @@ const (
 )
 
 var config pkg.Config
+var serverConfig server.Config
 
 // NewRootCommand returns the root CLI command with persistent flag handling.
 var rootCmd = &cobra.Command{
@@ -45,6 +47,8 @@ func Execute() {
 	rootCmd.PersistentFlags().StringVar(&config.Memo, FlagMemo, "Sent by økp4 faucet", "The memo description")
 	rootCmd.PersistentFlags().Uint64Var(&config.GasLimit, FlagGasLimit, 200000, "Gas limit")
 	rootCmd.PersistentFlags().BoolVar(&config.NoTLS, FlagNoTLS, false, "No encryption with the GRPC endpoint")
+	rootCmd.PersistentFlags().BoolVar(&serverConfig.EnableMetrics, FlagMetrics, false, "Enable metrics endpoint")
+	rootCmd.PersistentFlags().BoolVar(&serverConfig.EnableHealth, FlagHealth, false, "Enable health endpoint")
 	rootCmd.PersistentFlags().BoolVar(&config.TLSSkipVerify,
 		FlagTLSSkipVerify,
 		false,

cmd/start.go

@@ -30,7 +30,7 @@ func NewStartCommand() *cobra.Command {
 				log.Info().Msg("Server stopped")
 			}(faucet)
 
-			server.NewServer(faucet).Start(addr)
+			server.NewServer(serverConfig).Start(addr)
 		},
 	}
 

internal/server/router.go

@@ -2,20 +2,23 @@ package server
 
 import (
 	"okp4/cosmos-faucet/internal/server/handlers"
-	"okp4/cosmos-faucet/pkg/client"
 	"okp4/cosmos-faucet/pkg/server"
 )
 
-func (s *httpServer) createRoutes(faucet *client.Faucet) {
+func (s *httpServer) createRoutes(config Config) {
 	s.router.Use(handlers.PrometheusMiddleware)
 	s.router.Path("/").
 		Queries("address", "{address}").
-		HandlerFunc(server.NewSendRequestHandlerFn(faucet)).
-		Methods("GET")
-	s.router.Path("/health").
-		HandlerFunc(handlers.NewHealthRequestHandlerFunc()).
-		Methods("GET")
-	s.router.Path("/metrics").
-		Handler(handlers.NewMetricsRequestHandler()).
+		HandlerFunc(server.NewSendRequestHandlerFn(config.Faucet)).
 		Methods("GET")
+	if config.EnableHealth {
+		s.router.Path("/health").
+			HandlerFunc(handlers.NewHealthRequestHandlerFunc()).
+			Methods("GET")
+	}
+	if config.EnableMetrics {
+		s.router.Path("/metrics").
+			Handler(handlers.NewMetricsRequestHandler()).
+			Methods("GET")
+	}
 }

internal/server/server.go

@@ -8,6 +8,13 @@ import (
 	"github.com/rs/zerolog/log"
 )
 
+// Config holds config of the http server.
+type Config struct {
+	EnableMetrics bool `mapstructure:"metrics"`
+	EnableHealth  bool `mapstructure:"health"`
+	Faucet        *client.Faucet
+}
+
 // HTTPServer exposes server methods.
 type HTTPServer interface {
 	Start(string)
@@ -18,11 +25,11 @@ type httpServer struct {
 }
 
 // NewServer creates a new httpServer containing router.
-func NewServer(faucet *client.Faucet) HTTPServer {
+func NewServer(config Config) HTTPServer {
 	server := &httpServer{
 		router: mux.NewRouter().StrictSlash(true),
 	}
-	server.createRoutes(faucet)
+	server.createRoutes(config)
 	return server
 }