Rename addon to service

cmd/control/service.go

@@ -0,0 +1,100 @@
+package control
+
+import (
+	"fmt"
+	"log"
+
+	"github.com/codegangsta/cli"
+	"github.com/rancherio/os/config"
+	"github.com/rancherio/os/util"
+)
+
+func serviceSubCommands() []cli.Command {
+	return []cli.Command{
+		{
+			Name:   "enable",
+			Usage:  "turn on an service",
+			Action: enable,
+		},
+		{
+			Name:   "disable",
+			Usage:  "turn off an service",
+			Action: disable,
+		},
+		{
+			Name:   "list",
+			Usage:  "list services and state",
+			Action: list,
+		},
+	}
+}
+
+func disable(c *cli.Context) {
+	changed := false
+	cfg, err := config.LoadConfig()
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	for _, service := range c.Args() {
+		filtered := make([]string, 0, len(c.Args()))
+		for _, existing := range cfg.EnabledServices {
+			if existing != service {
+				filtered = append(filtered, existing)
+			}
+		}
+
+		if len(filtered) != len(c.Args()) {
+			cfg.EnabledServices = filtered
+			changed = true
+		}
+	}
+
+	if changed {
+		if err = cfg.Set("enabled_services", cfg.EnabledServices); err != nil {
+			log.Fatal(err)
+		}
+	}
+}
+
+func enable(c *cli.Context) {
+	changed := false
+	cfg, err := config.LoadConfig()
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	for _, service := range c.Args() {
+		if _, ok := cfg.Services[service]; ok && !util.Contains(cfg.EnabledServices, service) {
+			cfg.EnabledServices = append(cfg.EnabledServices, service)
+			changed = true
+		}
+	}
+
+	if changed {
+		if err = cfg.Set("enabled_services", cfg.EnabledServices); err != nil {
+			log.Fatal(err)
+		}
+	}
+}
+
+func list(c *cli.Context) {
+	cfg, err := config.LoadConfig()
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	enabled := map[string]bool{}
+
+	for _, service := range cfg.EnabledServices {
+		enabled[service] = true
+	}
+
+	for service, _ := range cfg.Services {
+		if _, ok := enabled[service]; ok {
+			fmt.Printf("%s enabled\n", service)
+		} else {
+			fmt.Printf("%s disabled\n", service)
+		}
+	}
+}