chore: new -b/--benchmark flag

Author: mirkobrombin

internal/cli/cli.go

@@ -13,6 +13,7 @@ import (
 )
 
 var tuiMode bool
+var benchMode bool
 
 // rootCmd represents the base command when called without any subcommands.
 var rootCmd = &cobra.Command{
@@ -37,6 +38,7 @@ func init() {
 	rootCmd.AddCommand(pluginsCmd)
 
 	startCmd.Flags().BoolVarP(&tuiMode, "tui", "t", false, "Enable TUI mode")
+	startCmd.Flags().BoolVarP(&benchMode, "bench", "b", false, "Enable benchmark mode")
 }
 
 var generateCmd = &cobra.Command{
@@ -158,7 +160,7 @@ func start(cmd *cobra.Command, args []string) {
 	}
 
 	fmt.Println("Starting servers...")
-	server.StartServers(configs, tuiMode)
+	server.StartServers(configs, tuiMode, benchMode)
 
 	// Wait indefinitely if not in TUI mode, the servers will keep running
 	// and loggers will keep writing to both the stdout and the log files.

internal/server/middleware/middleware.go

@@ -1,6 +1,7 @@
 package middleware
 
 import (
+	"fmt"
 	"net/http"
 	"time"
 
@@ -56,6 +57,33 @@ func TimeoutMiddleware(timeout time.Duration) MiddlewareFunc {
 	}
 }
 
+// BenchmarkMiddleware logs the duration of HTTP requests.
+func BenchmarkMiddleware() MiddlewareFunc {
+	return func(next http.Handler) http.Handler {
+		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			start := time.Now()
+
+			next.ServeHTTP(w, r)
+
+			duration := time.Since(start)
+			formattedDuration := formatDuration(duration)
+			fmt.Printf("\033[33;40m⏲ Benchmark: %s %s completed in %s\033[0m\n", r.Method, r.URL.Path, formattedDuration)
+		})
+	}
+}
+
+// formatDuration formats a time.Duration to a human-readable string.
+func formatDuration(d time.Duration) string {
+	switch {
+	case d < time.Millisecond:
+		return fmt.Sprintf("%dµs", d.Microseconds())
+	case d < time.Second:
+		return fmt.Sprintf("%.3fms", float64(d.Microseconds())/1000)
+	default:
+		return fmt.Sprintf("%.3fs", d.Seconds())
+	}
+}
+
 // responseWriter wraps http.ResponseWriter to capture the status code.
 type responseWriter struct {
 	http.ResponseWriter

internal/server/server.go

@@ -22,7 +22,7 @@ var (
 )
 
 // StartServers starts the servers based on the provided configurations.
-func StartServers(configs []config.SiteConfig, enableTUI bool) {
+func StartServers(configs []config.SiteConfig, enableTUI bool, enableBench bool) {
 	tuiMode = enableTUI
 
 	// FIXME: move all TUI related code out of this package, I do not feel
@@ -69,6 +69,11 @@ func StartServers(configs []config.SiteConfig, enableTUI bool) {
 	// Initialize the global middleware manager
 	mwManager := middleware.NewMiddlewareManager()
 
+	// Enable benchmark middleware if requested
+	if enableBench {
+		mwManager.Use(middleware.BenchmarkMiddleware())
+	}
+
 	// Initialize the plugins with the global middleware manager
 	pluginManager := plugin.GetPluginManagerInstance()
 	if err := pluginManager.InitPlugins(mwManager); err != nil {