/
createexamples.go
76 lines (64 loc) · 1.69 KB
/
createexamples.go
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
package main
import (
"flag"
"fmt"
"io/ioutil"
"os"
"strings"
"github.com/bisrael8191/systemdeps"
"github.com/coreos/go-systemd/dbus"
"github.com/coreos/go-systemd/unit"
)
func generateExampleUnit(processName string) *systemdeps.UnitFile {
return &systemdeps.UnitFile{
Options: []*unit.UnitOption{
{Section: "Unit", Name: "Description", Value: strings.Title(processName) + " example service"},
{Section: "Service", Name: "Type", Value: "simple"},
{Section: "Service", Name: "ExecStart", Value: "/bin/sleep infinity"},
{Section: "Service", Name: "Restart", Value: "on-failure"},
},
}
}
func main() {
var systemdPath string
var configFile string
flag.StringVar(&systemdPath, "systemdpath", os.Getenv("HOME")+"/.config/systemd/user/", "systemd path")
flag.StringVar(&configFile, "config", "dependencies.json", "path of dependency file")
flag.Parse()
if !strings.HasSuffix(systemdPath, "/") {
systemdPath = systemdPath + "/"
}
userSystemd := strings.Contains(systemdPath, "user")
processes, err := systemdeps.ReadDependencyFile(configFile)
if err != nil {
panic(err)
}
for _, p := range processes.Processes {
unitName := systemdPath + p.Name + ".service"
unitStr := generateExampleUnit(p.Name).String()
fmt.Println("Adding unit file:", unitName)
err := ioutil.WriteFile(unitName, []byte(unitStr), 0644)
if err != nil {
panic(err)
}
}
// Open systemd connection
var conn *dbus.Conn
if userSystemd {
conn, err = dbus.NewUserConnection()
if err != nil {
panic(err)
}
} else {
conn, err = dbus.NewSystemConnection()
if err != nil {
panic(err)
}
}
defer conn.Close()
// Reload all units
err = conn.Reload()
if err != nil {
panic(err)
}
}