147 lines
2.9 KiB
Go
147 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/restic/restic"
|
|
"github.com/restic/restic/backend"
|
|
)
|
|
|
|
type CmdKey struct{}
|
|
|
|
func init() {
|
|
_, err := parser.AddCommand("key",
|
|
"manage keys",
|
|
"The key command manages keys (passwords) of a repository",
|
|
&CmdKey{})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func listKeys(s restic.Server) error {
|
|
tab := NewTable()
|
|
tab.Header = fmt.Sprintf(" %-10s %-10s %-10s %s", "ID", "User", "Host", "Created")
|
|
tab.RowFormat = "%s%-10s %-10s %-10s %s"
|
|
|
|
plen, err := s.PrefixLength(backend.Key)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
done := make(chan struct{})
|
|
defer close(done)
|
|
|
|
for name := range s.List(backend.Key, done) {
|
|
k, err := restic.LoadKey(s, name)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "LoadKey() failed: %v\n", err)
|
|
continue
|
|
}
|
|
|
|
var current string
|
|
if name == s.Key().Name() {
|
|
current = "*"
|
|
} else {
|
|
current = " "
|
|
}
|
|
tab.Rows = append(tab.Rows, []interface{}{current, name[:plen],
|
|
k.Username, k.Hostname, k.Created.Format(TimeFormat)})
|
|
}
|
|
|
|
tab.Write(os.Stdout)
|
|
|
|
return nil
|
|
}
|
|
|
|
func addKey(s restic.Server) error {
|
|
pw := readPassword("RESTIC_NEWPASSWORD", "enter password for new key: ")
|
|
pw2 := readPassword("RESTIC_NEWPASSWORD", "enter password again: ")
|
|
|
|
if pw != pw2 {
|
|
return errors.New("passwords do not match")
|
|
}
|
|
|
|
id, err := restic.AddKey(s, pw, s.Key())
|
|
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 deleteKey(s restic.Server, name string) error {
|
|
if name == s.Key().Name() {
|
|
return errors.New("refusing to remove key currently used to access repository")
|
|
}
|
|
|
|
err := s.Remove(backend.Key, name)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Printf("removed key %v\n", name)
|
|
return nil
|
|
}
|
|
|
|
func changePassword(s restic.Server) error {
|
|
pw := readPassword("RESTIC_NEWPASSWORD", "enter password for new key: ")
|
|
pw2 := readPassword("RESTIC_NEWPASSWORD", "enter password again: ")
|
|
|
|
if pw != pw2 {
|
|
return errors.New("passwords do not match")
|
|
}
|
|
|
|
// add new key
|
|
id, err := restic.AddKey(s, pw, s.Key())
|
|
if err != nil {
|
|
return fmt.Errorf("creating new key failed: %v\n", err)
|
|
}
|
|
|
|
// remove old key
|
|
err = s.Remove(backend.Key, s.Key().Name())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Printf("saved new key as %s\n", id)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (cmd CmdKey) Usage() string {
|
|
return "[list|add|rm|change] [ID]"
|
|
}
|
|
|
|
func (cmd CmdKey) Execute(args []string) error {
|
|
if len(args) < 1 || (args[0] == "rm" && len(args) != 2) {
|
|
return fmt.Errorf("wrong number of arguments, Usage: %s", cmd.Usage())
|
|
}
|
|
|
|
s, err := OpenRepo()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
switch args[0] {
|
|
case "list":
|
|
return listKeys(s)
|
|
case "add":
|
|
return addKey(s)
|
|
case "rm":
|
|
id, err := backend.Find(s, backend.Key, args[1])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return deleteKey(s, id)
|
|
case "change":
|
|
return changePassword(s)
|
|
}
|
|
|
|
return nil
|
|
}
|