mirror of
https://github.com/dragonheim/gagent.git
synced 2025-02-23 01:09:55 -08:00
James Wells
1bcc682b7c
refactor: reduced usage of fmt in favor of logs and string concatication. refactor: minor re-ordering of the data structures to reduce storage space required.
61 lines
2.1 KiB
Go
61 lines
2.1 KiB
Go
package setup
|
|
|
|
import (
|
|
log "log"
|
|
sync "sync"
|
|
|
|
cty "github.com/zclconf/go-cty/cty"
|
|
|
|
gs "github.com/dragonheim/gagent/internal/gstructs"
|
|
|
|
hclwrite "github.com/hashicorp/hcl/v2/hclwrite"
|
|
)
|
|
|
|
/*
|
|
* Main is the entrypoint for the setup process
|
|
*/
|
|
func Main(wg *sync.WaitGroup, config gs.GagentConfig) {
|
|
log.Printf("[INFO] Starting setup\n")
|
|
defer wg.Done()
|
|
|
|
f := hclwrite.NewEmptyFile()
|
|
rootBody := f.Body()
|
|
rootBody.SetAttributeValue("name", cty.StringVal(config.Name))
|
|
rootBody.SetAttributeValue("mode", cty.StringVal(config.Mode))
|
|
rootBody.SetAttributeValue("uuid", cty.StringVal(config.UUID))
|
|
rootBody.SetAttributeValue("listenaddr", cty.StringVal("0.0.0.0"))
|
|
rootBody.SetAttributeValue("clientport", cty.NumberIntVal(int64(config.ClientPort)))
|
|
rootBody.SetAttributeValue("routerport", cty.NumberIntVal(int64(config.RouterPort)))
|
|
rootBody.SetAttributeValue("workerport", cty.NumberIntVal(int64(config.WorkerPort)))
|
|
rootBody.AppendNewline()
|
|
|
|
clientBlock1 := rootBody.AppendNewBlock("client", []string{config.Name})
|
|
clientBody1 := clientBlock1.Body()
|
|
/*
|
|
* clientBody1.AppendUnstructuredTokens(
|
|
* hclwrite.TokensForTraversal(hcl.Traversal{
|
|
* hcl.TraverseRoot{
|
|
* Name: hcl.CommentGenerator("comment"),
|
|
* },
|
|
* },
|
|
* ))
|
|
*/
|
|
clientBody1.SetAttributeValue("clientid", cty.StringVal(config.UUID))
|
|
rootBody.AppendNewline()
|
|
|
|
routerBlock1 := rootBody.AppendNewBlock("router", []string{config.Name})
|
|
routerBody1 := routerBlock1.Body()
|
|
routerBody1.SetAttributeValue("routerid", cty.StringVal(config.UUID))
|
|
routerBody1.SetAttributeValue("address", cty.StringVal("127.0.0.1"))
|
|
routerBody1.SetAttributeValue("clientport", cty.NumberIntVal(int64(config.ClientPort)))
|
|
routerBody1.SetAttributeValue("routerport", cty.NumberIntVal(int64(config.RouterPort)))
|
|
routerBody1.SetAttributeValue("workerport", cty.NumberIntVal(int64(config.WorkerPort)))
|
|
rootBody.AppendNewline()
|
|
|
|
workerBlock1 := rootBody.AppendNewBlock("worker", []string{config.Name})
|
|
workerBody1 := workerBlock1.Body()
|
|
workerBody1.SetAttributeValue("workerid", cty.StringVal(config.UUID))
|
|
rootBody.AppendNewline()
|
|
|
|
log.Printf("[DEBUG] Configuration file created;\n\n%s\n", f.Bytes())
|
|
}
|