Something went wrong. Try again.
Monorepo for Tangled
Something went wrong. Try again.
3.6 kB · 143 lines
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144package keyfetch
import ( "context" "encoding/json" "fmt" "io" "net/http" "os" "strings"
"github.com/urfave/cli/v3" "golang.org/x/crypto/ssh" "tangled.org/core/log")
func Command() *cli.Command { return &cli.Command{ Name: "keys", Usage: "fetch public keys from the knot server", Action: Run, Flags: []cli.Flag{ &cli.StringFlag{ Name: "output", Aliases: []string{"o"}, Usage: "output format (table, json, authorized-keys)", Value: "table", }, &cli.StringFlag{ Name: "internal-api", Usage: "internal API endpoint", Value: "http://localhost:5444", }, &cli.StringFlag{ Name: "git-dir", Usage: "base directory for git repos", Value: "/home/git", }, &cli.StringFlag{ Name: "log-path", Usage: "path to log file", Value: "/home/git/log", }, &cli.StringFlag{ Name: "guard-path", Usage: "path to the knot binary for the authorized_keys forced command (defaults to os.Executable)", }, &cli.BoolFlag{ Name: "secure-mode", Usage: "emit -secure-mode in the authorized_keys forced command", }, }, }}
func Run(ctx context.Context, cmd *cli.Command) error { l := log.FromContext(ctx)
internalApi := cmd.String("internal-api") gitDir := cmd.String("git-dir") logPath := cmd.String("log-path") output := cmd.String("output")
executablePath := cmd.String("guard-path") if executablePath == "" { var err error executablePath, err = os.Executable() if err != nil { l.Error("error getting path of executable", "error", err) return err } }
resp, err := http.Get(internalApi + "/keys") if err != nil { l.Error("error reaching internal API endpoint; is the knot server running?", "error", err) return err } defer resp.Body.Close()
body, err := io.ReadAll(resp.Body) if err != nil { l.Error("error reading response body", "error", err) return err }
var data []map[string]any err = json.Unmarshal(body, &data) if err != nil { l.Error("error unmarshalling response body", "error", err) return err }
switch output { case "json": prettyJSON, err := json.MarshalIndent(data, "", " ") if err != nil { l.Error("error pretty printing JSON", "error", err) return err }
if _, err := os.Stdout.Write(prettyJSON); err != nil { l.Error("error writing to stdout", "error", err) return err } case "authorized-keys": formatted := formatKeyData(executablePath, gitDir, logPath, internalApi, cmd.Bool("secure-mode"), data) _, err := os.Stdout.Write([]byte(formatted)) if err != nil { l.Error("error writing to stdout", "error", err) return err } case "table": fmt.Printf("%-40s %-40s\n", "DID", "KEY") fmt.Println(strings.Repeat("-", 80))
for _, entry := range data { did, _ := entry["did"].(string) key, _ := entry["key"].(string) fmt.Printf("%-40s %-40s\n", did, key) } } return nil}
func formatKeyData(executablePath, gitDir, logPath, endpoint string, secureMode bool, data []map[string]any) string { secureFlag := "" if secureMode { secureFlag = " -secure-mode" } var result string for _, entry := range data { raw, _ := entry["key"].(string) key, _, _, _, err := ssh.ParseAuthorizedKey([]byte(raw)) if err != nil { continue } result += fmt.Sprintf( `command="%s guard -git-dir %s -user %s -log-path %s -internal-api %s%s",no-port-forwarding,no-X11-forwarding,no-agent-forwarding,no-pty %s`+"\n", executablePath, gitDir, entry["did"], logPath, endpoint, secureFlag, ssh.MarshalAuthorizedKey(key)) } return result}