Merge pull request #5606 from BitGo/add-common-types-for-tao

feat(abstract-substrate): add common types/methods for TAO to abstract-substrate

modules/abstract-substrate/src/lib/iface.ts

@@ -4,6 +4,7 @@ import {
   TransactionType,
 } from '@bitgo/sdk-core';
 import { BaseTxInfo, DecodedUnsignedTx, TypeRegistry } from '@substrate/txwrapper-core/lib/types';
+import { Args } from '@substrate/txwrapper-core/lib/types/method';
 
 export { HexString } from '@polkadot/util/types';
 
@@ -30,6 +31,8 @@ export enum MethodNames {
    * @see https://polkadot.js.org/docs/substrate/extrinsics/#transferkeepalivedest-multiaddress-value-compactu128
    */
   TransferKeepAlive = 'transferKeepAlive',
+  AddStake = 'addStake',
+  RemoveStake = 'removeStake',
 }
 
 /**
@@ -71,11 +74,23 @@ export interface TransferAllArgs {
   keepAlive: boolean;
 }
 
+export interface AddStakeArgs extends Args {
+  amountStaked: number;
+  hotkey: string;
+  netuid: number;
+}
+
+export interface RemoveStakeArgs extends Args {
+  amountUnstaked: number;
+  hotkey: string;
+  netuid: number;
+}
+
 /**
  * Decoded TxMethod from a transaction hex
  */
 export interface TxMethod {
-  args: TransferArgs | TransferAllArgs;
+  args: TransferArgs | TransferAllArgs | AddStakeArgs | RemoveStakeArgs;
   name: MethodNames;
   pallet: string;
 }

modules/abstract-substrate/src/lib/index.ts

@@ -1,11 +1,12 @@
 import * as Constants from './constants';
 import * as Errors from './errors';
 import * as Interface from './iface';
+import * as Schema from './txnSchema';
 export { Utils, default as utils } from './utils';
 
 export { KeyPair } from './keyPair';
 export { SingletonRegistry } from './singletonRegistry';
 export { Transaction } from './transaction';
 export { TransactionBuilder } from './transactionBuilder';
 export { TransferBuilder } from './transferBuilder';
-export { Constants, Errors, Interface };
+export { Constants, Errors, Interface, Schema };

modules/abstract-substrate/src/lib/txnSchema.ts

@@ -39,3 +39,15 @@ export const TransferTransactionSchema = joi.object({
 export const TransferAllTransactionSchema = joi.object({
   to: addressSchema.required(),
 });
+
+export const StakeTransactionSchema = joi.object({
+  amountStaked: joi.number().required(),
+  hotkey: joi.string().required(),
+  netuid: joi.number().required(),
+});
+
+export const UnstakeTransactionSchema = joi.object({
+  amountUnstaked: joi.number().required(),
+  hotkey: joi.string().required(),
+  netuid: joi.number().required(),
+});