zkjson 0.7.1 → 0.7.2

package/esm/collection.js

@@ -1,4 +1,4 @@
-import { newMemEmptyTrie } from "./circomlibjs.js"
+import newMemEmptyTrie from "./newMemEmptyTrie.js"
 import { pad, toSignal, encode, toIndex } from "./encoder.js"
 import Doc from "./doc.js"
 

package/esm/collection_tree.js

@@ -0,0 +1,43 @@
+import newMemEmptyTrie from "./newMemEmptyTrie.js"
+import { pad, toSignal, encode, toIndex } from "./encoder.js"
+
+export default class CollectionTree {
+  constructor({
+    size_val = 256,
+    size_path = 32,
+    size_json = 256,
+    level = 184,
+    kv,
+  }) {
+    this.kv = kv
+    this.size_val = size_val
+    this.size_path = size_path
+    this.size_json = size_json
+    this.level = level
+  }
+  async init() {
+    this.tree = await newMemEmptyTrie(this.kv)
+  }
+  async insert(_key, _val) {
+    const doc = encode(_val)
+    const id = toIndex(_key)
+    const val = pad(toSignal(doc), this.size_json)
+    return await this.tree.insert(id, val)
+  }
+  async update(_key, _val) {
+    const doc = encode(_val)
+    const id = toIndex(_key)
+    const val = pad(toSignal(doc), this.size_json)
+    return await this.tree.update(id, val)
+  }
+
+  async delete(_key) {
+    const id = toIndex(_key)
+    return await this.tree.delete(id)
+  }
+
+  async get(_key) {
+    const id = toIndex(_key)
+    return await this.tree.find(id)
+  }
+}

package/esm/db.js

@@ -1,4 +1,4 @@
-import { newMemEmptyTrie } from "./circomlibjs.js"
+import newMemEmptyTrie from "./newMemEmptyTrie.js"
 import { groth16 } from "snarkjs"
 import { is, indexOf, range, isNil } from "ramda"
 import { pad, toSignal, encode, toIndex } from "./encoder.js"

package/esm/db_tree.js

@@ -0,0 +1,122 @@
+import newMemEmptyTrie from "./newMemEmptyTrie.js"
+import { is, indexOf, isNil } from "ramda"
+import Collection from "./collection_tree.js"
+
+export default class DBTree {
+  constructor({
+    size_val = 256,
+    size_path = 32,
+    level = 184,
+    size_json = 256,
+    level_col = 24,
+    kv,
+  }) {
+    this.kv = kv
+    this.level_col = level_col
+    this.size_val = size_val
+    this.size_path = size_path
+    this.level = level
+    this.size_json = size_json
+    this.count = 0
+    this.ids = {}
+    this.cols = []
+  }
+
+  async init() {
+    this.kvi = this.kv?.("__db__")
+    this.tree = await newMemEmptyTrie(this.kvi)
+    if (this.kvi) {
+      const count = this.kvi.get("__count__")
+      if (!isNil(count)) this.count = count
+    }
+  }
+  async exists(num) {
+    const ex = this.ids[num] || (await this.tree.find(num)).found
+    if (ex) this.ids[num] = true
+    return ex
+  }
+  async getID(num) {
+    if (!isNil(num)) {
+      if (await this.exists(num)) throw Error("id exists")
+      return num
+    } else {
+      while (await this.exists(this.count)) this.count++
+      return this.count
+    }
+  }
+  async addCollection(num) {
+    if (!isNil(num) && (!is(Number, num) || Math.round(num) !== num)) {
+      throw Error("id is not an integer")
+    }
+    let id = null
+    if (!isNil(num)) {
+      if (this.cols[num]) throw Error("collection exists")
+      const col = await this.tree.find(num)
+      if (col.found) throw Error("collection exists")
+      id = num
+    } else {
+      id = await this.getID(num)
+    }
+    const _col = new Collection({
+      size_val: this.size_val,
+      size_path: this.size_path,
+      level: this.level,
+      size_json: this.size_json,
+      kv: this.kv?.(`__dir__/${id}`),
+    })
+    await _col.init()
+    this.cols[id] = _col
+    this.ids[id] = true
+    const root = _col.tree.F.toObject(_col.tree.root).toString()
+    await this.tree.insert(id, [root])
+    if (id === this.count) {
+      this.count++
+      if (this.kvi) this.kvi.put("__count__", this.count)
+    }
+    return id
+  }
+  getColTree(col) {
+    const _col = this.cols[col]
+    if (!_col) throw Error("collection doesn't exist")
+    return _col
+  }
+  async insert(col, _key, _val) {
+    const _col = this.getColTree(col)
+    let update = false
+    let res_doc
+    if ((await _col.get(_key)).found) {
+      update = true
+      res_doc = await _col.update(_key, _val)
+    } else {
+      res_doc = await _col.insert(_key, _val)
+    }
+    const res_col = await this.updateDB(_col, col)
+    return { update, doc: res_doc, col: res_col, tree: _col.tree }
+  }
+  async updateDB(_col, col) {
+    const root = _col.tree.F.toObject(_col.tree.root).toString()
+    const colD = col
+    return await this.tree.update(colD, [root])
+  }
+  async update(col, _key, _val) {
+    const _col = this.getColTree(col)
+    const res_doc = await _col.update(_key, _val)
+    const res_col = await this.updateDB(_col, col)
+    return { doc: res_doc, col: res_col, tree: _col.tree }
+  }
+
+  async delete(col, _key) {
+    const _col = this.getColTree(col)
+    const res_doc = await _col.delete(_key)
+    const res_col = await this.updateDB(_col, col)
+    return { doc: res_doc, col: res_col, tree: _col.tree }
+  }
+
+  async get(col, _key) {
+    const _col = this.getColTree(col)
+    return await _col.get(_key)
+  }
+  async getCol(col) {
+    return await this.tree.find(col)
+  }
+}

package/esm/index.js

@@ -1,7 +1,9 @@
+import DBTree from "./db_tree.js"
+import CollectionTree from "./collection_tree.js"
 import DB from "./db.js"
 import Doc from "./doc.js"
 import Collection from "./collection.js"
 import NFT from "./nft.js"
-export { DB, Collection, Doc, NFT }
+export { DB, Collection, Doc, NFT, DBTree, CollectionTree }
 export * from "./encoder.js"
 export * from "./parse.js"

package/esm/newMemEmptyTrie.js

@@ -0,0 +1,11 @@
+import SMTMemDb from "./circomlibjs/SMTMemDb.js"
+import SMT from "./circomlibjs/SMT.js"
+import getHashes from "./circomlibjs/getHashes.js"
+
+export default async function newMemEmptyTrie(kv) {
+  const { hash0, hash1, F } = await getHashes()
+  const db = new SMTMemDb(F, kv)
+  const rt = await db.getRoot()
+  const smt = new SMT(db, rt, hash0, hash1, F)
+  return smt
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "zkjson",
-  "version": "0.7.1",
+  "version": "0.7.2",
   "description": "Zero Knowledge Provable JSON",
   "exports": {
     ".": {