diff options
Diffstat (limited to 'internal/cli/cli.go')
| -rw-r--r-- | internal/cli/cli.go | 35 |
1 files changed, 35 insertions, 0 deletions
diff --git a/internal/cli/cli.go b/internal/cli/cli.go index e9e1893..0752806 100644 --- a/internal/cli/cli.go +++ b/internal/cli/cli.go @@ -2,10 +2,14 @@ package cli import ( "context" + "errors" "flag" "fmt" "os" + "os/signal" + "syscall" + "codeberg.org/snonux/goprecords/internal/daemon" "codeberg.org/snonux/goprecords/internal/goprecords" "codeberg.org/snonux/goprecords/internal/version" ) @@ -17,6 +21,9 @@ func Execute(args []string) error { fmt.Println(version.Version) return nil } + if len(args) > 0 && (args[0] == "-daemon" || args[0] == "--daemon") { + return runDaemon(args[1:]) + } // No subcommand – treat args as flags for a direct report from files. if len(args) == 0 { @@ -118,3 +125,31 @@ func runReportFromFiles(args []string) error { func runTests() error { return goprecords.RunIntegrationTests("./fixtures") } + +func defaultListenFromEnv() string { + if s := os.Getenv("GOPRECORDS_LISTEN"); s != "" { + return s + } + return ":8080" +} + +func runDaemon(args []string) error { + fs := flag.NewFlagSet("daemon", flag.ExitOnError) + statsDir := fs.String("stats-dir", os.Getenv("GOPRECORDS_STATS_DIR"), "Uptimed stats directory (required; env GOPRECORDS_STATS_DIR)") + listen := fs.String("listen", defaultListenFromEnv(), "TCP listen address (env GOPRECORDS_LISTEN, default :8080)") + if err := fs.Parse(args); err != nil { + return err + } + if *statsDir == "" { + fmt.Fprintln(os.Stderr, "daemon: missing required flag: -stats-dir (or GOPRECORDS_STATS_DIR)") + fs.Usage() + return fmt.Errorf("missing -stats-dir") + } + ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM) + defer stop() + err := daemon.Run(ctx, daemon.Config{StatsDir: *statsDir, Addr: *listen}) + if err != nil && !errors.Is(err, context.Canceled) { + return err + } + return nil +} |
