@aztec/aztec.js 0.1.0-alpha37 → 0.1.0-alpha39

package/src/sandbox/index.ts

@@ -0,0 +1,83 @@
+import { Fr, PrivateKey } from '@aztec/circuits.js';
+import { ContractAbi } from '@aztec/foundation/abi';
+import { sleep } from '@aztec/foundation/sleep';
+
+import SchnorrAccountContractAbi from '../abis/schnorr_account_contract.json' assert { type: 'json' };
+import { AztecRPC, getAccountWallets, getSchnorrAccount } from '../index.js';
+
+export const INITIAL_SANDBOX_ENCRYPTION_KEYS = [
+  new PrivateKey(Buffer.from('b2803ec899f76f6b2ac011480d24028f1a29587f8a3a92f7ee9d48d8c085c284', 'hex')),
+  new PrivateKey(Buffer.from('6bb46e9a80da2ff7bfff71c2c50eaaa4b15f7ed5ad1ade4261b574ef80b0cdb0', 'hex')),
+  new PrivateKey(Buffer.from('0f6addf0da06c33293df974a565b03d1ab096090d907d98055a8b7f4954e120c', 'hex')),
+];
+
+export const INITIAL_SANDBOX_SIGNING_KEYS = INITIAL_SANDBOX_ENCRYPTION_KEYS;
+
+export const INITIAL_SANDBOX_SALTS = [Fr.ZERO, Fr.ZERO, Fr.ZERO];
+
+export const INITIAL_SANDBOX_ACCOUNT_CONTRACT_ABI = SchnorrAccountContractAbi;
+
+/**
+ * Gets a wallet for the Aztec accounts that are initially stored in the sandbox.
+ * @param aztecRpc - An instance of the Aztec RPC interface.
+ * @returns An AccountWallet implementation that includes all the accounts found.
+ */
+export async function getSandboxAccountsWallet(aztecRpc: AztecRPC) {
+  return await getAccountWallets(
+    aztecRpc,
+    INITIAL_SANDBOX_ACCOUNT_CONTRACT_ABI as ContractAbi,
+    INITIAL_SANDBOX_ENCRYPTION_KEYS,
+    INITIAL_SANDBOX_SIGNING_KEYS,
+    INITIAL_SANDBOX_SALTS,
+  );
+}
+
+/**
+ * Deploys the initial set of schnorr signature accounts to the sandbox
+ * @param aztecRpc - An instance of the Aztec RPC interface.
+ * @returns The set of deployed Account objects and associated private encryption keys
+ */
+export async function deployInitialSandboxAccounts(aztecRpc: AztecRPC) {
+  const accounts = INITIAL_SANDBOX_ENCRYPTION_KEYS.map((privateKey, i) => {
+    const account = getSchnorrAccount(aztecRpc, privateKey, INITIAL_SANDBOX_SIGNING_KEYS[i], INITIAL_SANDBOX_SALTS[i]);
+    return {
+      account,
+      privateKey,
+    };
+  });
+  // Attempt to get as much parallelism as possible
+  const deployMethods = await Promise.all(
+    accounts.map(async x => {
+      const deployMethod = await x.account.getDeployMethod();
+      await deployMethod.create({ contractAddressSalt: x.account.salt });
+      await deployMethod.simulate({});
+      return deployMethod;
+    }),
+  );
+  // Send tx together to try and get them in the same rollup
+  const sentTxs = deployMethods.map(dm => {
+    return dm.send();
+  });
+  await Promise.all(
+    sentTxs.map(async (tx, i) => {
+      const wallet = await accounts[i].account.getWallet();
+      return tx.wait({ wallet });
+    }),
+  );
+  return accounts;
+}
+
+/**
+ * Function to wait until the sandbox becomes ready for use.
+ * @param rpcServer - The rpc client connected to the sandbox.
+ */
+export const waitForSandbox = async (rpcServer: AztecRPC) => {
+  while (true) {
+    try {
+      await rpcServer.getNodeInfo();
+      break;
+    } catch (err) {
+      await sleep(1000);
+    }
+  }
+};

package/src/utils/abi_types.ts

@@ -0,0 +1,4 @@
+import { Fr } from '@aztec/circuits.js';
+
+/** Any type that can be converted into a field for a contract call. */
+export type FieldLike = Fr | bigint | number | { /** Converts to field */ toField: () => Fr };

package/src/utils/account.ts

@@ -6,7 +6,7 @@ import { AztecRPC, TxStatus } from '@aztec/types';
 
 import { SingleKeyAccountEntrypoint } from '../account/entrypoint/single_key_account_entrypoint.js';
 import { EntrypointWallet, Wallet } from '../aztec_rpc_client/wallet.js';
-import { ContractDeployer, EntrypointCollection, generatePublicKey } from '../index.js';
+import { ContractDeployer, EntrypointCollection, StoredKeyAccountEntrypoint, generatePublicKey } from '../index.js';
 
 /**
  * Creates an Aztec Account.
@@ -27,7 +27,7 @@ export async function createAccounts(
     const privKey = i == 0 && privateKey ? privateKey : PrivateKey.random();
     const publicKey = await generatePublicKey(privKey);
     const deploymentInfo = await getContractDeploymentInfo(accountContractAbi, [], salt, publicKey);
-    await aztecRpcClient.addAccount(privKey, deploymentInfo.address, deploymentInfo.partialAddress);
+    await aztecRpcClient.registerAccount(privKey, deploymentInfo.completeAddress);
     const contractDeployer = new ContractDeployer(accountContractAbi, aztecRpcClient, publicKey);
     const tx = contractDeployer.deploy().send({ contractAddressSalt: salt });
     await tx.isMined({ interval: 0.5 });
@@ -36,15 +36,20 @@ export async function createAccounts(
       throw new Error(`Deployment tx not mined (status is ${receipt.status})`);
     }
     const address = receipt.contractAddress!;
-    if (!address.equals(deploymentInfo.address)) {
+    if (!address.equals(deploymentInfo.completeAddress.address)) {
       throw new Error(
-        `Deployment address does not match for account contract (expected ${deploymentInfo.address.toString()} got ${address.toString()})`,
+        `Deployment address does not match for account contract (expected ${deploymentInfo.completeAddress.address.toString()} got ${address.toString()})`,
       );
     }
     logger(`Created account ${address.toString()} with public key ${publicKey.toString()}`);
     accountImpls.registerAccount(
       address,
-      new SingleKeyAccountEntrypoint(address, deploymentInfo.partialAddress, privKey, await Schnorr.new()),
+      new SingleKeyAccountEntrypoint(
+        address,
+        deploymentInfo.completeAddress.partialAddress,
+        privKey,
+        await Schnorr.new(),
+      ),
     );
   }
   return new EntrypointWallet(aztecRpcClient, accountImpls);
@@ -53,23 +58,38 @@ export async function createAccounts(
 /**
  * Gets the Aztec accounts that are stored in an Aztec RPC instance.
  * @param aztecRpcClient - An instance of the Aztec RPC interface.
- * @param numberOfAccounts - The number of accounts to fetch.
+ * @param accountContractAbi - The abi of the account contract used when the accounts were deployed
+ * @param privateKeys - The encryption private keys used to create the accounts.
+ * @param signingKeys - The signing private keys used to create the accounts.
+ * @param salts - The salt values used to create the accounts.
  * @returns An AccountWallet implementation that includes all the accounts found.
  */
-export async function getAccountWallet(
+export async function getAccountWallets(
   aztecRpcClient: AztecRPC,
   accountContractAbi: ContractAbi,
-  privateKey: PrivateKey,
-  salt: Fr,
+  privateKeys: PrivateKey[],
+  signingKeys: PrivateKey[],
+  salts: Fr[],
 ) {
+  if (privateKeys.length != salts.length || signingKeys.length != privateKeys.length) {
+    throw new Error('Keys and salts must be the same length');
+  }
   const accountCollection = new EntrypointCollection();
-  const publicKey = await generatePublicKey(privateKey);
-  const deploymentInfo = await getContractDeploymentInfo(accountContractAbi, [], salt, publicKey);
-  const address = deploymentInfo.address;
+  for (let i = 0; i < privateKeys.length; i++) {
+    const publicKey = await generatePublicKey(privateKeys[i]);
+    const signingPublicKey = await generatePublicKey(signingKeys[i]);
+    const deploymentInfo = await getContractDeploymentInfo(
+      accountContractAbi,
+      [signingPublicKey.x, signingPublicKey.y],
+      salts[i],
+      publicKey,
+    );
+    const address = deploymentInfo.completeAddress.address;
 
-  accountCollection.registerAccount(
-    address,
-    new SingleKeyAccountEntrypoint(address, deploymentInfo.partialAddress, privateKey, await Schnorr.new()),
-  );
+    accountCollection.registerAccount(
+      address,
+      new StoredKeyAccountEntrypoint(address, signingKeys[i], await Schnorr.new()),
+    );
+  }
   return new EntrypointWallet(aztecRpcClient, accountCollection);
 }

package/src/utils/index.ts

@@ -3,3 +3,4 @@ export * from './account.js';
 export * from './pub_key.js';
 export * from './l1_contracts.js';
 export * from './l2_contracts.js';
+export * from './abi_types.js';

package/dest/account/complete_address.d.ts

@@ -1,13 +0,0 @@
-import { AztecAddress, PartialAddress, PublicKey } from '@aztec/circuits.js';
-/** Address and preimages associated with an account. */
-export type CompleteAddress = {
-    /** Address of an account. Derived from the partial address and public key. */
-    address: AztecAddress;
-    /** Partial address of the account. Required for deriving the address from the encryption public key. */
-    partialAddress: PartialAddress;
-    /** Encryption public key associated with this address. */
-    publicKey: PublicKey;
-};
-/** Returns whether the argument looks like a CompleteAddress. */
-export declare function isCompleteAddress(obj: any): obj is CompleteAddress;
-//# sourceMappingURL=complete_address.d.ts.map

package/dest/account/complete_address.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"complete_address.d.ts","sourceRoot":"","sources":["../../src/account/complete_address.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAE7E,wDAAwD;AACxD,MAAM,MAAM,eAAe,GAAG;IAC5B,8EAA8E;IAC9E,OAAO,EAAE,YAAY,CAAC;IACtB,wGAAwG;IACxG,cAAc,EAAE,cAAc,CAAC;IAC/B,0DAA0D;IAC1D,SAAS,EAAE,SAAS,CAAC;CACtB,CAAC;AAEF,iEAAiE;AACjE,wBAAgB,iBAAiB,CAAC,GAAG,EAAE,GAAG,GAAG,GAAG,IAAI,eAAe,CAIlE"}

package/dest/account/complete_address.js

@@ -1,8 +0,0 @@
-/** Returns whether the argument looks like a CompleteAddress. */
-export function isCompleteAddress(obj) {
-    if (!obj)
-        return false;
-    const maybe = obj;
-    return !!maybe.address && !!maybe.partialAddress && !!maybe.publicKey;
-}
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiY29tcGxldGVfYWRkcmVzcy5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy9hY2NvdW50L2NvbXBsZXRlX2FkZHJlc3MudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IkFBWUEsaUVBQWlFO0FBQ2pFLE1BQU0sVUFBVSxpQkFBaUIsQ0FBQyxHQUFRO0lBQ3hDLElBQUksQ0FBQyxHQUFHO1FBQUUsT0FBTyxLQUFLLENBQUM7SUFDdkIsTUFBTSxLQUFLLEdBQUcsR0FBc0IsQ0FBQztJQUNyQyxPQUFPLENBQUMsQ0FBQyxLQUFLLENBQUMsT0FBTyxJQUFJLENBQUMsQ0FBQyxLQUFLLENBQUMsY0FBYyxJQUFJLENBQUMsQ0FBQyxLQUFLLENBQUMsU0FBUyxDQUFDO0FBQ3hFLENBQUMifQ==

package/src/account/complete_address.ts

@@ -1,18 +0,0 @@
-import { AztecAddress, PartialAddress, PublicKey } from '@aztec/circuits.js';
-
-/** Address and preimages associated with an account. */
-export type CompleteAddress = {
-  /** Address of an account. Derived from the partial address and public key. */
-  address: AztecAddress;
-  /** Partial address of the account. Required for deriving the address from the encryption public key. */
-  partialAddress: PartialAddress;
-  /** Encryption public key associated with this address. */
-  publicKey: PublicKey;
-};
-
-/** Returns whether the argument looks like a CompleteAddress. */
-export function isCompleteAddress(obj: any): obj is CompleteAddress {
-  if (!obj) return false;
-  const maybe = obj as CompleteAddress;
-  return !!maybe.address && !!maybe.partialAddress && !!maybe.publicKey;
-}