restic/cmd/khepri/cmd_key.go

125 lines
2.6 KiB
Go

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 = "%s%-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
}
var current string
if id.Equal(key.ID()) {
current = "*"
} else {
current = " "
}
tab.Rows = append(tab.Rows, []interface{}{current, 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 {
return errors.New("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 delete_key(be backend.Server, key *khepri.Key, id backend.ID) error {
if id.Equal(key.ID()) {
return errors.New("refusing to remove key currently used to access repository")
}
err := be.Remove(backend.Key, id)
if err != nil {
return err
}
fmt.Printf("removed key %v\n", id)
return nil
}
func change_password(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 {
return errors.New("passwords do not match")
}
// add new key
id, err := key.AddKey(be, pw)
if err != nil {
return fmt.Errorf("creating new key failed: %v\n", err)
}
// remove old key
err = be.Remove(backend.Key, key.ID())
if err != nil {
return 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 || (args[0] == "rm" && len(args) != 2) {
return errors.New("usage: key [list|add|rm|change] [ID]")
}
switch args[0] {
case "list":
return list_keys(be, key)
case "add":
return add_key(be, key)
case "rm":
id, err := backend.Find(be, backend.Key, args[1])
if err != nil {
return err
}
return delete_key(be, key, id)
case "change":
return change_password(be, key)
}
return nil
}