@aztec/txe 0.0.1-commit.d6f2b3f94 → 0.0.1-commit.d939eb5aa

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (60) hide show
  1. package/dest/index.d.ts +1 -1
  2. package/dest/index.d.ts.map +1 -1
  3. package/dest/index.js +88 -54
  4. package/dest/oracle/interfaces.d.ts +29 -28
  5. package/dest/oracle/interfaces.d.ts.map +1 -1
  6. package/dest/oracle/txe_oracle_public_context.d.ts +13 -13
  7. package/dest/oracle/txe_oracle_public_context.d.ts.map +1 -1
  8. package/dest/oracle/txe_oracle_public_context.js +12 -12
  9. package/dest/oracle/txe_oracle_top_level_context.d.ts +28 -23
  10. package/dest/oracle/txe_oracle_top_level_context.d.ts.map +1 -1
  11. package/dest/oracle/txe_oracle_top_level_context.js +130 -61
  12. package/dest/rpc_translator.d.ts +120 -82
  13. package/dest/rpc_translator.d.ts.map +1 -1
  14. package/dest/rpc_translator.js +384 -162
  15. package/dest/state_machine/archiver.d.ts +3 -3
  16. package/dest/state_machine/archiver.d.ts.map +1 -1
  17. package/dest/state_machine/archiver.js +7 -8
  18. package/dest/state_machine/dummy_p2p_client.d.ts +10 -6
  19. package/dest/state_machine/dummy_p2p_client.d.ts.map +1 -1
  20. package/dest/state_machine/dummy_p2p_client.js +18 -6
  21. package/dest/state_machine/global_variable_builder.d.ts +9 -4
  22. package/dest/state_machine/global_variable_builder.d.ts.map +1 -1
  23. package/dest/state_machine/global_variable_builder.js +9 -3
  24. package/dest/state_machine/index.d.ts +4 -2
  25. package/dest/state_machine/index.d.ts.map +1 -1
  26. package/dest/state_machine/index.js +8 -4
  27. package/dest/state_machine/mock_epoch_cache.d.ts +18 -3
  28. package/dest/state_machine/mock_epoch_cache.d.ts.map +1 -1
  29. package/dest/state_machine/mock_epoch_cache.js +35 -2
  30. package/dest/state_machine/synchronizer.d.ts +5 -5
  31. package/dest/state_machine/synchronizer.d.ts.map +1 -1
  32. package/dest/state_machine/synchronizer.js +3 -3
  33. package/dest/txe_session.d.ts +10 -6
  34. package/dest/txe_session.d.ts.map +1 -1
  35. package/dest/txe_session.js +92 -24
  36. package/dest/util/encoding.d.ts +71 -1
  37. package/dest/util/encoding.d.ts.map +1 -1
  38. package/dest/util/encoding.js +4 -0
  39. package/dest/util/txe_public_contract_data_source.d.ts +2 -3
  40. package/dest/util/txe_public_contract_data_source.d.ts.map +1 -1
  41. package/dest/util/txe_public_contract_data_source.js +6 -25
  42. package/package.json +15 -15
  43. package/src/index.ts +89 -52
  44. package/src/oracle/interfaces.ts +32 -31
  45. package/src/oracle/txe_oracle_public_context.ts +12 -12
  46. package/src/oracle/txe_oracle_top_level_context.ts +142 -103
  47. package/src/rpc_translator.ts +446 -183
  48. package/src/state_machine/archiver.ts +6 -5
  49. package/src/state_machine/dummy_p2p_client.ts +25 -9
  50. package/src/state_machine/global_variable_builder.ts +18 -3
  51. package/src/state_machine/index.ts +9 -2
  52. package/src/state_machine/mock_epoch_cache.ts +46 -3
  53. package/src/state_machine/synchronizer.ts +4 -4
  54. package/src/txe_session.ts +97 -72
  55. package/src/util/encoding.ts +5 -0
  56. package/src/util/txe_public_contract_data_source.ts +10 -38
  57. package/dest/util/txe_contract_store.d.ts +0 -12
  58. package/dest/util/txe_contract_store.d.ts.map +0 -1
  59. package/dest/util/txe_contract_store.js +0 -22
  60. package/src/util/txe_contract_store.ts +0 -36
@@ -17,7 +17,7 @@ export class TXEArchiver extends ArchiverDataSourceBase {
17
17
  private readonly updater = new ArchiverDataStoreUpdater(this.store);
18
18
 
19
19
  constructor(db: AztecAsyncKVStore) {
20
- const store = new KVArchiverDataStore(db, 9999, { epochDuration: 32 });
20
+ const store = new KVArchiverDataStore(db, 9999);
21
21
  super(store);
22
22
  }
23
23
 
@@ -76,15 +76,16 @@ export class TXEArchiver extends ArchiverDataSourceBase {
76
76
  proven: tipId,
77
77
  finalized: tipId,
78
78
  checkpointed: tipId,
79
+ proposedCheckpoint: tipId,
79
80
  };
80
81
  }
81
82
 
82
- public getL2SlotNumber(): Promise<SlotNumber | undefined> {
83
- throw new Error('TXE Archiver does not implement "getL2SlotNumber"');
83
+ public getSyncedL2SlotNumber(): Promise<SlotNumber | undefined> {
84
+ throw new Error('TXE Archiver does not implement "getSyncedL2SlotNumber"');
84
85
  }
85
86
 
86
- public getL2EpochNumber(): Promise<EpochNumber | undefined> {
87
- throw new Error('TXE Archiver does not implement "getL2EpochNumber"');
87
+ public getSyncedL2EpochNumber(): Promise<EpochNumber | undefined> {
88
+ throw new Error('TXE Archiver does not implement "getSyncedL2EpochNumber"');
88
89
  }
89
90
 
90
91
  public isEpochComplete(_epochNumber: EpochNumber): Promise<boolean> {
@@ -16,12 +16,12 @@ import type {
16
16
  StatusMessage,
17
17
  } from '@aztec/p2p';
18
18
  import type { EthAddress, L2BlockStreamEvent, L2Tips } from '@aztec/stdlib/block';
19
- import type { PeerInfo } from '@aztec/stdlib/interfaces/server';
20
- import type { BlockProposal, CheckpointAttestation, CheckpointProposal } from '@aztec/stdlib/p2p';
19
+ import type { ITxProvider, PeerInfo } from '@aztec/stdlib/interfaces/server';
20
+ import type { BlockProposal, CheckpointAttestation, CheckpointProposal, TopicType } from '@aztec/stdlib/p2p';
21
21
  import type { BlockHeader, Tx, TxHash } from '@aztec/stdlib/tx';
22
22
 
23
23
  export class DummyP2P implements P2P {
24
- public validate(_txs: Tx[]): Promise<void> {
24
+ public validateTxsReceivedInBlockProposal(_txs: Tx[]): Promise<void> {
25
25
  return Promise.resolve();
26
26
  }
27
27
 
@@ -41,6 +41,10 @@ export class DummyP2P implements P2P {
41
41
  throw new Error('DummyP2P does not implement "getPeers"');
42
42
  }
43
43
 
44
+ public getGossipMeshPeerCount(_topicType: TopicType): Promise<number> {
45
+ return Promise.resolve(0);
46
+ }
47
+
44
48
  public broadcastProposal(_proposal: BlockProposal): Promise<void> {
45
49
  throw new Error('DummyP2P does not implement "broadcastProposal"');
46
50
  }
@@ -57,8 +61,12 @@ export class DummyP2P implements P2P {
57
61
  throw new Error('DummyP2P does not implement "registerBlockProposalHandler"');
58
62
  }
59
63
 
60
- public registerCheckpointProposalHandler(_handler: P2PCheckpointReceivedCallback): void {
61
- throw new Error('DummyP2P does not implement "registerCheckpointProposalHandler"');
64
+ public registerValidatorCheckpointProposalHandler(_handler: P2PCheckpointReceivedCallback): void {
65
+ throw new Error('DummyP2P does not implement "registerValidatorCheckpointProposalHandler"');
66
+ }
67
+
68
+ public registerAllNodesCheckpointProposalHandler(_handler: P2PCheckpointReceivedCallback): void {
69
+ throw new Error('DummyP2P does not implement "registerAllNodesCheckpointProposalHandler"');
62
70
  }
63
71
 
64
72
  public requestTxs(_txHashes: TxHash[]): Promise<(Tx | undefined)[]> {
@@ -99,6 +107,10 @@ export class DummyP2P implements P2P {
99
107
  throw new Error('DummyP2P does not implement "iteratePendingTxs"');
100
108
  }
101
109
 
110
+ public iterateEligiblePendingTxs(): AsyncIterableIterator<Tx> {
111
+ throw new Error('DummyP2P does not implement "iterateEligiblePendingTxs"');
112
+ }
113
+
102
114
  public getPendingTxCount(): Promise<number> {
103
115
  throw new Error('DummyP2P does not implement "getPendingTxCount"');
104
116
  }
@@ -127,6 +139,10 @@ export class DummyP2P implements P2P {
127
139
  throw new Error('DummyP2P does not implement "isP2PClient"');
128
140
  }
129
141
 
142
+ public getTxProvider(): ITxProvider {
143
+ throw new Error('DummyP2P does not implement "getTxProvider"');
144
+ }
145
+
130
146
  public getTxsByHash(_txHashes: TxHash[]): Promise<Tx[]> {
131
147
  throw new Error('DummyP2P does not implement "getTxsByHash"');
132
148
  }
@@ -167,10 +183,6 @@ export class DummyP2P implements P2P {
167
183
  throw new Error('DummyP2P does not implement "hasTxsInPool"');
168
184
  }
169
185
 
170
- public addTxsToPool(_txs: Tx[]): Promise<number> {
171
- throw new Error('DummyP2P does not implement "addTxs"');
172
- }
173
-
174
186
  public getSyncedLatestBlockNum(): Promise<number> {
175
187
  throw new Error('DummyP2P does not implement "getSyncedLatestBlockNum"');
176
188
  }
@@ -212,4 +224,8 @@ export class DummyP2P implements P2P {
212
224
  public registerDuplicateAttestationCallback(_callback: P2PDuplicateAttestationCallback): void {
213
225
  throw new Error('DummyP2P does not implement "registerDuplicateAttestationCallback"');
214
226
  }
227
+
228
+ public hasBlockProposalsForSlot(_slot: SlotNumber): Promise<boolean> {
229
+ throw new Error('DummyP2P does not implement "hasBlockProposalsForSlot"');
230
+ }
215
231
  }
@@ -1,15 +1,29 @@
1
+ import type { SimulationOverridesPlan } from '@aztec/ethereum/contracts';
1
2
  import { BlockNumber, type SlotNumber } from '@aztec/foundation/branded-types';
3
+ import { times } from '@aztec/foundation/collection';
2
4
  import type { EthAddress } from '@aztec/foundation/eth-address';
3
5
  import type { AztecAddress } from '@aztec/stdlib/aztec-address';
4
- import { GasFees } from '@aztec/stdlib/gas';
6
+ import { FEE_ORACLE_LAG, GasFees } from '@aztec/stdlib/gas';
5
7
  import { makeGlobalVariables } from '@aztec/stdlib/testing';
6
- import { type CheckpointGlobalVariables, type GlobalVariableBuilder, GlobalVariables } from '@aztec/stdlib/tx';
8
+ import {
9
+ type CheckpointGlobalVariables,
10
+ type FeeProvider,
11
+ type GlobalVariableBuilder,
12
+ GlobalVariables,
13
+ } from '@aztec/stdlib/tx';
7
14
 
8
- export class TXEGlobalVariablesBuilder implements GlobalVariableBuilder {
15
+ /** Simple FeeProvider for TXE that returns zero fees. */
16
+ export class TXEFeeProvider implements FeeProvider {
9
17
  public getCurrentMinFees(): Promise<GasFees> {
10
18
  return Promise.resolve(new GasFees(0, 0));
11
19
  }
12
20
 
21
+ public getPredictedMinFees(): Promise<GasFees[]> {
22
+ return Promise.resolve(times(FEE_ORACLE_LAG, () => new GasFees(0, 0)));
23
+ }
24
+ }
25
+
26
+ export class TXEGlobalVariablesBuilder implements GlobalVariableBuilder {
13
27
  public buildGlobalVariables(
14
28
  _blockNumber: BlockNumber,
15
29
  _coinbase: EthAddress,
@@ -23,6 +37,7 @@ export class TXEGlobalVariablesBuilder implements GlobalVariableBuilder {
23
37
  _coinbase: EthAddress,
24
38
  _feeRecipient: AztecAddress,
25
39
  _slotNumber: SlotNumber,
40
+ _simulationOverridesPlan?: SimulationOverridesPlan,
26
41
  ): Promise<CheckpointGlobalVariables> {
27
42
  const vars = makeGlobalVariables();
28
43
  return Promise.resolve({
@@ -4,6 +4,7 @@ import { CheckpointNumber } from '@aztec/foundation/branded-types';
4
4
  import { Fr } from '@aztec/foundation/curves/bn254';
5
5
  import { createLogger } from '@aztec/foundation/log';
6
6
  import { type AnchorBlockStore, type ContractStore, ContractSyncService, type NoteStore } from '@aztec/pxe/server';
7
+ import { MessageContextService } from '@aztec/pxe/simulator';
7
8
  import { L2Block } from '@aztec/stdlib/block';
8
9
  import { Checkpoint, L1PublishedData, PublishedCheckpoint } from '@aztec/stdlib/checkpoint';
9
10
  import type { AztecNode } from '@aztec/stdlib/interfaces/client';
@@ -12,7 +13,7 @@ import { getPackageVersion } from '@aztec/stdlib/update-checker';
12
13
 
13
14
  import { TXEArchiver } from './archiver.js';
14
15
  import { DummyP2P } from './dummy_p2p_client.js';
15
- import { TXEGlobalVariablesBuilder } from './global_variable_builder.js';
16
+ import { TXEFeeProvider, TXEGlobalVariablesBuilder } from './global_variable_builder.js';
16
17
  import { MockEpochCache } from './mock_epoch_cache.js';
17
18
  import { TXESynchronizer } from './synchronizer.js';
18
19
 
@@ -26,6 +27,7 @@ export class TXEStateMachine {
26
27
  public archiver: TXEArchiver,
27
28
  public anchorBlockStore: AnchorBlockStore,
28
29
  public contractSyncService: ContractSyncService,
30
+ public messageContextService: MessageContextService,
29
31
  ) {}
30
32
 
31
33
  public static async create(
@@ -50,12 +52,15 @@ export class TXEStateMachine {
50
52
  undefined,
51
53
  undefined,
52
54
  undefined,
55
+ undefined,
53
56
  VERSION,
54
57
  CHAIN_ID,
55
58
  new TXEGlobalVariablesBuilder(),
59
+ new TXEFeeProvider(),
56
60
  new MockEpochCache(),
57
61
  getPackageVersion() ?? '',
58
62
  new TestCircuitVerifier(),
63
+ new TestCircuitVerifier(),
59
64
  undefined,
60
65
  log,
61
66
  );
@@ -67,7 +72,9 @@ export class TXEStateMachine {
67
72
  createLogger('txe:contract_sync'),
68
73
  );
69
74
 
70
- return new this(node, synchronizer, archiver, anchorBlockStore, contractSyncService);
75
+ const messageContextService = new MessageContextService(node);
76
+
77
+ return new this(node, synchronizer, archiver, anchorBlockStore, contractSyncService, messageContextService);
71
78
  }
72
79
 
73
80
  public async handleL2Block(block: L2Block) {
@@ -8,7 +8,7 @@ import { EmptyL1RollupConstants, type L1RollupConstants } from '@aztec/stdlib/ep
8
8
  * Since in TXE we don't validate transactions, mock suffices here.
9
9
  */
10
10
  export class MockEpochCache implements EpochCacheInterface {
11
- getCommittee(): Promise<EpochCommitteeInfo> {
11
+ getCommittee(_slot: SlotTag = 'now'): Promise<EpochCommitteeInfo> {
12
12
  return Promise.resolve({
13
13
  committee: undefined,
14
14
  seed: 0n,
@@ -17,6 +17,22 @@ export class MockEpochCache implements EpochCacheInterface {
17
17
  });
18
18
  }
19
19
 
20
+ getSlotNow(): SlotNumber {
21
+ return SlotNumber(0);
22
+ }
23
+
24
+ getTargetSlot(): SlotNumber {
25
+ return SlotNumber(0);
26
+ }
27
+
28
+ getEpochNow(): EpochNumber {
29
+ return EpochNumber.ZERO;
30
+ }
31
+
32
+ getTargetEpoch(): EpochNumber {
33
+ return EpochNumber.ZERO;
34
+ }
35
+
20
36
  getEpochAndSlotNow(): EpochAndSlot & { nowMs: bigint } {
21
37
  return {
22
38
  epoch: EpochNumber.ZERO,
@@ -26,15 +42,27 @@ export class MockEpochCache implements EpochCacheInterface {
26
42
  };
27
43
  }
28
44
 
29
- getEpochAndSlotInNextL1Slot(): EpochAndSlot & { now: bigint } {
45
+ getEpochAndSlotInNextL1Slot(): EpochAndSlot & { nowSeconds: bigint } {
30
46
  return {
31
47
  epoch: EpochNumber.ZERO,
32
48
  slot: SlotNumber(0),
33
49
  ts: 0n,
34
- now: 0n,
50
+ nowSeconds: 0n,
35
51
  };
36
52
  }
37
53
 
54
+ getTargetEpochAndSlotInNextL1Slot(): EpochAndSlot & { nowSeconds: bigint } {
55
+ return this.getEpochAndSlotInNextL1Slot();
56
+ }
57
+
58
+ isProposerPipeliningEnabled(): boolean {
59
+ return false;
60
+ }
61
+
62
+ pipeliningOffset(): number {
63
+ return 0;
64
+ }
65
+
38
66
  getProposerIndexEncoding(_epoch: EpochNumber, _slot: SlotNumber, _seed: bigint): `0x${string}` {
39
67
  return '0x00';
40
68
  }
@@ -50,6 +78,13 @@ export class MockEpochCache implements EpochCacheInterface {
50
78
  };
51
79
  }
52
80
 
81
+ getTargetAndNextSlot(): { targetSlot: SlotNumber; nextSlot: SlotNumber } {
82
+ return {
83
+ targetSlot: SlotNumber(0),
84
+ nextSlot: SlotNumber(0),
85
+ };
86
+ }
87
+
53
88
  getProposerAttesterAddressInSlot(_slot: SlotNumber): Promise<EthAddress | undefined> {
54
89
  return Promise.resolve(undefined);
55
90
  }
@@ -66,6 +101,14 @@ export class MockEpochCache implements EpochCacheInterface {
66
101
  return Promise.resolve([]);
67
102
  }
68
103
 
104
+ isEscapeHatchOpen(_epoch: EpochNumber): Promise<boolean> {
105
+ return Promise.resolve(false);
106
+ }
107
+
108
+ isEscapeHatchOpenAtSlot(_slot: SlotTag): Promise<boolean> {
109
+ return Promise.resolve(false);
110
+ }
111
+
69
112
  getL1Constants(): L1RollupConstants {
70
113
  return EmptyL1RollupConstants;
71
114
  }
@@ -1,7 +1,7 @@
1
1
  import { NUMBER_OF_L1_L2_MESSAGES_PER_ROLLUP } from '@aztec/constants';
2
2
  import { BlockNumber } from '@aztec/foundation/branded-types';
3
3
  import { Fr } from '@aztec/foundation/curves/bn254';
4
- import type { L2Block } from '@aztec/stdlib/block';
4
+ import type { BlockHash, L2Block } from '@aztec/stdlib/block';
5
5
  import type {
6
6
  MerkleTreeReadOperations,
7
7
  MerkleTreeWriteOperations,
@@ -33,12 +33,12 @@ export class TXESynchronizer implements WorldStateSynchronizer {
33
33
  }
34
34
 
35
35
  /**
36
- * Forces an immediate sync to an optionally provided minimum block number
36
+ * Forces an immediate sync to an optionally provided minimum block number.
37
37
  * @param targetBlockNumber - The target block number that we must sync to. Will download unproven blocks if needed to reach it.
38
- * @param skipThrowIfTargetNotReached - Whether to skip throwing if the target block number is not reached.
38
+ * @param blockHash - If provided, verifies the block at targetBlockNumber matches this hash.
39
39
  * @returns A promise that resolves with the block number the world state was synced to
40
40
  */
41
- public syncImmediate(_minBlockNumber?: BlockNumber, _skipThrowIfTargetNotReached?: boolean): Promise<BlockNumber> {
41
+ public syncImmediate(_minBlockNumber?: BlockNumber, _blockHash?: BlockHash): Promise<BlockNumber> {
42
42
  return Promise.resolve(this.blockNumber);
43
43
  }
44
44
 
@@ -3,11 +3,13 @@ import { Fr } from '@aztec/foundation/curves/bn254';
3
3
  import { type Logger, createLogger } from '@aztec/foundation/log';
4
4
  import { KeyStore } from '@aztec/key-store';
5
5
  import { openTmpStore } from '@aztec/kv-store/lmdb-v2';
6
- import type { ProtocolContract } from '@aztec/protocol-contracts';
7
6
  import {
8
7
  AddressStore,
9
8
  AnchorBlockStore,
9
+ CapsuleService,
10
10
  CapsuleStore,
11
+ ContractStore,
12
+ ContractSyncService,
11
13
  JobCoordinator,
12
14
  NoteService,
13
15
  NoteStore,
@@ -54,7 +56,6 @@ import { TXEArchiver } from './state_machine/archiver.js';
54
56
  import { TXEStateMachine } from './state_machine/index.js';
55
57
  import type { ForeignCallArgs, ForeignCallResult } from './util/encoding.js';
56
58
  import { TXEAccountStore } from './util/txe_account_store.js';
57
- import { TXEContractStore } from './util/txe_contract_store.js';
58
59
  import { getSingleTxBlockRequestHash, insertTxEffectIntoWorldTrees, makeTXEBlock } from './utils/block_creation.js';
59
60
  import { makeTxEffect } from './utils/tx_effect_creation.js';
60
61
 
@@ -113,6 +114,10 @@ export interface TXESessionStateHandler {
113
114
  enterPublicState(contractAddress?: AztecAddress): Promise<void>;
114
115
  enterPrivateState(contractAddress?: AztecAddress, anchorBlockNumber?: BlockNumber): Promise<PrivateContextInputs>;
115
116
  enterUtilityState(contractAddress?: AztecAddress): Promise<void>;
117
+
118
+ // TODO(F-335): Exposing the job info is abstraction breakage - drop the following 2 functions.
119
+ cycleJob(): Promise<string>;
120
+ getCurrentJob(): string;
116
121
  }
117
122
 
118
123
  /**
@@ -131,7 +136,7 @@ export class TXESession implements TXESessionStateHandler {
131
136
  | IPrivateExecutionOracle
132
137
  | IAvmExecutionOracle
133
138
  | ITxeExecutionOracle,
134
- private contractStore: TXEContractStore,
139
+ private contractStore: ContractStore,
135
140
  private noteStore: NoteStore,
136
141
  private keyStore: KeyStore,
137
142
  private addressStore: AddressStore,
@@ -146,14 +151,14 @@ export class TXESession implements TXESessionStateHandler {
146
151
  private chainId: Fr,
147
152
  private version: Fr,
148
153
  private nextBlockTimestamp: bigint,
154
+ private contractSyncService: ContractSyncService,
149
155
  ) {}
150
156
 
151
- static async init(protocolContracts: ProtocolContract[]) {
157
+ static async init(contractStore: ContractStore) {
152
158
  const store = await openTmpStore('txe-session');
153
159
 
154
160
  const addressStore = new AddressStore(store);
155
161
  const privateEventStore = new PrivateEventStore(store);
156
- const contractStore = new TXEContractStore(store);
157
162
  const noteStore = new NoteStore(store);
158
163
  const senderTaggingStore = new SenderTaggingStore(store);
159
164
  const recipientTaggingStore = new RecipientTaggingStore(store);
@@ -172,12 +177,6 @@ export class TXESession implements TXESessionStateHandler {
172
177
  noteStore,
173
178
  ]);
174
179
 
175
- // Register protocol contracts.
176
- for (const { contractClass, instance, artifact } of protocolContracts) {
177
- await contractStore.addContractArtifact(contractClass.id, artifact);
178
- await contractStore.addContractInstance(instance);
179
- }
180
-
181
180
  const archiver = new TXEArchiver(store);
182
181
  const anchorBlockStore = new AnchorBlockStore(store);
183
182
  const stateMachine = await TXEStateMachine.create(archiver, anchorBlockStore, contractStore, noteStore);
@@ -188,6 +187,9 @@ export class TXESession implements TXESessionStateHandler {
188
187
 
189
188
  const initialJobId = jobCoordinator.beginJob();
190
189
 
190
+ const logger = createLogger('txe:session');
191
+ const contractSyncService = new ContractSyncService(stateMachine.node, contractStore, noteStore, logger);
192
+
191
193
  const topLevelOracleHandler = new TXEOracleTopLevelContext(
192
194
  stateMachine,
193
195
  contractStore,
@@ -200,16 +202,16 @@ export class TXESession implements TXESessionStateHandler {
200
202
  senderAddressBookStore,
201
203
  capsuleStore,
202
204
  privateEventStore,
203
- initialJobId,
204
205
  nextBlockTimestamp,
205
206
  version,
206
207
  chainId,
207
208
  new Map(),
209
+ contractSyncService,
208
210
  );
209
- await topLevelOracleHandler.txeAdvanceBlocksBy(1);
211
+ await topLevelOracleHandler.advanceBlocksBy(1);
210
212
 
211
213
  return new TXESession(
212
- createLogger('txe:session'),
214
+ logger,
213
215
  stateMachine,
214
216
  topLevelOracleHandler,
215
217
  contractStore,
@@ -227,6 +229,7 @@ export class TXESession implements TXESessionStateHandler {
227
229
  version,
228
230
  chainId,
229
231
  nextBlockTimestamp,
232
+ contractSyncService,
230
233
  );
231
234
  }
232
235
 
@@ -261,6 +264,17 @@ export class TXESession implements TXESessionStateHandler {
261
264
  }
262
265
  }
263
266
 
267
+ getCurrentJob(): string {
268
+ return this.currentJobId;
269
+ }
270
+
271
+ /** Commits the current job and begins a new one. Returns the new job ID. */
272
+ async cycleJob(): Promise<string> {
273
+ await this.jobCoordinator.commitJob(this.currentJobId);
274
+ this.currentJobId = this.jobCoordinator.beginJob();
275
+ return this.currentJobId;
276
+ }
277
+
264
278
  async enterTopLevelState() {
265
279
  switch (this.state.name) {
266
280
  case 'PRIVATE': {
@@ -284,8 +298,7 @@ export class TXESession implements TXESessionStateHandler {
284
298
  }
285
299
 
286
300
  // Commit all staged stores from the job that was just completed, then begin a new job
287
- await this.jobCoordinator.commitJob(this.currentJobId);
288
- this.currentJobId = this.jobCoordinator.beginJob();
301
+ await this.cycleJob();
289
302
 
290
303
  this.oracleHandler = new TXEOracleTopLevelContext(
291
304
  this.stateMachine,
@@ -299,11 +312,11 @@ export class TXESession implements TXESessionStateHandler {
299
312
  this.senderAddressBookStore,
300
313
  this.capsuleStore,
301
314
  this.privateEventStore,
302
- this.currentJobId,
303
315
  this.nextBlockTimestamp,
304
316
  this.version,
305
317
  this.chainId,
306
318
  this.authwits,
319
+ this.contractSyncService,
307
320
  );
308
321
 
309
322
  this.state = { name: 'TOP_LEVEL' };
@@ -323,6 +336,7 @@ export class TXESession implements TXESessionStateHandler {
323
336
 
324
337
  await new NoteService(this.noteStore, this.stateMachine.node, anchorBlock!, this.currentJobId).syncNoteNullifiers(
325
338
  contractAddress,
339
+ await this.keyStore.getAccounts(),
326
340
  );
327
341
  const latestBlock = await this.stateMachine.node.getBlockHeader('latest');
328
342
 
@@ -339,30 +353,33 @@ export class TXESession implements TXESessionStateHandler {
339
353
  const taggingIndexCache = new ExecutionTaggingIndexCache();
340
354
 
341
355
  const utilityExecutor = this.utilityExecutorForContractSync(anchorBlock);
342
- this.oracleHandler = new PrivateExecutionOracle(
343
- Fr.ZERO,
344
- new TxContext(this.chainId, this.version, GasSettings.empty()),
345
- new CallContext(AztecAddress.ZERO, contractAddress, FunctionSelector.empty(), false),
346
- anchorBlock!,
356
+ this.oracleHandler = new PrivateExecutionOracle({
357
+ argsHash: Fr.ZERO,
358
+ txContext: new TxContext(this.chainId, this.version, GasSettings.empty()),
359
+ callContext: new CallContext(AztecAddress.ZERO, contractAddress, FunctionSelector.empty(), false),
360
+ anchorBlockHeader: anchorBlock!,
347
361
  utilityExecutor,
348
- [],
349
- [],
350
- new HashedValuesCache(),
362
+ authWitnesses: [],
363
+ capsules: [],
364
+ executionCache: new HashedValuesCache(),
351
365
  noteCache,
352
366
  taggingIndexCache,
353
- this.contractStore,
354
- this.noteStore,
355
- this.keyStore,
356
- this.addressStore,
357
- this.stateMachine.node,
358
- this.senderTaggingStore,
359
- this.recipientTaggingStore,
360
- this.senderAddressBookStore,
361
- this.capsuleStore,
362
- this.privateEventStore,
363
- this.stateMachine.contractSyncService,
364
- this.currentJobId,
365
- );
367
+ contractStore: this.contractStore,
368
+ noteStore: this.noteStore,
369
+ keyStore: this.keyStore,
370
+ addressStore: this.addressStore,
371
+ aztecNode: this.stateMachine.node,
372
+ senderTaggingStore: this.senderTaggingStore,
373
+ recipientTaggingStore: this.recipientTaggingStore,
374
+ senderAddressBookStore: this.senderAddressBookStore,
375
+ capsuleService: new CapsuleService(this.capsuleStore, await this.keyStore.getAccounts()),
376
+ privateEventStore: this.privateEventStore,
377
+ contractSyncService: this.stateMachine.contractSyncService,
378
+ l2TipsStore: this.stateMachine.node,
379
+ jobId: this.currentJobId,
380
+ scopes: await this.keyStore.getAccounts(),
381
+ messageContextService: this.stateMachine.messageContextService,
382
+ });
366
383
 
367
384
  // We store the note and tagging index caches fed into the PrivateExecutionOracle (along with some other auxiliary
368
385
  // data) in order to refer to it later, mimicking the way this object is used by the ContractFunctionSimulator. The
@@ -414,24 +431,28 @@ export class TXESession implements TXESessionStateHandler {
414
431
  this.stateMachine.node,
415
432
  anchorBlockHeader,
416
433
  this.currentJobId,
417
- ).syncNoteNullifiers(contractAddress);
434
+ ).syncNoteNullifiers(contractAddress, await this.keyStore.getAccounts());
418
435
 
419
- this.oracleHandler = new UtilityExecutionOracle(
436
+ this.oracleHandler = new UtilityExecutionOracle({
420
437
  contractAddress,
421
- [],
422
- [],
438
+ authWitnesses: [],
439
+ capsules: [],
423
440
  anchorBlockHeader,
424
- this.contractStore,
425
- this.noteStore,
426
- this.keyStore,
427
- this.addressStore,
428
- this.stateMachine.node,
429
- this.recipientTaggingStore,
430
- this.senderAddressBookStore,
431
- this.capsuleStore,
432
- this.privateEventStore,
433
- this.currentJobId,
434
- );
441
+ contractStore: this.contractStore,
442
+ noteStore: this.noteStore,
443
+ keyStore: this.keyStore,
444
+ addressStore: this.addressStore,
445
+ aztecNode: this.stateMachine.node,
446
+ recipientTaggingStore: this.recipientTaggingStore,
447
+ senderAddressBookStore: this.senderAddressBookStore,
448
+ capsuleService: new CapsuleService(this.capsuleStore, await this.keyStore.getAccounts()),
449
+ privateEventStore: this.privateEventStore,
450
+ messageContextService: this.stateMachine.messageContextService,
451
+ contractSyncService: this.contractSyncService,
452
+ l2TipsStore: this.stateMachine.node,
453
+ jobId: this.currentJobId,
454
+ scopes: await this.keyStore.getAccounts(),
455
+ });
435
456
 
436
457
  this.state = { name: 'UTILITY' };
437
458
  this.logger.debug(`Entered state ${this.state.name}`);
@@ -444,8 +465,8 @@ export class TXESession implements TXESessionStateHandler {
444
465
 
445
466
  // Note that while all public and private contexts do is build a single block that we then process when exiting
446
467
  // those, the top level context performs a large number of actions not captured in the following 'close' call. Among
447
- // others, it will create empty blocks (via `txeAdvanceBlocksBy` and `deploy`), create blocks with transactions via
448
- // `txePrivateCallNewFlow` and `txePublicCallNewFlow`, add accounts to PXE via `txeAddAccount`, etc. This is a
468
+ // others, it will create empty blocks (via `advanceBlocksBy` and `deploy`), create blocks with transactions via
469
+ // `privateCallNewFlow` and `publicCallNewFlow`, add accounts to PXE via `addAccount`, etc. This is a
449
470
  // slight inconsistency in the working model of this class, but is not too bad.
450
471
  // TODO: it's quite unfortunate that we need to capture the authwits created to later pass them again when the top
451
472
  // level context is re-created. This is because authwits create a temporary utility context that'd otherwise reset
@@ -499,29 +520,33 @@ export class TXESession implements TXESessionStateHandler {
499
520
  }
500
521
 
501
522
  private utilityExecutorForContractSync(anchorBlock: any) {
502
- return async (call: FunctionCall) => {
523
+ return async (call: FunctionCall, scopes: AztecAddress[]) => {
503
524
  const entryPointArtifact = await this.contractStore.getFunctionArtifactWithDebugMetadata(call.to, call.selector);
504
525
  if (entryPointArtifact.functionType !== FunctionType.UTILITY) {
505
526
  throw new Error(`Cannot run ${entryPointArtifact.functionType} function as utility`);
506
527
  }
507
528
 
508
529
  try {
509
- const oracle = new UtilityExecutionOracle(
510
- call.to,
511
- [],
512
- [],
513
- anchorBlock!,
514
- this.contractStore,
515
- this.noteStore,
516
- this.keyStore,
517
- this.addressStore,
518
- this.stateMachine.node,
519
- this.recipientTaggingStore,
520
- this.senderAddressBookStore,
521
- this.capsuleStore,
522
- this.privateEventStore,
523
- this.currentJobId,
524
- );
530
+ const oracle = new UtilityExecutionOracle({
531
+ contractAddress: call.to,
532
+ authWitnesses: [],
533
+ capsules: [],
534
+ anchorBlockHeader: anchorBlock!,
535
+ contractStore: this.contractStore,
536
+ noteStore: this.noteStore,
537
+ keyStore: this.keyStore,
538
+ addressStore: this.addressStore,
539
+ aztecNode: this.stateMachine.node,
540
+ recipientTaggingStore: this.recipientTaggingStore,
541
+ senderAddressBookStore: this.senderAddressBookStore,
542
+ capsuleService: new CapsuleService(this.capsuleStore, scopes),
543
+ privateEventStore: this.privateEventStore,
544
+ messageContextService: this.stateMachine.messageContextService,
545
+ contractSyncService: this.contractSyncService,
546
+ l2TipsStore: this.stateMachine.node,
547
+ jobId: this.currentJobId,
548
+ scopes,
549
+ });
525
550
  await new WASMSimulator()
526
551
  .executeUserCircuit(toACVMWitness(0, call.args), entryPointArtifact, new Oracle(oracle).toACIRCallback())
527
552
  .catch((err: Error) => {
@@ -3,6 +3,7 @@ import type { EthAddress } from '@aztec/foundation/eth-address';
3
3
  import { hexToBuffer } from '@aztec/foundation/string';
4
4
  import { type ContractArtifact, ContractArtifactSchema } from '@aztec/stdlib/abi';
5
5
  import { AztecAddress } from '@aztec/stdlib/aztec-address';
6
+ import { BlockHash } from '@aztec/stdlib/block';
6
7
  import { type ContractInstanceWithAddress, ContractInstanceWithAddressSchema } from '@aztec/stdlib/contract';
7
8
 
8
9
  import { z } from 'zod';
@@ -25,6 +26,10 @@ export function addressFromSingle(obj: ForeignCallSingle) {
25
26
  return new AztecAddress(fromSingle(obj));
26
27
  }
27
28
 
29
+ export function blockHashFromSingle(obj: ForeignCallSingle) {
30
+ return new BlockHash(fromSingle(obj));
31
+ }
32
+
28
33
  export function fromArray(obj: ForeignCallArray) {
29
34
  return obj.map(str => Fr.fromBuffer(hexToBuffer(str)));
30
35
  }