o1js/src/examples/zkapps/dex/happy-path-with-actions.ts

TypeScript framework for zk-SNARKs and zkApps

import { expect } from 'expect';
import { AccountUpdate, Mina, UInt64 } from 'o1js';
import { tic, toc } from '../../utils/tic-toc.node.js';
import {
  Dex,
  DexTokenHolder,
  addresses,
  getTokenBalances,
  keys,
  tokenIds,
} from './dex-with-actions.js';
import { TrivialCoin as TokenContract } from './erc20.js';

let proofsEnabled = true;
tic('Happy path with actions');
console.log();
let Local = await Mina.LocalBlockchain({
  proofsEnabled,
  enforceTransactionLimits: true,
});
Mina.setActiveInstance(Local);
let [feePayer] = Local.testAccounts;
let tx, balances, oldBalances;

if (proofsEnabled) {
  tic('compile (token)');
  await TokenContract.compile();
  toc();
  tic('compile (dex token holder)');
  await DexTokenHolder.compile();
  toc();
  tic('compile (dex main contract)');
  await Dex.compile();
  toc();
}

let tokenX = new TokenContract(addresses.tokenX);
let tokenY = new TokenContract(addresses.tokenY);
let dex = new Dex(addresses.dex);
let dexTokenHolderX = new DexTokenHolder(addresses.dex, tokenIds.X);
let dexTokenHolderY = new DexTokenHolder(addresses.dex, tokenIds.Y);

tic('deploy & init token contracts');
tx = await Mina.transaction(feePayer, async () => {
  await tokenX.deploy();
  await tokenY.deploy();

  // pay fees for creating 2 token contract accounts, and fund them so each can create 1 account themselves
  const accountFee = Mina.getNetworkConstants().accountCreationFee;
  let feePayerUpdate = AccountUpdate.fundNewAccount(feePayer, 2);
  feePayerUpdate.send({ to: tokenX.self, amount: accountFee });
  feePayerUpdate.send({ to: tokenY.self, amount: accountFee });
});
await tx.prove();
await tx.sign([feePayer.key, keys.tokenX, keys.tokenY]).send();
toc();
console.log('account updates length', tx.transaction.accountUpdates.length);

tic('deploy dex contracts');
tx = await Mina.transaction(feePayer, async () => {
  // pay fees for creating 3 dex accounts
  AccountUpdate.createSigned(feePayer).balance.subInPlace(
    Mina.getNetworkConstants().accountCreationFee.mul(3)
  );
  await dex.deploy();
  await dexTokenHolderX.deploy();
  await tokenX.approveAccountUpdate(dexTokenHolderX.self);
  await dexTokenHolderY.deploy();
  await tokenY.approveAccountUpdate(dexTokenHolderY.self);
});
await tx.prove();
await tx.sign([feePayer.key, keys.dex]).send();
toc();
console.log('account updates length', tx.transaction.accountUpdates.length);

tic('transfer tokens to user');
let USER_DX = 1_000n;
tx = await Mina.transaction(feePayer, async () => {
  // pay fees for creating 3 user accounts
  let au = AccountUpdate.fundNewAccount(feePayer, 3);
  au.send({ to: addresses.user, amount: 20e9 }); // give users MINA to pay fees
  await tokenX.transfer(addresses.tokenX, addresses.user, UInt64.from(USER_DX));
  await tokenY.transfer(addresses.tokenY, addresses.user, UInt64.from(USER_DX));
});
await tx.prove();
await tx.sign([feePayer.key, keys.tokenX, keys.tokenY]).send();
toc();
console.log('account updates length', tx.transaction.accountUpdates.length);

// this is done in advance to avoid account update limit in `supply`
tic("create user's lq token account");
tx = await Mina.transaction(addresses.user, async () => {
  AccountUpdate.fundNewAccount(addresses.user);
  await dex.createAccount();
});
await tx.prove();
await tx.sign([keys.user]).send();
toc();
console.log('account updates length', tx.transaction.accountUpdates.length);

[oldBalances, balances] = [balances, getTokenBalances()];
expect(balances.user.X).toEqual(USER_DX);
console.log(balances);

tic('supply liquidity');
tx = await Mina.transaction(addresses.user, async () => {
  await dex.supplyLiquidityBase(UInt64.from(USER_DX), UInt64.from(USER_DX));
});
await tx.prove();
await tx.sign([keys.user]).send();
toc();
console.log('account updates length', tx.transaction.accountUpdates.length);
[oldBalances, balances] = [balances, getTokenBalances()];
expect(balances.user.X).toEqual(0n);
console.log(balances);

tic('redeem liquidity, step 1');
let USER_DL = 100n;
tx = await Mina.transaction(addresses.user, async () => {
  await dex.redeemInitialize(UInt64.from(USER_DL));
});
await tx.prove();
await tx.sign([keys.user]).send();
toc();
console.log('account updates length', tx.transaction.accountUpdates.length);
console.log(getTokenBalances());

tic('redeem liquidity, step 2a (get back token X)');
tx = await Mina.transaction(addresses.user, async () => {
  await dexTokenHolderX.redeemLiquidityFinalize();
  await tokenX.approveAccountUpdate(dexTokenHolderX.self);
});
await tx.prove();
await tx.sign([keys.user]).send();
toc();
console.log('account updates length', tx.transaction.accountUpdates.length);
console.log(getTokenBalances());

tic('redeem liquidity, step 2b (get back token Y)');
tx = await Mina.transaction(addresses.user, async () => {
  await dexTokenHolderY.redeemLiquidityFinalize();
  await tokenY.approveAccountUpdate(dexTokenHolderY.self);
});
await tx.prove();
await tx.sign([keys.user]).send();
toc();
console.log('account updates length', tx.transaction.accountUpdates.length);
console.log(getTokenBalances());

[oldBalances, balances] = [balances, getTokenBalances()];
expect(balances.user.X).toEqual(USER_DL / 2n);

tic('swap 10 X for Y');
USER_DX = 10n;
tx = await Mina.transaction(addresses.user, async () => {
  await dex.swapX(UInt64.from(USER_DX));
});
await tx.prove();
await tx.sign([keys.user]).send();
toc();
console.log('account updates length', tx.transaction.accountUpdates.length);

[oldBalances, balances] = [balances, getTokenBalances()];
expect(balances.user.X).toEqual(oldBalances.user.X - USER_DX);
console.log(balances);

toc();
console.log('dex happy path with actions was successful! 🎉');