yanic/models/ansible.go

package models

// Ansible struct
type Ansible struct {
	Nodes []string `json:"nodes"`
	Meta  struct {
		HostVars map[string]*AnsibleHostVars `json:"hostvars,omitempty"`
	} `json:"_meta"`
}

// AnsibleHostVars new values for a node
type AnsibleHostVars struct {
	Address      string  `json:"ansible_ssh_host"`
	Hostname     string  `json:"node_name,omitempty"`
	Owner        string  `json:"owner,omitempty"`
	Channel24    uint32  `json:"radio24_channel,omitempty"`
	TxPower24    uint32  `json:"radio24_txpower,omitempty"`
	Channel5     uint32  `json:"radio5_channel,omitempty"`
	TxPower5     uint32  `json:"radio5_txpower,omitempty"`
	GeoLatitude  float64 `json:"geo_latitude,omitempty"`
	GeoLongitude float64 `json:"geo_longitude,omitempty"`
}

// GenerateAnsible but nodes and aliases together to a ansible change output
func GenerateAnsible(nodes *Nodes, aliases map[string]*Alias) *Ansible {
	ansible := &Ansible{Nodes: make([]string, 0)}
	ansible.Meta.HostVars = make(map[string]*AnsibleHostVars)
	for nodeid, alias := range aliases {
		if node := nodes.List[nodeid]; node != nil {

			ansible.Nodes = append(ansible.Nodes, nodeid)

			vars := &AnsibleHostVars{
				Hostname: alias.Hostname,
				Owner:    alias.Owner,
			}
			if node.Nodeinfo.Network.Addresses != nil {
				vars.Address = node.Nodeinfo.Network.Addresses[0]
			}
			if alias.Wireless != nil {
				vars.Channel24 = alias.Wireless.Channel24
				vars.TxPower24 = alias.Wireless.TxPower24
				vars.Channel5 = alias.Wireless.Channel5
				vars.TxPower5 = alias.Wireless.TxPower5
			}
			if alias.Location != nil {
				vars.GeoLatitude = alias.Location.Latitude
				vars.GeoLongitude = alias.Location.Longtitude
			}
			ansible.Meta.HostVars[nodeid] = vars

		}
	}
	return ansible
}