@dorafactory/maci-sdk 0.1.2 → 0.1.3-pre.10

package/src/libs/contract/contract.ts

@@ -2,6 +2,8 @@ import { OfflineSigner } from '@cosmjs/proto-signing';
 import { ContractParams } from '../../types';
 import {
   createAMaciClientBy,
+  createApiMaciClientBy,
+  createApiSaasClientBy,
   createContractClientByWallet,
   createMaciClientBy,
   createOracleMaciClientBy,
@@ -13,6 +15,7 @@ import {
   CreateMaciRoundParams,
   CreateOracleMaciRoundParams,
   CreateSaasOracleMaciRoundParams,
+  CreateApiSaasAmaciRoundParams,
 } from './types';
 import { getAMaciRoundCircuitFee, getContractParams } from './utils';
 import { QTR_LIB } from './vars';
@@ -27,9 +30,9 @@ export class Contract {
   public rpcEndpoint: string;
   public registryAddress: string;
   public saasAddress: string;
+  public apiSaasAddress: string;
   public maciCodeId: number;
   public oracleCodeId: number;
-  public saasOracleCodeId: number;
   public feegrantOperator: string;
   public whitelistBackendPubkey: string;
 
@@ -38,9 +41,9 @@ export class Contract {
     rpcEndpoint,
     registryAddress,
     saasAddress,
+    apiSaasAddress,
     maciCodeId,
     oracleCodeId,
-    saasOracleCodeId,
     feegrantOperator,
     whitelistBackendPubkey,
   }: ContractParams) {
@@ -48,9 +51,9 @@ export class Contract {
     this.rpcEndpoint = rpcEndpoint;
     this.registryAddress = registryAddress;
     this.saasAddress = saasAddress;
+    this.apiSaasAddress = apiSaasAddress;
     this.maciCodeId = maciCodeId;
     this.oracleCodeId = oracleCodeId;
-    this.saasOracleCodeId = saasOracleCodeId;
     this.feegrantOperator = feegrantOperator;
     this.whitelistBackendPubkey = whitelistBackendPubkey;
   }
@@ -65,10 +68,11 @@ export class Contract {
     description,
     link,
     maxVoter,
-    maxOption,
+    voteOptionMap,
     voiceCreditAmount,
     circuitType,
     preDeactivateRoot,
+    oracleWhitelistPubkey,
     fee = 'auto',
   }: CreateAMaciRoundParams & { signer: OfflineSigner }) {
     const start_time = (startVoting.getTime() * 10 ** 6).toString();
@@ -82,7 +86,7 @@ export class Contract {
     const requiredFee = getAMaciRoundCircuitFee(
       this.network,
       maxVoter,
-      maxOption
+      voteOptionMap.length
     );
 
     preDeactivateRoot = preDeactivateRoot || '0';
@@ -102,9 +106,10 @@ export class Contract {
           end_time,
         },
         maxVoter: maxVoter.toString(),
-        maxOption: maxOption.toString(),
+        voteOptionMap,
         certificationSystem: '0',
         circuitType,
+        oracleWhitelistPubkey,
       },
       fee,
       undefined,
@@ -1005,6 +1010,20 @@ export class Contract {
     });
   }
 
+  async apiMaciClient({
+    signer,
+    contractAddress,
+  }: {
+    signer: OfflineSigner;
+    contractAddress: string;
+  }) {
+    return createApiMaciClientBy({
+      rpcEndpoint: this.rpcEndpoint,
+      wallet: signer,
+      contractAddress,
+    });
+  }
+
   async saasClient({
     signer,
     contractAddress,
@@ -1019,7 +1038,593 @@ export class Contract {
     });
   }
 
+  async apiSaasClient({
+    signer,
+    contractAddress,
+  }: {
+    signer: OfflineSigner;
+    contractAddress: string;
+  }) {
+    return createApiSaasClientBy({
+      rpcEndpoint: this.rpcEndpoint,
+      wallet: signer,
+      contractAddress,
+    });
+  }
+
   async contractClient({ signer }: { signer: OfflineSigner }) {
     return createContractClientByWallet(this.rpcEndpoint, signer);
   }
+
+  async createApiSaasMaciRound({
+    signer,
+    operatorPubkey,
+    startVoting,
+    endVoting,
+    title,
+    description,
+    link,
+    maxVoter,
+    voteOptionMap,
+    whitelistBackendPubkey,
+    gasStation = false,
+    fee = 1.8,
+  }: CreateSaasOracleMaciRoundParams & { signer: OfflineSigner }) {
+    const startTime = (startVoting.getTime() * 1_000_000).toString();
+    const endTime = (endVoting.getTime() * 1_000_000).toString();
+
+    const client = await createApiSaasClientBy({
+      rpcEndpoint: this.rpcEndpoint,
+      wallet: signer,
+      contractAddress: this.apiSaasAddress,
+    });
+    const [operatorPubkeyX, operatorPubkeyY] = unpackPubKey(
+      BigInt(operatorPubkey)
+    );
+
+    const roundParams = {
+      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 createResponse;
+
+    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 = {
+        create_api_maci_round: {
+          certification_system: '0',
+          circuit_type: '0',
+          coordinator: roundParams.coordinator,
+          max_voters: roundParams.maxVoters.toString(),
+          round_info: roundParams.roundInfo,
+          start_time: roundParams.startTime,
+          end_time: roundParams.endTime,
+          vote_option_map: roundParams.voteOptionMap,
+          whitelist_backend_pubkey: roundParams.whitelistBackendPubkey,
+        },
+      };
+      const gasEstimation = await contractClient.simulate(
+        address,
+        [
+          {
+            typeUrl: '/cosmwasm.wasm.v1.MsgExecuteContract',
+            value: {
+              sender: address,
+              contract: this.apiSaasAddress,
+              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.apiSaasAddress,
+      };
+      createResponse = await client.createMaciRound(roundParams, grantFee);
+    } else if (gasStation && typeof fee === 'object') {
+      // When gasStation is true and fee is StdFee, add granter
+      const grantFee: StdFee = {
+        ...fee,
+        granter: this.apiSaasAddress,
+      };
+      createResponse = await client.createMaciRound(roundParams, grantFee);
+    } else {
+      createResponse = await client.createMaciRound(roundParams, fee);
+    }
+
+    let contractAddress = '';
+    createResponse.events.map((event) => {
+      if (event.type === 'wasm') {
+        let actionEvent = event.attributes.find(
+          (attr) => attr.key === 'action'
+        )!;
+        if (actionEvent.value === 'created_maci_round') {
+          contractAddress = event.attributes
+            .find((attr) => attr.key === 'round_addr')!
+            .value.toString();
+        }
+      }
+    });
+    return {
+      ...createResponse,
+      contractAddress,
+    };
+  }
+
+  async setApiSaasMaciRoundInfo({
+    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 createApiSaasClientBy({
+      rpcEndpoint: this.rpcEndpoint,
+      wallet: signer,
+      contractAddress: this.apiSaasAddress,
+    });
+
+    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.apiSaasAddress,
+              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.apiSaasAddress,
+      };
+      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.apiSaasAddress,
+      };
+      return client.setRoundInfo(
+        {
+          contractAddr: contractAddress,
+          roundInfo,
+        },
+        grantFee
+      );
+    }
+
+    return client.setRoundInfo(
+      {
+        contractAddr: contractAddress,
+        roundInfo,
+      },
+      fee
+    );
+  }
+
+  async setApiSaasMaciRoundVoteOptions({
+    signer,
+    contractAddress,
+    voteOptionMap,
+    gasStation = false,
+    fee = 1.8,
+  }: {
+    signer: OfflineSigner;
+    contractAddress: string;
+    voteOptionMap: string[];
+    gasStation?: boolean;
+    fee?: StdFee | 'auto' | number;
+  }) {
+    const client = await createApiSaasClientBy({
+      rpcEndpoint: this.rpcEndpoint,
+      wallet: signer,
+      contractAddress: this.apiSaasAddress,
+    });
+
+    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.apiSaasAddress,
+              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.apiSaasAddress,
+      };
+      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.apiSaasAddress,
+      };
+      return client.setVoteOptionsMap(
+        {
+          contractAddr: contractAddress,
+          voteOptionMap,
+        },
+        grantFee
+      );
+    }
+
+    return client.setVoteOptionsMap(
+      {
+        contractAddr: contractAddress,
+        voteOptionMap,
+      },
+      fee
+    );
+  }
+
+  async addApiSaasOperator({
+    signer,
+    operator,
+    gasStation = false,
+    fee = 1.8,
+  }: {
+    signer: OfflineSigner;
+    operator: string;
+    gasStation?: boolean;
+    fee?: StdFee | 'auto' | number;
+  }) {
+    const client = await createApiSaasClientBy({
+      rpcEndpoint: this.rpcEndpoint,
+      wallet: signer,
+      contractAddress: this.apiSaasAddress,
+    });
+
+    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.apiSaasAddress,
+              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.apiSaasAddress,
+      };
+      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.apiSaasAddress,
+      };
+      return client.addOperator({ operator }, grantFee);
+    }
+
+    return client.addOperator({ operator }, fee);
+  }
+
+  async removeApiSaasOperator({
+    signer,
+    operator,
+    gasStation = false,
+    fee = 1.8,
+  }: {
+    signer: OfflineSigner;
+    operator: string;
+    gasStation?: boolean;
+    fee?: StdFee | 'auto' | number;
+  }) {
+    const client = await createApiSaasClientBy({
+      rpcEndpoint: this.rpcEndpoint,
+      wallet: signer,
+      contractAddress: this.apiSaasAddress,
+    });
+
+    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.apiSaasAddress,
+              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.apiSaasAddress,
+      };
+      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.apiSaasAddress,
+      };
+      return client.removeOperator({ operator }, grantFee);
+    }
+
+    return client.removeOperator({ operator }, fee);
+  }
+
+  async isApiSaasOperator({
+    signer,
+    operator,
+  }: {
+    signer: OfflineSigner;
+    operator: string;
+  }) {
+    const client = await createApiSaasClientBy({
+      rpcEndpoint: this.rpcEndpoint,
+      wallet: signer,
+      contractAddress: this.apiSaasAddress,
+    });
+    return client.isOperator({ address: operator });
+  }
+
+  async createApiSaasAmaciRound({
+    signer,
+    operator,
+    startVoting,
+    endVoting,
+    title,
+    description,
+    link,
+    maxVoter,
+    voteOptionMap,
+    whitelist,
+    voiceCreditAmount,
+    preDeactivateRoot,
+    oracleWhitelistPubkey,
+    circuitType,
+    gasStation = false,
+    fee = 1.8,
+  }: CreateApiSaasAmaciRoundParams & { signer: OfflineSigner }) {
+    const startTime = (startVoting.getTime() * 1_000_000).toString();
+    const endTime = (endVoting.getTime() * 1_000_000).toString();
+
+    const client = await createApiSaasClientBy({
+      rpcEndpoint: this.rpcEndpoint,
+      wallet: signer,
+      contractAddress: this.apiSaasAddress,
+    });
+
+    const roundParams = {
+      certificationSystem: '0',
+      circuitType: circuitType.toString(),
+      maxVoter: maxVoter.toString(),
+      operator,
+      oracleWhitelistPubkey,
+      preDeactivateRoot: preDeactivateRoot || '0',
+      roundInfo: {
+        title,
+        description: description || '',
+        link: link || '',
+      },
+      voiceCreditAmount,
+      voteOptionMap,
+      votingTime: {
+        start_time: startTime,
+        end_time: endTime,
+      },
+      whitelist,
+    };
+
+    let createResponse;
+
+    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 = {
+        create_amaci_round: {
+          certification_system: '0',
+          circuit_type: circuitType.toString(),
+          max_voter: maxVoter.toString(),
+          operator,
+          oracle_whitelist_pubkey: oracleWhitelistPubkey,
+          pre_deactivate_root: preDeactivateRoot || '0',
+          round_info: roundParams.roundInfo,
+          voice_credit_amount: voiceCreditAmount,
+          vote_option_map: voteOptionMap,
+          voting_time: roundParams.votingTime,
+          whitelist,
+        },
+      };
+      const gasEstimation = await contractClient.simulate(
+        address,
+        [
+          {
+            typeUrl: '/cosmwasm.wasm.v1.MsgExecuteContract',
+            value: {
+              sender: address,
+              contract: this.apiSaasAddress,
+              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.apiSaasAddress,
+      };
+      createResponse = await client.createAmaciRound(roundParams, grantFee);
+    } else if (gasStation && typeof fee === 'object') {
+      // When gasStation is true and fee is StdFee, add granter
+      const grantFee: StdFee = {
+        ...fee,
+        granter: this.apiSaasAddress,
+      };
+      createResponse = await client.createAmaciRound(roundParams, grantFee);
+    } else {
+      createResponse = await client.createAmaciRound(roundParams, fee);
+    }
+
+    let contractAddress = '';
+    createResponse.events.map((event) => {
+      if (event.type === 'wasm') {
+        let actionEvent = event.attributes.find(
+          (attr) => attr.key === 'action'
+        );
+        if (actionEvent && actionEvent.value === 'created_round') {
+          const roundAddrEvent = event.attributes.find(
+            (attr) => attr.key === 'round_addr'
+          );
+          if (roundAddrEvent) {
+            contractAddress = roundAddrEvent.value.toString();
+          }
+        }
+      }
+    });
+    return {
+      ...createResponse,
+      contractAddress,
+    };
+  }
 }