From 6bca637513e065a33cadaccad97ada25eb7a6b00 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Paul=20B=C3=BCtow?= <pbuetow@mimecast.com>
Date: Sun, 16 Feb 2020 13:58:26 +0000
Subject: can upload script via drun

---
 cmd/drun/main.go | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

(limited to 'cmd')

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)
+}
-- 
cgit v1.2.3