@dorafactory/maci-sdk 0.1.0 → 0.1.2-pre.0

package/src/libs/contract/contract.ts

@@ -6,16 +6,19 @@ import {
   createMaciClientBy,
   createOracleMaciClientBy,
   createRegistryClientBy,
+  createSaasClientBy,
 } from './config';
 import {
   CreateAMaciRoundParams,
   CreateMaciRoundParams,
   CreateOracleMaciRoundParams,
+  CreateSaasOracleMaciRoundParams,
 } from './types';
 import { getAMaciRoundCircuitFee, getContractParams } from './utils';
 import { QTR_LIB } from './vars';
 import { MaciRoundType, MaciCertSystemType } from '../../types';
 import { unpackPubKey } from '../crypto';
+import { StdFee, GasPrice, calculateFee } from '@cosmjs/stargate';
 
 export const prefix = 'dora';
 
@@ -23,8 +26,10 @@ export class Contract {
   public network: 'mainnet' | 'testnet';
   public rpcEndpoint: string;
   public registryAddress: string;
+  public saasAddress: string;
   public maciCodeId: number;
   public oracleCodeId: number;
+  public saasOracleCodeId: number;
   public feegrantOperator: string;
   public whitelistBackendPubkey: string;
 
@@ -32,16 +37,20 @@ export class Contract {
     network,
     rpcEndpoint,
     registryAddress,
+    saasAddress,
     maciCodeId,
     oracleCodeId,
+    saasOracleCodeId,
     feegrantOperator,
     whitelistBackendPubkey,
   }: ContractParams) {
     this.network = network;
     this.rpcEndpoint = rpcEndpoint;
     this.registryAddress = registryAddress;
+    this.saasAddress = saasAddress;
     this.maciCodeId = maciCodeId;
     this.oracleCodeId = oracleCodeId;
+    this.saasOracleCodeId = saasOracleCodeId;
     this.feegrantOperator = feegrantOperator;
     this.whitelistBackendPubkey = whitelistBackendPubkey;
   }
@@ -197,11 +206,14 @@ export class Contract {
     title,
     description,
     link,
+    maxVoter,
     voteOptionMap,
     circuitType,
     whitelistEcosystem,
     whitelistSnapshotHeight,
     whitelistVotingPowerArgs,
+    whitelistBackendPubkey,
+    feegrantOperator,
     fee = 'auto',
   }: CreateOracleMaciRoundParams & { signer: OfflineSigner }) {
     const start_time = (startVoting.getTime() * 1_000_000).toString();
@@ -223,6 +235,7 @@ export class Contract {
       address,
       this.oracleCodeId,
       {
+        max_voters: maxVoter.toString(),
         round_info: { title, description: description || '', link: link || '' },
         voting_time: {
           start_time,
@@ -233,13 +246,17 @@ export class Contract {
           y: operatorPubkeyY.toString(),
         },
         vote_option_map: voteOptionMap,
-        whitelist_backend_pubkey: this.whitelistBackendPubkey,
+        whitelist_backend_pubkey: whitelistBackendPubkey
+          ? whitelistBackendPubkey
+          : this.whitelistBackendPubkey,
         whitelist_ecosystem: whitelistEcosystem,
         whitelist_snapshot_height: whitelistSnapshotHeight,
         whitelist_voting_power_args: whitelistVotingPowerArgs,
         circuit_type: maciVoteType,
         certification_system: maciCertSystem,
-        feegrant_operator: this.feegrantOperator,
+        feegrant_operator: feegrantOperator
+          ? feegrantOperator
+          : this.feegrantOperator,
       },
       `[Oracle MACI] ${title}`,
       fee
@@ -248,6 +265,526 @@ export class Contract {
     return instantiateResponse;
   }
 
+  async createSaasOracleMaciRound({
+    signer,
+    operatorPubkey,
+    startVoting,
+    endVoting,
+    title,
+    description,
+    link,
+    maxVoter,
+    voteOptionMap,
+    whitelistBackendPubkey,
+  }: CreateSaasOracleMaciRoundParams & { signer: OfflineSigner }) {
+    const startTime = (startVoting.getTime() * 1_000_000).toString();
+    const endTime = (endVoting.getTime() * 1_000_000).toString();
+
+    const client = await createSaasClientBy({
+      rpcEndpoint: this.rpcEndpoint,
+      wallet: signer,
+      contractAddress: this.saasAddress,
+    });
+    const [operatorPubkeyX, operatorPubkeyY] = unpackPubKey(
+      BigInt(operatorPubkey)
+    );
+
+    const createResponse = await client.createOracleMaciRound({
+      certificationSystem: '0',
+      circuitType: '0',
+      coordinator: {
+        x: operatorPubkeyX.toString(),
+        y: operatorPubkeyY.toString(),
+      },
+      maxVoters: maxVoter,
+      roundInfo: {
+        title,
+        description: description || '',
+        link: link || '',
+      },
+      startTime,
+      endTime,
+      voteOptionMap,
+      whitelistBackendPubkey:
+        whitelistBackendPubkey || this.whitelistBackendPubkey,
+    });
+
+    let contractAddress = '';
+    createResponse.events.map((event) => {
+      if (event.type === 'wasm') {
+        let actionEvent = event.attributes.find(
+          (attr) => attr.key === 'action'
+        )!;
+        if (actionEvent.value === 'created_oracle_maci_round') {
+          contractAddress = event.attributes
+            .find((attr) => attr.key === 'round_addr')!
+            .value.toString();
+        }
+      }
+    });
+    return {
+      ...createResponse,
+      contractAddress,
+    };
+  }
+
+  async setSaasOracleMaciRoundInfo({
+    signer,
+    contractAddress,
+    title,
+    description,
+    link,
+    gasStation = false,
+    fee = 1.8,
+  }: {
+    signer: OfflineSigner;
+    contractAddress: string;
+    title: string;
+    description: string;
+    link: string;
+    gasStation?: boolean;
+    fee?: StdFee | 'auto' | number;
+  }) {
+    const client = await createSaasClientBy({
+      rpcEndpoint: this.rpcEndpoint,
+      wallet: signer,
+      contractAddress: this.saasAddress,
+    });
+
+    const roundInfo = {
+      title,
+      description,
+      link,
+    };
+
+    if (gasStation && typeof fee !== 'object') {
+      // When gasStation is true and fee is not StdFee, we need to simulate first then add granter
+      const [{ address }] = await signer.getAccounts();
+      const contractClient = await this.contractClient({ signer });
+      const msg = {
+        set_round_info: {
+          contract_addr: contractAddress,
+          round_info: roundInfo,
+        },
+      };
+      const gasEstimation = await contractClient.simulate(
+        address,
+        [
+          {
+            typeUrl: '/cosmwasm.wasm.v1.MsgExecuteContract',
+            value: {
+              sender: address,
+              contract: this.saasAddress,
+              msg: new TextEncoder().encode(JSON.stringify(msg)),
+            },
+          },
+        ],
+        ''
+      );
+      const multiplier = typeof fee === 'number' ? fee : 1.8;
+      const gasPrice = GasPrice.fromString('10000000000peaka');
+      const calculatedFee = calculateFee(
+        Math.round(gasEstimation * multiplier),
+        gasPrice
+      );
+      const grantFee: StdFee = {
+        amount: calculatedFee.amount,
+        gas: calculatedFee.gas,
+        granter: this.saasAddress,
+      };
+      return client.setRoundInfo(
+        {
+          contractAddr: contractAddress,
+          roundInfo,
+        },
+        grantFee
+      );
+    } else if (gasStation && typeof fee === 'object') {
+      // When gasStation is true and fee is StdFee, add granter
+      const grantFee: StdFee = {
+        ...fee,
+        granter: this.saasAddress,
+      };
+      return client.setRoundInfo(
+        {
+          contractAddr: contractAddress,
+          roundInfo,
+        },
+        grantFee
+      );
+    }
+
+    return client.setRoundInfo(
+      {
+        contractAddr: contractAddress,
+        roundInfo,
+      },
+      fee
+    );
+  }
+
+  async setSaasOracleMaciRoundVoteOptions({
+    signer,
+    contractAddress,
+    voteOptionMap,
+    gasStation = false,
+    fee = 1.8,
+  }: {
+    signer: OfflineSigner;
+    contractAddress: string;
+    voteOptionMap: string[];
+    gasStation?: boolean;
+    fee?: StdFee | 'auto' | number;
+  }) {
+    const client = await createSaasClientBy({
+      rpcEndpoint: this.rpcEndpoint,
+      wallet: signer,
+      contractAddress: this.saasAddress,
+    });
+
+    if (gasStation && typeof fee !== 'object') {
+      // When gasStation is true and fee is not StdFee, we need to simulate first then add granter
+      const [{ address }] = await signer.getAccounts();
+      const contractClient = await this.contractClient({ signer });
+      const msg = {
+        set_vote_options_map: {
+          contract_addr: contractAddress,
+          vote_option_map: voteOptionMap,
+        },
+      };
+      const gasEstimation = await contractClient.simulate(
+        address,
+        [
+          {
+            typeUrl: '/cosmwasm.wasm.v1.MsgExecuteContract',
+            value: {
+              sender: address,
+              contract: this.saasAddress,
+              msg: new TextEncoder().encode(JSON.stringify(msg)),
+            },
+          },
+        ],
+        ''
+      );
+      const multiplier = typeof fee === 'number' ? fee : 1.8;
+      const gasPrice = GasPrice.fromString('10000000000peaka');
+      const calculatedFee = calculateFee(
+        Math.round(gasEstimation * multiplier),
+        gasPrice
+      );
+      const grantFee: StdFee = {
+        amount: calculatedFee.amount,
+        gas: calculatedFee.gas,
+        granter: this.saasAddress,
+      };
+      return client.setVoteOptionsMap(
+        {
+          contractAddr: contractAddress,
+          voteOptionMap,
+        },
+        grantFee
+      );
+    } else if (gasStation && typeof fee === 'object') {
+      // When gasStation is true and fee is StdFee, add granter
+      const grantFee: StdFee = {
+        ...fee,
+        granter: this.saasAddress,
+      };
+      return client.setVoteOptionsMap(
+        {
+          contractAddr: contractAddress,
+          voteOptionMap,
+        },
+        grantFee
+      );
+    }
+
+    return client.setVoteOptionsMap(
+      {
+        contractAddr: contractAddress,
+        voteOptionMap,
+      },
+      fee
+    );
+  }
+
+  async addSaasOperator({
+    signer,
+    operator,
+    gasStation = false,
+    fee = 1.8,
+  }: {
+    signer: OfflineSigner;
+    operator: string;
+    gasStation?: boolean;
+    fee?: StdFee | 'auto' | number;
+  }) {
+    const client = await createSaasClientBy({
+      rpcEndpoint: this.rpcEndpoint,
+      wallet: signer,
+      contractAddress: this.saasAddress,
+    });
+
+    if (gasStation && typeof fee !== 'object') {
+      // When gasStation is true and fee is not StdFee, we need to simulate first then add granter
+      const [{ address }] = await signer.getAccounts();
+      const contractClient = await this.contractClient({ signer });
+      const msg = {
+        add_operator: {
+          operator,
+        },
+      };
+      const gasEstimation = await contractClient.simulate(
+        address,
+        [
+          {
+            typeUrl: '/cosmwasm.wasm.v1.MsgExecuteContract',
+            value: {
+              sender: address,
+              contract: this.saasAddress,
+              msg: new TextEncoder().encode(JSON.stringify(msg)),
+            },
+          },
+        ],
+        ''
+      );
+      const multiplier = typeof fee === 'number' ? fee : 1.8;
+      const gasPrice = GasPrice.fromString('10000000000peaka');
+      const calculatedFee = calculateFee(
+        Math.round(gasEstimation * multiplier),
+        gasPrice
+      );
+      const grantFee: StdFee = {
+        amount: calculatedFee.amount,
+        gas: calculatedFee.gas,
+        granter: this.saasAddress,
+      };
+      return client.addOperator({ operator }, grantFee);
+    } else if (gasStation && typeof fee === 'object') {
+      // When gasStation is true and fee is StdFee, add granter
+      const grantFee: StdFee = {
+        ...fee,
+        granter: this.saasAddress,
+      };
+      return client.addOperator({ operator }, grantFee);
+    }
+
+    return client.addOperator({ operator }, fee);
+  }
+
+  async removeSaasOperator({
+    signer,
+    operator,
+    gasStation = false,
+    fee = 1.8,
+  }: {
+    signer: OfflineSigner;
+    operator: string;
+    gasStation?: boolean;
+    fee?: StdFee | 'auto' | number;
+  }) {
+    const client = await createSaasClientBy({
+      rpcEndpoint: this.rpcEndpoint,
+      wallet: signer,
+      contractAddress: this.saasAddress,
+    });
+
+    if (gasStation && typeof fee !== 'object') {
+      // When gasStation is true and fee is not StdFee, we need to simulate first then add granter
+      const [{ address }] = await signer.getAccounts();
+      const contractClient = await this.contractClient({ signer });
+      const msg = {
+        remove_operator: {
+          operator,
+        },
+      };
+      const gasEstimation = await contractClient.simulate(
+        address,
+        [
+          {
+            typeUrl: '/cosmwasm.wasm.v1.MsgExecuteContract',
+            value: {
+              sender: address,
+              contract: this.saasAddress,
+              msg: new TextEncoder().encode(JSON.stringify(msg)),
+            },
+          },
+        ],
+        ''
+      );
+      const multiplier = typeof fee === 'number' ? fee : 1.8;
+      const gasPrice = GasPrice.fromString('10000000000peaka');
+      const calculatedFee = calculateFee(
+        Math.round(gasEstimation * multiplier),
+        gasPrice
+      );
+      const grantFee: StdFee = {
+        amount: calculatedFee.amount,
+        gas: calculatedFee.gas,
+        granter: this.saasAddress,
+      };
+      return client.removeOperator({ operator }, grantFee);
+    } else if (gasStation && typeof fee === 'object') {
+      // When gasStation is true and fee is StdFee, add granter
+      const grantFee: StdFee = {
+        ...fee,
+        granter: this.saasAddress,
+      };
+      return client.removeOperator({ operator }, grantFee);
+    }
+
+    return client.removeOperator({ operator }, fee);
+  }
+
+  async isSaasOperator({
+    signer,
+    operator,
+  }: {
+    signer: OfflineSigner;
+    operator: string;
+  }) {
+    const client = await createSaasClientBy({
+      rpcEndpoint: this.rpcEndpoint,
+      wallet: signer,
+      contractAddress: this.saasAddress,
+    });
+    return client.isOperator({ address: operator });
+  }
+
+  async depositSaas({
+    signer,
+    amount,
+    gasStation = false,
+    fee = 1.8,
+  }: {
+    signer: OfflineSigner;
+    amount: string;
+    gasStation?: boolean;
+    fee?: StdFee | 'auto' | number;
+  }) {
+    const client = await createSaasClientBy({
+      rpcEndpoint: this.rpcEndpoint,
+      wallet: signer,
+      contractAddress: this.saasAddress,
+    });
+
+    const funds = [
+      {
+        denom: 'peaka',
+        amount: amount,
+      },
+    ];
+
+    if (gasStation && typeof fee !== 'object') {
+      // When gasStation is true and fee is not StdFee, we need to simulate first then add granter
+      const [{ address }] = await signer.getAccounts();
+      const contractClient = await this.contractClient({ signer });
+      const msg = {
+        deposit: {},
+      };
+      const gasEstimation = await contractClient.simulate(
+        address,
+        [
+          {
+            typeUrl: '/cosmwasm.wasm.v1.MsgExecuteContract',
+            value: {
+              sender: address,
+              contract: this.saasAddress,
+              msg: new TextEncoder().encode(JSON.stringify(msg)),
+              funds,
+            },
+          },
+        ],
+        ''
+      );
+      const multiplier = typeof fee === 'number' ? fee : 1.8;
+      const gasPrice = GasPrice.fromString('10000000000peaka');
+      const calculatedFee = calculateFee(
+        Math.round(gasEstimation * multiplier),
+        gasPrice
+      );
+      const grantFee: StdFee = {
+        amount: calculatedFee.amount,
+        gas: calculatedFee.gas,
+        granter: this.saasAddress,
+      };
+      return client.deposit(grantFee, undefined, funds);
+    } else if (gasStation && typeof fee === 'object') {
+      // When gasStation is true and fee is StdFee, add granter
+      const grantFee: StdFee = {
+        ...fee,
+        granter: this.saasAddress,
+      };
+      return client.deposit(grantFee, undefined, funds);
+    }
+
+    return client.deposit(fee, undefined, funds);
+  }
+
+  async withdrawSaas({
+    signer,
+    amount,
+    gasStation = false,
+    fee = 1.8,
+  }: {
+    signer: OfflineSigner;
+    amount: string;
+    gasStation?: boolean;
+    fee?: StdFee | 'auto' | number;
+  }) {
+    const client = await createSaasClientBy({
+      rpcEndpoint: this.rpcEndpoint,
+      wallet: signer,
+      contractAddress: this.saasAddress,
+    });
+
+    if (gasStation && typeof fee !== 'object') {
+      // When gasStation is true and fee is not StdFee, we need to simulate first then add granter
+      const [{ address }] = await signer.getAccounts();
+      const contractClient = await this.contractClient({ signer });
+      const msg = {
+        withdraw: {
+          amount,
+        },
+      };
+      const gasEstimation = await contractClient.simulate(
+        address,
+        [
+          {
+            typeUrl: '/cosmwasm.wasm.v1.MsgExecuteContract',
+            value: {
+              sender: address,
+              contract: this.saasAddress,
+              msg: new TextEncoder().encode(JSON.stringify(msg)),
+            },
+          },
+        ],
+        ''
+      );
+      const multiplier = typeof fee === 'number' ? fee : 1.8;
+      const gasPrice = GasPrice.fromString('10000000000peaka');
+      const calculatedFee = calculateFee(
+        Math.round(gasEstimation * multiplier),
+        gasPrice
+      );
+      const grantFee: StdFee = {
+        amount: calculatedFee.amount,
+        gas: calculatedFee.gas,
+        granter: this.saasAddress,
+      };
+      return client.withdraw({ amount }, grantFee);
+    } else if (gasStation && typeof fee === 'object') {
+      // When gasStation is true and fee is StdFee, add granter
+      const grantFee: StdFee = {
+        ...fee,
+        granter: this.saasAddress,
+      };
+      return client.withdraw({ amount }, grantFee);
+    }
+
+    return client.withdraw({ amount }, fee);
+  }
+
   async queryRoundInfo({
     signer,
     roundAddress,
@@ -321,6 +858,20 @@ export class Contract {
     });
   }
 
+  async saasClient({
+    signer,
+    contractAddress,
+  }: {
+    signer: OfflineSigner;
+    contractAddress: string;
+  }) {
+    return createSaasClientBy({
+      rpcEndpoint: this.rpcEndpoint,
+      wallet: signer,
+      contractAddress,
+    });
+  }
+
   async contractClient({ signer }: { signer: OfflineSigner }) {
     return createContractClientByWallet(this.rpcEndpoint, signer);
   }