[Sui 19/x] - sui contract manager additions (pyth-network#953)

* getSetDataSourcesPayload, getSetStalePriceThresholdPayload * use parent functions for SuiChain
dmshvetsov · Aug 1, 2023 · c890d35 · c890d35
1 parent ce66a36
commit c890d35
Showing 1 changed file with 15 additions and 40 deletions.
diff --git a/contract_manager/src/chains.ts b/contract_manager/src/chains.ts
@@ -9,6 +9,7 @@ import {
   AptosAuthorizeUpgradeContract,
   toChainId,
   SetDataSources,
+  SetValidPeriod,
   DataSource,
 } from "xc_admin_common";
 import { AptosClient } from "aptos";
@@ -66,6 +67,19 @@ export abstract class Chain extends Storable {
     return new SetDataSources(this.wormholeChainName, datasources).encode();
   }
 
+  /**
+   * Returns the payload for a governance SetStalePriceThreshold instruction for contracts deployed on this chain
+   * @param newValidStalePriceThreshold the new stale price threshold in seconds
+   */
+  generateGovernanceSetStalePriceThreshold(
+    newValidStalePriceThreshold: bigint
+  ): Buffer {
+    return new SetValidPeriod(
+      this.wormholeChainName,
+      newValidStalePriceThreshold
+    ).encode();
+  }
+
   /**
    * Returns the payload for a governance contract upgrade instruction for contracts deployed on this chain
    * @param upgradeInfo based on the contract type, this can be a contract address, codeId, package digest, etc.
@@ -192,54 +206,15 @@ export class SuiChain extends Chain {
     return SuiChain.type;
   }
 
-  //TODO: Move this logic to xc_admin_common
-  private wrapWithWormholeGovernancePayload(
-    actionVariant: number,
-    payload: Buffer
-  ): Buffer {
-    const actionVariantBuffer = Buffer.alloc(1);
-    actionVariantBuffer.writeUint8(actionVariant, 0);
-    const chainBuffer = Buffer.alloc(2);
-    chainBuffer.writeUint16BE(CHAINS["sui"], 0);
-    const result = Buffer.concat([
-      Buffer.from(
-        "0000000000000000000000000000000000000000000000000000000000000001",
-        "hex"
-      ),
-      actionVariantBuffer,
-      chainBuffer,
-      payload,
-    ]);
-    return result;
-  }
-
   /**
    * Returns the payload for a governance contract upgrade instruction for contracts deployed on this chain
    * @param digest hex string of the 32 byte digest for the new package without the 0x prefix
    */
   generateGovernanceUpgradePayload(digest: string): Buffer {
-    const upgrade = new SuiAuthorizeUpgradeContract(
+    return new SuiAuthorizeUpgradeContract(
       this.wormholeChainName,
       digest
     ).encode();
-    return this.wrapWithWormholeGovernancePayload(0, upgrade);
-  }
-
-  generateGovernanceSetFeePayload(fee: number, exponent: number): Buffer {
-    const setFee = new SetFee(
-      this.wormholeChainName,
-      BigInt(fee),
-      BigInt(exponent)
-    ).encode();
-    return this.wrapWithWormholeGovernancePayload(3, setFee);
-  }
-
-  generateGovernanceSetDataSources(datasources: DataSource[]): Buffer {
-    const setDataSource = new SetDataSources(
-      this.wormholeChainName,
-      datasources
-    ).encode();
-    return this.wrapWithWormholeGovernancePayload(2, setDataSource);
   }
 }