feat(cli): add deploy command

Author: saitho

ansible/inventory.go

@@ -4,12 +4,16 @@ import (
 	"io/ioutil"
 	"os"
 	"path/filepath"
+	"strings"
 	"text/template"
 )
 
 // InventoryOptions is a configuration used during creation of the temporary inventory file
 type InventoryOptions struct {
-	IPAddress string
+	IPAddress                  string
+	ProjectDefinitionFile      string
+	ProjectDefinitionFileName  string
+	TmpProjectDefinitionFolder string
 }
 
 // IPAddress sets the corresponding inventory option
@@ -19,9 +23,28 @@ func IPAddress(ipAddress string) InventoryOption {
 	}
 }
 
+// ProjectDefinitionFile sets the corresponding inventory option
+func ProjectDefinitionFile(projectDefinitionFile string) InventoryOption {
+	return func(args *InventoryOptions) {
+		args.ProjectDefinitionFile = projectDefinitionFile
+	}
+}
+
 // InventoryOption is a single inventory setting
 type InventoryOption func(*InventoryOptions)
 
+func copyFile(srcPath string, destPath string) error {
+	input, err := ioutil.ReadFile(srcPath)
+	if err != nil {
+		return err
+	}
+
+	if err = ioutil.WriteFile(destPath, input, 0644); err != nil {
+		return err
+	}
+	return nil
+}
+
 // CreateInventoryFile creates a temporary inventory file with the given settings and returns an absolute file path.
 // Note: make sure to remove the file when you don't need it anymore!
 func CreateInventoryFile(options ...InventoryOption) (string, error) {
@@ -41,15 +64,34 @@ func CreateInventoryFile(options ...InventoryOption) (string, error) {
 		return "", err
 	}
 
+	if len(opts.ProjectDefinitionFile) > 0 {
+		opts.TmpProjectDefinitionFolder, err = ioutil.TempDir("", "project_definitions")
+		if err != nil {
+			return "", err
+		}
+		projectDefinitionFile := filepath.Base(opts.ProjectDefinitionFile)
+
+		// Copy project definition file
+		err = copyFile(opts.ProjectDefinitionFile, filepath.Join(opts.TmpProjectDefinitionFolder, projectDefinitionFile))
+
+		opts.ProjectDefinitionFileName = projectDefinitionFile
+		opts.ProjectDefinitionFileName = strings.TrimSuffix(opts.ProjectDefinitionFileName, ".yml")
+		opts.ProjectDefinitionFileName = strings.TrimSuffix(opts.ProjectDefinitionFileName, ".yaml")
+	}
+
 	t, err := template.New("ansible_inventory").Parse(`# This file was generated by StackHead.
 ---
 all:
   vars:
     ansible_user: root
     ansible_connection: ssh
+    stackhead__config_folder: "{{ .TmpProjectDefinitionFolder }}"
   hosts:
     mackerel:
       ansible_host: {{ .IPAddress }}
+      stackhead:
+        applications:
+          - {{ .ProjectDefinitionFileName }}
 `)
 
 	if err = t.Execute(tmpFile, opts); err != nil {

commands/deploy.go

@@ -0,0 +1,48 @@
+package commands
+
+import (
+	"fmt"
+	"os"
+	"sync"
+
+	"github.com/spf13/cobra"
+
+	"github.com/getstackhead/stackhead/cli/ansible"
+	"github.com/getstackhead/stackhead/cli/routines"
+)
+
+// DeployApplication is a command object for Cobra that provides the deploy command
+var DeployApplication = &cobra.Command{
+	Use:     "deploy [path to project definition] [ipv4 address]",
+	Example: "deploy ./my_project_definition.yml 192.168.178.14",
+	Short:   "Deploy a project onto the target server",
+	Long:    `deploy will deploy the given project onto the server`,
+	Args:    cobra.ExactArgs(2),
+	Run: func(cmd *cobra.Command, args []string) {
+		routines.RunTask(
+			routines.Text(fmt.Sprintf("Deploying project \"%s\" onto server with IP \"%s\"", args[0], args[1])),
+			routines.Execute(func(wg *sync.WaitGroup, result chan routines.TaskResult) {
+				defer wg.Done()
+
+				// Generate Inventory file
+				inventoryFile, err := ansible.CreateInventoryFile(
+					ansible.IPAddress(args[1]),
+					ansible.ProjectDefinitionFile(args[0]),
+				)
+				if err == nil {
+					defer os.Remove(inventoryFile)
+					err = routines.ExecAnsiblePlaybook("application-deploy", inventoryFile)
+				}
+
+				taskResult := routines.TaskResult{
+					Error: err != nil,
+				}
+				if err != nil {
+					taskResult.Message = err.Error()
+				}
+
+				result <- taskResult
+			}),
+		)
+	},
+}

commands/setup.go

@@ -20,7 +20,7 @@ var SetupServer = &cobra.Command{
 	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.Text(fmt.Sprintf("Setting up server at IP \"%s\"", args[0])),
 			routines.Execute(func(wg *sync.WaitGroup, result chan routines.TaskResult) {
 				defer wg.Done()
 

main.go

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

stackhead/modules.go

@@ -21,7 +21,7 @@ func SplitModuleName(moduleName string) (string, string, string) {
 func GetModuleBaseName(moduleName string) string {
 	moduleName = RemoveVendor(moduleName)
 	moduleType := GetModuleType(moduleName)
-	return strings.TrimPrefix(moduleName, moduleType + "_")
+	return strings.TrimPrefix(moduleName, moduleType+"_")
 }
 
 // ExtractVendor returns the vendor name of a given module name (e.g. getstackhead in getstackhead.stackhead_webserver_nginx)