Add log format flag

Author: cedws

pkg/cmd/cmd.go

@@ -2,13 +2,16 @@ package cmd
 
 import (
 	"context"
+	"log/slog"
+	"os"
 
 	"github.com/alecthomas/kong"
 	"github.com/cedws/sshgate/pkg/sshgate"
 )
 
 type cli struct {
 	ListenAddr string `help:"Address to listen on" default:":2222"`
+	LogFormat  string `help:"Log format"`
 	Config     string `help:"Path to JSON config file"`
 
 	Serve      serveCmd      `cmd:"" default:"1" help:"Start the server"`
@@ -18,7 +21,20 @@ type cli struct {
 type serveCmd struct{}
 
 func (s *serveCmd) Run(c *cli) error {
-	config, err := sshgate.Open(c.Config)
+	var handler slog.Handler
+
+	switch c.LogFormat {
+	case "json":
+		handler = slog.NewJSONHandler(os.Stderr, nil)
+	case "text":
+		fallthrough
+	default:
+		handler = slog.NewTextHandler(os.Stderr, nil)
+	}
+
+	slog.SetDefault(slog.New(handler))
+
+	config, err := sshgate.ReadConfig(c.Config)
 	if err != nil {
 		return err
 	}

pkg/sshgate/config.go

@@ -48,7 +48,7 @@ func (c Config) Validate() error {
 	return nil
 }
 
-func Open(path string) (*Config, error) {
+func ReadConfig(path string) (*Config, error) {
 	data, err := os.ReadFile(path)
 	if err != nil {
 		return nil, fmt.Errorf("failed to read config file: %w", err)