gravitl/netmaker
1
1
Fork 0
mirror of https://github.com/gravitl/netmaker.git synced 2024-09-21 07:46:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

on delete, removes files and ends daemon if required, ifacedelta on key refresh

Browse source
This commit is contained in:
0xdcarns 2022-02-04 16:47:57 -05:00
parent 1f104f97f4
commit 731977533b
1 changed files with 13 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
netclient/functions/daemon.go
View file

@ -249,22 +249,31 @@ func NodeUpdate(client mqtt.Client, msg mqtt.Message) {
cfg.Node = newNode
switch newNode.Action {
case models.NODE_DELETE:
if err := RemoveLocalInstance(&cfg, cfg.Network); err != nil {
ncutils.PrintLog("error deleting local instance: "+err.Error(), 1)
return
}
if token := client.Unsubscribe(fmt.Sprintf("update/%s/%s", newNode.Network, newNode.ID), fmt.Sprintf("peers/%s/%s", newNode.Network, newNode.ID)); token.Wait() && token.Error() != nil {
ncutils.PrintLog("error unsubscribing during node deletion", 1)
}
if err := WipeLocal(cfg.Network); err != nil {
ncutils.PrintLog("error deleting local instance: "+err.Error(), 1)
ncutils.PrintLog("Please perform manual clean up", 1)
}
currNets, err := ncutils.GetSystemNetworks()
if err == nil && len(currNets) == 0 {
if err = RemoveLocalInstance(&cfg, cfg.Network); err != nil {
ncutils.PrintLog("Please perform manual clean up", 1)
}
os.Exit(0)
}
return
case models.NODE_UPDATE_KEY:
if err := UpdateKeys(&cfg, client); err != nil {
ncutils.PrintLog("err updating wireguard keys: "+err.Error(), 1)
}
ifaceDelta = true
case models.NODE_NOOP:
default:
}
//Save new config
cfg.Node.Action = models.NODE_NOOP
if err := config.Write(&cfg, cfg.Network); err != nil {
ncutils.PrintLog("error updating node configuration: "+err.Error(), 1)
}