wip, run-role

Author: tobscher

.kiss.yml

@@ -19,6 +19,28 @@ hosts:
         shell:
           command: echo "Foo"
 
+# Role specific tasks
+roles:
+  common:
+    tasks:
+      - task: Common task
+        shell:
+          command: whoami
+  docker:
+    with: [common]
+    tasks:
+      - task: install docker
+        description: Install docker
+        apt-get:
+          update: yes
+          packages: [docker.io]
+  db:
+    with: [common]
+    tasks:
+      - task: Install MySQL
+        apt-get:
+          update: yes
+          packages: [mysql-server]
 
 # Global tasks
 tasks:

commands/commands.go

@@ -2,4 +2,8 @@ package commands
 
 import "github.com/tobscher/kiss/logging"
 
-var logger = logging.GetLogger("kiss")
+var (
+	logger     = logging.GetLogger("kiss")
+	configFile string
+	verbose    bool
+)

commands/run-role.go

@@ -0,0 +1,56 @@
+package commands
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/spf13/cobra"
+	"github.com/tobscher/kiss/configuration"
+	"github.com/tobscher/kiss/core"
+	"github.com/tobscher/kiss/logging"
+)
+
+// NewRunRoleCommand creates a new command to run a task.
+func NewRunRoleCommand() *cobra.Command {
+	command := &cobra.Command{
+		Use:   "run-role",
+		Short: "Run a role",
+		Long:  "Run a role on all remote systems with that role.",
+		Run:   runRoleRun,
+	}
+	command.Flags().StringVar(&configFile, "config", ".kiss.yml", "path to config file")
+	command.Flags().BoolVarP(&verbose, "verbose", "v", false, "verbose output")
+
+	return command
+}
+
+func runRoleRun(cmd *cobra.Command, args []string) {
+	if verbose {
+		logger.SetLevel(logging.DEBUG)
+		core.SetLogLevel(logging.DEBUG)
+	}
+
+	config := configuration.Load(configFile)
+
+	if len(args) < 1 {
+		fmt.Printf("Error: Expected role-name: kiss run-role <role-name>\n")
+		os.Exit(1)
+	}
+
+	// Try to find task in global list
+	roleName := args[0]
+	role, ok := config.Roles[roleName]
+	if !ok {
+		fmt.Printf("Error: Role not found: %v\n", roleName)
+		os.Exit(1)
+	}
+
+	logger.Infof("Running role `%v`", roleName)
+	for _, host := range config.Hosts.WithRole(roleName) {
+		logger.Infof("Selecting host `%v`", host.Host)
+
+		for _, task := range role.Tasks {
+			runWithRunner(&task, &host)
+		}
+	}
+}

commands/run.go

@@ -11,11 +11,6 @@ import (
 	"github.com/tobscher/kiss/logging"
 )
 
-var (
-	configFile string
-	verbose    bool
-)
-
 // NewRunCommand creates a new command to run a task.
 func NewRunCommand() *cobra.Command {
 	command := &cobra.Command{

configuration/config.go

@@ -15,6 +15,7 @@ type Configuration struct {
 	Vars  map[string]string
 	Hosts HostCollection
 	Tasks TaskCollection
+	Roles RoleCollection
 }
 
 // Load loads configuration from the given path.

configuration/config_test.go

@@ -119,3 +119,20 @@ func TestTasks(t *testing.T) {
 		t.Error("Expected command to be go version.")
 	}
 }
+
+func TestRoles(t *testing.T) {
+	config := Load("./fixtures/example.yml")
+
+	if len(config.Roles) != 4 {
+		t.Errorf("Expected 3 roles got %v.", len(config.Roles))
+	}
+
+	docker := config.Roles["docker"]
+	if !arrayIncludes(docker.With, "common") {
+		t.Error("Expected role docker to be based of common.")
+	}
+
+	if len(docker.Tasks) != 2 {
+		t.Errorf("Expected role docker to have 2 tasks got %v.", len(docker.Tasks))
+	}
+}

configuration/fixtures/example.yml

@@ -29,6 +29,39 @@ hosts:
       - docker
       - db
 
+# Role specific tasks
+roles:
+  common:
+    tasks:
+      - task: Common task
+        shell:
+          command: whoami
+  docker:
+    with: [common]
+    tasks:
+      - task: install_docker
+        description: Install docker
+        apt-get:
+          update: yes
+          packages: [docker.io]
+      - task: run_docker
+        description: run docker daemon
+        shell:
+          command: docker -d &
+  db:
+    with: [common]
+    tasks:
+      - task: Install MySQL
+        apt-get:
+          update: yes
+          packages: [mysql-server]
+  web:
+    with: [common]
+    tasks:
+      - task: Install ruby
+        shell:
+          command: echo "Install ruby..."
+
 # Global tasks
 tasks:
   - task: Environment

configuration/helpers_test.go

@@ -0,0 +1,11 @@
+package configuration
+
+func arrayIncludes(haystack []string, needle string) bool {
+	for _, v := range haystack {
+		if v == needle {
+			return true
+		}
+	}
+
+	return false
+}

configuration/hosts.go

@@ -29,3 +29,18 @@ func (h HostCollection) Get(name string) *Host {
 
 	return nil
 }
+
+// WithRole returns all hosts with the given role name.
+func (h HostCollection) WithRole(roleName string) HostCollection {
+	var hosts HostCollection
+
+	for _, host := range h {
+		for _, role := range host.Roles {
+			if role == roleName {
+				hosts = append(hosts, host)
+			}
+		}
+	}
+
+	return hosts
+}

configuration/roles.go

@@ -0,0 +1,8 @@
+package configuration
+
+type RoleCollection map[string]Role
+
+type Role struct {
+	With  []string
+	Tasks TaskCollection
+}

main.go

@@ -6,6 +6,7 @@ func main() {
 	rootCmd := commands.NewRootCommand()
 	rootCmd.AddCommand(commands.NewVersionCommand(name, version))
 	rootCmd.AddCommand(commands.NewRunCommand())
+	rootCmd.AddCommand(commands.NewRunRoleCommand())
 	rootCmd.AddCommand(commands.NewTasksCommand())
 	rootCmd.AddCommand(commands.NewHostsCommand())
 

plugins/core/plugin.go

@@ -41,7 +41,7 @@ func Serve() {
 	for {
 		conn, err := l.Accept()
 		if err != nil {
-			log.Fatal(err)
+			logger.Fatal(err.Error())
 		}
 
 		go server.ServeCodec(jsonrpc.NewServerCodec(conn))