Merge pull request #913 from ava-labs/addSigToAllCred

chore: add sig to all credentials
ava-labs · Jan 6, 2025 · 06c8738 · 06c8738
2 parents ef80a85 + cba36c9
commit 06c8738
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 5 deletions.
diff --git a/examples/p-chain/etna/createChain.ts b/examples/p-chain/etna/createChain.ts
@@ -1,7 +1,8 @@
-import { addTxSignatures, pvm, utils } from '../../../src';
+import { pvm, utils } from '../../../src';
 import { setupEtnaExample } from './utils/etna-helper';
 import { testGenesisData } from '../../../src/fixtures/transactions';
 import { getEnvVars } from '../../utils/getEnvVars';
+import { addSigToAllCreds } from './utils/addSignatureToAllCred';
 
 /**
  * Create a new chain on the P-Chain.
@@ -38,10 +39,7 @@ const createChainTxExample = async () => {
     context,
   );
 
-  await addTxSignatures({
-    unsignedTx: tx,
-    privateKeys: [utils.hexToBuffer(PRIVATE_KEY)],
-  });
+  await addSigToAllCreds(tx, utils.hexToBuffer(PRIVATE_KEY));
 
   return pvmApi.issueSignedTx(tx.getSignedTx());
 };

diff --git a/examples/p-chain/etna/utils/addSignatureToAllCred.ts b/examples/p-chain/etna/utils/addSignatureToAllCred.ts
@@ -0,0 +1,18 @@
+import { secp256k1, type UnsignedTx } from '../../../../src';
+
+export const addSigToAllCreds = async (
+  unsignedTx: UnsignedTx,
+  privateKey: Uint8Array,
+) => {
+  const unsignedBytes = unsignedTx.toBytes();
+  const publicKey = secp256k1.getPublicKey(privateKey);
+
+  if (!unsignedTx.hasPubkey(publicKey)) {
+    return;
+  }
+  const signature = await secp256k1.sign(unsignedBytes, privateKey);
+
+  for (let i = 0; i < unsignedTx.getCredentials().length; i++) {
+    unsignedTx.addSignatureAt(signature, i, 0);
+  }
+};