Add credentials

mprimi · Jun 29, 2022 · 9218307 · 9218307
1 parent 8744fc4
commit 9218307
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 3 deletions.
diff --git a/commands/natsclient.go b/commands/natsclient.go
@@ -14,10 +14,27 @@ var singleton struct {
 }
 
 func getJSContext() (nats.JetStreamContext, error)  {
-
   if singleton.nc == nil {
     log.debug("Connecting to %s", options.natsURL)
-    nc, err := nats.Connect(options.natsURL)
+
+    natsOpts := []nats.Option{
+      nats.Name("Nasefa"),
+      nats.DisconnectHandler(func(nc *nats.Conn) {
+        log.debug("Client disconnected")
+      }),
+      nats.ReconnectHandler(func(nc *nats.Conn) {
+        log.debug("Client reconnected")
+      }),
+      nats.ClosedHandler(func(nc *nats.Conn) {
+        log.debug("Client closed")
+      }),
+    }
+
+    if options.credentials != "" {
+      natsOpts = append(natsOpts, nats.UserCredentials(options.credentials))
+    }
+
+    nc, err := nats.Connect(options.natsURL, natsOpts...)
     if err != nil {
       return nil, err
     }

diff --git a/commands/options.go b/commands/options.go
@@ -14,13 +14,15 @@ const (
 
 var options struct {
   natsURL             string
+  credentials         string
   debug               bool
   quiet               bool
   noEmojis            bool
 }
 
 func RegisterTopLevelFlags() {
-  flag.StringVar(&options.natsURL, "natsURL", defaultNatsURL, "NATS server URL ")
+  flag.StringVar(&options.natsURL, "natsURL", defaultNatsURL, "NATS server URL (may include username, password, token)")
+  flag.StringVar(&options.credentials, "creds", "", "Path to credentials file")
   flag.BoolVar(&options.debug, "debug", defaultDebug, "Print debug statements")
   flag.BoolVar(&options.quiet, "quiet", defaultQuiet, "Quiet, only print fatal errors")
   flag.BoolVar(&options.noEmojis, "noEmoji", defaultNoEmojis, "Disable emojis in console messages")