@human-protocol/sdk 1.0.0 → 1.0.2

package/src/types.ts

@@ -2,6 +2,7 @@ import {
   Escrow,
   EscrowFactory,
   HMToken,
+  Staking,
 } from '@human-protocol/core/typechain-types';
 import { ethers } from 'ethers';
 
@@ -567,6 +568,14 @@ export type ContractData = {
    * HMToken contract instance
    */
   hmToken?: HMToken;
+  /**
+   * Staking contract address
+   */
+  stakingAddr?: string;
+  /**
+   * Staking contract instance
+   */
+  staking?: Staking;
 };
 
 /**
@@ -627,6 +636,10 @@ export type JobArguments = {
   /**
    * Factory contract address
    */
+  stakingAddr?: string;
+  /**
+   * Staking contract address
+   */
   factoryAddr?: string;
   /**
    * Escrow contract address

package/src/utils.ts

@@ -7,6 +7,10 @@ import {
   EscrowFactory__factory,
   HMToken,
   HMToken__factory,
+  Staking,
+  Staking__factory,
+  RewardPool,
+  RewardPool__factory,
 } from '@human-protocol/core/typechain-types';
 
 /**
@@ -31,16 +35,18 @@ export const getHmToken = async (
  * **Deploy EscrowFactory contract**
  *
  * @param {string} hmTokenAddr HMToken address
+ * @param {string} stakingAddr Staking address
  * @param {ethers.Signer | undefined} signer Deployer signer
  * @returns {Promise<EscrowFactory>} Deployed contract instance
  */
 export const deployEscrowFactory = async (
   hmTokenAddr: string,
+  stakingAddr: string,
   signer?: ethers.Signer
 ): Promise<EscrowFactory> => {
   const factory = new EscrowFactory__factory(signer);
 
-  const contract = await factory.deploy(hmTokenAddr);
+  const contract = await factory.deploy(hmTokenAddr, stakingAddr);
 
   return contract;
 };
@@ -81,6 +87,66 @@ export const getEscrow = async (
   return contract;
 };
 
+/**
+ * **Deploy Staking contract**
+ *
+ * @param {string} hmTokenAddr HMToken address
+ * @param {number} minimumStake Minimum amount to stake
+ * @param {number} lockPeriod Lock period after unstake
+ * @param {ethers.Signer | undefined} signer Deployer signer
+ * @returns {Promise<Staking>} Deployed contract instance
+ */
+export const deployStaking = async (
+  hmTokenAddr: string,
+  minimumStake: number,
+  lockPeriod: number,
+  signer?: ethers.Signer
+): Promise<Staking> => {
+  const staking = new Staking__factory(signer);
+  const contract = await staking.deploy(hmTokenAddr, minimumStake, lockPeriod);
+
+  return contract;
+};
+
+/**
+ * **Get Staking contract instance at given address**
+ *
+ * @param {string} stakingAddr Staking contract address
+ * @param {ethers.Signer | undefined} signer Deployer signer
+ * @returns {Promise<Staking>} Attached contract instance
+ */
+export const getStaking = async (
+  stakingAddr: string,
+  signer?: ethers.Signer
+): Promise<Staking> => {
+  const factory = new Staking__factory(signer);
+
+  const contract = await factory.attach(stakingAddr);
+
+  return contract;
+};
+
+/**
+ * **Deploy RewardPool contract**
+ *
+ * @param {string} hmTokenAddr HMToken address
+ * @param {string} stakingAddr Staking address
+ * @param {number} fee Reward fee of the protocol
+ * @param {ethers.Signer | undefined} signer Deployer signer
+ * @returns {Promise<Staking>} Deployed contract instance
+ */
+export const deployRewardPool = async (
+  hmTokenAddr: string,
+  stakingAddr: string,
+  fee: number,
+  signer?: ethers.Signer
+): Promise<RewardPool> => {
+  const rewardPool = new RewardPool__factory(signer);
+  const contract = await rewardPool.deploy(hmTokenAddr, stakingAddr, fee);
+
+  return contract;
+};
+
 /**
  * **Get specific amount representation in given decimals**
  *