feat(public): only deploy/register public_dispatch

yarn-project/circuits.js/src/contract/contract_class.ts

@@ -1,10 +1,10 @@
 import { type ContractArtifact, type FunctionArtifact, FunctionSelector, FunctionType } from '@aztec/foundation/abi';
 import { Fr } from '@aztec/foundation/fields';
-import { type ContractClass, type ContractClassWithId } from '@aztec/types/contracts';
+import { type ContractClass, type ContractClassWithId, type PublicFunction } from '@aztec/types/contracts';
 
+import { PUBLIC_DISPATCH_SELECTOR } from '../constants.gen.js';
 import { computeArtifactHash } from './artifact_hash.js';
 import { type ContractClassIdPreimage, computeContractClassIdWithPreimage } from './contract_class_id.js';
-import { packBytecode } from './public_bytecode.js';
 
 /** Contract artifact including its artifact hash */
 type ContractArtifactWithHash = ContractArtifact & { artifactHash: Fr };
@@ -17,15 +17,27 @@ export function getContractClassFromArtifact(
   artifact: ContractArtifact | ContractArtifactWithHash,
 ): ContractClassWithId & ContractClassIdPreimage {
   const artifactHash = 'artifactHash' in artifact ? artifact.artifactHash : computeArtifactHash(artifact);
-  const publicFunctions: ContractClass['publicFunctions'] = artifact.functions
+  const artifactPublicFunctions: ContractClass['publicFunctions'] = artifact.functions
     .filter(f => f.functionType === FunctionType.PUBLIC)
     .map(f => ({
       selector: FunctionSelector.fromNameAndParameters(f.name, f.parameters),
       bytecode: f.bytecode,
     }))
     .sort(cmpFunctionArtifacts);
 
-  const packedBytecode = packBytecode(publicFunctions);
+  let packedBytecode = Buffer.alloc(0);
+  let dispatchFunction: PublicFunction | undefined = undefined;
+  if (artifactPublicFunctions.length > 0) {
+    dispatchFunction = artifactPublicFunctions.find(f =>
+      f.selector.equals(FunctionSelector.fromField(new Fr(PUBLIC_DISPATCH_SELECTOR))),
+    );
+    if (!dispatchFunction) {
+      throw new Error(
+        'A contract with public functions should define a public_dispatch(Field) function as its public entrypoint.',
+      );
+    }
+    packedBytecode = dispatchFunction.bytecode;
+  }
 
   const privateFunctions: ContractClass['privateFunctions'] = artifact.functions
     .filter(f => f.functionType === FunctionType.PRIVATE)
@@ -35,7 +47,8 @@ export function getContractClassFromArtifact(
   const contractClass: ContractClass = {
     version: 1,
     artifactHash,
-    publicFunctions,
+    // TODO(https://github.com/AztecProtocol/aztec-packages/issues/8985): Remove public functions.
+    publicFunctions: dispatchFunction ? [dispatchFunction] : [],
     packedBytecode,
     privateFunctions,
   };

yarn-project/circuits.js/src/contract/events/contract_class_registered_event.ts

@@ -1,15 +1,14 @@
-import { bufferFromFields } from '@aztec/foundation/abi';
+import { FunctionSelector, bufferFromFields } from '@aztec/foundation/abi';
 import { type AztecAddress } from '@aztec/foundation/aztec-address';
 import { toBigIntBE } from '@aztec/foundation/bigint-buffer';
 import { Fr } from '@aztec/foundation/fields';
 import { BufferReader } from '@aztec/foundation/serialize';
-import { type ContractClassPublic } from '@aztec/types/contracts';
+import { type ContractClassPublic, type PublicFunction } from '@aztec/types/contracts';
 
 import chunk from 'lodash.chunk';
 
-import { REGISTERER_CONTRACT_CLASS_REGISTERED_MAGIC_VALUE } from '../../constants.gen.js';
+import { PUBLIC_DISPATCH_SELECTOR, REGISTERER_CONTRACT_CLASS_REGISTERED_MAGIC_VALUE } from '../../constants.gen.js';
 import { computeContractClassId, computePublicBytecodeCommitment } from '../contract_class_id.js';
-import { unpackBytecode } from '../public_bytecode.js';
 
 /** Event emitted from the ContractClassRegisterer. */
 export class ContractClassRegisteredEvent {
@@ -73,12 +72,21 @@ export class ContractClassRegisteredEvent {
       throw new Error(`Unexpected contract class version ${this.version}`);
     }
 
+    // TODO(https://github.com/AztecProtocol/aztec-packages/issues/8985): Remove public functions.
+    const publicFunctions: PublicFunction[] = [];
+    if (this.packedPublicBytecode.length > 0) {
+      publicFunctions.push({
+        selector: FunctionSelector.fromField(new Fr(PUBLIC_DISPATCH_SELECTOR)),
+        bytecode: this.packedPublicBytecode,
+      });
+    }
+
     return {
       id: this.contractClassId,
       artifactHash: this.artifactHash,
       packedBytecode: this.packedPublicBytecode,
       privateFunctionsRoot: this.privateFunctionsRoot,
-      publicFunctions: unpackBytecode(this.packedPublicBytecode),
+      publicFunctions: publicFunctions,
       version: this.version,
       privateFunctions: [],
       unconstrainedFunctions: [],

yarn-project/circuits.js/src/contract/index.ts

@@ -8,6 +8,5 @@ export * from './events/contract_instance_deployed_event.js';
 export * from './events/private_function_broadcasted_event.js';
 export * from './events/unconstrained_function_broadcasted_event.js';
 export * from './private_function.js';
-export * from './public_bytecode.js';
 export * from './private_function_membership_proof.js';
 export * from './unconstrained_function_membership_proof.js';

yarn-project/circuits.js/src/tests/factories.ts

@@ -138,7 +138,6 @@ import {
   computeContractClassId,
   computePublicBytecodeCommitment,
   makeRecursiveProof,
-  packBytecode,
 } from '../index.js';
 import { ContentCommitment, NUM_BYTES_PER_SHA256 } from '../structs/content_commitment.js';
 import { Gas } from '../structs/gas.js';
@@ -1362,7 +1361,7 @@ export function makeContractClassPublic(seed = 0): ContractClassPublic {
   const artifactHash = fr(seed + 1);
   const publicFunctions = makeTuple(3, makeContractClassPublicFunction, seed + 2);
   const privateFunctionsRoot = fr(seed + 3);
-  const packedBytecode = packBytecode(publicFunctions);
+  const packedBytecode = publicFunctions[0].bytecode;
   const publicBytecodeCommitment = computePublicBytecodeCommitment(packedBytecode);
   const id = computeContractClassId({ artifactHash, privateFunctionsRoot, publicBytecodeCommitment });
   return {

yarn-project/types/src/contracts/contract_class.ts

@@ -15,9 +15,10 @@ export interface ContractClass {
   artifactHash: Fr;
   /** List of individual private functions, constructors included. */
   privateFunctions: PrivateFunction[];
-  /** List of individual public functions. Should be removed once we switch to the AVM where all public bytecode is bundled together. */
+  // TODO(https://github.com/AztecProtocol/aztec-packages/issues/8985): Remove public functions.
+  /** Contains the public_dispatch function (and only that) if there's any public code in the contract. */
   publicFunctions: PublicFunction[];
-  /** Packed bytecode representation of the AVM bytecode for all public functions in this contract. Unused for now, see `publicFunctions`. */
+  /** Bytecode for the public_dispatch function, or empty. */
   packedBytecode: Buffer;
 }