cli (wip)

cli.js

@@ -0,0 +1,109 @@
+#!/usr/bin/env node
+const snarkjs = require("snarkjs");
+const bigInt = snarkjs.bigInt;
+const utils = require("./scripts/utils");
+const merkleTree = require('./lib/MerkleTree');
+const contract = require("truffle-contract");
+const Mixer = contract(require('./build/contracts/Mixer.json'));
+
+const sender = "";//accounts[0];
+const amount = "1 ether";
+
+async function deposit() {
+  let deposit = {
+    nullifier: utils.rbigint(31),
+    secret: utils.rbigint(31),
+  };
+  const preimage = Buffer.concat([deposit.nullifier.leInt2Buff(32), deposit.secret.leInt2Buff(32)]);
+  deposit.commitment = utils.pedersenHash(preimage);
+
+  console.log("Submitting deposit transaction");
+  const mixer = await Mixer.deployed();
+  await mixer.deposit(deposit.commitment, { value: amount, from: sender });
+
+  return preimage.toString('hex');
+}
+
+async function withdraw(note, receiver) {
+  let buf = Buffer.from(note.slice(2), "hex");
+  let deposit = {
+    nullifier: bigInt.leBuff2int(buf.slice(0, 32)),
+    secret: bigInt.leBuff2int(buf.slice(32, 64)),
+  };
+
+  console.log("Getting current state from mixer contract");
+  const mixer = await Mixer.deployed();
+
+  const {root, path_elements, path_index} = await tree.path(1);
+  // Circuit input
+  const input = {
+    // public
+    root: root,
+    nullifier: deposit.nullifier,
+    receiver: bigInt(receiver),
+    fee: bigInt(0),
+
+    // private
+    secret: deposit.secret,
+    pathElements: path_elements,
+    pathIndex: path_index,
+  };
+
+  console.log("Generating SNARK proof");
+  const { pi_a, pi_b, pi_c, publicSignals } = await utils.snarkProof(input);
+
+  console.log("Submitting withdraw transaction");
+  await mixer.withdraw(pi_a, pi_b, pi_c, publicSignals, { from: sender })
+}
+
+function printHelp() {
+  console.log(`
+Usage:
+  Submit a deposit from default eth account and return the resulting note 
+  $ ./cli.js deposit
+  
+  Withdraw a note to 'receiver' account
+  $ ./cli.js withdraw <note <receiver
+  
+Example:
+  $ ./cli.js deposit
+  ...
+  Your note: 0x1941fa999e2b4bfeec3ce53c2440c3bc991b1b84c9bb650ea19f8331baf621001e696487e2a2ee54541fa12f49498d71e24d00b1731a8ccd4f5f5126f3d9f400
+  
+  $ ./cli.js withdraw 0x1941fa999e2b4bfeec3ce53c2440c3bc991b1b84c9bb650ea19f8331baf621001e696487e2a2ee54541fa12f49498d71e24d00b1731a8ccd4f5f5126f3d9f400 0xee6249BA80596A4890D1BD84dbf5E4322eA4E7f0
+`);
+  process.exit(0);
+}
+
+(async () => {
+  const dep = await deposit();
+  console.log(`Your note: 0x${dep}`);
+
+  const acc = "0xee6249BA80596A4890D1BD84dbf5E4322eA4E7f0";//accounts[1];
+  await withdraw(dep, acc);
+})();
+
+
+// const args = process.argv.slice(2);
+// if (args.length === 0) {
+//   printHelp();
+// }
+//
+// switch (args[0]) {
+//   case 'deposit':
+//     if (args.length === 1)
+//       deposit().then(() => process.exit(0));
+//     else
+//       printHelp();
+//     break;
+//
+//   case 'withdraw':
+//     if (args.length === 3 && /^[0-9a-fA-F]{128}$/.test(args[1]) && /^[0-9a-fA-F]{64}$/.test(args[2]))
+//       withdraw(args[1], args[2]).then(() => process.exit(0));
+//     else
+//       printHelp();
+//     break;
+//
+//   default:
+//     printHelp();
+// }