package main import ( "encoding/json" "errors" "fmt" "os" "github.com/fd0/khepri" "github.com/fd0/khepri/backend" ) func list_keys(be backend.Server, key *khepri.Key) error { tab := NewTable() tab.Header = fmt.Sprintf("%-10s %-10s %-10s %s", "ID", "User", "Host", "Created") tab.RowFormat = "%-10s %-10s %-10s %s" plen, err := backend.PrefixLength(be, backend.Key) if err != nil { return err } backend.Each(be, backend.Key, func(id backend.ID, data []byte, err error) { k := khepri.Key{} err = json.Unmarshal(data, &k) if err != nil { return } tab.Rows = append(tab.Rows, []interface{}{id[:plen], k.Username, k.Hostname, k.Created.Format(TimeFormat)}) }) tab.Print(os.Stdout) return nil } func add_key(be backend.Server, key *khepri.Key) error { pw := readPassword("KHEPRI_NEWPASSWORD", "enter password for new key: ") pw2 := readPassword("KHEPRI_NEWPASSWORD", "enter password again: ") if pw != pw2 { errx(1, "passwords do not match") } id, err := key.AddKey(be, pw) if err != nil { return fmt.Errorf("creating new key failed: %v\n", err) } fmt.Printf("saved new key as %s\n", id) return nil } func commandKey(be backend.Server, key *khepri.Key, args []string) error { if len(args) < 1 { return errors.New("usage: key [list|add]") } switch args[0] { case "list": return list_keys(be, key) case "add": return add_key(be, key) } return nil }