feat: enable profiling with local PXE in cli-wallet

AztecProtocol · Dec 13, 2024 · 9defad0 · 9defad0
1 parent 52b541a
commit 9defad0
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/yarn-project/cli-wallet/src/bin/index.ts b/yarn-project/cli-wallet/src/bin/index.ts
@@ -88,9 +88,16 @@ async function main() {
     )
     .hook('preSubcommand', async command => {
       const { dataDir, remotePxe, nodeUrl } = command.optsWithGlobals();
+
       if (!remotePxe) {
         debugLogger.info('Using local PXE service');
-        await pxeWrapper.init(nodeUrl, join(dataDir, 'pxe'));
+        const subcommand = command.args[0];
+        const isProfiling = command.args.includes('--profile');
+        const enableProving = subcommand === 'simulate' && isProfiling;
+
+        await pxeWrapper.init(nodeUrl, join(dataDir, 'pxe'), {
+          proverEnabled: enableProving,
+        });
       }
       db.init(AztecLmdbStore.open(dataDir));
     });

diff --git a/yarn-project/cli-wallet/src/utils/pxe_wrapper.ts b/yarn-project/cli-wallet/src/utils/pxe_wrapper.ts
@@ -1,5 +1,5 @@
 import { type PXE, createAztecNodeClient } from '@aztec/circuit-types';
-import { createPXEService, getPXEServiceConfig } from '@aztec/pxe';
+import { type PXEServiceConfig, createPXEService, getPXEServiceConfig } from '@aztec/pxe';
 
 /*
  * Wrapper class for PXE service, avoids initialization issues due to
@@ -12,9 +12,9 @@ export class PXEWrapper {
     return PXEWrapper.pxe;
   }
 
-  async init(nodeUrl: string, dataDir: string) {
+  async init(nodeUrl: string, dataDir: string, overridePXEServiceConfig?: Partial<PXEServiceConfig>) {
     const aztecNode = createAztecNodeClient(nodeUrl);
-    const pxeConfig = getPXEServiceConfig();
+    const pxeConfig = Object.assign(getPXEServiceConfig(), overridePXEServiceConfig);
     pxeConfig.dataDirectory = dataDir;
     PXEWrapper.pxe = await createPXEService(aztecNode, pxeConfig);
   }