/
cloudinit.go
65 lines (56 loc) · 1.79 KB
/
cloudinit.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
package machine
import (
"bytes"
"html/template"
"log"
"github.com/alfg/openencoder/api/config"
)
const userDataTmpl = `
#cloud-config
package_upgrade: false
write_files:
- path: "/opt/.env"
content: |
REDIS_HOST={{.CloudinitRedisHost}}
REDIS_PORT={{.CloudinitRedisPort}}
DATABASE_HOST={{.CloudinitDatabaseHost}}
DATABASE_PORT={{.CloudinitDatabasePort}}
DATABASE_USER={{.CloudinitDatabaseUser}}
DATABASE_PASSWORD={{.CloudinitDatabasePassword}}
DATABASE_NAME={{.CloudinitDatabaseName}}
runcmd:
- docker run -d --env-file /opt/.env --rm {{.CloudinitWorkerImage}} worker
`
// UserData defines the userdata used for cloud-init.
type UserData struct {
AWSAccessKey string
AWSSecretKey string
SlackWebhook string
CloudinitRedisHost string
CloudinitRedisPort int
CloudinitDatabaseHost string
CloudinitDatabasePort int
CloudinitDatabaseUser string
CloudinitDatabasePassword string
CloudinitDatabaseName string
CloudinitWorkerImage string
}
func createUserData() string {
data := &UserData{
CloudinitRedisHost: config.Get().CloudinitRedisHost,
CloudinitRedisPort: config.Get().CloudinitRedisPort,
CloudinitDatabaseHost: config.Get().CloudinitDatabaseHost,
CloudinitDatabasePort: config.Get().CloudinitDatabasePort,
CloudinitDatabaseUser: config.Get().CloudinitDatabaseUser,
CloudinitDatabasePassword: config.Get().CloudinitDatabasePassword,
CloudinitDatabaseName: config.Get().CloudinitDatabaseName,
CloudinitWorkerImage: config.Get().CloudinitWorkerImage,
}
var tpl bytes.Buffer
t := template.Must(template.New("userdata").Parse(userDataTmpl))
if err := t.Execute(&tpl, data); err != nil {
log.Println(err)
}
result := tpl.String()
return result
}