1const Verifier = artifacts.require("../../contracts/Verifier");
 2const Miksi = artifacts.require("../../contracts/Miksi.sol");
 3
 4const chai = require("chai");
 5const expect = chai.expect;
 6
 7const fs = require("fs");
 8const { groth } = require('snarkjs');
 9const { stringifyBigInts, unstringifyBigInts } = require('ffjavascript').utils;
10const WitnessCalculatorBuilder = require("circom_runtime").WitnessCalculatorBuilder;
11const circomlib = require("circomlib");
12
13contract("miksi", (accounts) => {
14
15
16  const {
17    0: owner,
18    1: addr1, // used for the deposit
19    2: addr2, // used for the withdraw
20    3: addr3,
21  } = accounts;
22
23  let insVerifier;
24  let insMiksi;
25
26  before(async () => {
27
28    insVerifier = await Verifier.new();
29    insMiksi = await Miksi.new(insVerifier.address);
30  });
31
32  it("miksi flow", async () => {
33    const secret = "123456789";
34
35    const coinCode = "0"; // refearing to ETH
36    const ethAmount = '0.5';
37    const amount = web3.utils.toWei(ethAmount, 'ether');
38
39    let balance_wei = await web3.eth.getBalance(addr1);
40    console.log("Balance at " + addr1, web3.utils.fromWei(balance_wei, 'ether'));
41    expect(balance_wei).to.be.equal('100000000000000000000');
42
43    const poseidon = circomlib.poseidon.createHash(6, 8, 57);
44    const commitment = poseidon([coinCode, amount, secret]).toString();
45
46    // deposit
47    console.log("Deposit of " + ethAmount + " ETH from " + addr1);
48    await insMiksi.deposit(coinCode, commitment, {from: addr1, value: amount});
49
50    balance_wei = await web3.eth.getBalance(addr1);
51    console.log("Balance at " + addr1, web3.utils.fromWei(balance_wei, 'ether'));
52    expect(balance_wei).to.be.equal('99499141300000000000');
53
54    // getDeposit data
55    const res = await insMiksi.getDeposit(commitment);
56    expect(res[0].toString()).to.be.equal(coinCode);
57    expect(res[1].toString()).to.be.equal(amount);
58
59
60    // calculate witness
61    const wasm = await fs.promises.readFile("./build/withdraw.wasm");
62    const input = unstringifyBigInts({
63      "coinCode": coinCode,
64      "amount": amount,
65      "commitment": commitment,
66      "secret": secret,
67      "address": addr2
68    });
69    const options = {};
70    console.log("Calculate witness");
71    const wc = await WitnessCalculatorBuilder(wasm, options);
72    const w = await wc.calculateWitness(input);
73    const witness = unstringifyBigInts(stringifyBigInts(w));
74
75    // generate zkproof of commitment using snarkjs (as is a test)
76    const provingKey = unstringifyBigInts(JSON.parse(fs.readFileSync("./build/proving_key.json", "utf8")));
77
78    console.log("Generate zkSNARK proof");
79    const {proof, publicSignals} = groth.genProof(provingKey, witness);
80
81    // withdraw
82    console.log("Withdraw of " + ethAmount + " ETH to " + addr2);
83    const resW = await insMiksi.withdraw(
84      commitment, 
85      addr2,
86      [proof.pi_a[0].toString(), proof.pi_a[1].toString()],
87      [
88        [proof.pi_b[0][1].toString(), proof.pi_b[0][0].toString()],
89        [proof.pi_b[1][1].toString(), proof.pi_b[1][0].toString()]
90      ],
91      [proof.pi_c[0].toString(), proof.pi_c[1].toString()]
92    );
93    // console.log("resW", resW);
94
95    balance_wei = await web3.eth.getBalance(addr2);
96    console.log("Balance at " + addr2, web3.utils.fromWei(balance_wei, 'ether'));
97    expect(balance_wei).to.be.equal('100500000000000000000');
98  });
99});