Switched to using cobra for better cli support

This commit is contained in:
2026-05-31 17:04:39 -05:00
parent 2e00798d2d
commit 9b665d2df6
5 changed files with 49 additions and 16 deletions
+1 -1
View File
@@ -15,7 +15,7 @@ help:
@echo " make clean - Remove built artifacts"
run:
go run $(CMD_DIR) -config $(CONFIG)
go run $(CMD_DIR) --config $(CONFIG)
build:
mkdir -p $(BIN_DIR)
+30 -8
View File
@@ -2,7 +2,7 @@ package main
import (
"context"
"flag"
"fmt"
"log"
"net/http"
"os"
@@ -10,6 +10,8 @@ import (
"syscall"
"time"
"github.com/spf13/cobra"
"scratchbox/internal/cleanup"
"scratchbox/internal/config"
httpapi "scratchbox/internal/http"
@@ -17,24 +19,43 @@ import (
)
func main() {
configPath := flag.String("config", "", "path to config file")
flag.Parse()
cmd := newRootCmd()
if err := cmd.Execute(); err != nil {
log.Fatalf("command failed: %v", err)
}
}
cfg, err := config.Load(*configPath)
func newRootCmd() *cobra.Command {
var configPath string
cmd := &cobra.Command{
Use: "pastebin",
Short: "Run the scratchbox service",
SilenceUsage: true,
RunE: func(_ *cobra.Command, _ []string) error {
return run(configPath)
},
}
cmd.Flags().StringVarP(&configPath, "config", "c", "", "path to config file")
return cmd
}
func run(configPath string) error {
cfg, err := config.Load(configPath)
if err != nil {
log.Fatalf("config load failed: %v", err)
return fmt.Errorf("config load failed: %w", err)
}
logger := log.New(os.Stdout, "[pastebin] ", log.LstdFlags|log.LUTC)
store, err := storage.NewFilesystemStore(cfg.Storage.DataDir)
if err != nil {
logger.Fatalf("storage init failed: %v", err)
return fmt.Errorf("storage init failed: %w", err)
}
server, err := httpapi.NewServer(cfg, store, logger)
if err != nil {
logger.Fatalf("http init failed: %v", err)
return fmt.Errorf("http init failed: %w", err)
}
cleanupWorker := cleanup.NewWorker(store, cfg.Storage.CleanupIntervalParsed, logger)
@@ -60,6 +81,7 @@ func main() {
logger.Printf("listening on %s", cfg.Server.ListenAddr)
if err := httpServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
logger.Fatalf("http server exited with error: %v", err)
return fmt.Errorf("http server exited with error: %w", err)
}
return nil
}
+4 -7
View File
@@ -4,7 +4,6 @@ import (
"bytes"
"context"
"encoding/json"
"flag"
"io"
"log"
"net"
@@ -167,8 +166,6 @@ func TestMainHelperProcess(t *testing.T) {
if os.Getenv("GO_WANT_MAIN_HELPER") != "1" {
return
}
flag.CommandLine = flag.NewFlagSet(os.Args[0], flag.ExitOnError)
mode := os.Getenv("MAIN_HELPER_MODE")
switch mode {
case "success":
@@ -183,16 +180,16 @@ func TestMainHelperProcess(t *testing.T) {
_ = proc.Signal(os.Interrupt)
}
}()
os.Args = []string{"pastebin", "-config", cfgPath}
os.Args = []string{"pastebin", "--config", cfgPath}
main()
os.Exit(0)
case "bad-config":
os.Args = []string{"pastebin", "-config", "/no/such/config.yaml"}
os.Args = []string{"pastebin", "--config", "/no/such/config.yaml"}
main()
os.Exit(0)
case "storage-fail":
cfgPath := os.Getenv("MAIN_HELPER_CONFIG")
os.Args = []string{"pastebin", "-config", cfgPath}
os.Args = []string{"pastebin", "--config", cfgPath}
main()
os.Exit(0)
case "http-fail":
@@ -200,7 +197,7 @@ func TestMainHelperProcess(t *testing.T) {
if err := os.Chdir(t.TempDir()); err != nil {
os.Exit(2)
}
os.Args = []string{"pastebin", "-config", cfgPath}
os.Args = []string{"pastebin", "--config", cfgPath}
main()
os.Exit(0)
default:
+6
View File
@@ -3,6 +3,12 @@ module scratchbox
go 1.25.4
require (
github.com/spf13/cobra v1.10.1
golang.org/x/time v0.15.0
gopkg.in/yaml.v3 v3.0.1
)
require (
github.com/inconshreveable/mousetrap v1.1.0 // indirect
github.com/spf13/pflag v1.0.9 // indirect
)
+8
View File
@@ -1,3 +1,11 @@
github.com/cpuguy83/go-md2man/v2 v2.0.6/go.mod h1:oOW0eioCTA6cOiMLiUPZOpcVxMig6NIQQ7OS05n1F4g=
github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
github.com/spf13/cobra v1.10.1 h1:lJeBwCfmrnXthfAupyUTzJ/J4Nc1RsHC/mSRU2dll/s=
github.com/spf13/cobra v1.10.1/go.mod h1:7SmJGaTHFVBY0jW4NXGluQoLvhqFQM+6XSKD+P4XaB0=
github.com/spf13/pflag v1.0.9 h1:9exaQaMOCwffKiiiYk6/BndUBv+iRViNW+4lEMi0PvY=
github.com/spf13/pflag v1.0.9/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
golang.org/x/time v0.15.0 h1:bbrp8t3bGUeFOx08pvsMYRTCVSMk89u4tKbNOZbp88U=
golang.org/x/time v0.15.0/go.mod h1:Y4YMaQmXwGQZoFaVFk4YpCt4FLQMYKZe9oeV/f4MSno=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=