@smoothbricks/lmao 0.1.0

package/dist/lib/sqlite/sqlite-writer.js

@@ -0,0 +1,78 @@
+import { buildAddColumnSql, buildInsertParams, buildInsertSql, extractSqliteColumnsFromTableInfo, getActiveUserFields, getInsertStatementCacheKey, getMissingSchemaColumns, isSqliteDuplicateColumnError, parseSqliteTableInfoRows, SPANS_TABLE_INFO_SQL, SPANS_TABLE_INIT_SQL, walkSpanSegments, } from './sqlite-common.js';
+export class SQLiteTraceWriter {
+    db;
+    knownColumns = new Set();
+    insertStmtCache = new Map();
+    constructor(db) {
+        this.db = db;
+        this.init();
+    }
+    init() {
+        // Parallel Bun test workers share one trace DB, so wait for short SQLite writer locks instead of failing setup.
+        this.db.exec('PRAGMA busy_timeout = 10000');
+        this.db.exec(SPANS_TABLE_INIT_SQL);
+        this.refreshKnownColumns();
+    }
+    refreshKnownColumns() {
+        this.knownColumns.clear();
+        // WHY: PRAGMA row shape changes would make later ALTER/INSERT logic drift silently, so validate here before
+        // mutating the writer's cached schema view.
+        const rows = parseSqliteTableInfoRows(this.db.prepare(SPANS_TABLE_INFO_SQL).all());
+        for (const column of extractSqliteColumnsFromTableInfo(rows)) {
+            this.knownColumns.add(column.name);
+        }
+    }
+    /** Ensure user-defined schema columns exist in the spans table */
+    ensureColumns(schema) {
+        for (const column of getMissingSchemaColumns(schema, this.knownColumns)) {
+            try {
+                this.db.exec(buildAddColumnSql(column));
+            }
+            catch (error) {
+                if (!isSqliteDuplicateColumnError(error, column.name)) {
+                    throw error;
+                }
+            }
+            this.refreshKnownColumns();
+            this.knownColumns.add(column.name);
+        }
+    }
+    writeSegmentRows(segment) {
+        this.ensureColumns(segment.buffer._logSchema);
+        const activeUserFields = getActiveUserFields(segment.buffer._logSchema, this.knownColumns);
+        const insertStmt = this.getInsertStatement(activeUserFields);
+        for (let row = 0; row < segment.buffer._writeIndex; row++) {
+            insertStmt.run(...buildInsertParams(segment, row, activeUserFields));
+        }
+    }
+    flushAllSegments(rootBuffer) {
+        for (const segment of walkSpanSegments(rootBuffer)) {
+            this.writeSegmentRows(segment);
+        }
+    }
+    getInsertStatement(activeUserFields) {
+        const key = getInsertStatementCacheKey(activeUserFields);
+        const cached = this.insertStmtCache.get(key);
+        if (cached) {
+            return cached;
+        }
+        const stmt = this.db.prepare(buildInsertSql(activeUserFields));
+        this.insertStmtCache.set(key, stmt);
+        return stmt;
+    }
+    /** Write a root SpanBuffer tree to the database */
+    flush(rootBuffer) {
+        this.db.exec('BEGIN IMMEDIATE');
+        try {
+            this.flushAllSegments(rootBuffer);
+            this.db.exec('COMMIT');
+        }
+        catch (error) {
+            this.db.exec('ROLLBACK');
+            throw error;
+        }
+    }
+    close() {
+        this.db.close();
+    }
+}

package/dist/lib/testing/bun-harness.d.ts

@@ -0,0 +1,150 @@
+import * as bunTest from 'bun:test';
+import { afterAll as _afterAll, afterEach as _afterEach, beforeAll as _beforeAll, beforeEach as _beforeEach, describe as _describe, expect as _expect, it as _it } from 'bun:test';
+import type { SpanContext } from '../opContext/spanContextTypes.js';
+import type { OpContextBinding, OpContextOf } from '../opContext/types.js';
+import { S } from '../schema/builder.js';
+import { LogSchema } from '../schema/LogSchema.js';
+import type { SchemaFields } from '../schema/types.js';
+import type { SQLiteWriterConfig } from '../sqlite/sqlite-writer.js';
+import type { Tracer } from '../tracer.js';
+type HarnessSpanContext<B extends OpContextBinding> = SpanContext<OpContextOf<B>>;
+type BunTestHarnessBuiltins = {
+    describe: ReturnType<typeof S.category>;
+};
+type HarnessSchema<TExt extends SchemaFields> = BunTestHarnessBuiltins & TExt;
+type BunDescribe = typeof _describe;
+type BunIt = typeof _it;
+export interface BunTestModuleShape {
+    describe: BunDescribe;
+    it: BunIt;
+    test: BunIt;
+    [key: string]: unknown;
+}
+type ExtendBindingLogSchema<B extends OpContextBinding, TExt extends SchemaFields> = B extends OpContextBinding<infer T, infer FF, infer Deps, infer UserCtx> ? OpContextBinding<T extends LogSchema<infer Fields> ? LogSchema<Fields & TExt> : never, FF, Deps, UserCtx> : never;
+export type BunTestSetupOptions<TExt extends SchemaFields = Record<never, never>> = {
+    /**
+     * Optional SQLite output path.
+     *
+     * Omit this to run traced tests without DB persistence.
+     */
+    sqlite?: SQLiteWriterConfig;
+    /** Optional verbose stdout tracing; defaults to env flag detection. */
+    verbose?: boolean;
+    /** Optional test-only schema columns merged into span tag/log methods. */
+    extraTestColumns?: TExt;
+};
+export interface BunTestTracerInstance<B extends OpContextBinding> {
+    setup(): void;
+    useTestSpan(): HarnessSpanContext<B>;
+    getTracer(): Tracer<B>;
+    createBunTestMock<TModule extends BunTestModuleShape>(bunTestModule: TModule): TModule;
+}
+export interface BunTestSuiteTracer<B extends OpContextBinding> {
+    useTestTracer: BunTestTracerInstance<B>;
+    useTestSpan(): HarnessSpanContext<B>;
+    setupBunTestSuiteTracing(): void;
+}
+export type BunTestSuiteUseTestTracer<B extends OpContextBinding, TExt extends SchemaFields = Record<never, never>> = ReturnType<typeof makeBunTestSuiteTracer<B, TExt>>['useTestTracer'];
+export type TestTracer<B extends OpContextBinding, TExt extends SchemaFields = Record<never, never>> = BunTestSuiteUseTestTracer<B, TExt>;
+type ActiveBunTestTracer = BunTestTracerInstance<OpContextBinding>;
+/** Returns the active suite tracer set by installBunTestTracing(), or null if none is active. */
+export declare function getActiveSuiteTracer(): ActiveBunTestTracer | null;
+export declare function installBunTestTracing(tracer: ActiveBunTestTracer): void;
+export declare function makeBunTestSuiteTracer<B extends OpContextBinding>(binding: B, options?: Omit<BunTestSetupOptions, 'extraTestColumns'>): BunTestSuiteTracer<ExtendBindingLogSchema<B, BunTestHarnessBuiltins>>;
+export declare function makeBunTestSuiteTracer<B extends OpContextBinding, TExt extends SchemaFields = Record<never, never>>(binding: B, options?: BunTestSetupOptions<TExt>): BunTestSuiteTracer<ExtendBindingLogSchema<B, HarnessSchema<TExt>>>;
+/**
+ * Define a per-package test tracer — metadata for preload discovery + typed span accessor.
+ *
+ * Returns plain data that `autoSetupBunTestTracing` reads at discovery time, plus a
+ * `useTestSpan()` that threads the package-local binding through the generic
+ * `useTestSpan<B>()` helper.
+ *
+ * WHY generic threading: `autoSetupBunTestTracing` creates the active tracer from the
+ * same `opContext` binding passed here, so the exported helper can preserve the real
+ * span type without a local assertion bridge.
+ */
+export declare function defineTestTracer<B extends OpContextBinding>(binding: B, options?: {
+    extraTestColumns?: SchemaFields;
+}): {
+    useTestSpan: () => SpanContext<OpContextOf<B>>;
+    opContext: B;
+    extraTestColumns: SchemaFields | undefined;
+};
+/**
+ * Create an instance-scoped Bun test tracer harness.
+ *
+ * Unlike initTraceTestRun(), this keeps tracer/root/ALS state isolated per instance.
+ */
+export declare function makeTestTracer<B extends OpContextBinding>(binding: B, options?: Omit<BunTestSetupOptions, 'extraTestColumns'>): BunTestTracerInstance<ExtendBindingLogSchema<B, BunTestHarnessBuiltins>>;
+export declare function makeTestTracer<B extends OpContextBinding, TExt extends SchemaFields = Record<never, never>>(binding: B, options?: BunTestSetupOptions<TExt>): BunTestTracerInstance<ExtendBindingLogSchema<B, HarnessSchema<TExt>>>;
+/** Initialize the root tracer for the entire bun test run. Call once in preload. */
+export declare function initTraceTestRun<B extends OpContextBinding>(opContext: B, options?: BunTestSetupOptions): void;
+/**
+ * Create a bun:test mock replacement that wraps it()/test() in trace spans.
+ *
+ * Call this from the preload file and pass the result to mock.module:
+ * ```typescript
+ * mock.module('bun:test', () => createBunTestMock(bunTest));
+ * ```
+ *
+ * @param bunTestModule - The original bun:test namespace (`import * as bunTest from 'bun:test'`)
+ */
+export declare function createBunTestMock<TModule extends BunTestModuleShape>(bunTestModule: TModule): TModule;
+/**
+ * Get the current span context from AsyncLocalStorage (inside an it() block).
+ *
+ * Package-local tracer modules pass their own exported span-context type here so
+ * `useTestSpan()` preserves the suite-local schema extension instead of falling
+ * back to the erased global singleton shape.
+ */
+export declare function useTestSpan<TCtx extends SpanContext<OpContextOf<OpContextBinding>>>(): TCtx;
+/**
+ * Get the root tracer instance.
+ *
+ * The optional type parameter narrows the return type for consumers that know
+ * the concrete binding (e.g. `getTracer<typeof opContext>()`). The underlying
+ * tracer is always the one created by `installBunTestTracing()` or `initTraceTestRun()`.
+ */
+export declare function getTracer(): Tracer<OpContextBinding>;
+/**
+ * Wrapped describe — tracks describe nesting for the standalone export path.
+ * describe() callbacks run synchronously (just registering tests).
+ */
+export declare function describe(name: string, fn: () => void): void;
+export declare namespace describe {
+    var skip: bunTest.Describe<[]>;
+    var todo: bunTest.Describe<[]>;
+}
+/** Wrapped it — creates a child span of the root trace for the test case */
+export declare function it(name: string, fn: () => void | Promise<void>): void;
+export declare namespace it {
+    var skip: bunTest.Test<[]>;
+    var todo: bunTest.Test<[]>;
+}
+export { _afterAll as afterAll, _afterEach as afterEach, _beforeAll as beforeAll, _beforeEach as beforeEach, _expect as expect, };
+export type AutoSetupOptions = {
+    /** Absolute path to the packages directory to scan for test-suite-tracer.ts files. */
+    packagesDir: string;
+};
+/**
+ * Auto-discover package test-suite-tracers and create the root tracer for the
+ * current bun test run.
+ *
+ * When Bun is running from inside `packages/<name>`, prefer that package's own
+ * `src/test-suite-tracer.ts` and avoid importing unrelated packages. Repo-root
+ * runs still fall back to the full scan so cross-package test runs keep the
+ * merged-schema behavior.
+ *
+ * Per-package `useTestSpan()` delegates to the global `useTestSpan()` from this
+ * module which checks `_activeSuiteTracer` — set by the root preload via
+ * `installBunTestTracing()`.
+ *
+ * @example
+ * ```ts
+ * // test-trace-preload.ts (monorepo root)
+ * import { autoSetupBunTestTracing } from './packages/lmao/src/lib/testing/bun-harness.js';
+ * await autoSetupBunTestTracing({ packagesDir: './packages' });
+ * ```
+ */
+export declare function autoSetupBunTestTracing(options: AutoSetupOptions): Promise<boolean>;
+//# sourceMappingURL=bun-harness.d.ts.map

package/dist/lib/testing/bun-harness.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bun-harness.d.ts","sourceRoot":"","sources":["../../../src/lib/testing/bun-harness.ts"],"names":[],"mappings":"AA4CA,OAAO,KAAK,OAAO,MAAM,UAAU,CAAC;AACpC,OAAO,EACL,QAAQ,IAAI,SAAS,EACrB,SAAS,IAAI,UAAU,EACvB,SAAS,IAAI,UAAU,EACvB,UAAU,IAAI,WAAW,EACzB,QAAQ,IAAI,SAAS,EACrB,MAAM,IAAI,OAAO,EACjB,EAAE,IAAI,GAAG,EAEV,MAAM,UAAU,CAAC;AAIlB,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kCAAkC,CAAC;AACpE,OAAO,KAAK,EAA4C,gBAAgB,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACrH,OAAO,EAAE,CAAC,EAAE,MAAM,sBAAsB,CAAC;AACzC,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAEvD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAErE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AA8B3C,KAAK,kBAAkB,CAAC,CAAC,SAAS,gBAAgB,IAAI,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;AAClF,KAAK,sBAAsB,GAAG;IAAE,QAAQ,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAA;CAAE,CAAC;AAC1E,KAAK,aAAa,CAAC,IAAI,SAAS,YAAY,IAAI,sBAAsB,GAAG,IAAI,CAAC;AAC9E,KAAK,WAAW,GAAG,OAAO,SAAS,CAAC;AACpC,KAAK,KAAK,GAAG,OAAO,GAAG,CAAC;AAOxB,MAAM,WAAW,kBAAkB;IACjC,QAAQ,EAAE,WAAW,CAAC;IACtB,EAAE,EAAE,KAAK,CAAC;IACV,IAAI,EAAE,KAAK,CAAC;IACZ,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AASD,KAAK,sBAAsB,CAAC,CAAC,SAAS,gBAAgB,EAAE,IAAI,SAAS,YAAY,IAC/E,CAAC,SAAS,gBAAgB,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,EAAE,MAAM,IAAI,EAAE,MAAM,OAAO,CAAC,GACpE,gBAAgB,CAAC,CAAC,SAAS,SAAS,CAAC,MAAM,MAAM,CAAC,GAAG,SAAS,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,GACzG,KAAK,CAAC;AAEZ,MAAM,MAAM,mBAAmB,CAAC,IAAI,SAAS,YAAY,GAAG,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI;IAClF;;;;OAIG;IACH,MAAM,CAAC,EAAE,kBAAkB,CAAC;IAC5B,uEAAuE;IACvE,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,0EAA0E;IAC1E,gBAAgB,CAAC,EAAE,IAAI,CAAC;CACzB,CAAC;AA2IF,MAAM,WAAW,qBAAqB,CAAC,CAAC,SAAS,gBAAgB;IAC/D,KAAK,IAAI,IAAI,CAAC;IACd,WAAW,IAAI,kBAAkB,CAAC,CAAC,CAAC,CAAC;IACrC,SAAS,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC;IACvB,iBAAiB,CAAC,OAAO,SAAS,kBAAkB,EAAE,aAAa,EAAE,OAAO,GAAG,OAAO,CAAC;CACxF;AAED,MAAM,WAAW,kBAAkB,CAAC,CAAC,SAAS,gBAAgB;IAC5D,aAAa,EAAE,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACxC,WAAW,IAAI,kBAAkB,CAAC,CAAC,CAAC,CAAC;IACrC,wBAAwB,IAAI,IAAI,CAAC;CAClC;AAED,MAAM,MAAM,yBAAyB,CACnC,CAAC,SAAS,gBAAgB,EAC1B,IAAI,SAAS,YAAY,GAAG,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAC9C,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC;AAExE,MAAM,MAAM,UAAU,CACpB,CAAC,SAAS,gBAAgB,EAC1B,IAAI,SAAS,YAAY,GAAG,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAC9C,yBAAyB,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;AAEvC,KAAK,mBAAmB,GAAG,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;AAInE,iGAAiG;AACjG,wBAAgB,oBAAoB,IAAI,mBAAmB,GAAG,IAAI,CAEjE;AAED,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,mBAAmB,GAAG,IAAI,CAIvE;AAED,wBAAgB,sBAAsB,CAAC,CAAC,SAAS,gBAAgB,EAC/D,OAAO,EAAE,CAAC,EACV,OAAO,CAAC,EAAE,IAAI,CAAC,mBAAmB,EAAE,kBAAkB,CAAC,GACtD,kBAAkB,CAAC,sBAAsB,CAAC,CAAC,EAAE,sBAAsB,CAAC,CAAC,CAAC;AACzE,wBAAgB,sBAAsB,CAAC,CAAC,SAAS,gBAAgB,EAAE,IAAI,SAAS,YAAY,GAAG,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,EACjH,OAAO,EAAE,CAAC,EACV,OAAO,CAAC,EAAE,mBAAmB,CAAC,IAAI,CAAC,GAClC,kBAAkB,CAAC,sBAAsB,CAAC,CAAC,EAAE,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAatE;;;;;;;;;;GAUG;AACH,wBAAgB,gBAAgB,CAAC,CAAC,SAAS,gBAAgB,EACzD,OAAO,EAAE,CAAC,EACV,OAAO,CAAC,EAAE;IAAE,gBAAgB,CAAC,EAAE,YAAY,CAAA;CAAE,GAC5C;IACD,WAAW,EAAE,MAAM,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;IAC/C,SAAS,EAAE,CAAC,CAAC;IACb,gBAAgB,EAAE,YAAY,GAAG,SAAS,CAAC;CAC5C,CAWA;AAED;;;;GAIG;AACH,wBAAgB,cAAc,CAAC,CAAC,SAAS,gBAAgB,EACvD,OAAO,EAAE,CAAC,EACV,OAAO,CAAC,EAAE,IAAI,CAAC,mBAAmB,EAAE,kBAAkB,CAAC,GACtD,qBAAqB,CAAC,sBAAsB,CAAC,CAAC,EAAE,sBAAsB,CAAC,CAAC,CAAC;AAC5E,wBAAgB,cAAc,CAAC,CAAC,SAAS,gBAAgB,EAAE,IAAI,SAAS,YAAY,GAAG,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,EACzG,OAAO,EAAE,CAAC,EACV,OAAO,CAAC,EAAE,mBAAmB,CAAC,IAAI,CAAC,GAClC,qBAAqB,CAAC,sBAAsB,CAAC,CAAC,EAAE,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAkOzE,oFAAoF;AACpF,wBAAgB,gBAAgB,CAAC,CAAC,SAAS,gBAAgB,EAAE,SAAS,EAAE,CAAC,EAAE,OAAO,CAAC,EAAE,mBAAmB,GAAG,IAAI,CA2D9G;AAED;;;;;;;;;GASG;AACH,wBAAgB,iBAAiB,CAAC,OAAO,SAAS,kBAAkB,EAAE,aAAa,EAAE,OAAO,GAAG,OAAO,CAwErG;AAED;;;;;;GAMG;AACH,wBAAgB,WAAW,CAAC,IAAI,SAAS,WAAW,CAAC,WAAW,CAAC,gBAAgB,CAAC,CAAC,KAAK,IAAI,CAAC;AAW7F;;;;;;GAMG;AACH,wBAAgB,SAAS,IAAI,MAAM,CAAC,gBAAgB,CAAC,CAOpD;AAED;;;GAGG;AACH,wBAAgB,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,IAAI,QASpD;yBATe,QAAQ;;;;AA6ExB,4EAA4E;AAC5E,wBAAgB,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAerE;yBAfe,EAAE;;;;AAyBlB,OAAO,EACL,SAAS,IAAI,QAAQ,EACrB,UAAU,IAAI,SAAS,EACvB,UAAU,IAAI,SAAS,EACvB,WAAW,IAAI,UAAU,EACzB,OAAO,IAAI,MAAM,GAClB,CAAC;AAMF,MAAM,MAAM,gBAAgB,GAAG;IAC7B,sFAAsF;IACtF,WAAW,EAAE,MAAM,CAAC;CACrB,CAAC;AAEF;;;;;;;;;;;;;;;;;;;GAmBG;AACH,wBAAsB,uBAAuB,CAAC,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,OAAO,CAAC,CA8DzF"}