git.r.bdr.sh - rbdr/mobius/blob - cmd/mobius-hotline-client/main.go

   1 package main
   2
   3 import (
   4         "context"
   5         "flag"
   6         "fmt"
   7         "github.com/jhalter/mobius/hotline"
   8         "github.com/rivo/tview"
   9         "log/slog"
  10         "os"
  11         "os/signal"
  12         "runtime"
  13         "syscall"
  14 )
  15
  16 var logLevels = map[string]slog.Level{
  17         "debug": slog.LevelDebug,
  18         "info":  slog.LevelInfo,
  19         "warn":  slog.LevelWarn,
  20         "error": slog.LevelError,
  21 }
  22
  23 func main() {
  24         _, cancelRoot := context.WithCancel(context.Background())
  25
  26         sigChan := make(chan os.Signal, 1)
  27         signal.Notify(sigChan, syscall.SIGTERM, syscall.SIGINT, os.Interrupt)
  28
  29         configDir := flag.String("config", defaultConfigPath(), "Path to config root")
  30         version := flag.Bool("version", false, "print version and exit")
  31         logLevel := flag.String("log-level", "info", "Log level")
  32         logFile := flag.String("log-file", "", "output logs to file")
  33
  34         flag.Parse()
  35
  36         if *version {
  37                 fmt.Printf("v%s\n", hotline.VERSION)
  38                 os.Exit(0)
  39         }
  40
  41         // init DebugBuffer
  42         db := &hotline.DebugBuffer{
  43                 TextView: tview.NewTextView(),
  44         }
  45
  46         // Add file logger if optional log-file flag was passed
  47         if *logFile != "" {
  48                 f, err := os.OpenFile(*logFile,
  49                         os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
  50                 defer func() { _ = f.Close() }()
  51                 if err != nil {
  52                         panic(err)
  53                 }
  54         }
  55
  56         logger := slog.New(slog.NewTextHandler(db, &slog.HandlerOptions{Level: logLevels[*logLevel]}))
  57         logger.Info("Started Mobius client", "Version", hotline.VERSION)
  58
  59         go func() {
  60                 sig := <-sigChan
  61                 logger.Info("Stopping client", "signal", sig.String())
  62                 cancelRoot()
  63         }()
  64
  65         client := hotline.NewUIClient(*configDir, logger)
  66         client.DebugBuf = db
  67         client.UI.Start()
  68 }
  69
  70 func defaultConfigPath() (cfgPath string) {
  71         switch runtime.GOOS {
  72         case "windows":
  73                 cfgPath = "mobius-client-config.yaml"
  74         case "darwin":
  75                 if _, err := os.Stat("/usr/local/etc/mobius-client-config.yaml"); err == nil {
  76                         cfgPath = "/usr/local/etc/mobius-client-config.yaml"
  77                 } else if _, err := os.Stat("/opt/homebrew/etc/mobius-client-config.yaml"); err == nil {
  78                         cfgPath = "/opt/homebrew/etc/mobius-client-config.yaml"
  79                 }
  80         case "linux":
  81                 cfgPath = "/usr/local/etc/mobius-client-config.yaml"
  82         default:
  83                 fmt.Printf("unsupported OS")
  84         }
  85
  86         return cfgPath
  87 }