@aztec/stdlib 0.0.0-test.1 → 0.0.1-fake-ceab37513c

package/src/snapshots/types.ts

@@ -0,0 +1,70 @@
+import type { EthAddress } from '@aztec/foundation/eth-address';
+import { type ZodFor, schemas } from '@aztec/foundation/schemas';
+
+import { z } from 'zod';
+
+export const SnapshotDataKeys = [
+  'archiver',
+  'nullifier-tree',
+  'public-data-tree',
+  'note-hash-tree',
+  'archive-tree',
+  'l1-to-l2-message-tree',
+] as const;
+
+export type SnapshotDataKeys = (typeof SnapshotDataKeys)[number];
+
+export type SnapshotDataUrls = Record<SnapshotDataKeys, string>;
+
+export type SnapshotMetadata = {
+  l2BlockNumber: number;
+  l2BlockHash: string;
+  l1BlockNumber: number;
+  timestamp: number;
+  dataUrls: SnapshotDataUrls;
+  schemaVersions: { archiver: number; worldState: number };
+};
+
+export type SnapshotsIndexMetadata = {
+  l1ChainId: number;
+  rollupVersion: number;
+  rollupAddress: EthAddress;
+};
+
+export type SnapshotsIndex = SnapshotsIndexMetadata & {
+  snapshots: SnapshotMetadata[];
+};
+
+export type UploadSnapshotMetadata = Pick<SnapshotMetadata, 'l2BlockNumber' | 'l2BlockHash' | 'l1BlockNumber'> &
+  Pick<SnapshotsIndex, 'l1ChainId' | 'rollupVersion' | 'rollupAddress'>;
+
+export const SnapshotsIndexSchema = z.object({
+  l1ChainId: z.number(),
+  rollupVersion: z.number(),
+  rollupAddress: schemas.EthAddress,
+  snapshots: z.array(
+    z.object({
+      l2BlockNumber: z.number(),
+      l2BlockHash: z.string(),
+      l1BlockNumber: z.number(),
+      timestamp: z.number(),
+      schemaVersions: z.object({
+        archiver: z.number(),
+        worldState: z.number(),
+      }),
+      dataUrls: z
+        .record(z.enum(SnapshotDataKeys), z.string())
+        // See https://stackoverflow.com/questions/77958464/zod-record-with-required-keys
+        .refine((obj): obj is Required<typeof obj> => SnapshotDataKeys.every(key => !!obj[key])),
+    }),
+  ),
+}) satisfies ZodFor<SnapshotsIndex>;
+
+export const UploadSnapshotMetadataSchema = z.object({
+  l2BlockNumber: z.number(),
+  l2BlockHash: z.string(),
+  l1BlockNumber: z.number(),
+  l1ChainId: z.number(),
+  rollupVersion: z.number(),
+  rollupAddress: schemas.EthAddress,
+}) satisfies ZodFor<UploadSnapshotMetadata>;

package/src/snapshots/upload.ts

@@ -0,0 +1,64 @@
+import { fromEntries, getEntries, pick } from '@aztec/foundation/collection';
+import { jsonStringify } from '@aztec/foundation/json-rpc';
+import { isoDate } from '@aztec/foundation/string';
+import type { FileStore } from '@aztec/stdlib/file-store';
+
+import { getBasePath, getSnapshotIndex, getSnapshotIndexPath } from './download.js';
+import type { SnapshotDataKeys, SnapshotMetadata, SnapshotsIndex, UploadSnapshotMetadata } from './types.js';
+
+export async function uploadSnapshotData(
+  localPaths: Record<SnapshotDataKeys, string>,
+  schemaVersions: SnapshotMetadata['schemaVersions'],
+  metadata: UploadSnapshotMetadata,
+  store: FileStore,
+  opts: { pathFor?: (key: SnapshotDataKeys) => string; private?: boolean } = {},
+): Promise<SnapshotMetadata> {
+  const timestamp = Date.now();
+  const date = isoDate();
+  const basePath = getBasePath(metadata);
+  const targetPathFor =
+    opts.pathFor ?? ((key: SnapshotDataKeys) => `${basePath}/${key}-${date}-${metadata.l2BlockHash}.db`);
+
+  const dataUrls = fromEntries(
+    await Promise.all(
+      getEntries(localPaths).map(
+        async ([key, path]) =>
+          [key, await store.upload(targetPathFor(key), path, { compress: true, public: !opts.private })] as const,
+      ),
+    ),
+  );
+
+  return {
+    ...pick(metadata, 'l1BlockNumber', 'l2BlockHash', 'l2BlockNumber'),
+    schemaVersions,
+    timestamp,
+    dataUrls,
+  };
+}
+
+export async function uploadSnapshotToIndex(
+  localPaths: Record<SnapshotDataKeys, string>,
+  schemaVersions: SnapshotMetadata['schemaVersions'],
+  metadata: UploadSnapshotMetadata,
+  store: FileStore,
+): Promise<SnapshotMetadata> {
+  const newSnapshotMetadata = await uploadSnapshotData(localPaths, schemaVersions, metadata, store);
+  const snapshotsIndex = (await getSnapshotIndex(metadata, store)) ?? createEmptyIndex(metadata);
+  snapshotsIndex.snapshots.unshift(newSnapshotMetadata);
+
+  await store.save(getSnapshotIndexPath(metadata), Buffer.from(jsonStringify(snapshotsIndex, true)), {
+    public: true, // Make the index publicly accessible
+    compress: false, // Ensure index.json is not gzipped
+    metadata: { ['Cache-control']: 'no-store' }, // Do not cache object versions
+  });
+  return newSnapshotMetadata;
+}
+
+function createEmptyIndex(
+  metadata: Pick<SnapshotsIndex, 'l1ChainId' | 'rollupVersion' | 'rollupAddress'>,
+): SnapshotsIndex {
+  return {
+    ...pick(metadata, 'l1ChainId', 'rollupVersion', 'rollupAddress'),
+    snapshots: [],
+  };
+}

package/src/stats/stats.ts

@@ -81,6 +81,8 @@ export type ClientCircuitName =
   | 'private-kernel-reset'
   | 'private-kernel-tail'
   | 'private-kernel-tail-to-public'
+  | 'hiding-kernel-to-rollup'
+  | 'hiding-kernel-to-public'
   | 'app-circuit';
 
 export type ServerCircuitName =
@@ -92,6 +94,7 @@ export type ServerCircuitName =
   | 'block-root-rollup'
   | 'single-tx-block-root-rollup'
   | 'empty-block-root-rollup'
+  | 'padding-block-root-rollup'
   | 'block-merge-rollup'
   | 'root-rollup'
   | 'avm-circuit'
@@ -199,9 +202,9 @@ export type L2BlockHandledStats = {
   /** Total duration in ms. */
   duration: number;
   /** Pending block number. */
-  unfinalisedBlockNumber: bigint;
+  unfinalizedBlockNumber: bigint;
   /** Proven block number. */
-  finalisedBlockNumber: bigint;
+  finalizedBlockNumber: bigint;
   /** Oldest historic block number. */
   oldestHistoricBlock: bigint;
 } & L2BlockStats;
@@ -220,12 +223,12 @@ export type TxStats = {
   nullifierCount: number;
   /** Number of private logs */
   privateLogCount: number;
-  /** How many classes were registered through the canonical class registerer. */
-  classRegisteredCount: number;
+  /** How many classes were published to the canonical class registry. */
+  classPublishedCount: number;
   /** Serialized size of contract class logs in fields. */
   contractClassLogSize: number;
   /** How this tx pays for its fee */
-  feePaymentMethod: 'fee_juice' | 'fpc_public' | 'fpc_private';
+  feePaymentMethod: 'fee_juice' | 'fpc';
 };
 
 /**