diff options
Diffstat (limited to 'cmd')
| -rw-r--r-- | cmd/drun/main.go | 16 |
1 files changed, 15 insertions, 1 deletions
diff --git a/cmd/drun/main.go b/cmd/drun/main.go index b1936d4..1b88b1b 100644 --- a/cmd/drun/main.go +++ b/cmd/drun/main.go @@ -3,6 +3,7 @@ package main import ( "context" "flag" + "io/ioutil" "os" "github.com/mimecast/dtail/internal/clients" @@ -67,7 +68,7 @@ func main() { ServersStr: serversStr, Discovery: discovery, UserName: userName, - What: command, + What: readCommand(command), TrustAllHosts: trustAllHosts, } @@ -80,3 +81,16 @@ func main() { logger.Flush() os.Exit(status) } + +func readCommand(command string) string { + if _, err := os.Stat(command); os.IsNotExist(err) { + return command + } + + bytes, err := ioutil.ReadFile(command) + if err != nil { + panic(err) + } + + return string(bytes) +} |
