Browse Source

minor cleanups

0xdcarns 3 years ago
parent
commit
90cbbfdf4a
5 changed files with 17 additions and 46 deletions
  1. 0 1
      controllers/nodeGrpcController.go
  2. 0 1
      docker/Dockerfile-netmaker-slim
  3. 3 0
      main.go
  4. 0 19
      models/node.go
  5. 14 25
      serverctl/serverctl.go

+ 0 - 1
controllers/nodeGrpcController.go

@@ -33,7 +33,6 @@ func (s *NodeServiceServer) ReadNode(ctx context.Context, req *nodepb.Object) (*
 		return nil, err
 	}
 	node.Update(&node)
-	node.SetDefaults()
 	response := &nodepb.Object{
 		Data: string(nodeData),
 		Type: nodepb.NODE_TYPE,

+ 0 - 1
docker/Dockerfile-netmaker-slim

@@ -11,7 +11,6 @@ RUN GOOS=linux GOARCH=amd64 CGO_ENABLED=1 /usr/local/go/bin/go build -ldflags="-
 
 WORKDIR /app/netclient
 
-
 RUN GOOS=linux GOARCH=amd64 CGO_ENABLED=0 /usr/local/go/bin/go build -ldflags="-w -s" -o netclient main.go
 #second stage
 

+ 3 - 0
main.go

@@ -53,6 +53,9 @@ func initialize() { // Client Mode Prereq Check
 		if uid != 0 {
 			log.Fatal("To run in client mode requires root privileges. Either disable client mode or run with sudo.")
 		}
+		if err := serverctl.InitServerNetclient(); err != nil {
+			log.Fatal("Did not find netclient to use CLIENT_MODE")
+		}
 	}
 
 	if servercfg.IsDNSMode() {

+ 0 - 19
models/node.go

@@ -532,25 +532,6 @@ func GetAllNodes() ([]Node, error) {
 	return nodes, nil
 }
 
-func GetNode(macaddress string, network string) (Node, error) {
-
-	var node Node
-
-	key, err := GetID(macaddress, network)
-	if err != nil {
-		return node, err
-	}
-	data, err := database.FetchRecord(database.NODES_TABLE_NAME, key)
-	if err != nil {
-		return node, err
-	}
-	if err = json.Unmarshal([]byte(data), &node); err != nil {
-		return node, err
-	}
-
-	return node, err
-}
-
 func GetID(macaddress string, network string) (string, error) {
 	if macaddress == "" || network == "" {
 		return "", errors.New("unable to get record key")

+ 14 - 25
serverctl/serverctl.go

@@ -100,8 +100,7 @@ func RemoveNetwork(network string) (bool, error) {
 
 }
 
-func HandleContainedClient() error {
-	log.SetFlags(log.Flags() &^ (log.Llongfile | log.Lshortfile))
+func InitServerNetclient() error {
 	netclientDir := ncutils.GetNetclientPath()
 	netclientPath := ncutils.GetNetclientPathSpecific()
 	_, err := os.Stat(netclientDir)
@@ -123,11 +122,19 @@ func HandleContainedClient() error {
 		log.Println("could not change netclient binary permissions")
 		return err
 	}
+	return nil
+}
+
+func HandleContainedClient() error {
+	log.SetFlags(log.Flags() &^ (log.Llongfile | log.Lshortfile))
 
+	netclientPath := ncutils.GetNetclientPathSpecific()
 	checkinCMD := exec.Command(netclientPath+"netclient", "checkin", "-n", "all")
-	checkinCMD.Stdout = os.Stdout
+	if servercfg.GetVerbose() >= 2 {
+		checkinCMD.Stdout = os.Stdout
+	}
 	checkinCMD.Stderr = os.Stderr
-	err = checkinCMD.Start()
+	err := checkinCMD.Start()
 	if err != nil {
 		if servercfg.GetVerbose() >= 2 {
 			log.Println(err)
@@ -151,36 +158,18 @@ func AddNetwork(network string) (bool, error) {
 		log.Println("could not get public IP.")
 		return false, err
 	}
-	netclientDir := ncutils.GetNetclientPath()
 	netclientPath := ncutils.GetNetclientPathSpecific()
-	_, err = os.Stat(netclientDir)
-	if os.IsNotExist(err) {
-		os.Mkdir(netclientDir, 744)
-	} else if err != nil {
-		log.Println("could not find or create", netclientDir)
-		return false, err
-	}
+
 	token, err := functions.CreateServerToken(network)
 	if err != nil {
 		log.Println("could not create server token for " + network)
 		return false, err
 	}
-	_, err = os.Stat(netclientPath + "netclient")
-	if os.IsNotExist(err) {
-		err = InstallNetclient()
-		if err != nil {
-			return false, err
-		}
-	}
-	err = os.Chmod(netclientPath+"netclient", 0755)
-	if err != nil {
-		log.Println("could not change netclient directory permissions")
-		return false, err
-	}
+
 	functions.PrintUserLog(models.NODE_SERVER_NAME, "executing network join: "+netclientPath+"netclient "+"join "+"-t "+token+" -name "+models.NODE_SERVER_NAME+" -endpoint "+pubip, 0)
 	var joinCMD *exec.Cmd
 	if servercfg.IsClientMode() == "contained" {
-		joinCMD = exec.Command(netclientPath+"netclient", "join", "-t", token, "-name", models.NODE_SERVER_NAME, "-endpoint", pubip, "-daemon", "off")
+		joinCMD = exec.Command(netclientPath+"netclient", "join", "-t", token, "-name", models.NODE_SERVER_NAME, "-endpoint", pubip, "-daemon", "off", "-dnson", "no")
 	} else {
 		joinCMD = exec.Command(netclientPath+"netclient", "join", "-t", token, "-name", models.NODE_SERVER_NAME, "-endpoint", pubip)
 	}