GODT-35: New pmapi client and manager using resty

pkg/pmapi/client_keys.go

@@ -0,0 +1,70 @@
+package pmapi
+
+import (
+	"context"
+
+	"github.com/pkg/errors"
+)
+
+// Unlock unlocks all the user and address keys using the given passphrase, creating user and address keyrings.
+// If the keyrings are already present, they are not recreated.
+func (c *client) Unlock(ctx context.Context, passphrase []byte) (err error) {
+	c.keyRingLock.Lock()
+	defer c.keyRingLock.Unlock()
+
+	// FIXME(conman): Should this be done as part of NewClient somehow?
+
+	return c.unlock(ctx, passphrase)
+}
+
+// unlock unlocks the user's keys but without locking the keyring lock first.
+// Should only be used internally by methods that first lock the lock.
+func (c *client) unlock(ctx context.Context, passphrase []byte) (err error) {
+	if _, err = c.CurrentUser(ctx); err != nil {
+		return
+	}
+
+	if c.userKeyRing == nil {
+		if err = c.unlockUser(passphrase); err != nil {
+			return errors.Wrap(err, "failed to unlock user")
+		}
+	}
+
+	for _, address := range c.addresses {
+		if c.addrKeyRing[address.ID] == nil {
+			if err = c.unlockAddress(passphrase, address); err != nil {
+				return errors.Wrap(err, "failed to unlock address")
+			}
+		}
+	}
+
+	return
+}
+
+func (c *client) ReloadKeys(ctx context.Context, passphrase []byte) (err error) {
+	c.keyRingLock.Lock()
+	defer c.keyRingLock.Unlock()
+
+	c.clearKeys()
+
+	return c.unlock(ctx, passphrase)
+}
+
+func (c *client) clearKeys() {
+	if c.userKeyRing != nil {
+		c.userKeyRing.ClearPrivateParams()
+		c.userKeyRing = nil
+	}
+
+	for id, kr := range c.addrKeyRing {
+		if kr != nil {
+			kr.ClearPrivateParams()
+		}
+		delete(c.addrKeyRing, id)
+	}
+}
+
+func (c *client) IsUnlocked() bool {
+	// FIXME(conman): Better way to check? we don't currently check address keys.
+	return c.userKeyRing != nil
+}