@economicagents/graph 0.1.0

package/dist/store.js

@@ -0,0 +1,288 @@
+/**
+ * SQLite store for economic graph. Uses better-sqlite3 (optional dependency).
+ * Falls back to sql.js (pure JS) when better-sqlite3 native bindings are unavailable.
+ */
+import { createRequire } from "module";
+import { mkdirSync } from "fs";
+import { join } from "path";
+const require = createRequire(import.meta.url);
+let db = null;
+let dbPath = null;
+let sqliteAvailable = null;
+let sqlJsFallback = null;
+/** For tests: set sql.js as fallback when better-sqlite3 native bindings are unavailable. */
+export function setSqlJsFallback(SQL) {
+    sqlJsFallback = SQL;
+}
+function wrapSqlJsDb(nativeDb) {
+    return {
+        exec(sql) {
+            nativeDb.exec(sql);
+        },
+        prepare(sql) {
+            return {
+                run: (...args) => {
+                    const stmt = nativeDb.prepare(sql);
+                    stmt.bind(args);
+                    while (stmt.step()) {
+                        /* consume rows */
+                    }
+                    stmt.free();
+                    return { changes: 1 };
+                },
+                get: (...args) => {
+                    const stmt = nativeDb.prepare(sql);
+                    stmt.bind(args);
+                    const row = stmt.step() ? stmt.getAsObject() : undefined;
+                    stmt.free();
+                    return row;
+                },
+                all: (...args) => {
+                    const stmt = nativeDb.prepare(sql);
+                    stmt.bind(args);
+                    const rows = [];
+                    while (stmt.step())
+                        rows.push(stmt.getAsObject());
+                    stmt.free();
+                    return rows;
+                },
+            };
+        },
+        close: () => nativeDb.close(),
+    };
+}
+/** Exported for tests: true when better-sqlite3 (disk persistence) is active; false when using sql.js fallback. */
+export function isUsingBetterSqlite3() {
+    if (sqliteAvailable === null)
+        isSqliteAvailable();
+    return sqliteAvailable === true;
+}
+/** Exported for tests: whether SQLite (better-sqlite3 or sql.js fallback) is available. */
+export function isSqliteAvailable() {
+    if (sqliteAvailable === null) {
+        try {
+            const Database = require("better-sqlite3");
+            const testDb = new Database(":memory:");
+            testDb.close();
+            sqliteAvailable = true;
+        }
+        catch {
+            sqliteAvailable = false;
+        }
+    }
+    if (sqliteAvailable)
+        return true;
+    return sqlJsFallback !== null;
+}
+function getDb(graphPath) {
+    const resolvedPath = join(graphPath, "graph.db");
+    if (db && dbPath === resolvedPath)
+        return db;
+    if (db) {
+        db.close();
+        db = null;
+        dbPath = null;
+    }
+    if (!isSqliteAvailable()) {
+        throw new Error("better-sqlite3 or sql.js required for graph. Install: npm install better-sqlite3");
+    }
+    try {
+        if (sqliteAvailable) {
+            const Database = require("better-sqlite3");
+            mkdirSync(graphPath, { recursive: true });
+            db = new Database(resolvedPath);
+            dbPath = resolvedPath;
+        }
+        else if (sqlJsFallback) {
+            // sql.js creates in-memory DB; graphPath is ignored. Use better-sqlite3 for production persistence.
+            const nativeDb = new sqlJsFallback.Database();
+            db = wrapSqlJsDb(nativeDb);
+            dbPath = resolvedPath;
+        }
+        else {
+            throw new Error("No SQLite backend available");
+        }
+        initSchema(db);
+        return db;
+    }
+    catch (err) {
+        const msg = err instanceof Error ? err.message : String(err);
+        throw new Error(`Graph database init failed: ${msg}. Install better-sqlite3 for production: npm install better-sqlite3`);
+    }
+}
+function initSchema(database) {
+    database.exec(`
+    CREATE TABLE IF NOT EXISTS accounts (
+      address TEXT PRIMARY KEY,
+      owner TEXT NOT NULL,
+      firstSeenBlock INTEGER NOT NULL
+    );
+
+    CREATE TABLE IF NOT EXISTS payments (
+      id INTEGER PRIMARY KEY AUTOINCREMENT,
+      fromAddr TEXT NOT NULL,
+      toAddr TEXT NOT NULL,
+      amount TEXT NOT NULL,
+      token TEXT NOT NULL,
+      blockNumber INTEGER NOT NULL,
+      txHash TEXT NOT NULL,
+      logIndex INTEGER,
+      source TEXT NOT NULL
+    );
+    CREATE INDEX IF NOT EXISTS idx_payments_from ON payments(fromAddr);
+    CREATE INDEX IF NOT EXISTS idx_payments_to ON payments(toAddr);
+    CREATE INDEX IF NOT EXISTS idx_payments_block ON payments(blockNumber);
+
+    CREATE TABLE IF NOT EXISTS user_ops (
+      userOpHash TEXT PRIMARY KEY,
+      sender TEXT NOT NULL,
+      success INTEGER NOT NULL,
+      actualGasCost TEXT,
+      blockNumber INTEGER,
+      txHash TEXT
+    );
+
+    CREATE TABLE IF NOT EXISTS facilities (
+      address TEXT PRIMARY KEY,
+      lender TEXT NOT NULL,
+      borrower TEXT NOT NULL,
+      firstSeenBlock INTEGER NOT NULL
+    );
+
+    CREATE TABLE IF NOT EXISTS escrows (
+      address TEXT PRIMARY KEY,
+      consumer TEXT NOT NULL,
+      provider TEXT NOT NULL,
+      firstSeenBlock INTEGER NOT NULL
+    );
+
+    CREATE TABLE IF NOT EXISTS splitters (
+      address TEXT PRIMARY KEY,
+      token TEXT NOT NULL,
+      firstSeenBlock INTEGER NOT NULL
+    );
+
+    CREATE TABLE IF NOT EXISTS slas (
+      address TEXT PRIMARY KEY,
+      provider TEXT NOT NULL,
+      consumer TEXT NOT NULL,
+      firstSeenBlock INTEGER NOT NULL
+    );
+
+    CREATE TABLE IF NOT EXISTS sla_events (
+      id INTEGER PRIMARY KEY AUTOINCREMENT,
+      slaAddress TEXT NOT NULL,
+      eventType TEXT NOT NULL,
+      provider TEXT NOT NULL,
+      consumer TEXT,
+      amount TEXT,
+      requestHash TEXT,
+      blockNumber INTEGER NOT NULL,
+      txHash TEXT
+    );
+    CREATE INDEX IF NOT EXISTS idx_sla_events_sla ON sla_events(slaAddress);
+    CREATE INDEX IF NOT EXISTS idx_sla_events_provider ON sla_events(provider);
+
+    CREATE TABLE IF NOT EXISTS sync_state (
+      contract_key TEXT PRIMARY KEY,
+      lastBlock INTEGER NOT NULL
+    );
+  `);
+}
+export function ensureGraphDir(graphPath) {
+    mkdirSync(graphPath, { recursive: true });
+}
+export function getDatabase(graphPath) {
+    return getDb(graphPath);
+}
+export function closeDatabase() {
+    if (db) {
+        db.close();
+        db = null;
+        dbPath = null;
+    }
+}
+export function insertAccount(database, address, owner, firstSeenBlock) {
+    database
+        .prepare("INSERT OR IGNORE INTO accounts (address, owner, firstSeenBlock) VALUES (?, ?, ?)")
+        .run(address.toLowerCase(), owner.toLowerCase(), firstSeenBlock);
+}
+export function insertPayment(database, fromAddr, toAddr, amount, token, blockNumber, txHash, logIndex, source) {
+    database
+        .prepare(`INSERT INTO payments (fromAddr, toAddr, amount, token, blockNumber, txHash, logIndex, source)
+       VALUES (?, ?, ?, ?, ?, ?, ?, ?)`)
+        .run(fromAddr.toLowerCase(), toAddr.toLowerCase(), amount, token.toLowerCase(), blockNumber, txHash, logIndex, source);
+}
+export function insertUserOp(database, userOpHash, sender, success, actualGasCost, blockNumber, txHash) {
+    database
+        .prepare(`INSERT OR REPLACE INTO user_ops (userOpHash, sender, success, actualGasCost, blockNumber, txHash)
+       VALUES (?, ?, ?, ?, ?, ?)`)
+        .run(userOpHash, sender.toLowerCase(), success ? 1 : 0, actualGasCost, blockNumber, txHash);
+}
+export function insertFacility(database, address, lender, borrower, firstSeenBlock) {
+    database
+        .prepare("INSERT OR IGNORE INTO facilities (address, lender, borrower, firstSeenBlock) VALUES (?, ?, ?, ?)")
+        .run(address.toLowerCase(), lender.toLowerCase(), borrower.toLowerCase(), firstSeenBlock);
+}
+export function insertEscrow(database, address, consumer, provider, firstSeenBlock) {
+    database
+        .prepare("INSERT OR IGNORE INTO escrows (address, consumer, provider, firstSeenBlock) VALUES (?, ?, ?, ?)")
+        .run(address.toLowerCase(), consumer.toLowerCase(), provider.toLowerCase(), firstSeenBlock);
+}
+export function insertSplitter(database, address, token, firstSeenBlock) {
+    database
+        .prepare("INSERT OR IGNORE INTO splitters (address, token, firstSeenBlock) VALUES (?, ?, ?)")
+        .run(address.toLowerCase(), token.toLowerCase(), firstSeenBlock);
+}
+export function insertSLA(database, address, provider, consumer, firstSeenBlock) {
+    database
+        .prepare("INSERT OR IGNORE INTO slas (address, provider, consumer, firstSeenBlock) VALUES (?, ?, ?, ?)")
+        .run(address.toLowerCase(), provider.toLowerCase(), consumer.toLowerCase(), firstSeenBlock);
+}
+export function getSyncState(database, key) {
+    const row = database
+        .prepare("SELECT lastBlock FROM sync_state WHERE contract_key = ?")
+        .get(key);
+    return row?.lastBlock ?? null;
+}
+export function setSyncState(database, key, lastBlock) {
+    database
+        .prepare("INSERT OR REPLACE INTO sync_state (contract_key, lastBlock) VALUES (?, ?)")
+        .run(key, lastBlock);
+}
+export function getAccountAddresses(database) {
+    const rows = database
+        .prepare("SELECT address FROM accounts")
+        .all();
+    return rows.map((r) => r.address);
+}
+export function getFacilityAddresses(database) {
+    const rows = database
+        .prepare("SELECT address FROM facilities")
+        .all();
+    return rows.map((r) => r.address);
+}
+export function getEscrowAddresses(database) {
+    const rows = database
+        .prepare("SELECT address FROM escrows")
+        .all();
+    return rows.map((r) => r.address);
+}
+export function getSplitterAddresses(database) {
+    const rows = database
+        .prepare("SELECT address FROM splitters")
+        .all();
+    return rows.map((r) => r.address);
+}
+export function getSLAAddresses(database) {
+    const rows = database
+        .prepare("SELECT address FROM slas")
+        .all();
+    return rows.map((r) => r.address);
+}
+export function insertSLAEvent(database, slaAddress, eventType, provider, consumer, amount, requestHash, blockNumber, txHash) {
+    database
+        .prepare(`INSERT INTO sla_events (slaAddress, eventType, provider, consumer, amount, requestHash, blockNumber, txHash)
+       VALUES (?, ?, ?, ?, ?, ?, ?, ?)`)
+        .run(slaAddress.toLowerCase(), eventType, provider.toLowerCase(), consumer?.toLowerCase() ?? null, amount, requestHash, blockNumber, txHash);
+}

package/dist/types.d.ts

@@ -0,0 +1,24 @@
+import type { Address } from "viem";
+export interface GraphConfig {
+    rpcUrl: string;
+    chainId: number;
+    graphPath: string;
+    aepAccountFactoryAddress: Address;
+    entryPointAddress: Address;
+    creditFacilityFactoryAddress?: Address;
+    escrowFactoryAddress?: Address;
+    revenueSplitterFactoryAddress?: Address;
+    slaFactoryAddress?: Address;
+    usdcAddress: Address;
+}
+export interface SyncResult {
+    accountsAdded: number;
+    paymentsAdded: number;
+    userOpsAdded: number;
+    creditEventsAdded: number;
+    escrowEventsAdded: number;
+    splitterEventsAdded: number;
+    slaEventsAdded: number;
+}
+export type PaymentSource = "transfer" | "credit_draw" | "credit_repay" | "escrow_fund" | "escrow_release" | "splitter_distribute";
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAEpC,MAAM,WAAW,WAAW;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,wBAAwB,EAAE,OAAO,CAAC;IAClC,iBAAiB,EAAE,OAAO,CAAC;IAC3B,4BAA4B,CAAC,EAAE,OAAO,CAAC;IACvC,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B,6BAA6B,CAAC,EAAE,OAAO,CAAC;IACxC,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,WAAW,EAAE,OAAO,CAAC;CACtB;AAED,MAAM,WAAW,UAAU;IACzB,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,iBAAiB,EAAE,MAAM,CAAC;IAC1B,mBAAmB,EAAE,MAAM,CAAC;IAC5B,cAAc,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,MAAM,aAAa,GACrB,UAAU,GACV,aAAa,GACb,cAAc,GACd,aAAa,GACb,gBAAgB,GAChB,qBAAqB,CAAC"}

package/dist/types.js

@@ -0,0 +1 @@
+export {};

package/package.json

@@ -0,0 +1,56 @@
+{
+  "name": "@economicagents/graph",
+  "version": "0.1.0",
+  "description": "Economic graph for AEP - transaction graph, payments, credit events, credit scoring, recommendations",
+  "license": "Apache-2.0",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/economicagents/AEP.git"
+  },
+  "bugs": "https://github.com/economicagents/AEP/issues",
+  "homepage": "https://github.com/economicagents/AEP#readme",
+  "keywords": [
+    "aep",
+    "agent",
+    "graph",
+    "analytics",
+    "credit-score"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "type": "module",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "bin": {
+    "aep-graph": "./dist/cli.js"
+  },
+  "dependencies": {
+    "viem": "^2.21.0"
+  },
+  "optionalDependencies": {
+    "better-sqlite3": "^12.1.0"
+  },
+  "devDependencies": {
+    "@types/better-sqlite3": "^7.6.11",
+    "sql.js": "^1.14.0",
+    "typescript": "^5.6.0",
+    "vitest": "^2.1.0"
+  },
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "sync": "node dist/cli.js sync",
+    "test": "vitest run",
+    "test:watch": "vitest"
+  }
+}