Browse Source

bug squashin

worker-9 4 years ago
parent
commit
e94e67c7f3
2 changed files with 7 additions and 7 deletions
  1. 4 7
      netclient/functions/checkin.go
  2. 3 0
      netclient/wireguard/kernel.go

+ 4 - 7
netclient/functions/checkin.go

@@ -98,11 +98,6 @@ func checkNodeActions(node *models.Node, network string, servercfg config.Server
 			log.Println("Unable to process reset keys request:", err)
 			log.Println("Unable to process reset keys request:", err)
 			return ""
 			return ""
 		}
 		}
-		node.Action = models.NODE_NOOP
-		if err = config.ModConfig(node); err != nil {
-			return ""
-		}
-		Push(network)
 		return ""
 		return ""
 	}
 	}
 	if node.Action == models.NODE_DELETE {
 	if node.Action == models.NODE_DELETE {
@@ -127,12 +122,10 @@ func checkNodeActions(node *models.Node, network string, servercfg config.Server
 func CheckConfig(cliconf config.ClientConfig) error {
 func CheckConfig(cliconf config.ClientConfig) error {
 
 
 	network := cliconf.Network
 	network := cliconf.Network
-	// node := server.GetNode(network)
 	cfg, err := config.ReadConfig(network)
 	cfg, err := config.ReadConfig(network)
 	if err != nil {
 	if err != nil {
 		return err
 		return err
 	}
 	}
-	// nodecfg := cfg.Node
 	servercfg := cfg.Server
 	servercfg := cfg.Server
 
 
 	newNode, err := Pull(network, false)
 	newNode, err := Pull(network, false)
@@ -231,6 +224,10 @@ func Pull(network string, manual bool) (*models.Node, error) {
 		if err != nil {
 		if err != nil {
 			return &resNode, err
 			return &resNode, err
 		}
 		}
+	} else {
+		if err = wireguard.SetWGConfig(network, true); err != nil {
+			return nil, err
+		}
 	}
 	}
 	setDNS(&resNode, servercfg, &cfg.Node)
 	setDNS(&resNode, servercfg, &cfg.Node)
 
 

+ 3 - 0
netclient/wireguard/kernel.go

@@ -342,6 +342,9 @@ func SetWGKeyConfig(network string, serveraddr string) error {
 	if err != nil {
 	if err != nil {
 		return err
 		return err
 	}
 	}
+	if node.Action == models.NODE_UPDATE_KEY {
+		node.Action = models.NODE_NOOP
+	}
 	err = config.ModConfig(&node)
 	err = config.ModConfig(&node)
 	if err != nil {
 	if err != nil {
 		return err
 		return err