feat(contract_manager): support near wormhole contract and upgrade payload

Author: Riateche

contract_manager/src/chains.ts

@@ -898,8 +898,12 @@ export class NearChain extends Chain {
     };
   }
 
-  generateGovernanceUpgradePayload(upgradeInfo: unknown): Buffer {
-    throw new Error("unsupported");
+  /**
+   * Returns the payload for a governance contract upgrade instruction for contracts deployed on this chain
+   * @param codeHash hex string of the 32 byte code hash for the new contract without the 0x prefix
+   */
+  generateGovernanceUpgradePayload(codeHash: string): Buffer {
+    return new UpgradeContract256Bit(this.wormholeChainName, codeHash).encode();
   }
 
   async getAccountAddress(privateKey: PrivateKey): Promise<string> {

contract_manager/src/contracts/near.ts

@@ -8,8 +8,82 @@ import {
 } from "../base";
 import { Chain, NearChain } from "../chains";
 import * as nearAPI from "near-api-js";
-import * as bs58 from "bs58";
 import { BN } from "fuels";
+import { WormholeContract } from "./wormhole";
+
+export class NearWormholeContract extends WormholeContract {
+  static type = "NearWormholeContract";
+
+  constructor(public chain: NearChain, public address: string) {
+    super();
+  }
+
+  getId(): string {
+    return `${this.chain.getId()}__${this.address.replace(/-|\./g, "_")}`;
+  }
+
+  getChain(): NearChain {
+    return this.chain;
+  }
+
+  getType(): string {
+    return NearWormholeContract.type;
+  }
+
+  static fromJson(
+    chain: Chain,
+    parsed: { type: string; address: string }
+  ): NearWormholeContract {
+    if (parsed.type !== NearWormholeContract.type)
+      throw new Error("Invalid type");
+    if (!(chain instanceof NearChain))
+      throw new Error(`Wrong chain type ${chain}`);
+    return new NearWormholeContract(chain, parsed.address);
+  }
+
+  toJson(): KeyValueConfig {
+    return {
+      chain: this.chain.getId(),
+      address: this.address,
+      type: NearWormholeContract.type,
+    };
+  }
+
+  async upgradeGuardianSets(
+    senderPrivateKey: PrivateKey,
+    vaa: Buffer
+  ): Promise<TxResult> {
+    const senderAddress = await this.chain.getAccountAddress(senderPrivateKey);
+    const account = await this.chain.getNearAccount(
+      senderAddress,
+      senderPrivateKey
+    );
+    const outcome = await account.functionCall({
+      contractId: this.address,
+      methodName: "submit_vaa",
+      args: { vaa: vaa.toString("hex") },
+      gas: new BN(300e12),
+      attachedDeposit: new BN(1e12),
+    });
+    return { id: outcome.transaction.hash, info: outcome };
+  }
+
+  getCurrentGuardianSetIndex(): Promise<number> {
+    throw new Error(
+      "near wormhole contract doesn't implement getCurrentGuardianSetIndex method"
+    );
+  }
+  getChainId(): Promise<number> {
+    throw new Error(
+      "near wormhole contract doesn't implement getChainId method"
+    );
+  }
+  getGuardianSet(): Promise<string[]> {
+    throw new Error(
+      "near wormhole contract doesn't implement getGuardianSet method"
+    );
+  }
+}
 
 export class NearPriceFeedContract extends PriceFeedContract {
   public static type = "NearPriceFeedContract";
@@ -18,6 +92,26 @@ export class NearPriceFeedContract extends PriceFeedContract {
     super();
   }
 
+  getId(): string {
+    return `${this.chain.getId()}__${this.address.replace(/-|\./g, "_")}`;
+  }
+
+  getType(): string {
+    return NearPriceFeedContract.type;
+  }
+
+  getChain(): NearChain {
+    return this.chain;
+  }
+
+  toJson(): KeyValueConfig {
+    return {
+      chain: this.chain.getId(),
+      address: this.address,
+      type: NearPriceFeedContract.type,
+    };
+  }
+
   static fromJson(
     chain: Chain,
     parsed: { type: string; address: string }
@@ -31,10 +125,6 @@ export class NearPriceFeedContract extends PriceFeedContract {
     return new NearPriceFeedContract(chain, parsed.address);
   }
 
-  getChain(): NearChain {
-    return this.chain;
-  }
-
   async getContractNearAccount(
     senderPrivateKey?: PrivateKey
   ): Promise<nearAPI.Account> {
@@ -113,8 +203,11 @@ export class NearPriceFeedContract extends PriceFeedContract {
     if (vaas.length === 0) {
       throw new Error("no vaas specified");
     }
-    const address = await this.chain.getAccountAddress(senderPrivateKey);
-    const account = await this.chain.getNearAccount(address, senderPrivateKey);
+    const senderAddress = await this.chain.getAccountAddress(senderPrivateKey);
+    const account = await this.chain.getNearAccount(
+      senderAddress,
+      senderPrivateKey
+    );
     let results = [];
     for (let vaa of vaas) {
       const outcome = await account.functionCall({
@@ -124,7 +217,6 @@ export class NearPriceFeedContract extends PriceFeedContract {
         gas: new BN(300e12),
         attachedDeposit: new BN(1e12),
       });
-      console.log("outcome", outcome);
       results.push({ id: outcome.transaction.hash, info: outcome });
     }
     if (results.length === 1) {
@@ -137,6 +229,25 @@ export class NearPriceFeedContract extends PriceFeedContract {
     }
   }
 
+  async executeGovernanceInstruction(
+    senderPrivateKey: PrivateKey,
+    vaa: Buffer
+  ): Promise<TxResult> {
+    const senderAddress = await this.chain.getAccountAddress(senderPrivateKey);
+    const account = await this.chain.getNearAccount(
+      senderAddress,
+      senderPrivateKey
+    );
+    const outcome = await account.functionCall({
+      contractId: this.address,
+      methodName: "execute_governance_instruction",
+      args: { vaa: vaa.toString("hex") },
+      gas: new BN(300e12),
+      attachedDeposit: new BN(1e12),
+    });
+    return { id: outcome.transaction.hash, info: outcome };
+  }
+
   getBaseUpdateFee(): Promise<{ amount: string; denom?: string }> {
     throw new Error("near contract doesn't implement getBaseUpdateFee method");
   }
@@ -145,27 +256,9 @@ export class NearPriceFeedContract extends PriceFeedContract {
       "near contract doesn't implement getLastExecutedGovernanceSequence method"
     );
   }
-
-  executeGovernanceInstruction(
-    senderPrivateKey: PrivateKey,
-    vaa: Buffer
-  ): Promise<TxResult> {
-    throw new Error("Method not implemented.");
-  }
   getGovernanceDataSource(): Promise<DataSource> {
-    throw new Error("Method not implemented.");
-  }
-  getId(): string {
-    return `${this.chain.getId()}_${this.address.replace(/-|\./g, "_")}`;
-  }
-  getType(): string {
-    return NearPriceFeedContract.type;
-  }
-  toJson(): KeyValueConfig {
-    return {
-      chain: this.chain.getId(),
-      address: this.address,
-      type: NearPriceFeedContract.type,
-    };
+    throw new Error(
+      "near contract doesn't implement getGovernanceDataSource method"
+    );
   }
 }

contract_manager/src/store.ts

@@ -36,7 +36,7 @@ import {
   StarknetPriceFeedContract,
   StarknetWormholeContract,
 } from "./contracts/starknet";
-import { NearPriceFeedContract } from "./contracts/near";
+import { NearPriceFeedContract, NearWormholeContract } from "./contracts/near";
 
 export class Store {
   public chains: Record<string, Chain> = { global: new GlobalChain() };
@@ -160,6 +160,7 @@ export class Store {
       [TonPriceFeedContract.type]: TonPriceFeedContract,
       [TonWormholeContract.type]: TonWormholeContract,
       [NearPriceFeedContract.type]: NearPriceFeedContract,
+      [NearWormholeContract.type]: NearWormholeContract,
     };
     this.getYamlFiles(`${this.path}/contracts/`).forEach((yamlFile) => {
       const parsedArray = parse(readFileSync(yamlFile, "utf-8"));

contract_manager/store/contracts/NearWormholeContracts.yaml

@@ -0,0 +1,6 @@
+- chain: near
+  address: contract.wormhole_crypto.near
+  type: NearWormholeContract
+- chain: near_testnet
+  address: wormhole.wormhole.testnet
+  type: NearWormholeContract

governance/xc_admin/packages/xc_admin_common/src/governance_payload/UpgradeContract.ts

@@ -34,7 +34,7 @@ export class CosmosUpgradeContract extends PythGovernanceActionImpl {
   }
 }
 
-// Used by Aptos, Sui and Starknet
+// Used by Aptos, Sui, Near, and Starknet
 export class UpgradeContract256Bit extends PythGovernanceActionImpl {
   static layout: BufferLayout.Structure<Readonly<{ hash: string }>> =
     BufferLayout.struct([BufferLayoutExt.hexBytes(32, "hash")]);