From 22e4fc56cca0a8c466bf09a6f529573063ce5cb6 Mon Sep 17 00:00:00 2001
From: Bryan McNulty <bryanmcnulty@protonmail.com>
Date: Thu, 17 Apr 2025 02:26:28 -0500
Subject: Validate output flag before method execution

---
 cmd/args.go | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

(limited to 'cmd/args.go')

diff --git a/cmd/args.go b/cmd/args.go
index 50e7c74..b2b6fe2 100644
--- a/cmd/args.go
+++ b/cmd/args.go
@@ -6,6 +6,7 @@ import (
   "fmt"
   "github.com/spf13/cobra"
   "github.com/spf13/pflag"
+  "os"
 )
 
 func registerRpcFlags(cmd *cobra.Command) {
@@ -122,5 +123,20 @@ func argsOutput(methods ...string) func(cmd *cobra.Command, args []string) error
       as = append(as, argsSmbClient())
     }
   }
-  return args(as...)
+
+  return args(append(as, func(*cobra.Command, []string) (err error) {
+
+    if outputPath != "" {
+      if outputPath == "-" {
+        exec.Output.Writer = os.Stdout
+
+      } else if outputPath != "" {
+
+        if exec.Output.Writer, err = os.OpenFile(outputPath, os.O_WRONLY|os.O_CREATE, 0644); err != nil {
+          log.Fatal().Err(err).Msg("Failed to open output file")
+        }
+      }
+    }
+    return
+  })...)
 }
-- 
cgit v1.2.3