fix: deploy aleo programs

Author: troykessler

typescript/aleo-sdk/src/clients/provider.ts

@@ -5,6 +5,8 @@ import {
   Plaintext,
   Program,
   ProgramManager,
+  getOrInitConsensusVersionTestHeights,
+  initThreadPool,
 } from '@provablehq/sdk';
 import { BigNumber } from 'bignumber.js';
 
@@ -14,12 +16,8 @@ import { mailbox } from '../artifacts.js';
 import { formatAddress, getMessageKey } from '../utils/helper.js';
 import { AleoTransaction } from '../utils/types.js';
 
-// TODO: make denom in AltVM optional
-// TODO: add remove destination gas config method in AltVM
-// TODO: only allow domainId in createMailox in AltVM
-// TODO: add createNoopHook method in AltVM
-// TODO: add getTokenMetadata method in AltVM
-// TODO: don't allow routes in create routing ism
+getOrInitConsensusVersionTestHeights('0,1,2,3,4,5,6,7,8,9,10');
+await initThreadPool();
 
 export class AleoProvider implements AltVM.IProvider {
   protected readonly aleoClient: AleoNetworkClient;
@@ -39,6 +37,16 @@ export class AleoProvider implements AltVM.IProvider {
     this.aleoClient = new AleoNetworkClient(rpcUrls[0]);
   }
 
+  protected getProgramSalt(address: string): string {
+    // get the program salt by hashing the address with bhp256 and take the first 12
+    // characters from the hex hash
+    return new BHP256()
+      .hash(Plaintext.fromString(address).toBitsLe())
+      .toBytesLe()
+      .reduce((acc, b) => acc + b.toString(16).padStart(2, '0'), '')
+      .slice(0, 12);
+  }
+
   // ### QUERY BASE ###
 
   async isHealthy() {
@@ -120,7 +128,7 @@ export class AleoProvider implements AltVM.IProvider {
     let res;
     try {
       res = await this.aleoClient.getProgramMappingPlaintext(
-        req.mailboxAddress,
+        `mailbox_${this.getProgramSalt(req.mailboxAddress)}.aleo`,
         'mailbox',
         'true',
       );
@@ -584,7 +592,7 @@ export class AleoProvider implements AltVM.IProvider {
     req: AltVM.ReqCreateMailbox,
   ): Promise<AleoTransaction> {
     return {
-      programName: 'mailbox.aleo',
+      programName: '',
       functionName: 'init',
       priorityFee: 0,
       privateFee: false,
@@ -596,7 +604,7 @@ export class AleoProvider implements AltVM.IProvider {
     req: AltVM.ReqSetDefaultIsm,
   ): Promise<AleoTransaction> {
     return {
-      programName: 'mailbox.aleo',
+      programName: `mailbox_${this.getProgramSalt(req.mailboxAddress)}.aleo`,
       functionName: 'set_default_ism',
       priorityFee: 0,
       privateFee: false,
@@ -608,7 +616,7 @@ export class AleoProvider implements AltVM.IProvider {
     req: AltVM.ReqSetDefaultHook,
   ): Promise<AleoTransaction> {
     return {
-      programName: 'mailbox.aleo',
+      programName: `mailbox_${this.getProgramSalt(req.mailboxAddress)}.aleo`,
       functionName: 'set_default_hook',
       priorityFee: 0,
       privateFee: false,
@@ -620,7 +628,7 @@ export class AleoProvider implements AltVM.IProvider {
     req: AltVM.ReqSetRequiredHook,
   ): Promise<AleoTransaction> {
     return {
-      programName: 'mailbox.aleo',
+      programName: `mailbox_${this.getProgramSalt(req.mailboxAddress)}.aleo`,
       functionName: 'set_required_hook',
       priorityFee: 0,
       privateFee: false,
@@ -632,7 +640,7 @@ export class AleoProvider implements AltVM.IProvider {
     req: AltVM.ReqSetMailboxOwner,
   ): Promise<AleoTransaction> {
     return {
-      programName: 'mailbox.aleo',
+      programName: `mailbox_${this.getProgramSalt(req.mailboxAddress)}.aleo`,
       functionName: 'set_owner',
       priorityFee: 0,
       privateFee: false,
@@ -743,15 +751,23 @@ export class AleoProvider implements AltVM.IProvider {
   }
 
   async getCreateMerkleTreeHookTransaction(
-    _req: AltVM.ReqCreateMerkleTreeHook,
+    req: AltVM.ReqCreateMerkleTreeHook,
   ): Promise<AleoTransaction> {
-    // TODO: replace default mailbox id with real one
     return {
       programName: 'hook_manager.aleo',
       functionName: 'init_merkle_tree',
       priorityFee: 0,
       privateFee: false,
-      inputs: [Program.fromString(mailbox).address().to_string()],
+      inputs: [
+        Program.fromString(
+          mailbox.replaceAll(
+            'mailbox.aleo',
+            `mailbox_${this.getProgramSalt(req.mailboxAddress)}.aleo`,
+          ),
+        )
+          .address()
+          .to_string(),
+      ],
     };
   }
 
@@ -803,12 +819,19 @@ export class AleoProvider implements AltVM.IProvider {
     });
 
     return {
-      programName: 'validator_announce.aleo',
+      programName: '',
       functionName: 'init',
       priorityFee: 0,
       privateFee: false,
       inputs: [
-        Program.fromString(mailbox).address().to_string(),
+        Program.fromString(
+          mailbox.replaceAll(
+            'mailbox.aleo',
+            `mailbox_${this.getProgramSalt(req.mailboxAddress)}.aleo`,
+          ),
+        )
+          .address()
+          .to_string(),
         `${localDomain}u32`,
       ],
     };

typescript/aleo-sdk/src/clients/signer.ts

@@ -4,7 +4,6 @@ import {
   NetworkRecordProvider,
   Program,
   ProgramManager,
-  ProgramManagerBase,
 } from '@provablehq/sdk';
 
 import { AltVM } from '@hyperlane-xyz/utils';
@@ -54,8 +53,25 @@ export class AleoSigner
     this.programManager.setAccount(this.aleoAccount);
   }
 
-  private async deployProgram(programName: string): Promise<Program[]> {
-    const programs = loadProgramsInDeployOrder(programName);
+  private async deployProgram(programName: string): Promise<{
+    address: string;
+    program: Program;
+  }> {
+    let programs = loadProgramsInDeployOrder(programName);
+
+    const address = new Account().address().to_string();
+    const salt = this.getProgramSalt(address);
+
+    programs = programs.map((p) => {
+      return Program.fromString(
+        p
+          .toString()
+          .replaceAll(
+            /(mailbox|dispatch_proxy|validator_announce)\.aleo/g,
+            (_, p1) => `${p1}_${salt}.aleo`,
+          ),
+      );
+    });
 
     for (const program of programs) {
       const isDeployed = await this.isProgramDeployed(program);
@@ -66,20 +82,19 @@ export class AleoSigner
         continue;
       }
 
-      const fee = await ProgramManagerBase.estimateDeploymentFee(
-        program.toString(),
-      );
-
       const txId = await this.programManager.deploy(
         program.toString(),
-        Math.ceil(Number(fee) / 10 ** 6),
+        0,
         false,
       );
 
       await this.aleoClient.waitForTransactionConfirmation(txId);
     }
 
-    return programs;
+    return {
+      address,
+      program: programs[programs.length - 1],
+    };
   }
 
   getSignerAddress(): string {
@@ -123,19 +138,20 @@ export class AleoSigner
   async createMailbox(
     req: Omit<AltVM.ReqCreateMailbox, 'signer'>,
   ): Promise<AltVM.ResCreateMailbox> {
-    await this.deployProgram('dispatch_proxy');
+    const { address } = await this.deployProgram('dispatch_proxy');
 
-    // TODO: init with correct mailbox id
     const tx = await this.getCreateMailboxTransaction({
       signer: this.getSignerAddress(),
       ...req,
     });
 
+    tx.programName = `mailbox_${this.getProgramSalt(address)}.aleo`;
+
     const txId = await this.programManager.execute(tx);
     await this.aleoClient.waitForTransactionConfirmation(txId);
 
     return {
-      mailboxAddress: 'mailbox.aleo',
+      mailboxAddress: address,
     };
   }
 
@@ -482,19 +498,20 @@ export class AleoSigner
   async createValidatorAnnounce(
     req: Omit<AltVM.ReqCreateValidatorAnnounce, 'signer'>,
   ): Promise<AltVM.ResCreateValidatorAnnounce> {
-    await this.deployProgram('validator_announce');
+    const { address } = await this.deployProgram('validator_announce');
 
-    // TODO: init with correct validator announce and mailbox id
     const tx = await this.getCreateValidatorAnnounceTransaction({
       signer: this.getSignerAddress(),
       ...req,
     });
 
+    tx.programName = `validator_announce_${this.getProgramSalt(address)}.aleo`;
+
     const txId = await this.programManager.execute(tx);
     await this.aleoClient.waitForTransactionConfirmation(txId);
 
     return {
-      validatorAnnounceId: 'validator_announce.aleo',
+      validatorAnnounceId: address,
     };
   }
 

typescript/aleo-sdk/src/deploy.ts

@@ -1,32 +1,34 @@
-import { AleoProvider } from './clients/provider.js';
 import { AleoSigner } from './clients/signer.js';
 
 const main = async () => {
-  const localnetRpc = 'http://localhost:3030';
-  const provider = await AleoProvider.connect([localnetRpc], '');
+  try {
+    const localnetRpc = 'http://localhost:3030';
 
-  const latestBlockHeight = await provider.getHeight();
-  console.log('latestBlockHeight: ', latestBlockHeight);
+    // test private key with funds
+    const privateKey =
+      'APrivateKey1zkp8CZNn3yeCseEtxuVPbDCwSyhGW6yZKUYKfgXmcpoGPWH';
+    const signer = await AleoSigner.connectWithSigner(
+      [localnetRpc],
+      privateKey,
+    );
 
-  // test private key with funds
-  const privateKey =
-    'APrivateKey1zkp8CZNn3yeCseEtxuVPbDCwSyhGW6yZKUYKfgXmcpoGPWH';
-  const signer = await AleoSigner.connectWithSigner([localnetRpc], privateKey);
+    const address = signer.getSignerAddress();
+    console.log('signer address: ', address);
 
-  const address = signer.getSignerAddress();
-  console.log('signer address: ', address);
+    const balance = await signer.getBalance({
+      address,
+      denom: '',
+    });
+    console.log('signer credits balance: ', balance);
 
-  const balance = await signer.getBalance({
-    address,
-    denom: '',
-  });
-  console.log('signer credits balance: ', balance);
-
-  const { mailboxAddress } = await signer.createMailbox({
-    domainId: 1337,
-    defaultIsmAddress: '',
-  });
-  console.log('mailboxAddress', mailboxAddress);
+    const { mailboxAddress } = await signer.createMailbox({
+      domainId: 1337,
+      defaultIsmAddress: '',
+    });
+    console.log('mailboxAddress', mailboxAddress);
+  } catch (err) {
+    console.log(err);
+  }
 };
 
 main();