Local ledger implementation in typescript

Author: bleepbloopsify

CHANGELOG.md

@@ -45,6 +45,7 @@ This project adheres to
   https://github.com/o1-labs/o1js/pull/2436
 - Improved the runtime table API with a `RuntimeTable` class with better
   readability https://github.com/o1-labs/o1js/pull/2402
+- Internal `Mina.LocalBlockchain` type exported via `Mina`. https://github.com/o1-labs/o1js/pull/2538
 
 ### Fixed
 

src/lib/ledger/ledger.ts

@@ -0,0 +1,90 @@
+import { TokenId, ZkappCommand } from '../mina/v1/account-update.js';
+import { Account, newAccount } from '../mina/v1/account.js';
+import { NetworkValue } from '../mina/v1/precondition.js';
+import { Account as AccountV2 } from '../mina/v2/account.js';
+import { InitialApplyState, checkAndApplyAccountUpdate } from '../mina/v2/zkapp-logic.js';
+import { PublicKey } from '../provable/crypto/signature.js';
+import { UInt64 } from '../provable/int.js';
+import { Field } from '../provable/wrapped.js';
+export const DefaultTokenId = 1n;
+
+export type AccountId = { publicKey: PublicKey; tokenId?: Field };
+
+export class Ledger {
+  _nextLocation: bigint;
+  _accounts: Map<bigint, Account>;
+  _locations: Map<AccountId, bigint>;
+
+  constructor() {
+    this._nextLocation = 1n;
+    this._accounts = new Map<bigint, Account>();
+    this._locations = new Map<AccountId, bigint>();
+  }
+
+  static create(): Ledger {
+    return new Ledger();
+  }
+
+  addAccount(publicKey: PublicKey, balance: bigint | number | string): void {
+    const accountId = { publicKey, tokenId: TokenId.default };
+    const location = (() => {
+      const existing = this._locations.get(accountId);
+      if (existing !== undefined) {
+        throw new Error('account with public key already exists');
+      }
+
+      const newLocation = this._nextLocation;
+      this._nextLocation += 1n;
+
+      return newLocation;
+    })();
+
+    const account = newAccount(accountId);
+    account.balance = UInt64.from(balance);
+
+    this._locations.set(accountId, location);
+    this._accounts.set(location, account);
+  }
+
+  getAccount(publicKey: PublicKey, tokenId: Field = TokenId.default): Account | undefined {
+    const accountId = { publicKey, tokenId };
+    const location = this._locations.get(accountId);
+    if (location === undefined) {
+      return undefined;
+    }
+
+    return undefined;
+  }
+
+  applyTransaction(transaction: ZkappCommand, networkState: NetworkValue): void {
+    for (const update of transaction.accountUpdates) {
+      const { body, authorization } = update;
+
+      if (body.authorizationKind.isProved && !authorization.proof) {
+        throw Error(
+          `The actual authorization does not match the expected authorization kind. Did you forget to invoke \`await tx.prove()\`?`
+        );
+      }
+
+      const accountId: AccountId = { publicKey: body.publicKey, tokenId: body.tokenId };
+      const account = (() => {
+        const location = this._locations.get(accountId);
+        if (location === undefined) {
+          throw new Error('account not found');
+        }
+
+        return this._accounts.get(location);
+      })();
+      if (account === undefined) {
+        throw new Error('account not found');
+      }
+
+      const accountV2 = AccountV2.fromGeneric(account);
+
+
+      /// WIP here
+
+      checkAndApplyAccountUpdate(networkState, accountV2, update, InitialApplyState);
+    }
+  }
+}

src/lib/mina/v1/local-blockchain.ts

@@ -61,7 +61,11 @@ namespace TestPublicKey {
   }
 }
 
+<<<<<<< Updated upstream
 export type LocalBlockchain = Awaited<ReturnType<typeof LocalBlockchain>>;
+=======
+export type LocalBlockchain = Awaited<ReturnType<typeof LocalBlockchain>>
+>>>>>>> Stashed changes
 
 /**
  * A mock Mina blockchain running locally and useful for testing.

src/lib/mina/v2/zkapp-logic.ts

@@ -1,28 +1,33 @@
+import { Bool } from '../../provable/bool.js';
+import { PublicKey } from '../../provable/crypto/signature.js';
+import { Field } from '../../provable/field.js';
+import { Int64, Sign, UInt32, UInt64 } from '../../provable/int.js';
+import { ZkappConstants } from '../v1/constants.js';
 import { AccountUpdate } from './account-update.js';
 import { Account } from './account.js';
 import { AuthorizationLevel } from './authorization.js';
 import { Update } from './core.js';
 import { Permissions } from './permissions.js';
 import {
-  Preconditions,
   EpochDataPreconditions,
   EpochLedgerPreconditions,
+  Preconditions,
 } from './preconditions.js';
 import { StateLayout, StateUpdates, StateValues } from './state.js';
 import { ZkappFeePayment } from './transaction.js';
 import { ChainView, EpochData, EpochLedgerData } from './views.js';
-import { Bool } from '../../provable/bool.js';
-import { Field } from '../../provable/field.js';
-import { Int64, Sign, UInt64, UInt32 } from '../../provable/int.js';
-import { PublicKey } from '../../provable/crypto/signature.js';
-import { ZkappConstants } from '../v1/constants.js';
 
-export { checkAndApplyAccountUpdate, checkAndApplyFeePayment, ApplyState };
+export { ApplyState, checkAndApplyAccountUpdate, checkAndApplyFeePayment };
 
 type ApplyResult<T> = ({ status: 'Applied' } & T) | { status: 'Failed'; errors: Error[] };
 
 type ApplyState<T> = { status: 'Alive'; value: T } | { status: 'Dead' };
 
+export const InitialApplyState = {
+  status: 'Alive',
+  value: Int64.zero,
+};
+
 function updateApplyState<T>(
   applyState: ApplyState<T>,
   errors: Error[],