gravitl/netmaker
1
1
Fork 0
mirror of https://github.com/gravitl/netmaker.git synced 2025-09-06 05:04:27 +08:00
Code Issues Projects Releases Packages Wiki Activity

move logic to serverctl.setNetworkDefaults()

Browse source
This commit is contained in:
Matthew R. Kasun 2022-09-30 15:16:13 -04:00
parent 856cd0a15a
commit 1de2989b84
2 changed files with 26 additions and 36 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
main.go
View file

@ -12,7 +12,6 @@ import (
"os/signal"
"runtime/debug"
"strconv"
"strings"
"sync"
"syscall"
"time"
@ -75,38 +74,6 @@ func initialize() { // Client Mode Prereq Check
if err = database.InitializeDatabase(); err != nil {
logger.FatalLog("Error connecting to database")
}
//update network defaults
networks, _ := logic.GetNetworks()
for _, network := range networks {
update := false
newNet := network
newNet.SetDefaults()
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()
if err := logic.SaveNetwork(&network); err != nil {
logger.Log(0, "error saving networks during with defaults:", err.Error())
}
}
}
logger.Log(0, "database successfully connected")
if err = logic.AddServerIDIfNotPresent(); err != nil {
logger.Log(1, "failed to save server ID")

29
serverctl/serverctl.go
View file

@ -132,9 +132,32 @@ func setNetworkDefaults() error {
logger.Log(0, "could not initialize NetworkUsers on network", net.NetID)
}
pro.AddProNetDefaults(&net)
_, _, _, _, _, _, err = logic.UpdateNetwork(&net, &net)
if err != nil {
logger.Log(0, "could not set defaults on network", net.NetID)
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)
}
}
}
return nil