Merge pull request #1064 from threefoldtech/development_stash_account_methods

add stash account functions

Author: rawdaGastan

clients/tfchain-client-go/twin.go

@@ -182,3 +182,57 @@ func (s *Substrate) UpdateTwin(identity Identity, relay string, pk []byte) (uint
 
 	return s.GetTwinByPubKey(identity.PublicKey())
 }
+
+// BondTwinAccount bonds a twin with a stash/savings account
+// The stash account should initiate the bond between the twin and the account address
+func (s *Substrate) BondTwinAccount(identity Identity, twinID uint32) error {
+	cl, meta, err := s.GetClient()
+	if err != nil {
+		return err
+	}
+
+	c, err := types.NewCall(meta, "TfgridModule.bond_twin_account", twinID)
+	if err != nil {
+		return errors.Wrap(err, "failed to create call")
+	}
+
+	if _, err := s.Call(cl, meta, identity, c); err != nil {
+		return errors.Wrap(err, "failed to bond twin account")
+	}
+
+	return nil
+}
+
+// GetTwinBondedAccount gets the bonded/stash account for a twin
+func (s *Substrate) GetTwinBondedAccount(twinID uint32) (*AccountID, error) {
+	cl, meta, err := s.GetClient()
+	if err != nil {
+		return nil, err
+	}
+
+	bytes, err := Encode(twinID)
+	if err != nil {
+		return nil, errors.Wrap(err, "substrate: encoding error building query arguments")
+	}
+
+	key, err := types.CreateStorageKey(meta, "TfgridModule", "TwinBoundedAccountID", bytes, nil)
+	if err != nil {
+		return nil, errors.Wrap(err, "failed to create substrate query key")
+	}
+
+	raw, err := cl.RPC.State.GetStorageRawLatest(key)
+	if err != nil {
+		return nil, errors.Wrap(err, "failed to lookup twin bonded account")
+	}
+
+	if len(*raw) == 0 {
+		return nil, nil // No bonded account found
+	}
+
+	var accountID AccountID
+	if err := Decode(*raw, &accountID); err != nil {
+		return nil, errors.Wrap(err, "failed to decode account")
+	}
+
+	return &accountID, nil
+}

clients/tfchain-client-go/twin_test.go

@@ -23,3 +23,37 @@ func TestTwin(t *testing.T) {
 
 	require.Equal(t, uint32(twin.ID), id)
 }
+
+func TestBondTwinAccount(t *testing.T) {
+	cl := startLocalConnection(t)
+	defer cl.Close()
+
+	twinID := assertCreateTwin(t, cl, AccountBob)
+
+	// Get Alice Stash identity to act as the bonded account
+	stashUser := Accounts[AccountAliceStash]
+	stashIdentity, err := NewIdentityFromSr25519Phrase(stashUser.Phrase)
+	require.NoError(t, err)
+
+	err = cl.BondTwinAccount(stashIdentity, twinID)
+	require.NoError(t, err)
+
+	bondedAccount, err := cl.GetTwinBondedAccount(twinID)
+	require.NoError(t, err)
+	require.NotNil(t, bondedAccount)
+
+	stashAccount, err := FromAddress(stashUser.Address)
+	require.NoError(t, err)
+	require.Equal(t, stashAccount.PublicKey(), bondedAccount.PublicKey())
+}
+
+func TestGetTwinBondedAccount_NotBonded(t *testing.T) {
+	cl := startLocalConnection(t)
+	defer cl.Close()
+
+	twinID := assertCreateTwin(t, cl, AccountAlice)
+
+	bondedAccount, err := cl.GetTwinBondedAccount(twinID)
+	require.NoError(t, err)
+	require.Nil(t, bondedAccount)
+}