netmaker/cli/cmd/node/update.go

package node

import (
	"encoding/json"
	"log"
	"net"
	"os"
	"strings"
	"time"

	"github.com/gravitl/netmaker/cli/functions"
	"github.com/gravitl/netmaker/models"
	"github.com/spf13/cobra"
)

var nodeUpdateCmd = &cobra.Command{
	Use:   "update [NETWORK NAME] [NODE ID]",
	Args:  cobra.ExactArgs(2),
	Short: "Update a Node",
	Long:  `Update a Node`,
	Run: func(cmd *cobra.Command, args []string) {
		var (
			node        = &models.Node{}
			networkName = args[0]
			nodeID      = args[1]
		)
		if nodeDefinitionFilePath != "" {
			content, err := os.ReadFile(nodeDefinitionFilePath)
			if err != nil {
				log.Fatal("Error when opening file: ", err)
			}
			if err := json.Unmarshal(content, node); err != nil {
				log.Fatal(err)
			}
		} else {
			if address != "" {
				if _, addr, err := net.ParseCIDR(address); err != nil {
					log.Fatal(err)
				} else {
					node.Address = *addr
				}
			}
			if address6 != "" {
				if _, addr6, err := net.ParseCIDR(address6); err != nil {
					log.Fatal(err)
				} else {
					node.Address6 = *addr6
				}
			}
			if localAddress != "" {
				if _, localAddr, err := net.ParseCIDR(localAddress); err != nil {
					log.Fatal(err)
				} else {
					node.LocalAddress = *localAddr
					node.IsLocal = true
				}
			}
			node.PostUp = postUp
			node.PostDown = postDown
			node.PersistentKeepalive = time.Duration(time.Second * time.Duration(keepAlive))
			if relayAddrs != "" {
				node.RelayAddrs = strings.Split(relayAddrs, ",")
			}
			if egressGatewayRanges != "" {
				node.EgressGatewayRanges = strings.Split(egressGatewayRanges, ",")
			}
			node.ExpirationDateTime = time.Unix(int64(expirationDateTime), 0)
			if defaultACL {
				node.DefaultACL = "yes"
			}
			node.DNSOn = dnsOn
			node.Connected = !disconnect
		}
		functions.PrettyPrint(functions.UpdateNode(networkName, nodeID, node))
	},
}

func init() {
	nodeUpdateCmd.Flags().StringVar(&nodeDefinitionFilePath, "file", "", "Filepath of updated node definition in JSON")
	nodeUpdateCmd.Flags().StringVar(&address, "ipv4_addr", "", "IPv4 address of the node")
	nodeUpdateCmd.Flags().StringVar(&address6, "ipv6_addr", "", "IPv6 address of the node")
	nodeUpdateCmd.Flags().StringVar(&localAddress, "local_addr", "", "Locally reachable address of the node")
	nodeUpdateCmd.Flags().StringVar(&name, "name", "", "Node name")
	nodeUpdateCmd.Flags().StringVar(&postUp, "post_up", "", "Commands to run after node is up `;` separated")
	nodeUpdateCmd.Flags().StringVar(&postDown, "post_down", "", "Commands to run after node is down `;` separated")
	nodeUpdateCmd.Flags().IntVar(&keepAlive, "keep_alive", 0, "Interval in which packets are sent to keep connections open with peers")
	nodeUpdateCmd.Flags().StringVar(&relayAddrs, "relay_addrs", "", "Addresses for relaying connections if node acts as a relay")
	nodeUpdateCmd.Flags().StringVar(&egressGatewayRanges, "egress_addrs", "", "Addresses for egressing traffic if node acts as an egress")
	nodeUpdateCmd.Flags().IntVar(&expirationDateTime, "expiry", 0, "UNIX timestamp after which node will lose access to the network")
	nodeUpdateCmd.Flags().BoolVar(&defaultACL, "acl", false, "Enable default ACL ?")
	nodeUpdateCmd.Flags().BoolVar(&dnsOn, "dns", false, "Setup DNS entries for peers locally ?")
	nodeUpdateCmd.Flags().BoolVar(&disconnect, "disconnect", false, "Disconnect from the network ?")
	rootCmd.AddCommand(nodeUpdateCmd)
}