From 45e900178776924f79288a4ea3dfd902bd9cc7e2 Mon Sep 17 00:00:00 2001
From: Juan Font Alonso <juanfontalonso@gmail.com>
Date: Sat, 1 May 2021 20:05:10 +0200
Subject: [PATCH] Register cmd under node

---
 README.md                  |  2 +-
 api.go                     |  2 +-
 cmd/headscale/headscale.go | 10 +++-------
 3 files changed, 5 insertions(+), 9 deletions(-)

diff --git a/README.md b/README.md
index 0fe65805..17016974 100644
--- a/README.md
+++ b/README.md
@@ -78,7 +78,7 @@ Suggestions/PRs welcomed!
 
 8. In the server, register your machine to a namespace with the CLI
   ```shell
-  ./headscale -n myfirstnamespace register YOURMACHINEKEY
+  ./headscale -n myfirstnamespace node register YOURMACHINEKEY
   ```
 
 ## Configuration reference
diff --git a/api.go b/api.go
index 721b65ab..f875259c 100644
--- a/api.go
+++ b/api.go
@@ -43,7 +43,7 @@ func (h *Headscale) RegisterWebAPI(c *gin.Context) {
 
 	<p>
 		<code>
-			<b>headscale register %s</b>
+			<b>headscale -n NAMESPACE node register %s</b>
 		</code>
 	</p>
 
diff --git a/cmd/headscale/headscale.go b/cmd/headscale/headscale.go
index 3ea3a00a..51df310d 100644
--- a/cmd/headscale/headscale.go
+++ b/cmd/headscale/headscale.go
@@ -86,7 +86,6 @@ func main() {
 	headscaleCmd.AddCommand(cli.NamespaceCmd)
 	headscaleCmd.AddCommand(cli.NodeCmd)
 	headscaleCmd.AddCommand(cli.PreauthkeysCmd)
-	headscaleCmd.AddCommand(cli.RegisterCmd)
 	headscaleCmd.AddCommand(cli.RoutesCmd)
 	headscaleCmd.AddCommand(cli.ServeCmd)
 	headscaleCmd.AddCommand(versionCmd)
@@ -103,12 +102,6 @@ func main() {
 		log.Fatalf(err.Error())
 	}
 
-	cli.RegisterCmd.PersistentFlags().StringP("namespace", "n", "", "Namespace")
-	err = cli.RegisterCmd.MarkPersistentFlagRequired("namespace")
-	if err != nil {
-		log.Fatalf(err.Error())
-	}
-
 	cli.RoutesCmd.PersistentFlags().StringP("namespace", "n", "", "Namespace")
 	err = cli.RoutesCmd.MarkPersistentFlagRequired("namespace")
 	if err != nil {
@@ -118,6 +111,9 @@ func main() {
 	cli.NamespaceCmd.AddCommand(cli.CreateNamespaceCmd)
 	cli.NamespaceCmd.AddCommand(cli.ListNamespacesCmd)
 
+	cli.NodeCmd.AddCommand(cli.ListNodesCmd)
+	cli.NodeCmd.AddCommand(cli.RegisterCmd)
+
 	cli.RoutesCmd.AddCommand(cli.ListRoutesCmd)
 	cli.RoutesCmd.AddCommand(cli.EnableRouteCmd)