feat(cli): add setup command

Author: saitho

ansible/inventory.go

@@ -0,0 +1,60 @@
+package ansible
+
+import (
+	"io/ioutil"
+	"os"
+	"path/filepath"
+	"text/template"
+)
+
+type InventoryOptions struct {
+	IpAddress string
+}
+
+func IpAddress(text string) InventoryOption {
+	return func(args *InventoryOptions) {
+		args.IpAddress = text
+	}
+}
+
+type InventoryOption func(*InventoryOptions)
+
+func CreateInventoryFile(options ...InventoryOption) (string, error) {
+	opts := &InventoryOptions{}
+	for _, setter := range options {
+		setter(opts)
+	}
+
+	tmpFile, err := ioutil.TempFile("", "inventory")
+	if err != nil {
+		return "", err
+	}
+
+	filePath, err := filepath.Abs(tmpFile.Name())
+	if err != nil {
+		os.Remove(tmpFile.Name())
+		return "", err
+	}
+
+	t, err := template.New("ansible_inventory").Parse(`# This file was generated by StackHead.
+---
+all:
+  vars:
+    ansible_user: root
+    ansible_connection: ssh
+  hosts:
+    mackerel:
+      ansible_host: {{ .IpAddress }}
+`)
+
+	if err = t.Execute(tmpFile, opts); err != nil {
+		return "", err
+	}
+
+	// Close the file
+	if err := tmpFile.Close(); err != nil {
+		return "", err
+	}
+
+	return filePath, nil
+}

commands/setup.go

@@ -0,0 +1,44 @@
+package commands
+
+import (
+	"fmt"
+	"os"
+	"sync"
+
+	"github.com/spf13/cobra"
+
+	"github.com/getstackhead/stackhead/cli/ansible"
+	"github.com/getstackhead/stackhead/cli/routines"
+)
+
+var SetupServer = &cobra.Command{
+	Use:     "setup [ipv4 address]",
+	Example: "setup 192.168.178.14",
+	Short:   "Prepare a server for deployment",
+	Long:    `setup will install all required software on a server. You are then able to deploy projects onto it.`,
+	Args:    cobra.ExactArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		routines.RunTask(
+			routines.Text(fmt.Sprintf("Deploying to server at IP \"%s\"", args[0])),
+			routines.Execute(func(wg *sync.WaitGroup, result chan routines.TaskResult) {
+				defer wg.Done()
+
+				// Generate Inventory file
+				inventoryFile, err := ansible.CreateInventoryFile(ansible.IpAddress(args[0]))
+				if err == nil {
+					defer os.Remove(inventoryFile)
+					err = routines.ExecAnsiblePlaybook("server-provision", inventoryFile)
+				}
+
+				taskResult := routines.TaskResult{
+					Error: err != nil,
+				}
+				if err != nil {
+					taskResult.Message = err.Error()
+				}
+
+				result <- taskResult
+			}),
+		)
+	},
+}

commands/validate.go

@@ -10,10 +10,11 @@ import (
 )
 
 var Validate = &cobra.Command{
-	Use:   "validate [path to project definition file]",
-	Short: "Validate a project definition file",
-	Long:  `validate is used to make sure your project definition file meets the StackHead project definition syntax.`,
-	Args:  cobra.ExactArgs(1),
+	Use:     "validate [path to project definition file]",
+	Example: "validate ./my-project-definition.yml",
+	Short:   "Validate a project definition file",
+	Long:    `validate is used to make sure your project definition file meets the StackHead project definition syntax.`,
+	Args:    cobra.ExactArgs(1),
 	Run: func(cmd *cobra.Command, args []string) {
 		result, err := jsonschema.ValidateFile(args[0])
 

main.go

@@ -44,6 +44,7 @@ func init() {
 
 	rootCmd.AddCommand(commands.Validate)
 	rootCmd.AddCommand(commands.Init)
+	rootCmd.AddCommand(commands.SetupServer)
 }
 
 // initConfig reads in config file and ENV variables if set.

routines/exec.go

@@ -36,3 +36,11 @@ func ExecAnsibleGalaxy(args ...string) error {
 	args = append(args, "-p "+collectionDir[0])
 	return Exec("ansible-galaxy", args...)
 }
+
+func ExecAnsiblePlaybook(playbookName string, inventoryPath string) error {
+	stackHeadLocation, err := ansible.GetStackHeadCollectionLocation()
+	if err != nil {
+		return err
+	}
+	return Exec("ansible-playbook", stackHeadLocation+"/playbooks/"+playbookName+".yml", "-i", inventoryPath)
+}