mirror of
https://github.com/juanfont/headscale.git
synced 2024-12-20 19:09:07 +01:00
108 lines
2.4 KiB
Go
108 lines
2.4 KiB
Go
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/hako/durafmt"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var PreauthkeysCmd = &cobra.Command{
|
|
Use: "preauthkeys",
|
|
Short: "Handle the preauthkeys in Headscale",
|
|
}
|
|
|
|
var ListPreAuthKeys = &cobra.Command{
|
|
Use: "list",
|
|
Short: "List the preauthkeys for this namespace",
|
|
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)
|
|
}
|
|
keys, err := h.GetPreAuthKeys(n)
|
|
if strings.HasPrefix(o, "json") {
|
|
JsonOutput(keys, err, o)
|
|
return
|
|
}
|
|
|
|
if err != nil {
|
|
fmt.Printf("Error getting the list of keys: %s\n", err)
|
|
return
|
|
}
|
|
for _, k := range *keys {
|
|
expiration := "-"
|
|
if k.Expiration != nil {
|
|
expiration = k.Expiration.Format("2006-01-02 15:04:05")
|
|
}
|
|
|
|
var reusable string
|
|
if k.Ephemeral {
|
|
reusable = "N/A"
|
|
} else {
|
|
reusable = fmt.Sprintf("%v", k.Reusable)
|
|
}
|
|
|
|
fmt.Printf(
|
|
"key: %s, namespace: %s, reusable: %s, ephemeral: %v, expiration: %s, created_at: %s\n",
|
|
k.Key,
|
|
k.Namespace.Name,
|
|
reusable,
|
|
k.Ephemeral,
|
|
expiration,
|
|
k.CreatedAt.Format("2006-01-02 15:04:05"),
|
|
)
|
|
}
|
|
},
|
|
}
|
|
|
|
var CreatePreAuthKeyCmd = &cobra.Command{
|
|
Use: "create",
|
|
Short: "Creates a new preauthkey in the specified namespace",
|
|
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)
|
|
}
|
|
reusable, _ := cmd.Flags().GetBool("reusable")
|
|
ephemeral, _ := cmd.Flags().GetBool("ephemeral")
|
|
|
|
e, _ := cmd.Flags().GetString("expiration")
|
|
var expiration *time.Time
|
|
if e != "" {
|
|
duration, err := durafmt.ParseStringShort(e)
|
|
if err != nil {
|
|
log.Fatalf("Error parsing expiration: %s", err)
|
|
}
|
|
exp := time.Now().UTC().Add(duration.Duration())
|
|
expiration = &exp
|
|
}
|
|
|
|
k, err := h.CreatePreAuthKey(n, reusable, ephemeral, expiration)
|
|
if strings.HasPrefix(o, "json") {
|
|
JsonOutput(k, err, o)
|
|
return
|
|
}
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
fmt.Printf("Key: %s\n", k.Key)
|
|
},
|
|
}
|