Add --user-data-from-file flag to server create command (#88)

Author: thetechnick

CHANGES.md

@@ -4,6 +4,7 @@
 
 * Print dates in local time
 * Do not echo token when creating a context
+* Add `--user-data-from-file` flag to `hcloud server create` command
 
 ## v1.2.0
 

Gopkg.lock

@@ -4,7 +4,7 @@
 
 [[constraint]]
   name = "github.com/hetznercloud/hcloud-go"
-  version = "1.2.0"
+  version = "1.3.0"
 
 [[constraint]]
   name = "github.com/pelletier/go-toml"

Gopkg.toml

@@ -2,6 +2,8 @@ package cli
 
 import (
 	"fmt"
+	"io/ioutil"
+	"os"
 
 	"github.com/hetznercloud/hcloud-go/hcloud"
 	"github.com/spf13/cobra"
@@ -47,6 +49,9 @@ func newServerCreateCommand(cli *CLI) *cobra.Command {
 	cmd.Flag("ssh-key").Annotations = map[string][]string{
 		cobra.BashCompCustom: {"__hcloud_sshkey_names"},
 	}
+
+	cmd.Flags().String("user-data-from-file", "", "Read user data from specified file (use - to read from stdin)")
+
 	return cmd
 }
 
@@ -80,6 +85,7 @@ func optsFromFlags(cli *CLI, flags *pflag.FlagSet) (opts hcloud.ServerCreateOpts
 	image, _ := flags.GetString("image")
 	location, _ := flags.GetString("location")
 	datacenter, _ := flags.GetString("datacenter")
+	userDataFile, _ := flags.GetString("user-data-from-file")
 	sshKeys, _ := flags.GetStringSlice("ssh-key")
 
 	opts = hcloud.ServerCreateOpts{
@@ -91,6 +97,20 @@ func optsFromFlags(cli *CLI, flags *pflag.FlagSet) (opts hcloud.ServerCreateOpts
 			Name: image,
 		},
 	}
+
+	if userDataFile != "" {
+		var data []byte
+		if userDataFile == "-" {
+			data, err = ioutil.ReadAll(os.Stdin)
+		} else {
+			data, err = ioutil.ReadFile(userDataFile)
+		}
+		if err != nil {
+			return
+		}
+		opts.UserData = string(data)
+	}
+
 	for _, sshKeyIDOrName := range sshKeys {
 		var sshKey *hcloud.SSHKey
 		sshKey, _, err = cli.Client().SSHKey.Get(cli.Context, sshKeyIDOrName)