@aztec-labs/stdlib 6.0.0-rc.3 → 6.0.0

package/src/file-store/s3.ts

@@ -5,7 +5,9 @@ import {
   PutObjectCommand,
   S3Client,
 } from '@aws-sdk/client-s3';
+import { getEnv } from '@aztec-labs/foundation/config/env';
 import { type Logger, createLogger } from '@aztec-labs/foundation/log';
+import { Buffer } from 'buffer';
 import { createReadStream, createWriteStream } from 'fs';
 import { mkdir, mkdtemp, stat, unlink } from 'fs/promises';
 import { tmpdir } from 'os';
@@ -37,7 +39,7 @@ export class S3FileStore implements FileStore {
     private readonly log: Logger = createLogger('stdlib:s3-file-store'),
   ) {
     this.endpoint = opts.endpoint;
-    this.region = this.endpoint ? 'auto' : (process.env.AWS_REGION ?? process.env.AWS_DEFAULT_REGION ?? 'us-east-1');
+    this.region = this.endpoint ? 'auto' : (getEnv().AWS_REGION ?? getEnv().AWS_DEFAULT_REGION ?? 'us-east-1');
     this.publicBaseUrl = opts.publicBaseUrl;
 
     const clientOptions: any = {};

package/src/gas/gas.ts

@@ -1,4 +1,5 @@
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import { schemas } from '@aztec-labs/foundation/schemas';
 import {
   BufferReader,
@@ -8,7 +9,6 @@ import {
   serializeToSink,
 } from '@aztec-labs/foundation/serialize';
 import type { FieldsOf } from '@aztec-labs/foundation/types';
-import { inspect } from 'util';
 import { z } from 'zod';
 
 import type { UInt32 } from '../types/shared.js';

package/src/gas/gas_fees.ts

@@ -1,8 +1,8 @@
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import { schemas } from '@aztec-labs/foundation/schemas';
 import { BufferReader, BufferSink, FieldReader, serializeToFields } from '@aztec-labs/foundation/serialize';
 import type { FieldsOf } from '@aztec-labs/foundation/types';
-import { inspect } from 'util';
 import { z } from 'zod';
 
 import type { UInt128 } from '../types/shared.js';

package/src/ha-signing/config.ts

@@ -16,6 +16,9 @@ import {
 import type { ZodFor } from '@aztec-labs/foundation/schemas';
 import { z } from 'zod';
 
+/** Default retention for signed duties, in hours. */
+export const DEFAULT_SIGNED_DUTIES_RETENTION_HOURS = 24;
+
 /**
  * Base signing protection configuration shared by both HA (Postgres) and local (LMDB) signers.
  */
@@ -40,7 +43,7 @@ export type BaseSignerConfig = {
   signerCallTimeoutMs?: number;
   /** Maximum age of a stuck duty in ms (defaults to 2x hardcoded Aztec slot duration if not set) */
   maxStuckDutiesAgeMs?: number;
-  /** Optional: clean up old duties after this many hours (disabled if not set) */
+  /** Duty retention in hours from signing start, with a minimum of two slots (defaults to 24). */
   cleanupOldDutiesAfterHours?: number;
 } & Pick<L1ContractAddresses, 'rollupAddress'>;
 
@@ -72,8 +75,9 @@ export const baseSignerConfigMappings: ConfigMappingsType<BaseSignerConfig> = {
   },
   cleanupOldDutiesAfterHours: {
     env: 'VALIDATOR_HA_OLD_DUTIES_MAX_AGE_H',
-    description: 'Optional: clean up old duties after this many hours (disabled if not set)',
-    ...optionalNumberConfigHelper(),
+    description: 'Duty retention in hours from signing start (must be positive; minimum two slots)',
+    defaultValue: DEFAULT_SIGNED_DUTIES_RETENTION_HOURS,
+    parseEnv: (value: string) => z.coerce.number().int().positive().parse(value),
   },
   ...pickL1ContractAddressMappings('rollupAddress'),
 };
@@ -84,7 +88,7 @@ export const BaseSignerConfigSchema = z.object({
   peerSigningTimeoutMs: z.number().min(0),
   signerCallTimeoutMs: z.number().min(0).optional(),
   maxStuckDutiesAgeMs: z.number().min(0).optional(),
-  cleanupOldDutiesAfterHours: z.number().min(0).optional(),
+  cleanupOldDutiesAfterHours: z.number().positive().optional(),
   ...pickL1ContractAddressesSchema('rollupAddress'),
 }) satisfies ZodFor<BaseSignerConfig>;
 

package/src/ha-signing/index.ts

@@ -1,6 +1,7 @@
 export {
   type BaseSignerConfig,
   BaseSignerConfigSchema,
+  DEFAULT_SIGNED_DUTIES_RETENTION_HOURS,
   baseSignerConfigMappings,
   type ValidatorHASignerConfig,
   ValidatorHASignerConfigSchema,

package/src/interfaces/aztec-node-admin.ts

@@ -1,5 +1,6 @@
 import type { L1ContractAddresses } from '@aztec-labs/ethereum/l1-contract-addresses';
 import { createSafeJsonRpcClient, defaultFetch } from '@aztec-labs/foundation/json-rpc/client';
+import { isValidLogLevelSpec } from '@aztec-labs/foundation/log';
 import { z } from 'zod';
 
 import type { ApiSchemaFor } from '../schemas/schemas.js';
@@ -94,6 +95,12 @@ export type AztecNodeAdminConfig = Omit<ValidatorClientFullConfig, keyof L1Contr
     skipIncomingProposals?: boolean;
     // Keep in sync with P2PConfig.preferredPeers (circular dep prevents Pick<P2PConfig, ...> here)
     preferredPeers?: string[];
+    /**
+     * Log level spec in `LOG_LEVEL` env var format (e.g. `info` or `debug;trace:sequencer,p2p`). Setting it updates
+     * all loggers in the node's main thread immediately; worker threads and spawned processes keep their own
+     * configuration. Not persisted: a restart reverts to the `LOG_LEVEL` env var.
+     */
+    logLevel?: string;
   };
 
 export const AztecNodeAdminConfigSchema = SequencerConfigSchema.merge(ProverConfigSchema)
@@ -111,6 +118,7 @@ export const AztecNodeAdminConfigSchema = SequencerConfigSchema.merge(ProverConf
       maxPendingTxCount: z.number(),
       skipIncomingProposals: z.boolean().optional(),
       preferredPeers: z.array(z.string()).optional(),
+      logLevel: z.string().refine(isValidLogLevelSpec, 'Invalid log level spec').optional(),
     }),
   );
 

package/src/interfaces/configs.ts

@@ -102,6 +102,11 @@ export interface SequencerConfig {
   injectUnrecoverableSignatureAttestation?: boolean;
   /** Inject a non-proposer attestation slot in yParity (v ∈ {0, 1}) form in the packed L1 tuple (for testing only) */
   injectYParityAttestation?: boolean;
+  /**
+   * Publish the checkpoint with a fake attestation first, then invalidate it and publish it again with the collected
+   * attestations within the same slot (for testing only)
+   */
+  republishAfterFakeAttestation?: boolean;
   /** Whether to run in fisherman mode: builds blocks on every slot for validation without publishing */
   fishermanMode?: boolean;
   /** Shuffle attestation ordering to create invalid ordering (for testing only) */
@@ -185,6 +190,7 @@ export const SequencerConfigSchema = zodFor<SequencerConfig>()(
     injectHighSValueAttestation: z.boolean().optional(),
     injectUnrecoverableSignatureAttestation: z.boolean().optional(),
     injectYParityAttestation: z.boolean().optional(),
+    republishAfterFakeAttestation: z.boolean().optional(),
     fishermanMode: z.boolean().optional(),
     shuffleAttestationOrdering: z.boolean().optional(),
     blockDurationMs: z.number().positive().optional(),

package/src/kernel/claimed_length_array.ts

@@ -1,4 +1,5 @@
 import type { Fr } from '@aztec-labs/foundation/curves/bn254';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import {
   BufferReader,
   FieldReader,
@@ -7,7 +8,6 @@ import {
   serializeToBuffer,
   serializeToFields,
 } from '@aztec-labs/foundation/serialize';
-import { inspect } from 'util';
 
 export class ClaimedLengthArray<T extends Serializable, N extends number> {
   constructor(

package/src/kernel/hints/transient_data_squashing_hint.ts

@@ -1,6 +1,6 @@
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import { BufferReader, FieldReader, serializeToBuffer } from '@aztec-labs/foundation/serialize';
-import { inspect } from 'util';
 
 export class TransientDataSquashingHint {
   constructor(

package/src/kernel/log_hash.ts

@@ -1,7 +1,7 @@
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import { BufferReader, BufferSink, FieldReader, serializeToSink } from '@aztec-labs/foundation/serialize';
 import type { FieldsOf } from '@aztec-labs/foundation/types';
-import { inspect } from 'util';
 
 import { AztecAddress } from '../aztec-address/index.js';
 

package/src/kernel/private_log_data.ts

@@ -1,8 +1,8 @@
 import { PRIVATE_LOG_DATA_LENGTH } from '@aztec-labs/constants';
 import type { Fr } from '@aztec-labs/foundation/curves/bn254';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import { BufferReader, FieldReader, serializeToBuffer, serializeToFields } from '@aztec-labs/foundation/serialize';
 import type { FieldsOf } from '@aztec-labs/foundation/types';
-import { inspect } from 'util';
 
 import { AztecAddress } from '../aztec-address/index.js';
 import { PrivateLog } from '../logs/private_log.js';

package/src/kernel/private_to_avm_accumulated_data.ts

@@ -7,6 +7,7 @@ import {
 import { type FieldsOf, makeTuple } from '@aztec-labs/foundation/array';
 import { arraySerializedSizeOfNonEmpty } from '@aztec-labs/foundation/collection';
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import { schemas } from '@aztec-labs/foundation/schemas';
 import {
   BufferReader,
@@ -17,7 +18,6 @@ import {
   serializeToFields,
   serializeToSink,
 } from '@aztec-labs/foundation/serialize';
-import { inspect } from 'util';
 import { z } from 'zod';
 
 import { ScopedL2ToL1Message } from '../messaging/l2_to_l1_message.js';

package/src/kernel/private_to_public_accumulated_data.ts

@@ -10,6 +10,7 @@ import {
 import { type FieldsOf, makeTuple } from '@aztec-labs/foundation/array';
 import { arraySerializedSizeOfNonEmpty } from '@aztec-labs/foundation/collection';
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import {
   BufferReader,
   BufferSink,
@@ -18,7 +19,6 @@ import {
   serializeToFields,
   serializeToSink,
 } from '@aztec-labs/foundation/serialize';
-import { inspect } from 'util';
 
 import { PrivateLog } from '../logs/private_log.js';
 import { ScopedL2ToL1Message } from '../messaging/l2_to_l1_message.js';

package/src/kernel/private_to_rollup_accumulated_data.ts

@@ -9,6 +9,7 @@ import {
 import { type FieldsOf, makeTuple } from '@aztec-labs/foundation/array';
 import { arraySerializedSizeOfNonEmpty } from '@aztec-labs/foundation/collection';
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import { bufferSchemaFor } from '@aztec-labs/foundation/schemas';
 import {
   BufferReader,
@@ -18,7 +19,6 @@ import {
   serializeToSink,
 } from '@aztec-labs/foundation/serialize';
 import { bufferToHex, hexToBuffer } from '@aztec-labs/foundation/string';
-import { inspect } from 'util';
 
 import { PrivateLog } from '../logs/private_log.js';
 import { ScopedL2ToL1Message } from '../messaging/l2_to_l1_message.js';

package/src/kernel/private_validation_requests.ts

@@ -3,9 +3,9 @@ import {
   MAX_NOTE_HASH_READ_REQUESTS_PER_TX,
   MAX_NULLIFIER_READ_REQUESTS_PER_TX,
 } from '@aztec-labs/constants';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import { BufferReader, serializeToBuffer } from '@aztec-labs/foundation/serialize';
 import { bufferToHex, hexToBuffer } from '@aztec-labs/foundation/string';
-import { inspect } from 'util';
 
 import { ScopedKeyValidationRequestAndSeparator } from '../kernel/hints/scoped_key_validation_request_and_separator.js';
 import { ClaimedLengthArray, ClaimedLengthArrayFromBuffer } from './claimed_length_array.js';

package/src/kernel/public_call_request.ts

@@ -1,5 +1,6 @@
 import { COUNTED_PUBLIC_CALL_REQUEST_LENGTH, PUBLIC_CALL_REQUEST_LENGTH } from '@aztec-labs/constants';
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import { schemas } from '@aztec-labs/foundation/schemas';
 import {
   BufferReader,
@@ -9,7 +10,6 @@ import {
   serializeToSink,
 } from '@aztec-labs/foundation/serialize';
 import type { FieldsOf } from '@aztec-labs/foundation/types';
-import { inspect } from 'util';
 import { z } from 'zod';
 
 import { AztecAddress } from '../aztec-address/index.js';

package/src/keys/public_keys.ts

@@ -14,6 +14,7 @@ import { schemas } from '@aztec-labs/foundation/schemas';
 import { BufferReader, FieldReader, serializeToBuffer } from '@aztec-labs/foundation/serialize';
 import { bufferToHex, withoutHexPrefix } from '@aztec-labs/foundation/string';
 import type { FieldsOf } from '@aztec-labs/foundation/types';
+import { Buffer } from 'buffer';
 import { z } from 'zod';
 
 import { PublicKey, hashPublicKey } from './public_key.js';

package/src/logs/contract_class_log.ts

@@ -1,6 +1,7 @@
 import { CONTRACT_CLASS_LOG_LENGTH, CONTRACT_CLASS_LOG_SIZE_IN_FIELDS } from '@aztec-labs/constants';
 import { poseidon2Hash } from '@aztec-labs/foundation/crypto/poseidon';
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import { schemas } from '@aztec-labs/foundation/schemas';
 import {
   BufferReader,
@@ -10,7 +11,6 @@ import {
   serializeToSink,
 } from '@aztec-labs/foundation/serialize';
 import type { FieldsOf } from '@aztec-labs/foundation/types';
-import { inspect } from 'util';
 import { z } from 'zod';
 
 import { AztecAddress } from '../aztec-address/index.js';

package/src/logs/log_cursor.ts

@@ -1,5 +1,6 @@
 import { BlockNumber, BlockNumberSchema } from '@aztec-labs/foundation/branded-types';
 import { BufferReader, numToUInt32BE } from '@aztec-labs/foundation/serialize';
+import { Buffer } from 'buffer';
 import { z } from 'zod';
 
 import { schemas } from '../schemas/schemas.js';

package/src/logs/private_log.ts

@@ -2,6 +2,7 @@ import { PRIVATE_LOG_LENGTH, PRIVATE_LOG_SIZE_IN_FIELDS } from '@aztec-labs/cons
 import { type FieldsOf, makeTuple } from '@aztec-labs/foundation/array';
 import { padArrayEnd } from '@aztec-labs/foundation/collection';
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import { schemas } from '@aztec-labs/foundation/schemas';
 import {
   BufferReader,
@@ -11,7 +12,6 @@ import {
   serializeToFields,
   serializeToSink,
 } from '@aztec-labs/foundation/serialize';
-import { inspect } from 'util';
 import { z } from 'zod';
 
 export class PrivateLog {

package/src/logs/public_log.ts

@@ -1,9 +1,9 @@
 import { FLAT_PUBLIC_LOGS_PAYLOAD_LENGTH, PUBLIC_LOG_HEADER_LENGTH } from '@aztec-labs/constants';
 import type { FieldsOf } from '@aztec-labs/foundation/array';
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import { type ZodFor, schemas } from '@aztec-labs/foundation/schemas';
 import { BufferReader, FieldReader, serializeToBuffer } from '@aztec-labs/foundation/serialize';
-import { inspect } from 'util';
 import { z } from 'zod';
 
 import { AztecAddress } from '../aztec-address/index.js';

package/src/messaging/l1_to_l2_message.ts

@@ -5,6 +5,7 @@ import { Fr } from '@aztec-labs/foundation/curves/bn254';
 import { BufferReader, serializeToBuffer } from '@aztec-labs/foundation/serialize';
 import { bufferToHex } from '@aztec-labs/foundation/string';
 import { SiblingPath } from '@aztec-labs/foundation/trees';
+import { Buffer } from 'buffer';
 
 import type { AztecAddress } from '../aztec-address/index.js';
 import type { BlockParameter } from '../block/block_parameter.js';

package/src/messaging/l2_to_l1_membership.ts

@@ -2,6 +2,7 @@ import { MAX_CHECKPOINTS_PER_EPOCH, OUT_HASH_TREE_LEAF_COUNT } from '@aztec-labs
 import { type EpochNumber, EpochNumberSchema } from '@aztec-labs/foundation/branded-types';
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
 import { SiblingPath, UnbalancedMerkleTreeCalculator, computeUnbalancedShaRoot } from '@aztec-labs/foundation/trees';
+import { Buffer } from 'buffer';
 import { z } from 'zod';
 
 import type { AztecNode } from '../interfaces/aztec-node.js';

package/src/messaging/out_hash.ts

@@ -2,6 +2,7 @@ import { OUT_HASH_TREE_LEAF_COUNT } from '@aztec-labs/constants';
 import { padArrayEnd } from '@aztec-labs/foundation/collection';
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
 import { computeBalancedShaRoot, computeUnbalancedShaRoot, computeWonkyShaRoot } from '@aztec-labs/foundation/trees';
+import { Buffer } from 'buffer';
 
 /**
  * Computes the out hash of a tx. It's the root of the unbalanced tree of all the L2-to-L1 messages emitted by the tx.

package/src/note/note_dao.ts

@@ -4,6 +4,7 @@ import { BufferReader, serializeToBuffer } from '@aztec-labs/foundation/serializ
 import { AztecAddress } from '@aztec-labs/stdlib/aztec-address';
 import { Note } from '@aztec-labs/stdlib/note';
 import { TxHash } from '@aztec-labs/stdlib/tx';
+import { Buffer } from 'buffer';
 
 /**
  * A Note Data Access Object, representing a note that was committed to the note hash tree, holding all of the

package/src/proofs/chonk_proof.ts

@@ -6,6 +6,7 @@ import { randomBytes } from '@aztec-labs/foundation/crypto/random';
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
 import { bufferSchemaFor } from '@aztec-labs/foundation/schemas';
 import { BufferReader, BufferSink } from '@aztec-labs/foundation/serialize';
+import { Buffer } from 'buffer';
 
 /**
  * Serialization format detection for ChonkProof is value-based on the leading uint32:

package/src/proofs/proof.ts

@@ -2,7 +2,8 @@ import { PAIRING_POINTS_SIZE } from '@aztec-labs/constants';
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
 import { BufferReader, serializeToBuffer } from '@aztec-labs/foundation/serialize';
 import { bufferToHex, hexToBuffer } from '@aztec-labs/foundation/string';
-import { strict as assert } from 'assert';
+import { assert } from '@aztec-labs/foundation/validation';
+import { Buffer } from 'buffer';
 
 const EMPTY_PROOF_SIZE = 42;
 

package/src/rollup/block_headers_hash.ts

@@ -2,6 +2,7 @@ import { DomainSeparator } from '@aztec-labs/constants';
 import { poseidon2HashWithSeparator } from '@aztec-labs/foundation/crypto/poseidon';
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
 import { computeUnbalancedMerkleTreeRootAsync } from '@aztec-labs/foundation/trees';
+import { Buffer } from 'buffer';
 
 import type { BlockHeader } from '../tx/block_header.js';
 

package/src/rollup/checkpoint_constant_data.ts

@@ -1,9 +1,9 @@
 import { SlotNumber } from '@aztec-labs/foundation/branded-types';
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
 import { EthAddress } from '@aztec-labs/foundation/eth-address';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import { BufferReader, serializeToBuffer } from '@aztec-labs/foundation/serialize';
 import type { FieldsOf } from '@aztec-labs/foundation/types';
-import { inspect } from 'util';
 
 import { AztecAddress } from '../aztec-address/index.js';
 import { GasFees } from '../gas/gas_fees.js';

package/src/rollup/checkpoint_header.ts

@@ -3,11 +3,11 @@ import { SlotNumber } from '@aztec-labs/foundation/branded-types';
 import { sha256ToField } from '@aztec-labs/foundation/crypto/sha256';
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
 import { EthAddress } from '@aztec-labs/foundation/eth-address';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import type { ZodFor } from '@aztec-labs/foundation/schemas';
 import { BufferReader, bigintToUInt64BE, serializeToBuffer } from '@aztec-labs/foundation/serialize';
 import { bufferToHex, hexToBuffer } from '@aztec-labs/foundation/string';
 import type { FieldsOf } from '@aztec-labs/foundation/types';
-import { inspect } from 'util';
 import { z } from 'zod';
 
 import { AztecAddress } from '../aztec-address/index.js';

package/src/rollup/checkpoint_root_rollup_private_inputs.ts

@@ -6,6 +6,7 @@ import { bufferSchemaFor } from '@aztec-labs/foundation/schemas';
 import { BufferReader, type Tuple, serializeToBuffer } from '@aztec-labs/foundation/serialize';
 import { bufferToHex, hexToBuffer } from '@aztec-labs/foundation/string';
 import type { FieldsOf } from '@aztec-labs/foundation/types';
+import { Buffer } from 'buffer';
 
 import { ParityPublicInputs } from '../parity/parity_public_inputs.js';
 import { ProofData, type RollupHonkProofData, type UltraHonkProofData } from '../proofs/proof_data.js';

package/src/slashing/votes.ts

@@ -3,6 +3,7 @@ import { padArrayEnd } from '@aztec-labs/foundation/collection';
 import { EthAddress } from '@aztec-labs/foundation/eth-address';
 import { type Logger, createLogger } from '@aztec-labs/foundation/log';
 import type { PartialBy } from '@aztec-labs/foundation/types';
+import { Buffer } from 'buffer';
 
 import { getEpochForOffense } from './helpers.js';
 import type { Offense, ValidatorSlashVote } from './types.js';

package/src/snapshots/download.ts

@@ -1,6 +1,7 @@
 import { fromEntries, getEntries, maxBy } from '@aztec-labs/foundation/collection';
 import { jsonParseWithSchema } from '@aztec-labs/foundation/json-rpc';
 import type { ReadOnlyFileStore } from '@aztec-labs/stdlib/file-store';
+import { Buffer } from 'buffer';
 import { createReadStream, createWriteStream } from 'fs';
 import fs from 'fs/promises';
 import pathMod from 'path';
@@ -94,29 +95,37 @@ export async function downloadSnapshot(
   snapshot: Pick<SnapshotMetadata, 'dataUrls'>,
   localPaths: Record<SnapshotDataKeys, string>,
   store: ReadOnlyFileStore,
+  keys: readonly SnapshotDataKeys[] = SnapshotDataKeys,
 ): Promise<void> {
-  await Promise.all(
-    getEntries(localPaths).map(async ([key, path]) => {
-      await fs.mkdir(pathMod.dirname(path), { recursive: true });
+  const results = await Promise.allSettled(
+    getEntries(localPaths)
+      .filter(([key]) => keys.includes(key))
+      .map(async ([key, path]) => {
+        await fs.mkdir(pathMod.dirname(path), { recursive: true });
 
-      const tmpPath = `${path}.download`;
-      try {
-        const url = snapshot.dataUrls[key];
-        await store.download(url, tmpPath);
+        const tmpPath = `${path}.download`;
+        try {
+          const url = snapshot.dataUrls[key];
+          await store.download(url, tmpPath);
 
-        const isGzip = await detectGzip(tmpPath);
+          const isGzip = await detectGzip(tmpPath);
 
-        const read = createReadStream(tmpPath);
-        const write = createWriteStream(path);
-        if (isGzip) {
-          const gunzip = createGunzip();
-          await pipeline(read, gunzip, write);
-        } else {
-          await pipeline(read, write);
+          const read = createReadStream(tmpPath);
+          const write = createWriteStream(path);
+          if (isGzip) {
+            const gunzip = createGunzip();
+            await pipeline(read, gunzip, write);
+          } else {
+            await pipeline(read, write);
+          }
+        } finally {
+          await fs.unlink(tmpPath).catch(() => undefined);
         }
-      } finally {
-        await fs.unlink(tmpPath).catch(() => undefined);
-      }
-    }),
+      }),
   );
+  for (const result of results) {
+    if (result.status === 'rejected') {
+      throw result.reason;
+    }
+  }
 }

package/src/snapshots/index.ts

@@ -1,3 +1,9 @@
 export * from './types.js';
 export * from './upload.js';
-export { getBasePath, downloadSnapshot, getLatestSnapshotMetadata, makeSnapshotPaths } from './download.js';
+export {
+  getBasePath,
+  downloadSnapshot,
+  getSnapshotIndex,
+  getLatestSnapshotMetadata,
+  makeSnapshotPaths,
+} from './download.js';

package/src/snapshots/upload.ts

@@ -2,6 +2,7 @@ import { fromEntries, getEntries, pick } from '@aztec-labs/foundation/collection
 import { jsonStringify } from '@aztec-labs/foundation/json-rpc';
 import { isoDate } from '@aztec-labs/foundation/string';
 import type { FileStore } from '@aztec-labs/stdlib/file-store';
+import { Buffer } from 'buffer';
 
 import { getBasePath, getSnapshotIndex, getSnapshotIndexPath } from './download.js';
 import type { SnapshotDataKeys, SnapshotMetadata, SnapshotsIndex, UploadSnapshotMetadata } from './types.js';

package/src/tests/factories.ts

@@ -52,6 +52,7 @@ import { GrumpkinScalar, Point } from '@aztec-labs/foundation/curves/grumpkin';
 import { EthAddress } from '@aztec-labs/foundation/eth-address';
 import type { Bufferable, Serializable, Tuple } from '@aztec-labs/foundation/serialize';
 import { MembershipWitness } from '@aztec-labs/foundation/trees';
+import { Buffer } from 'buffer';
 
 import { FunctionSelector } from '../abi/function_selector.js';
 import {

package/src/tests/mocks.ts

@@ -18,6 +18,7 @@ import { Secp256k1Signer } from '@aztec-labs/foundation/crypto/secp256k1-signer'
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
 import { EthAddress } from '@aztec-labs/foundation/eth-address';
 import { Signature } from '@aztec-labs/foundation/eth-signature';
+import { Buffer } from 'buffer';
 import { type TypedDataDefinition, hashTypedData } from 'viem';
 
 import type { ContractArtifact } from '../abi/abi.js';

package/src/trees/append_only_tree_snapshot.ts

@@ -1,9 +1,9 @@
 import { TreeLeafIndex, TreeLeafIndexSchema } from '@aztec-labs/foundation/branded-types';
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import { schemas } from '@aztec-labs/foundation/schemas';
 import { BufferReader, BufferSink, FieldReader } from '@aztec-labs/foundation/serialize';
 import { bufferToHex, hexToBuffer } from '@aztec-labs/foundation/string';
-import { inspect } from 'util';
 import { z } from 'zod';
 
 /**

package/src/trees/nullifier_leaf.ts

@@ -3,6 +3,7 @@ import { Fr } from '@aztec-labs/foundation/curves/bn254';
 import { schemas } from '@aztec-labs/foundation/schemas';
 import { BufferReader } from '@aztec-labs/foundation/serialize';
 import type { IndexedTreeLeaf, IndexedTreeLeafPreimage } from '@aztec-labs/foundation/trees';
+import { Buffer } from 'buffer';
 import { z } from 'zod';
 
 /**

package/src/trees/public_data_leaf.ts

@@ -4,6 +4,7 @@ import { Fr } from '@aztec-labs/foundation/curves/bn254';
 import { schemas } from '@aztec-labs/foundation/schemas';
 import { BufferReader, serializeToBuffer } from '@aztec-labs/foundation/serialize';
 import type { IndexedTreeLeaf, IndexedTreeLeafPreimage } from '@aztec-labs/foundation/trees';
+import { Buffer } from 'buffer';
 import { z } from 'zod';
 
 /**

package/src/tx/block_header.ts

@@ -3,6 +3,7 @@ import { BlockNumber, SlotNumber } from '@aztec-labs/foundation/branded-types';
 import { poseidon2HashWithSeparator } from '@aztec-labs/foundation/crypto/poseidon';
 import { randomInt } from '@aztec-labs/foundation/crypto/random';
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import { type ZodFor, schemas } from '@aztec-labs/foundation/schemas';
 import {
   BufferReader,
@@ -13,7 +14,6 @@ import {
 } from '@aztec-labs/foundation/serialize';
 import { bufferToHex, hexToBuffer } from '@aztec-labs/foundation/string';
 import type { FieldsOf } from '@aztec-labs/foundation/types';
-import { inspect } from 'util';
 import { z } from 'zod';
 
 import { BlockHash } from '../block/block_hash.js';

package/src/tx/call_context.ts

@@ -1,8 +1,8 @@
 import { CALL_CONTEXT_LENGTH } from '@aztec-labs/constants';
 import type { Fr } from '@aztec-labs/foundation/curves/bn254';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import { BufferReader, FieldReader, serializeToBuffer, serializeToFields } from '@aztec-labs/foundation/serialize';
 import type { FieldsOf } from '@aztec-labs/foundation/types';
-import { inspect } from 'util';
 import { z } from 'zod';
 
 import { FunctionSelector } from '../abi/index.js';

package/src/tx/global_variables.ts

@@ -3,6 +3,7 @@ import { BlockNumber, BlockNumberSchema, SlotNumber } from '@aztec-labs/foundati
 import { randomInt } from '@aztec-labs/foundation/crypto/random';
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
 import { EthAddress } from '@aztec-labs/foundation/eth-address';
+import { inspect } from '@aztec-labs/foundation/inspect';
 import { jsonStringify } from '@aztec-labs/foundation/json-rpc';
 import {
   BufferReader,
@@ -12,7 +13,6 @@ import {
   serializeToSink,
 } from '@aztec-labs/foundation/serialize';
 import type { FieldsOf } from '@aztec-labs/foundation/types';
-import { inspect } from 'util';
 import { z } from 'zod';
 
 import { AztecAddress } from '../aztec-address/index.js';

package/src/tx/profiling.ts

@@ -2,6 +2,7 @@ import { CircuitKind } from '@aztec-foundation/bb.js';
 
 import { Fr } from '@aztec-labs/foundation/curves/bn254';
 import { type ZodFor, optional, schemas } from '@aztec-labs/foundation/schemas';
+import { Buffer } from 'buffer';
 import { z } from 'zod';
 
 import { AztecAddress } from '../aztec-address/index.js';