/
assembler.go
60 lines (53 loc) · 1.25 KB
/
assembler.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
package main
import (
"os"
"fmt"
"io/ioutil"
"gopkg.in/yaml.v2"
"github.com/gliderlabs/sigil"
_ "github.com/gliderlabs/sigil/builtin"
)
var config = make(map[string]interface{})
var services = make(map[string]interface{})
func main() {
data, err := ioutil.ReadFile(os.Args[1])
fatal(err)
var obj map[string]interface{}
fatal(yaml.Unmarshal(data, &obj))
if value, ok := obj["config"]; ok {
exports := value.(map[interface{}]interface{})
for k,v := range exports {
config[k.(string)] = v
}
}
if value, ok := obj["services"]; ok {
svcs := value.([]interface{})
for _, v := range svcs {
load(v.(string))
}
}
yml, err := yaml.Marshal(services)
fatal(err)
fmt.Println(string(yml))
}
func load(name string) {
data, err := ioutil.ReadFile(name + ".yml")
fatal(err)
rendered, err := sigil.Execute(data, config, name)
fatal(err)
var service map[string]interface{}
fatal(yaml.Unmarshal(rendered.Bytes(), &service))
if value, ok := service["extra"]; ok {
exports := value.(map[interface{}]interface{})
for k,v := range exports {
config[k.(string)] = v
}
delete(service, "extra")
}
services[name] = service
}
func fatal(err error) {
if (err != nil) {
panic(err)
}
}