feat(target_chains/sui): add iota cli lib

ali-behjati · ali-behjati · commit 0526bd88585e · 2025-02-27T13:29:26.000+01:00
diff --git a/target_chains/sui/cli-iota/package.json b/target_chains/sui/cli-iota/package.json
@@ -1,7 +1,7 @@
 {
   "name": "pyth-iota-cli",
   "version": "0.1.0",
-  "description": "Pyth Sui Integration Cli tools",
+  "description": "Pyth IOTA Integration Cli tools",
   "main": "index.js",
   "license": "Apache-2.0",
   "scripts": {
diff --git a/target_chains/sui/cli-iota/src/cli.ts b/target_chains/sui/cli-iota/src/cli.ts
@@ -3,13 +3,13 @@ import { hideBin } from "yargs/helpers";
 import {
   DefaultStore,
   getDefaultDeploymentConfig,
-  SuiChain,
-  SuiPriceFeedContract,
+  IotaChain,
+  IotaPriceFeedContract,
 } from "@pythnetwork/contract-manager";
 import { PriceServiceConnection } from "@pythnetwork/price-service-client";
 import { execSync } from "child_process";
 import { initPyth, publishPackage } from "./pyth_deploy";
-import { Ed25519Keypair } from "@mysten/sui/keypairs/ed25519";
+import { Ed25519Keypair } from "@iota/iota-sdk/keypairs/ed25519";
 import { resolve } from "path";
 import {
   buildForBytecodeAndDigest,
@@ -26,15 +26,16 @@ const OPTIONS = {
   contract: {
     type: "string",
     demandOption: true,
-    desc: "Contract to use for the command (e.g sui_testnet_0xe8c2ddcd5b10e8ed98e53b12fcf8f0f6fd9315f810ae61fa4001858851f21c88)",
+    desc: "Contract to use for the command (e.g FIXME)",
   },
   path: {
     type: "string",
     default: "../../contracts",
-    desc: "Path to the sui contracts, will use ../../contracts by default",
+    desc: "Path to the iota contracts, will use ../../contracts by default",
   },
   endpoint: {
     type: "string",
+    default: "https://hermes.pyth.network",
     desc: "Price service endpoint to use, defaults to https://hermes.pyth.network for mainnet and https://hermes-beta.pyth.network for testnet",
   },
   "feed-id": {
@@ -44,24 +45,14 @@ const OPTIONS = {
   },
 } as const;
 
-function getContract(contractId: string): SuiPriceFeedContract {
-  const contract = DefaultStore.contracts[contractId] as SuiPriceFeedContract;
+function getContract(contractId: string): IotaPriceFeedContract {
+  const contract = DefaultStore.contracts[contractId] as IotaPriceFeedContract;
   if (!contract) {
     throw new Error(`Contract ${contractId} not found`);
   }
   return contract;
 }
 
-function getPriceService(
-  contract: SuiPriceFeedContract,
-  endpointOverride: string | undefined
-): PriceServiceConnection {
-  const defaultEndpoint = contract.getChain().isMainnet()
-    ? "https://hermes.pyth.network"
-    : "https://hermes-beta.pyth.network";
-  return new PriceServiceConnection(endpointOverride || defaultEndpoint);
-}
-
 yargs(hideBin(process.argv))
   .command(
     "create",
@@ -80,7 +71,7 @@ yargs(hideBin(process.argv))
     },
     async (argv) => {
       const contract = getContract(argv.contract);
-      const priceService = getPriceService(contract, argv.endpoint);
+      const priceService = new PriceServiceConnection(argv.endpoint);
       const feedIds = argv["feed-id"] as string[];
       const vaas = await priceService.getLatestVaas(feedIds);
       const digest = await contract.executeCreatePriceFeed(
@@ -106,7 +97,7 @@ yargs(hideBin(process.argv))
     },
     async (argv) => {
       const contract = getContract(argv.contract);
-      const priceService = getPriceService(contract, argv.endpoint);
+      const priceService = new PriceServiceConnection(argv.endpoint);
       const feedIds = await priceService.getPriceFeedIds();
       const BATCH_SIZE = 10;
       for (let i = 0; i < feedIds.length; i += BATCH_SIZE) {
@@ -138,7 +129,7 @@ yargs(hideBin(process.argv))
         digest: number[];
       } = JSON.parse(
         execSync(
-          `sui move build --dump-bytecode-as-base64 --path ${__dirname}/${argv.path} 2> /dev/null`,
+          `iota move build --dump-bytecode-as-base64 --path ${__dirname}/${argv.path} 2> /dev/null`,
           {
             encoding: "utf-8",
           }
@@ -158,26 +149,26 @@ yargs(hideBin(process.argv))
           chain: {
             type: "string",
             demandOption: true,
-            desc: "Chain to deploy the code to. Can be sui_mainnet or sui_testnet",
+            desc: "Chain to deploy the code to. Can be iota_mainnet or iota_testnet",
           },
           path: OPTIONS.path,
         })
         .usage(
-          "$0 deploy --private-key <private-key> --chain [sui_mainnet|sui_testnet] --path <path-to-contracts>"
+          "$0 deploy --private-key <private-key> --chain [iota_mainnet|iota_testnet] --path <path-to-contracts>"
         );
     },
     async (argv) => {
       const walletPrivateKey = argv["private-key"];
-      const chain = DefaultStore.chains[argv.chain] as SuiChain;
+      const chain = DefaultStore.chains[argv.chain] as IotaChain;
       const keypair = Ed25519Keypair.fromSecretKey(
-        Buffer.from(walletPrivateKey, "hex")
+        new Uint8Array(Buffer.from(walletPrivateKey, "hex"))
       );
       const result = await publishPackage(
         keypair,
         chain.getProvider(),
         argv.path
       );
-      const deploymentType = chain.isMainnet() ? "stable" : "beta";
+      const deploymentType = "stable";
       const config = getDefaultDeploymentConfig(deploymentType);
       await initPyth(
         keypair,
@@ -206,7 +197,7 @@ yargs(hideBin(process.argv))
     },
     async (argv) => {
       const contract = getContract(argv.contract);
-      const priceService = getPriceService(contract, argv.endpoint);
+      const priceService = new PriceServiceConnection(argv.endpoint);
       const feedIds = argv["feed-id"] as string[];
       const vaas = await priceService.getLatestVaas(feedIds);
       const digest = await contract.executeUpdatePriceFeedWithFeeds(
@@ -239,7 +230,7 @@ yargs(hideBin(process.argv))
     async (argv) => {
       const contract = getContract(argv.contract);
       const keypair = Ed25519Keypair.fromSecretKey(
-        Buffer.from(argv["private-key"], "hex")
+        new Uint8Array(Buffer.from(argv["private-key"], "hex"))
       );
 
       const pythContractsPath = resolve(`${__dirname}/${argv.path}`);
diff --git a/target_chains/sui/cli-iota/src/pyth_deploy.ts b/target_chains/sui/cli-iota/src/pyth_deploy.ts
@@ -1,15 +1,20 @@
-import { Transaction } from "@mysten/sui/transactions";
+import { Transaction } from "@iota/iota-sdk/transactions";
 
-import { MIST_PER_SUI, normalizeSuiObjectId, fromB64 } from "@mysten/sui/utils";
+import {
+  NANOS_PER_IOTA,
+  normalizeIotaObjectId,
+  fromB64,
+} from "@iota/iota-sdk/utils";
 
-import { Ed25519Keypair } from "@mysten/sui/dist/cjs/keypairs/ed25519";
+import { Ed25519Keypair } from "@iota/iota-sdk/keypairs/ed25519";
 import { execSync } from "child_process";
 import { DataSource } from "@pythnetwork/xc-admin-common";
-import { SuiClient } from "@mysten/sui/client";
+import { IotaClient } from "@iota/iota-sdk/client";
+import { bcs } from "@iota/iota-sdk/bcs";
 
 export async function publishPackage(
   keypair: Ed25519Keypair,
-  provider: SuiClient,
+  provider: IotaClient,
   packagePath: string
 ): Promise<{ packageId: string; upgradeCapId: string; deployerCapId: string }> {
   // Build contracts
@@ -18,7 +23,7 @@ export async function publishPackage(
     dependencies: string[];
   } = JSON.parse(
     execSync(
-      `sui move build --dump-bytecode-as-base64 --path ${__dirname}/${packagePath} 2> /dev/null`,
+      `iota move build --dump-bytecode-as-base64 --path ${__dirname}/${packagePath} 2> /dev/null`,
       {
         encoding: "utf-8",
       }
@@ -31,17 +36,17 @@ export async function publishPackage(
   // const transactionBlock = new TransactionBlock();
   const txb = new Transaction();
 
-  txb.setGasBudget(MIST_PER_SUI / 2n); // 0.5 SUI
+  txb.setGasBudget(NANOS_PER_IOTA / 2n); // 0.5 SUI
 
   const [upgradeCap] = txb.publish({
     modules: buildOutput.modules.map((m: string) => Array.from(fromB64(m))),
     dependencies: buildOutput.dependencies.map((d: string) =>
-      normalizeSuiObjectId(d)
+      normalizeIotaObjectId(d)
     ),
   });
 
   // Transfer upgrade capability to deployer
-  txb.transferObjects([upgradeCap], txb.pure.address(keypair.toSuiAddress()));
+  txb.transferObjects([upgradeCap], txb.pure.address(keypair.toIotaAddress()));
 
   // Execute transactions
   const result = await provider.signAndExecuteTransaction({
@@ -97,7 +102,7 @@ export async function publishPackage(
 
 export async function initPyth(
   keypair: Ed25519Keypair,
-  provider: SuiClient,
+  provider: IotaClient,
   pythPackageId: string,
   deployerCapId: string,
   upgradeCapId: string,
@@ -109,19 +114,31 @@ export async function initPyth(
   const tx = new Transaction();
 
   const baseUpdateFee = tx.pure.u64(1);
-  const dataSourceEmitterAddresses = tx.pure.arguments(
-    config.dataSources.map((dataSource) => [
-      ...Buffer.from(dataSource.emitterAddress, "hex"),
-    ])
+  const dataSourceEmitterAddresses = tx.pure(
+    bcs
+      .vector(bcs.vector(bcs.u8()))
+      .serialize(
+        config.dataSources.map((dataSource) => [
+          ...Buffer.from(dataSource.emitterAddress, "hex"),
+        ])
+      )
   );
-  const dataSourceEmitterChainIds = tx.pure.arguments(
-    config.dataSources.map((dataSource) => dataSource.emitterChain)
+  const dataSourceEmitterChainIds = tx.pure(
+    bcs
+      .vector(bcs.u64())
+      .serialize(
+        config.dataSources.map((dataSource) => dataSource.emitterChain)
+      )
   );
-  const governanceEmitterAddress = tx.pure.arguments([
-    ...Buffer.from(config.governanceDataSource.emitterAddress, "hex"),
-  ]);
-  const governanceEmitterChainId = tx.pure.arguments(
-    config.governanceDataSource.emitterChain
+  const governanceEmitterAddress = tx.pure(
+    bcs
+      .vector(bcs.u8())
+      .serialize([
+        ...Buffer.from(config.governanceDataSource.emitterAddress, "hex"),
+      ])
+  );
+  const governanceEmitterChainId = tx.pure(
+    bcs.u64().serialize(config.governanceDataSource.emitterChain)
   );
   const stalePriceThreshold = tx.pure.u64(60);
   tx.moveCall({
@@ -138,7 +155,7 @@ export async function initPyth(
     ],
   });
 
-  tx.setGasBudget(MIST_PER_SUI / 10n); // 0.1 sui
+  tx.setGasBudget(NANOS_PER_IOTA / 10n); // 0.1 sui
 
   let result = await provider.signAndExecuteTransaction({
     signer: keypair,
diff --git a/target_chains/sui/cli-iota/src/upgrade_pyth.ts b/target_chains/sui/cli-iota/src/upgrade_pyth.ts
@@ -1,10 +1,14 @@
-import { Transaction } from "@mysten/sui/transactions";
-import { fromB64, MIST_PER_SUI, normalizeSuiObjectId } from "@mysten/sui/utils";
-import { SuiClient } from "@mysten/sui/client";
-import { Ed25519Keypair } from "@mysten/sui/keypairs/ed25519";
+import { Transaction } from "@iota/iota-sdk/transactions";
+import {
+  fromB64,
+  NANOS_PER_IOTA,
+  normalizeIotaObjectId,
+} from "@iota/iota-sdk/utils";
+import { IotaClient } from "@iota/iota-sdk/client";
+import { Ed25519Keypair } from "@iota/iota-sdk/keypairs/ed25519";
 
 import { execSync } from "child_process";
-import { SuiPriceFeedContract } from "@pythnetwork/contract-manager";
+import { IotaPriceFeedContract } from "@pythnetwork/contract-manager";
 
 export function buildForBytecodeAndDigest(packagePath: string) {
   const buildOutput: {
@@ -20,19 +24,19 @@ export function buildForBytecodeAndDigest(packagePath: string) {
   return {
     modules: buildOutput.modules.map((m: string) => Array.from(fromB64(m))),
     dependencies: buildOutput.dependencies.map((d: string) =>
-      normalizeSuiObjectId(d)
+      normalizeIotaObjectId(d)
     ),
     digest: Buffer.from(buildOutput.digest),
   };
 }
 
 export async function upgradePyth(
   keypair: Ed25519Keypair,
-  provider: SuiClient,
+  provider: IotaClient,
   modules: number[][],
   dependencies: string[],
   signedVaa: Buffer,
-  contract: SuiPriceFeedContract
+  contract: IotaPriceFeedContract
 ) {
   const pythPackage = await contract.getPackageId(contract.stateId);
 
@@ -64,7 +68,7 @@ export async function upgradePyth(
     arguments: [tx.object(contract.stateId), upgradeReceipt],
   });
 
-  tx.setGasBudget(MIST_PER_SUI / 4n); // 0.25 SUI
+  tx.setGasBudget(NANOS_PER_IOTA / 4n); // 0.25 SUI
 
   return provider.signAndExecuteTransaction({
     signer: keypair,
@@ -78,9 +82,9 @@ export async function upgradePyth(
 
 export async function migratePyth(
   keypair: Ed25519Keypair,
-  provider: SuiClient,
+  provider: IotaClient,
   signedUpgradeVaa: Buffer,
-  contract: SuiPriceFeedContract,
+  contract: IotaPriceFeedContract,
   pythPackageOld: string
 ) {
   const pythPackage = await contract.getPackageId(contract.stateId);
@@ -98,7 +102,7 @@ export async function migratePyth(
     arguments: [tx.object(contract.stateId), verificationReceipt as any],
   });
 
-  tx.setGasBudget(MIST_PER_SUI / 10n); //0.1 SUI
+  tx.setGasBudget(NANOS_PER_IOTA / 10n); //0.1 SUI
 
   return provider.signAndExecuteTransaction({
     signer: keypair,
diff --git a/target_chains/sui/cli/src/cli.ts b/target_chains/sui/cli/src/cli.ts
diff --git a/target_chains/sui/cli/src/pyth_deploy.ts b/target_chains/sui/cli/src/pyth_deploy.ts

Original file line number	Diff line number	Diff line change
`@@ -1,7 +1,7 @@`
`1`	`1`	`{`
`2`	`2`	`"name": "pyth-iota-cli",`
`3`	`3`	`"version": "0.1.0",`
`4`		`- "description": "Pyth Sui Integration Cli tools",`
	`4`	`+ "description": "Pyth IOTA Integration Cli tools",`
`5`	`5`	`"main": "index.js",`
`6`	`6`	`"license": "Apache-2.0",`
`7`	`7`	`"scripts": {`