-
Notifications
You must be signed in to change notification settings - Fork 4.6k
/
manifest.go
80 lines (66 loc) · 1.89 KB
/
manifest.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
77
78
79
80
/*
Copyright 2019 The Kubernetes Authors.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package systemd
import (
"bytes"
"fmt"
)
// Manifest defines a systemd unit
type Manifest struct {
sections []*section
}
// section defines a section of the unit i.e. Unit, Service etc
type section struct {
key string
content string
entries []string
}
// Set adds a key/pair to the a section in the systemd manifest
func (m *Manifest) Set(name, key, value string) {
s := m.getSection(name)
s.entries = append(s.entries, fmt.Sprintf("%s=%s\n", key, value))
}
// SetSection sets the raw content of a section
func (m *Manifest) SetSection(name, content string) {
m.getSection(name).content = content
}
// getSection checks if a section already exists
func (m *Manifest) getSection(key string) *section {
for _, s := range m.sections {
if s.key == key {
return s
}
}
// create a new section for this manifest
s := §ion{key: key, entries: make([]string, 0)}
m.sections = append(m.sections, s)
return s
}
// Render is responsible for generating the final unit
func (m *Manifest) Render() string {
var b bytes.Buffer
size := len(m.sections) - 1
for i, section := range m.sections {
b.WriteString(fmt.Sprintf("[%s]\n", section.key))
if section.content != "" {
b.WriteString(section.content)
}
for _, x := range section.entries {
b.WriteString(x)
}
if i < size {
b.WriteString("\n")
}
}
return b.String()
}