brewservice/main.go at master · jraede/brewservice · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
package main

import (
	"fmt"
	"github.com/codegangsta/cli"
	"os"
)

func fatal(args ...interface{}) {
	fmt.Println(args...)
	os.Exit(1)
}

func main() {
	dirname := "/usr/local/Cellar/"

	app := cli.NewApp()

	app.Name = "brewservice"
	app.Usage = "Easily start and stop services installed via homebrew"
	app.Version = "0.1.0"

	app.Commands = []cli.Command{
		{
			Name:    "update",
			Aliases: []string{"u"},
			Usage:   "Parse homebrew folders for plist files and cache them for later",
			Action: func(c *cli.Context) {
				cache, err := update(dirname)
				if err != nil {
					fatal(err)
				}

				err = cache.Save()
				if err != nil {
					fatal(err)
				}

				fmt.Printf("Loaded %d service(s) from %s\n", len(cache), dirname)
			},
		},
		{
			Name:    "list",
			Aliases: []string{"l"},
			Usage:   "List available services",
			Action: func(c *cli.Context) {

				cache := loadCacheOrFail()
				for service, _ := range cache {
					fmt.Println("* " + service)
				}
			},
		},
		{
			Name:  "start",
			Usage: "Start a service",
			Action: func(c *cli.Context) {
				cache := loadCacheOrFail()
				serviceName := c.Args().First()
				if service, ok := cache[serviceName]; ok {
					err := service.Start()
					if err != nil {
						fatal(err)
					}
					fmt.Println(serviceName + " started")
				} else {
					fatal("Service not found")
				}
			},
		},
		{
			Name:  "stop",
			Usage: "Stop a service",
			Action: func(c *cli.Context) {
				cache := loadCacheOrFail()
				serviceName := c.Args().First()
				if service, ok := cache[serviceName]; ok {
					err := service.Stop()
					if err != nil {
						fatal(err)
					}
					fmt.Println(serviceName + " stopped")
				} else {
					fatal("Service not found")
				}
			},
		},
		{
			Name:  "restart",
			Usage: "Restart a service",
			Action: func(c *cli.Context) {
				cache := loadCacheOrFail()
				serviceName := c.Args().First()
				if service, ok := cache[serviceName]; ok {
					err := service.Restart()
					if err != nil {
						fatal(err)
					}
					fmt.Println(serviceName + " restarted")
				} else {
					fatal("Service not found")
				}
			},
		},
		{
			Name:  "status",
			Usage: "View status of all services",
			Action: func(c *cli.Context) {
				statuses, err := getStatus()
				if err != nil {
					fatal(err)
				}

				for service, status := range statuses {
					fmt.Printf("* %s: %s\n", service, status)
				}
			},
		},
	}

	app.Run(os.Args)

}