mirror of
https://github.com/gravitl/netmaker.git
synced 2024-11-14 05:40:50 +08:00
115 lines
3.3 KiB
Go
115 lines
3.3 KiB
Go
package serverctl
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/gravitl/netmaker/database"
|
|
"github.com/gravitl/netmaker/logger"
|
|
"github.com/gravitl/netmaker/logic"
|
|
"github.com/gravitl/netmaker/logic/acls"
|
|
"github.com/gravitl/netmaker/logic/acls/nodeacls"
|
|
"github.com/gravitl/netmaker/logic/pro"
|
|
)
|
|
|
|
const (
|
|
// NETMAKER_BINARY_NAME - name of netmaker binary
|
|
NETMAKER_BINARY_NAME = "netmaker"
|
|
)
|
|
|
|
func SetDefaults() error {
|
|
if err := setNodeDefaults(); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := setNetworkDefaults(); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := setUserDefaults(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// setNodeDefaults - runs through each node and set defaults
|
|
func setNodeDefaults() error {
|
|
// upgraded systems will not have ACL's set, which is why we need this function
|
|
nodes, err := logic.GetAllNodes()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for i := range nodes {
|
|
logic.SetNodeDefaults(&nodes[i])
|
|
logic.UpdateNode(&nodes[i], &nodes[i])
|
|
currentNodeACL, err := nodeacls.FetchNodeACL(nodeacls.NetworkID(nodes[i].Network), nodeacls.NodeID(nodes[i].ID.String()))
|
|
if (err != nil && (database.IsEmptyRecord(err) || strings.Contains(err.Error(), "no node ACL present"))) || currentNodeACL == nil {
|
|
if _, err = nodeacls.CreateNodeACL(nodeacls.NetworkID(nodes[i].Network), nodeacls.NodeID(nodes[i].ID.String()), acls.Allowed); err != nil {
|
|
logger.Log(1, "could not create a default ACL for node", nodes[i].ID.String())
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func setNetworkDefaults() error {
|
|
// upgraded systems will not have NetworkUsers's set, which is why we need this function
|
|
networks, err := logic.GetNetworks()
|
|
if err != nil && !database.IsEmptyRecord(err) {
|
|
return err
|
|
}
|
|
for _, network := range networks {
|
|
if err = pro.InitializeNetworkUsers(network.NetID); err != nil {
|
|
logger.Log(0, "could not initialize NetworkUsers on network", network.NetID)
|
|
}
|
|
pro.AddProNetDefaults(&network)
|
|
update := false
|
|
newNet := network
|
|
if strings.Contains(network.NetID, ".") {
|
|
newNet.NetID = strings.ReplaceAll(network.NetID, ".", "")
|
|
newNet.DefaultInterface = strings.ReplaceAll(network.DefaultInterface, ".", "")
|
|
update = true
|
|
}
|
|
if strings.ContainsAny(network.NetID, "ABCDEFGHIJKLMNOPQRSTUVWXYZ") {
|
|
newNet.NetID = strings.ToLower(network.NetID)
|
|
newNet.DefaultInterface = strings.ToLower(network.DefaultInterface)
|
|
update = true
|
|
}
|
|
if update {
|
|
newNet.SetDefaults()
|
|
if err := logic.SaveNetwork(&newNet); err != nil {
|
|
logger.Log(0, "error saving networks during initial update:", err.Error())
|
|
}
|
|
if err := logic.DeleteNetwork(network.NetID); err != nil {
|
|
logger.Log(0, "error deleting old network:", err.Error())
|
|
}
|
|
} else {
|
|
network.SetDefaults()
|
|
_, _, _, _, _, err = logic.UpdateNetwork(&network, &network)
|
|
if err != nil {
|
|
logger.Log(0, "could not set defaults on network", network.NetID)
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func setUserDefaults() error {
|
|
users, err := logic.GetUsers()
|
|
if err != nil && !database.IsEmptyRecord(err) {
|
|
return err
|
|
}
|
|
for _, user := range users {
|
|
updateUser, err := logic.GetUser(user.UserName)
|
|
if err != nil {
|
|
logger.Log(0, "could not update user", updateUser.UserName)
|
|
}
|
|
logic.SetUserDefaults(updateUser)
|
|
copyUser := updateUser
|
|
copyUser.Password = ""
|
|
if _, err = logic.UpdateUser(copyUser, updateUser); err != nil {
|
|
logger.Log(0, "could not update user", updateUser.UserName)
|
|
}
|
|
}
|
|
return nil
|
|
}
|