2021-04-07 00:33:01 +08:00
|
|
|
package serverctl
|
|
|
|
|
|
|
|
import (
|
2021-10-14 03:15:20 +08:00
|
|
|
"strings"
|
2021-09-07 03:53:16 +08:00
|
|
|
|
2022-02-06 02:12:07 +08:00
|
|
|
"github.com/gravitl/netmaker/database"
|
2021-12-07 04:31:08 +08:00
|
|
|
"github.com/gravitl/netmaker/logger"
|
2021-10-13 22:02:04 +08:00
|
|
|
"github.com/gravitl/netmaker/logic"
|
2022-03-14 08:58:30 +08:00
|
|
|
"github.com/gravitl/netmaker/logic/acls"
|
|
|
|
"github.com/gravitl/netmaker/logic/acls/nodeacls"
|
2022-09-14 03:41:23 +08:00
|
|
|
"github.com/gravitl/netmaker/logic/pro"
|
2021-04-07 00:33:01 +08:00
|
|
|
)
|
|
|
|
|
2022-02-19 00:11:21 +08:00
|
|
|
const (
|
|
|
|
// NETMAKER_BINARY_NAME - name of netmaker binary
|
|
|
|
NETMAKER_BINARY_NAME = "netmaker"
|
|
|
|
)
|
|
|
|
|
2022-08-31 23:38:29 +08:00
|
|
|
func SetDefaults() error {
|
|
|
|
if err := setNodeDefaults(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2022-09-14 03:41:23 +08:00
|
|
|
if err := setNetworkDefaults(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2022-09-14 21:58:01 +08:00
|
|
|
if err := setUserDefaults(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2022-08-31 23:38:29 +08:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// setNodeDefaults - runs through each node and set defaults
|
|
|
|
func setNodeDefaults() error {
|
2022-03-14 08:58:30 +08:00
|
|
|
// upgraded systems will not have ACL's set, which is why we need this function
|
2022-03-14 09:35:22 +08:00
|
|
|
nodes, err := logic.GetAllNodes()
|
2022-03-14 08:58:30 +08:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2022-03-14 09:35:22 +08:00
|
|
|
for i := range nodes {
|
2022-08-31 23:38:29 +08:00
|
|
|
logic.SetNodeDefaults(&nodes[i])
|
|
|
|
logic.UpdateNode(&nodes[i], &nodes[i])
|
2022-12-21 04:29:09 +08:00
|
|
|
currentNodeACL, err := nodeacls.FetchNodeACL(nodeacls.NetworkID(nodes[i].Network), nodeacls.NodeID(nodes[i].ID.String()))
|
2022-03-14 09:35:22 +08:00
|
|
|
if (err != nil && (database.IsEmptyRecord(err) || strings.Contains(err.Error(), "no node ACL present"))) || currentNodeACL == nil {
|
2022-12-21 04:29:09 +08:00
|
|
|
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())
|
2022-03-14 08:58:30 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2022-03-14 09:35:22 +08:00
|
|
|
return nil
|
2022-03-14 08:58:30 +08:00
|
|
|
}
|
2022-09-14 03:41:23 +08:00
|
|
|
|
|
|
|
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 _, net := range networks {
|
|
|
|
if err = pro.InitializeNetworkUsers(net.NetID); err != nil {
|
2022-09-14 04:32:11 +08:00
|
|
|
logger.Log(0, "could not initialize NetworkUsers on network", net.NetID)
|
|
|
|
}
|
|
|
|
pro.AddProNetDefaults(&net)
|
2022-10-01 03:16:13 +08:00
|
|
|
update := false
|
|
|
|
newNet := net
|
|
|
|
if strings.Contains(net.NetID, ".") {
|
|
|
|
newNet.NetID = strings.ReplaceAll(net.NetID, ".", "")
|
|
|
|
newNet.DefaultInterface = strings.ReplaceAll(net.DefaultInterface, ".", "")
|
|
|
|
update = true
|
|
|
|
}
|
|
|
|
if strings.ContainsAny(net.NetID, "ABCDEFGHIJKLMNOPQRSTUVWXYZ") {
|
|
|
|
newNet.NetID = strings.ToLower(net.NetID)
|
|
|
|
newNet.DefaultInterface = strings.ToLower(net.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(net.NetID); err != nil {
|
|
|
|
logger.Log(0, "error deleting old network:", err.Error())
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
net.SetDefaults()
|
|
|
|
_, _, _, _, _, _, err = logic.UpdateNetwork(&net, &net)
|
|
|
|
if err != nil {
|
|
|
|
logger.Log(0, "could not set defaults on network", net.NetID)
|
|
|
|
}
|
2022-09-14 03:41:23 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
2022-09-14 21:58:01 +08:00
|
|
|
|
|
|
|
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)
|
2022-09-15 02:44:52 +08:00
|
|
|
copyUser := updateUser
|
|
|
|
copyUser.Password = ""
|
|
|
|
if _, err = logic.UpdateUser(copyUser, updateUser); err != nil {
|
2022-09-14 21:58:01 +08:00
|
|
|
logger.Log(0, "could not update user", updateUser.UserName)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|