mirror of
https://github.com/juanfont/headscale.git
synced 2025-03-01 00:21:18 +01:00
2dfd42f80c
This commit is trying to DRY up the initiation of the gRPC client in each command: It renames the function to CLI instead of GRPC as it actually set up a CLI client, not a generic grpc client It also moves the configuration of address, timeout (which is now consistent) and api to use Viper, allowing users to set it via env vars and configuration file
90 lines
2.3 KiB
Go
90 lines
2.3 KiB
Go
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
v1 "github.com/juanfont/headscale/gen/go/headscale/v1"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(debugCmd)
|
|
|
|
createNodeCmd.Flags().StringP("name", "", "", "Name")
|
|
err := createNodeCmd.MarkFlagRequired("name")
|
|
if err != nil {
|
|
log.Fatal().Err(err).Msg("")
|
|
}
|
|
createNodeCmd.Flags().StringP("namespace", "n", "", "Namespace")
|
|
err = createNodeCmd.MarkFlagRequired("namespace")
|
|
if err != nil {
|
|
log.Fatal().Err(err).Msg("")
|
|
}
|
|
createNodeCmd.Flags().StringP("key", "k", "", "Key")
|
|
err = createNodeCmd.MarkFlagRequired("key")
|
|
if err != nil {
|
|
log.Fatal().Err(err).Msg("")
|
|
}
|
|
createNodeCmd.Flags().StringSliceP("route", "r", []string{}, "List (or repeated flags) of routes to advertise")
|
|
|
|
debugCmd.AddCommand(createNodeCmd)
|
|
}
|
|
|
|
var debugCmd = &cobra.Command{
|
|
Use: "debug",
|
|
Short: "debug and testing commands",
|
|
Long: "debug contains extra commands used for debugging and testing headscale",
|
|
}
|
|
|
|
var createNodeCmd = &cobra.Command{
|
|
Use: "create-node",
|
|
Short: "Create a node (machine) that can be registered with `nodes register <>` command",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
output, _ := cmd.Flags().GetString("output")
|
|
|
|
namespace, err := cmd.Flags().GetString("namespace")
|
|
if err != nil {
|
|
ErrorOutput(err, fmt.Sprintf("Error getting namespace: %s", err), output)
|
|
return
|
|
}
|
|
|
|
ctx, client, conn, cancel := getHeadscaleCLIClient()
|
|
defer cancel()
|
|
defer conn.Close()
|
|
|
|
name, err := cmd.Flags().GetString("name")
|
|
if err != nil {
|
|
ErrorOutput(err, fmt.Sprintf("Error getting node from flag: %s", err), output)
|
|
return
|
|
}
|
|
|
|
machineKey, err := cmd.Flags().GetString("key")
|
|
if err != nil {
|
|
ErrorOutput(err, fmt.Sprintf("Error getting key from flag: %s", err), output)
|
|
return
|
|
}
|
|
|
|
routes, err := cmd.Flags().GetStringSlice("route")
|
|
if err != nil {
|
|
ErrorOutput(err, fmt.Sprintf("Error getting routes from flag: %s", err), output)
|
|
return
|
|
}
|
|
|
|
request := &v1.DebugCreateMachineRequest{
|
|
Key: machineKey,
|
|
Name: name,
|
|
Namespace: namespace,
|
|
Routes: routes,
|
|
}
|
|
|
|
response, err := client.DebugCreateMachine(ctx, request)
|
|
if err != nil {
|
|
ErrorOutput(err, fmt.Sprintf("Cannot create machine: %s", err), output)
|
|
return
|
|
}
|
|
|
|
SuccessOutput(response.Machine, "Machine created", output)
|
|
},
|
|
}
|