package cli import ( "fmt" "log" "strings" "github.com/spf13/cobra" ) func init() { rootCmd.AddCommand(routesCmd) routesCmd.PersistentFlags().StringP("namespace", "n", "", "Namespace") err := routesCmd.MarkPersistentFlagRequired("namespace") if err != nil { log.Fatalf(err.Error()) } routesCmd.AddCommand(listRoutesCmd) routesCmd.AddCommand(enableRouteCmd) } var routesCmd = &cobra.Command{ Use: "routes", Short: "Manage the routes of Headscale", } var listRoutesCmd = &cobra.Command{ Use: "list NODE", Short: "List the routes exposed by this node", Args: func(cmd *cobra.Command, args []string) error { if len(args) < 1 { return fmt.Errorf("Missing parameters") } return nil }, Run: func(cmd *cobra.Command, args []string) { n, err := cmd.Flags().GetString("namespace") if err != nil { log.Fatalf("Error getting namespace: %s", err) } o, _ := cmd.Flags().GetString("output") h, err := getHeadscaleApp() if err != nil { log.Fatalf("Error initializing: %s", err) } routes, err := h.GetNodeRoutes(n, args[0]) if strings.HasPrefix(o, "json") { JsonOutput(routes, err, o) return } if err != nil { fmt.Println(err) return } fmt.Println(routes) }, } var enableRouteCmd = &cobra.Command{ Use: "enable node-name route", Short: "Allows exposing a route declared by this node to the rest of the nodes", Args: func(cmd *cobra.Command, args []string) error { if len(args) < 2 { return fmt.Errorf("Missing parameters") } return nil }, Run: func(cmd *cobra.Command, args []string) { n, err := cmd.Flags().GetString("namespace") if err != nil { log.Fatalf("Error getting namespace: %s", err) } o, _ := cmd.Flags().GetString("output") h, err := getHeadscaleApp() if err != nil { log.Fatalf("Error initializing: %s", err) } route, err := h.EnableNodeRoute(n, args[0], args[1]) if strings.HasPrefix(o, "json") { JsonOutput(route, err, o) return } if err != nil { fmt.Println(err) return } fmt.Printf("Enabled route %s\n", route) }, }