Add UI master key setup and change flows

appstate/state.go

@@ -32,6 +32,11 @@ type LockableSession interface {
 	Unlock(vault.MasterKey) error
 }
 
+type MasterKeyChangeableSession interface {
+	CurrentSession
+	ChangeMasterKey(vault.MasterKey) error
+}
+
 type SaveableSession interface {
 	CurrentSession
 	Save() error
@@ -419,6 +424,20 @@ func (s *State) Unlock(key vault.MasterKey) error {
 	return session.Unlock(key)
 }
 
+func (s *State) ChangeMasterKey(key vault.MasterKey) error {
+	session, ok := s.Session.(MasterKeyChangeableSession)
+	if !ok {
+		return fmt.Errorf("session does not support master key changes")
+	}
+
+	if err := session.ChangeMasterKey(key); err != nil {
+		return err
+	}
+
+	s.Dirty = true
+	return nil
+}
+
 func (s *State) EnterGroup(name string) {
 	s.CurrentPath = append(append([]string(nil), s.CurrentPath...), name)
 	s.SelectedEntryID = ""

appstate/state_test.go

@@ -696,6 +696,25 @@ func TestUnlockRestoresVaultVisibility(t *testing.T) {
 	}
 }
 
+func TestChangeMasterKeyMarksStateDirty(t *testing.T) {
+	t.Parallel()
+
+	sess := &lifecycleStubSession{}
+	state := State{Session: sess}
+
+	key := vault.MasterKey{Password: "correct horse battery staple"}
+	if err := state.ChangeMasterKey(key); err != nil {
+		t.Fatalf("ChangeMasterKey() error = %v", err)
+	}
+
+	if got := sess.changedKey; got.Password != key.Password {
+		t.Fatalf("changedKey = %#v, want %#v", got, key)
+	}
+	if !state.Dirty {
+		t.Fatal("Dirty = false, want true after ChangeMasterKey")
+	}
+}
+
 func TestEnterGroupAppendsPathAndClearsSelection(t *testing.T) {
 	t.Parallel()
 
@@ -929,6 +948,7 @@ type lifecycleStubSession struct {
 	openPath    string
 	saveAsPath  string
 	remotePath  string
+	changedKey  vault.MasterKey
 }
 
 func (s *lifecycleStubSession) Current() (vault.Model, error) {
@@ -954,3 +974,8 @@ func (s *lifecycleStubSession) OpenRemote(_ webdav.Client, path string, _ vault.
 	s.remotePath = path
 	return nil
 }
+
+func (s *lifecycleStubSession) ChangeMasterKey(key vault.MasterKey) error {
+	s.changedKey = key
+	return nil
+}