@calp-pro/dex-db 1.0.4 → 1.0.6

package/dex_db.js

@@ -3,84 +3,69 @@
 const rl = require('readline')
 
 function dex_db(pairs = []) {
-    const trie = {
-        pairs: {},
-        tokens: {}
-    }
-    const P = []//pairs
-    const T = []//tokens
-    const p2tt = []// [[it0, it1], ...]
+    const aP = []
+    const aT = []
+    const P = new Map()//pairs
+    const T = new Map()//tokens
+    const p2tt = []// [p1.it0, p1.it1, p2.it0, p2, it1, ...]
     const t2pt = []// [ [[ip, it], [ip, it], ...], ...]
 
-    const set = (address, index, trie) =>
-        address.split('').reduce(
-            (node, letter, i, a, last = i == a.length - 1) => {
-                if (last)
-                    node[letter] = index
-                else
-                    node[letter] ??= {}
-                return node[letter]
-            },
-            trie
-        )
-
-    const get = (address, trie) => {
-        const letters = address.split('')
-        let node = trie
-        for (let i = 0, letter; i < letters.length; i++) {
-            letter = letters[i]
-            if (node[letter] == undefined) return -1
-            node = node[letter]
-        }
-        return node
-    }
-
     const index = ([pair, token0, token1]) => {
-        var ip = get(pair, trie.pairs)
-        if (ip == -1) {
-            ip = P.length
-            P[ip] = pair
-            set(pair, ip, trie.pairs)
+        var ip = P.get(pair)
+        if (ip == undefined) {
+            P.set(pair, ip = P.size)
+            aP.push(pair)
         }
-        var it0 = get(token0, trie.tokens)
-        if (it0 == -1) {
-            it0 = T.length
-            T[it0] = token0
-            set(token0, it0, trie.tokens)
+
+        var it0 = T.get(token0)
+        if (it0 == undefined) {
+            T.set(token0, it0 = T.size)
+            aT.push(token1)
         }
-        var it1 = get(token1, trie.tokens)
-        if (it1 == -1) {
-            it1 = T.length
-            T[it1] = token1
-            set(token1, it1, trie.tokens)
+
+        var it1 = T.get(token1)
+        if (it1 == undefined) {
+            T.set(token1, it1 = T.size)
+            aT.push(token1)
         }
-        p2tt[ip] ??= []
-        p2tt[ip][0] = it0
-        p2tt[ip][1] = it1
+
+        p2tt[ip * 2] = it0
+        p2tt[ip * 2 + 1] = it1
         
         if (t2pt[it0])
-            t2pt[it0].push([ip, it1])
+            t2pt[it0].push(ip, it1)
         else
-            t2pt[it0] = [[ip, it1]]
+            t2pt[it0] = [ip, it1]
             
         if (t2pt[it1])
-            t2pt[it1].push([ip, it0])
+            t2pt[it1].push(ip, it0)
         else
-            t2pt[it1] = [[ip, it0]]
+            t2pt[it1] = [ip, it0]
     }
 
     const find_pairs_with_token = token => {
-        const it = get(token, trie.tokens)
-        if (it == -1) return []
-        return t2pt[it].map(pt => P[pt[0]])
+        const pairs = []
+        const it = T.get(token)
+        if (it == undefined) return pairs
+        for (var i = 0; i < t2pt[it].length; i += 2)
+            pairs.push(
+                aP[t2pt[it][i]]
+            )
+        return pairs
     }
 
     const find_pairs_with_tokens = (token0, token1) => {
-        const it0 = get(token0, trie.tokens)
-        if (it0 == -1) return []
-        const it1 = get(token1, trie.tokens)
-        if (it1 == -1) return []
-        return t2pt[it0].filter(pt => pt[1] == it1).map(pt => P[pt[0]])
+        const pairs = []
+        const it0 = T.get(token0)
+        if (it0 == undefined) return pairs
+        const it1 = T.get(token1)
+        if (it1 == undefined) return pairs
+        for (var i = 0; i < t2pt[it0].length; i += 2)
+            if (t2pt[it0][i + 1] == it1)
+                pairs.push(
+                    aP[t2pt[it0][i]]
+                )
+        return pairs
     }
     
     pairs.forEach(index)
@@ -89,7 +74,7 @@ function dex_db(pairs = []) {
         index,
         find_pairs_with_token,
         find_pairs_with_tokens,
-        //save, TODO 
+        //save, TODO
         //load, TODO
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@calp-pro/dex-db",
-  "version": "1.0.4",
+  "version": "1.0.6",
   "description": "Database designed for handle relation pair/pool address to token0 & token1 addresses",
   "keywords": [
     "DEX",
@@ -18,12 +18,23 @@
   "author": "Vladimir Spirin (spirin.vladimir@gmail.com)",
   "type": "commonjs",
   "main": "dex_db.js",
+  "module": "src/index.mjs",
+  "exports": {
+    ".": {
+      "types": "./index.d.ts",
+      "require": "./dex_db.js",
+      "import": "./index.mjs",
+      "default": "./dex_db.js"
+    }
+  },
+  "types": "index.d.ts",
   "scripts": {
     "test": "node --test test.*"
   },
   "devDependencies": {
     "pancakeswap-dump": "^1.0.8",
-    "sushiswap-dump": "^1.0.3"
+    "sushiswap-dump": "^1.0.3",
+    "uniswap-v2-dump": "^2.0.24"
   },
   "publishConfig": {
     "provenance": true

package/test-uniswap-v2.js

@@ -0,0 +1,32 @@
+const { describe, before, it } = require('node:test')
+const assert = require('node:assert/strict')
+const uniswap_v2_dump = require('uniswap-v2-dump')
+const dex_db = require('./dex_db')
+
+describe('Uniswap v2', () => {
+    var db
+
+    before(() =>
+        uniswap_v2_dump.load({workers: 0})
+        .then(pairs =>
+            db = dex_db(
+                pairs.map(({pair, token0, token1}) =>
+                    [pair, token0, token1]
+                )
+            )
+        )
+    )
+
+    it('Get all pairs with BAT at Uniswap v2', () => {
+        const pairs = db.find_pairs_with_token(
+            '0x0d8775f648430679a709e98d2b0cb6250d2887ef'/*BAT*/
+        )
+        assert.ok(pairs.length >= 40)
+        assert.equal(pairs[0], '0xb6909b960dbbe7392d405429eb2b3649752b4838', 'BAT/WETH https://etherscan.io/address/0xaf4b3145ca0cadbd9454f5815ef5d221f828507e')
+        assert.equal(pairs[1], '0x6929abd7931d0243777d3cd147fe863646a752ba', 'BAT/DAI https://etherscan.io/address/0x6929abd7931d0243777d3cd147fe863646a752ba')
+    })
+
+})
+
+
+