zkjson 0.1.7 → 0.1.8

package/doc.js

@@ -1,16 +1,54 @@
+const snarkjs = require("snarkjs")
+const { resolve } = require("path")
 const { pad, toSignal, encode, encodePath, encodeVal } = require("./encoder")
 
 module.exports = class Doc {
-  constructor({ size_val = 5, size_path = 5, size_json = 256 }) {
+  constructor({ size_val = 5, size_path = 5, size_json = 256, wasm, zkey }) {
     this.size_val = size_val
     this.size_path = size_path
     this.size_json = size_json
+    this.wasm = wasm
+    this.zkey = zkey
   }
-  async getInputs({ json, path, val }) {
+  async getInputs({ json, path }) {
     return {
       json: pad(toSignal(encode(json)), this.size_json),
       path: pad(toSignal(encodePath(path)), this.size_path),
-      val: pad(toSignal(encodeVal(val)), this.size_val),
+      val: pad(toSignal(encodeVal(this.getVal(json, path))), this.size_val),
     }
   }
+  _getVal(j, p) {
+    if (p.length === 0) {
+      return j
+    } else {
+      const sp = p[0].split("[")
+      for (let v of sp) {
+        if (/]$/.test(v)) {
+          j = j[v.replace(/]$/, "") * 1]
+        } else {
+          j = j[v]
+        }
+      }
+      return this._getVal(j, p.slice(1))
+    }
+  }
+  getVal(j, p) {
+    if (p === "") return j
+    return this._getVal(j, p.split("."))
+  }
+  async genProof(json, path) {
+    const inputs = await this.getInputs(json, path)
+    const { proof, publicSignals } = await snarkjs.groth16.fullProve(
+      inputs,
+      this.wasm,
+      this.zkey
+    )
+    return [
+      ...proof.pi_a.slice(0, 2),
+      ...proof.pi_b[0].slice(0, 2).reverse(),
+      ...proof.pi_b[1].slice(0, 2).reverse(),
+      ...proof.pi_c.slice(0, 2),
+      ...publicSignals,
+    ]
+  }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "zkjson",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "description": "Zero Knowledge Provable JSON",
   "main": "index.js",
   "license": "MIT",