can upload script via drun

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)
+}