TheArchive/headscale
1
0
Fork 0
mirror of https://github.com/juanfont/headscale.git synced 2024-11-23 18:15:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move the namespace argument back to a flag for the share and unshare

Browse Source 
commands.
This commit is contained in:
Ward Vandewege 2021-10-27 17:51:42 -04:00
parent f9ece0087d
commit d086cf4691
2 changed files with 27 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
cmd/headscale/cli/nodes.go
View File

@ -19,14 +19,28 @@ func init() {
rootCmd.AddCommand(nodeCmd)
listNodesCmd.Flags().StringP("namespace", "n", "", "Filter by namespace")
nodeCmd.AddCommand(listNodesCmd)
registerNodeCmd.Flags().StringP("namespace", "n", "", "Filter by namespace")
registerNodeCmd.Flags().StringP("namespace", "n", "", "Namespace")
err := registerNodeCmd.MarkFlagRequired("namespace")
if err != nil {
log.Fatalf(err.Error())
}
nodeCmd.AddCommand(registerNodeCmd)
nodeCmd.AddCommand(deleteNodeCmd)
shareMachineCmd.Flags().StringP("namespace", "n", "", "Namespace")
err = shareMachineCmd.MarkFlagRequired("namespace")
if err != nil {
log.Fatalf(err.Error())
}
nodeCmd.AddCommand(shareMachineCmd)
unshareMachineCmd.Flags().StringP("namespace", "n", "", "Namespace")
err = unshareMachineCmd.MarkFlagRequired("namespace")
if err != nil {
log.Fatalf(err.Error())
}
nodeCmd.AddCommand(unshareMachineCmd)
}
@ -199,6 +213,11 @@ var deleteNodeCmd = &cobra.Command{
}
func sharingWorker(cmd *cobra.Command, args []string) (*headscale.Headscale, string, *headscale.Machine, *headscale.Namespace) {
n, err := cmd.Flags().GetString("namespace")
if err != nil {
log.Fatalf("Error getting namespace: %s", err)
}
output, _ := cmd.Flags().GetString("output")
h, err := getHeadscaleApp()
@ -206,9 +225,9 @@ func sharingWorker(cmd *cobra.Command, args []string) (*headscale.Headscale, str
log.Fatalf("Error initializing: %s", err)
}
namespace, err := h.GetNamespace(args[1])
namespace, err := h.GetNamespace(n)
if err != nil {
log.Fatalf("Error fetching namespace %s: %s", args[1], err)
log.Fatalf("Error fetching namespace %s: %s", n, err)
}
id, err := strconv.Atoi(args[0])
@ -224,10 +243,10 @@ func sharingWorker(cmd *cobra.Command, args []string) (*headscale.Headscale, str
}
var shareMachineCmd = &cobra.Command{
Use: "share ID namespace",
Use: "share ID",
Short: "Shares a node from the current namespace to the specified one",
Args: func(cmd *cobra.Command, args []string) error {
if len(args) < 2 {
if len(args) < 1 {
return fmt.Errorf("missing parameters")
}
return nil
@ -249,10 +268,10 @@ var shareMachineCmd = &cobra.Command{
}
var unshareMachineCmd = &cobra.Command{
Use: "unshare ID namespace",
Use: "unshare ID",
Short: "Unshares a node from the specified namespace",
Args: func(cmd *cobra.Command, args []string) error {
if len(args) < 2 {
if len(args) < 1 {
return fmt.Errorf("missing parameters")
}
return nil

2
integration_test.go
View File

@ -493,7 +493,7 @@ func (s *IntegrationTestSuite) TestSharedNodes() {
result, err := executeCommand(
&headscale,
[]string{"headscale", "nodes", "share", fmt.Sprint(machine.ID), "main"},
[]string{"headscale", "nodes", "share", fmt.Sprint(machine.ID), "-n", "main"},
[]string{},
)
assert.Nil(s.T(), err)