dal/client/Bunding.ts

/**
 * This file is responsible for binding the C library to the Bun runtime.
 */
import { dlopen, FFIType, suffix, ptr, toBuffer } from "bun:ffi";
import { join } from "path";

const libname = `clib.${suffix}`;
const libpath = join("clib", libname);

const {
  symbols: { InitSQLite, CreateRowIterator, NextRow, GetLen, Free, Cleanup },
} = dlopen(libpath, {
  InitSQLite: {
    args: [FFIType.cstring],
    returns: FFIType.void,
  },
  CreateRowIterator: {
    args: [FFIType.cstring, FFIType.i32],
    returns: FFIType.i32,
  },
  NextRow: {
    args: [FFIType.i32],
    returns: FFIType.ptr,
  },
  GetLen: {
    args: [FFIType.i32],
    returns: FFIType.i32,
  },
  Free: {
    args: [FFIType.ptr],
    returns: FFIType.void,
  },
  Cleanup: {
    args: [FFIType.i32],
    returns: FFIType.void,
  },
});

function initSQLite(pragmas: Buffer) {
  InitSQLite(ptr(pragmas));
}

function rowIterator(buf: Buffer) {
  const iter = CreateRowIterator(ptr(buf), buf.length);
  const next = () => {
    const pointer = NextRow(iter);
    if (pointer === null) {
      return null;
    }
    const buf = Buffer.from(toBuffer(pointer, 0, GetLen(iter)));
    Free(pointer);
    return buf;
  };

  const cleanup = () => {
    Cleanup(iter);
  };

  return {
    next,
    cleanup,
  };
}

export default {
  initSQLite,
  rowIterator,
};