summaryrefslogtreecommitdiff
path: root/internal/cli/cli.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/cli/cli.go')
-rw-r--r--internal/cli/cli.go35
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
+}