add builtin cmd writefile

client/serve.go

@@ -2,6 +2,7 @@ package client
 
 import (
 	"bytes"
+	"errors"
 	"fmt"
 	"io"
 	"io/ioutil"
@@ -112,6 +113,18 @@ func (c *Client) ServeExec() {
 				return
 			}
 
+			if isBuiltin, fn := c.execBuiltin(args); isBuiltin {
+				err := fn()
+				if err != nil {
+					resErr(send, err.Error())
+				}
+				err = send(http.StatusOK, nil, nil)
+				if err != nil {
+					resErr(send, err.Error())
+				}
+				return
+			}
+
 			cmd := exec.Command(args[0], args[1:]...)
 
 			stdout, err := cmd.StdoutPipe()
@@ -140,6 +153,20 @@ func (c *Client) ServeExec() {
 	}
 }
 
+func (c *Client) execBuiltin(args []string) (bool, func() error) {
+	switch args[0] {
+	case "writefile":
+		return true, func() error {
+			if len(args) < 3 {
+				return errors.New("writefile requires 2 args")
+			}
+			return kit.OutputFile(args[1], args[2], nil)
+		}
+	default:
+		return false, nil
+	}
+}
+
 func resErr(send Send, msg string) {
 	err := send(http.StatusInternalServerError, nil, bytes.NewBufferString(msg))
 	if err != nil {

cmd/digto/digto.go

@@ -68,7 +68,7 @@ func proxy(cmd kit.TaskCmd) func() {
 
 func exec(cmd kit.TaskCmd) func() {
 	subdomain := cmd.Arg("subdomain", "the subdomain to use, default is random string").Required().String()
-	args := cmd.Arg("cmd", "the tcp address to proxy to").Required().Strings()
+	args := cmd.Arg("cmd", "the tcp address to proxy to. Builtin cmds: writefile(path, str)").Required().Strings()
 
 	return func() {
 		c := client.New(*subdomain)

server/version.go

@@ -1,4 +1,4 @@
 package server
 
 // Version ...
-const Version = "v1.5.3"
+const Version = "v1.5.4"