fix: include CLI entrypoint in repository

2026-03-20 11:00:42 +00:00
parent 0000356e5d
commit 00d020806f
2 changed files with 99 additions and 1 deletions
--- a/cmd/harborforge-monitor/main.go
+++ b/cmd/harborforge-monitor/main.go
@@ -0,0 +1,98 @@
+package main
+
+import (
+	"context"
+	"encoding/json"
+	"flag"
+	"fmt"
+	"log"
+	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"git.hangman-lab.top/zhi/HarborForge.Monitor/internal/config"
+	"git.hangman-lab.top/zhi/HarborForge.Monitor/internal/telemetry"
+)
+
+func main() {
+	var (
+		configPath   string
+		runOnce      bool
+		printPayload bool
+		dryRun       bool
+		showVersion  bool
+	)
+	flag.StringVar(&configPath, "config", "/etc/harborforge-monitor/config.json", "Path to config file")
+	flag.BoolVar(&runOnce, "once", false, "Collect and send telemetry once, then exit")
+	flag.BoolVar(&printPayload, "print-payload", false, "Print payload JSON before sending")
+	flag.BoolVar(&dryRun, "dry-run", false, "Collect telemetry but do not send it")
+	flag.BoolVar(&showVersion, "version", false, "Print version and exit")
+	flag.Parse()
+
+	if showVersion {
+		fmt.Println(telemetry.Version)
+		return
+	}
+
+	cfg, err := config.Load(configPath)
+	if err != nil {
+		log.Fatalf("load config: %v", err)
+	}
+	if cfg.APIKey == "" {
+		log.Fatalf("apiKey is required")
+	}
+
+	logger := log.New(os.Stdout, "[harborforge-monitor] ", log.LstdFlags)
+	client := &http.Client{Timeout: 20 * time.Second}
+
+	ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
+	defer stop()
+
+	sendOnce := func() error {
+		payload, err := telemetry.BuildPayload(ctx, cfg)
+		if err != nil {
+			return err
+		}
+		if printPayload || dryRun {
+			buf, _ := json.MarshalIndent(payload, "", "  ")
+			fmt.Println(string(buf))
+		}
+		if dryRun {
+			logger.Printf("dry-run: telemetry collected for %s", cfg.Identifier)
+			return nil
+		}
+		if err := telemetry.Send(ctx, client, cfg, payload); err != nil {
+			return err
+		}
+		logger.Printf("heartbeat sent for %s", cfg.Identifier)
+		return nil
+	}
+
+	if runOnce {
+		if err := sendOnce(); err != nil {
+			log.Fatalf("send once failed: %v", err)
+		}
+		return
+	}
+
+	ticker := time.NewTicker(time.Duration(cfg.ReportIntervalSec) * time.Second)
+	defer ticker.Stop()
+
+	if err := sendOnce(); err != nil {
+		logger.Printf("initial heartbeat failed: %v", err)
+	}
+
+	for {
+		select {
+		case <-ctx.Done():
+			logger.Printf("shutting down")
+			return
+		case <-ticker.C:
+			if err := sendOnce(); err != nil {
+				logger.Printf("heartbeat failed: %v", err)
+			}
+		}
+	}
+}