@pineliner/odb-client 1.1.5 → 1.2.0

package/dist/index.cjs

@@ -2138,6 +2138,417 @@ var __webpack_exports__ = {};
             return this.client.httpClient.getNamespaces();
         }
     }
+    function toHrana(v) {
+        if (null == v) return {
+            type: 'null'
+        };
+        if ('bigint' == typeof v) return {
+            type: 'integer',
+            value: v.toString()
+        };
+        if ('number' == typeof v) return Number.isInteger(v) ? {
+            type: 'integer',
+            value: String(v)
+        } : {
+            type: 'float',
+            value: v
+        };
+        if ('string' == typeof v) return {
+            type: 'text',
+            value: v
+        };
+        if (v instanceof Uint8Array) return {
+            type: 'blob',
+            base64: Buffer.from(v).toString('base64')
+        };
+        if ('boolean' == typeof v) return {
+            type: 'integer',
+            value: v ? '1' : '0'
+        };
+        return {
+            type: 'text',
+            value: String(v)
+        };
+    }
+    function fromHrana(v) {
+        switch(v.type){
+            case 'null':
+                return null;
+            case 'integer':
+                {
+                    const n = Number(v.value);
+                    return Number.isSafeInteger(n) ? n : BigInt(v.value);
+                }
+            case 'float':
+                return v.value;
+            case 'text':
+                return v.value;
+            case 'blob':
+                return new Uint8Array(Buffer.from(v.base64, 'base64'));
+        }
+    }
+    function toBatch(steps) {
+        return {
+            steps: steps.map((s, i)=>({
+                    condition: 0 === i ? void 0 : {
+                        type: 'ok',
+                        step: i - 1
+                    },
+                    stmt: {
+                        sql: s.sql,
+                        args: (s.params ?? []).map(toHrana)
+                    },
+                    expect: s.expect ? {
+                        affected: s.expect
+                    } : void 0
+                }))
+        };
+    }
+    class FrameWriter {
+        socket;
+        pending = Buffer.alloc(0);
+        constructor(socket){
+            this.socket = socket;
+        }
+        send(payload) {
+            const header = Buffer.allocUnsafe(4);
+            header.writeUInt32LE(payload.length, 0);
+            const frame = Buffer.concat([
+                header,
+                payload
+            ]);
+            this.pending = this.pending.length ? Buffer.concat([
+                this.pending,
+                frame
+            ]) : frame;
+            this.flush();
+        }
+        flush() {
+            while(this.pending.length > 0){
+                const n = this.socket.write(this.pending);
+                if (n <= 0) return;
+                this.pending = n < this.pending.length ? this.pending.subarray(n) : Buffer.alloc(0);
+            }
+        }
+    }
+    class FrameReader {
+        buf = Buffer.alloc(0);
+        push(chunk) {
+            this.buf = this.buf.length ? Buffer.concat([
+                this.buf,
+                chunk
+            ]) : chunk;
+            const out = [];
+            while(this.buf.length >= 4){
+                const len = this.buf.readUInt32LE(0);
+                if (this.buf.length < 4 + len) break;
+                out.push(this.buf.subarray(4, 4 + len));
+                this.buf = this.buf.subarray(4 + len);
+            }
+            return out;
+        }
+    }
+    class SocketClient {
+        target;
+        socket;
+        reader = new FrameReader();
+        writer;
+        pending = new Map();
+        nextId = 1;
+        ready;
+        constructor(target){
+            this.target = target;
+            this.ready = this.connect();
+            this.ready.catch(()=>{});
+        }
+        async connect() {
+            const Bun = globalThis.Bun;
+            if (!Bun?.connect) throw new Error('odblite-socket backend requires Bun (Bun.connect)');
+            const opts = this.target.unix ? {
+                unix: this.target.unix
+            } : {
+                hostname: this.target.hostname ?? '127.0.0.1',
+                port: this.target.port ?? 8787
+            };
+            this.socket = await Bun.connect({
+                ...opts,
+                socket: {
+                    data: (_s, chunk)=>{
+                        for (const frame of this.reader.push(chunk)){
+                            const msg = JSON.parse(frame.toString('utf8'));
+                            const resolve = this.pending.get(msg.id);
+                            if (resolve) {
+                                this.pending.delete(msg.id);
+                                resolve(msg.results);
+                            }
+                        }
+                    },
+                    drain: (_s)=>this.writer.flush(),
+                    close: ()=>{
+                        this.socket = void 0;
+                    }
+                }
+            });
+            this.writer = new FrameWriter(this.socket);
+        }
+        async pipeline(tenant, requests) {
+            if (!this.socket) {
+                this.ready = this.connect();
+                this.ready.catch(()=>{});
+            }
+            await this.ready;
+            const id = this.nextId++;
+            return new Promise((resolve)=>{
+                this.pending.set(id, resolve);
+                this.writer.send(Buffer.from(JSON.stringify({
+                    id,
+                    tenant,
+                    requests
+                }), 'utf8'));
+            });
+        }
+        close() {
+            this.socket?.end?.();
+            this.socket = void 0;
+        }
+    }
+    function toQueryResult(res) {
+        if ('error' === res.type) throw new Error(res.error.message);
+        const r = res.response.result;
+        const rows = r.rows.map((row)=>{
+            const obj = {};
+            r.cols.forEach((c, i)=>{
+                obj[c.name ?? `c${i}`] = fromHrana(row[i] ?? {
+                    type: 'null'
+                });
+            });
+            return obj;
+        });
+        return {
+            rows,
+            rowsAffected: r.affected_row_count,
+            lastInsertRowid: r.last_insert_rowid ? Number(r.last_insert_rowid) : void 0
+        };
+    }
+    class OdbliteSocketConnection {
+        client;
+        tenant;
+        databaseName;
+        databaseHash;
+        sql;
+        buffer;
+        constructor(client, tenant, buffer){
+            this.client = client;
+            this.tenant = tenant;
+            this.databaseName = tenant;
+            this.databaseHash = tenant;
+            this.buffer = buffer;
+            const self = this;
+            this.sql = function(strings, ...values) {
+                if (strings && 'object' == typeof strings && 'raw' in strings) {
+                    const text = strings.reduce((a, s, i)=>a + s + (i < values.length ? '?' : ''), '');
+                    const returns = /^\s*SELECT/i.test(text) || /\bRETURNING\b/i.test(text);
+                    return (returns ? self.query(text, values) : self.execute(text, values)).then((r)=>r.rows);
+                }
+                throw new Error('sql() fragment helper not implemented in odblite-socket');
+            };
+        }
+        async one(requests) {
+            const [res] = await this.client.pipeline(this.tenant, requests);
+            if (!res) throw new Error('odblite-socket: empty pipeline response');
+            return res;
+        }
+        async query(sql, params = []) {
+            return toQueryResult(await this.one([
+                {
+                    type: 'execute',
+                    stmt: {
+                        sql,
+                        args: params.map(toHrana)
+                    }
+                }
+            ]));
+        }
+        async execute(sql, params = []) {
+            const s = 'string' == typeof sql ? sql : sql.sql;
+            const p = 'string' == typeof sql ? params : sql.args ?? [];
+            if (this.buffer) {
+                this.buffer.push({
+                    sql: s,
+                    params: p
+                });
+                return {
+                    rows: [],
+                    rowsAffected: 0
+                };
+            }
+            return toQueryResult(await this.one([
+                {
+                    type: 'execute',
+                    stmt: {
+                        sql: s,
+                        args: p.map(toHrana)
+                    }
+                }
+            ]));
+        }
+        async atomic(steps) {
+            const res = await this.one([
+                {
+                    type: 'batch',
+                    batch: toBatch(steps)
+                }
+            ]);
+            if ('error' === res.type) throw new Error(res.error.message);
+            return res.response.result;
+        }
+        prepare(sql) {
+            return {
+                execute: async (params = [])=>this.execute(sql, params),
+                all: async (params = [])=>(await this.query(sql, params)).rows,
+                get: async (params = [])=>(await this.query(sql, params)).rows[0] ?? null
+            };
+        }
+        async transaction(fn) {
+            if (this.buffer) return fn(this);
+            const buffer = [];
+            const out = await fn(new OdbliteSocketConnection(this.client, this.tenant, buffer));
+            if (buffer.length) {
+                const res = await this.atomic(buffer.map((b)=>({
+                        sql: b.sql,
+                        params: b.params
+                    })));
+                const err = res.step_errors?.find((e)=>e);
+                if (err) throw new Error(`transaction batch failed: ${err.message}`);
+            }
+            return out;
+        }
+        begin(fn) {
+            return this.transaction(fn);
+        }
+        savepoint(fn) {
+            return this.transaction(fn);
+        }
+        async ensureNamespaces(namespaces) {
+            return {
+                success: true,
+                databaseName: this.tenant,
+                namespaces
+            };
+        }
+        async getNamespaces() {
+            const ns = (await this.query("SELECT name FROM pragma_database_list WHERE name != 'main'")).rows.map((r)=>r.name);
+            return {
+                success: true,
+                databaseName: this.tenant,
+                exists: true,
+                namespaces: ns,
+                registered: true,
+                registeredNamespaces: ns
+            };
+        }
+        async close() {}
+    }
+    class OdbliteSocketAdapter {
+        type = 'odblite-socket';
+        client;
+        constructor(config){
+            this.client = new SocketClient(config);
+        }
+        async connect(config) {
+            const tenant = config?.databaseHash ?? config?.databaseName;
+            if (!tenant) throw new Error('odblite-socket: databaseHash (or databaseName) required');
+            return new OdbliteSocketConnection(this.client, String(tenant));
+        }
+        async disconnect() {
+            this.client.close();
+        }
+        async isHealthy() {
+            try {
+                await Promise.race([
+                    this.client.pipeline('_health', [
+                        {
+                            type: 'get_autocommit'
+                        }
+                    ]),
+                    new Promise((_, rej)=>setTimeout(()=>rej(new Error('probe timeout')), 1500))
+                ]);
+                return true;
+            } catch  {
+                return false;
+            }
+        }
+    }
+    const DEFAULT_UNIX_SOCKET = '/tmp/odblite.sock';
+    function parseSocketTarget(s) {
+        const v = (s ?? '').trim();
+        if (!v) return {
+            unix: DEFAULT_UNIX_SOCKET
+        };
+        if (v.includes('/')) return {
+            unix: v
+        };
+        const m = v.match(/^(?:([^:]+):)?(\d+)$/);
+        if (m) return {
+            hostname: m[1] || '127.0.0.1',
+            port: Number(m[2])
+        };
+        return {
+            unix: v
+        };
+    }
+    class OdbliteAutoAdapter {
+        type = 'odblite';
+        chosen;
+        constructor(opts){
+            this.chosen = this.pick(opts);
+        }
+        async pick(opts) {
+            const hasBun = void 0 !== globalThis.Bun && !!globalThis.Bun.connect;
+            if (hasBun) for (const target of this.candidates(opts))try {
+                const socket = new OdbliteSocketAdapter(target);
+                if (await socket.isHealthy()) {
+                    console.log(`[odb-client] transport → SOCKET (router_v2 @ ${target.unix ?? `${target.hostname}:${target.port}`})`);
+                    return socket;
+                }
+                await socket.disconnect();
+            } catch  {}
+            console.log('[odb-client] transport → HTTP (no router_v2 socket reachable)');
+            return new ODBLiteAdapter(opts.http);
+        }
+        candidates(opts) {
+            if (opts.socket?.unix || opts.socket?.port) return [
+                opts.socket
+            ];
+            const env = globalThis.process?.env?.ODB_SOCKET;
+            if (env) return [
+                parseSocketTarget(env)
+            ];
+            const out = [
+                {
+                    unix: DEFAULT_UNIX_SOCKET
+                }
+            ];
+            try {
+                const host = new URL(opts.http?.serviceUrl).hostname;
+                const port = Number(globalThis.process?.env?.ODB_SOCKET_PORT ?? 8787);
+                if (host) out.push({
+                    hostname: host,
+                    port
+                });
+            } catch  {}
+            return out;
+        }
+        async connect(config) {
+            return (await this.chosen).connect(config);
+        }
+        async disconnect(tenantId) {
+            return (await this.chosen).disconnect(tenantId);
+        }
+        async isHealthy() {
+            return (await this.chosen).isHealthy();
+        }
+    }
     function parseSQL(sqlContent, options = {}) {
         const separatePragma = options.separatePragma ?? true;
         const statements = splitStatements(sqlContent);
@@ -2249,7 +2660,14 @@ var __webpack_exports__ = {};
                     break;
                 case 'odblite':
                     if (!config.odblite) throw new Error('odblite config required for odblite backend');
-                    this.adapter = new ODBLiteAdapter(config.odblite);
+                    this.adapter = new OdbliteAutoAdapter({
+                        socket: config.odbliteSocket,
+                        http: config.odblite
+                    });
+                    break;
+                case 'odblite-socket':
+                    if (!config.odbliteSocket) throw new Error('odbliteSocket config required for odblite-socket backend');
+                    this.adapter = new OdbliteSocketAdapter(config.odbliteSocket);
                     break;
                 default:
                     throw new Error(`Unknown backend type: ${config.backend}`);
@@ -2367,6 +2785,12 @@ var __webpack_exports__ = {};
                         databaseName: `${this.config.databasePath}_${name}`,
                         ...this.config.odblite
                     };
+                case 'odblite-socket':
+                    return {
+                        databaseName: name,
+                        databaseHash: name,
+                        ...this.config.odbliteSocket
+                    };
                 default:
                     throw new Error(`Unknown backend: ${this.config.backend}`);
             }