@observablehq/notebook-kit 1.1.0-rc.14 → 1.1.0-rc.16

package/dist/bin/notebooks.js

@@ -16,12 +16,18 @@ switch (command) {
         await run(process.argv.slice(3));
         break;
     }
+    case "query": {
+        const { default: run } = await import("./query.js");
+        await run(process.argv.slice(3));
+        break;
+    }
     default: {
         console.log(`usage: notebooks <command>
 
   preview      start the preview server
   build        generate a static site
   download     download an Observable Notebook as HTML
+  query        run a database query
   help         print usage information
   version      print the version
 `);

package/dist/bin/query.d.ts

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+export default function run(args?: string[]): Promise<void>;

package/dist/bin/query.js

@@ -0,0 +1,59 @@
+#!/usr/bin/env node
+import { parseArgs } from "node:util";
+import { getDatabase, getDatabaseConfig, getQueryCachePath } from "../src/databases/index.js";
+import { dirname } from "node:path/win32";
+import { mkdir, writeFile } from "node:fs/promises";
+import { join } from "node:path";
+if (process.argv[1] === import.meta.filename)
+    run();
+export default async function run(args) {
+    const { values, positionals } = parseArgs({
+        args,
+        allowPositionals: true,
+        allowNegative: true,
+        options: {
+            root: {
+                type: "string",
+                default: "."
+            },
+            database: {
+                type: "string"
+            },
+            help: {
+                type: "boolean",
+                short: "h"
+            }
+        }
+    });
+    if (values.help || !values.database) {
+        console.log(`usage: notebooks query <...query>
+
+  --database <name>        name of the database
+  --root <dir>             path to the root directory; defaults to cwd
+  -h, --help               show this message
+`);
+        return;
+    }
+    // Parse positionals into query template arguments.
+    const strings = [];
+    const params = [];
+    for (let i = 0; i < positionals.length; ++i) {
+        if (i & 1)
+            params.push(JSON.parse(positionals[i]));
+        else
+            strings.push(positionals[i]);
+    }
+    process.chdir(values.root);
+    const config = await getDatabaseConfig(".", values.database);
+    const database = await getDatabase(config);
+    const results = await database.call(null, strings, ...params);
+    const cachePath = await getQueryCachePath(".", values.database, strings, ...params);
+    await mkdir(dirname(cachePath), { recursive: true });
+    await writeFile(cachePath, JSON.stringify(results, replace));
+    console.log(join(values.root, cachePath));
+}
+// Force dates to be serialized as ISO 8601 UTC, undoing this:
+// https://github.com/snowflakedb/snowflake-connector-nodejs/blob/a9174fb7/lib/connection/result/sf_timestamp.js#L177-L179
+function replace(key, value) {
+    return this[key] instanceof Date ? Date.prototype.toJSON.call(this[key]) : value;
+}

package/dist/package.json

@@ -5,7 +5,7 @@
         "type": "git",
         "url": "git+https://github.com/observablehq/notebook-kit.git"
     },
-    "version": "1.1.0-rc.14",
+    "version": "1.1.0-rc.16",
     "type": "module",
     "scripts": {
         "test": "vitest",

package/dist/src/databases/duckdb.d.ts

@@ -1,2 +1,2 @@
-import type { DatabaseContext, DuckDBConfig, QueryTemplateFunction } from "./index.js";
-export default function duckdb({ path, options }: DuckDBConfig, context: DatabaseContext): QueryTemplateFunction;
+import type { DuckDBConfig, QueryTemplateFunction } from "./index.js";
+export default function duckdb({ path, options }: DuckDBConfig): QueryTemplateFunction;

package/dist/src/databases/duckdb.js

@@ -1,13 +1,9 @@
 import { DuckDBConnection, DuckDBInstance } from "@duckdb/node-api";
 import { BIGINT, BIT, BLOB, BOOLEAN, DATE, DOUBLE, FLOAT, HUGEINT, INTEGER, INTERVAL, SMALLINT, TIME, TIMESTAMP, TIMESTAMP_MS, TIMESTAMP_NS, TIMESTAMP_S, TIMESTAMPTZ, TINYINT, UBIGINT, UHUGEINT, UINTEGER, USMALLINT, UTINYINT, UUID, VARCHAR, VARINT } from "@duckdb/node-api"; // prettier-ignore
-import { join } from "node:path";
-export default function duckdb({ path, options }, context) {
-    if (path !== undefined)
-        path = join(context.cwd, path);
+export default function duckdb({ path, options }) {
     return async (strings, ...params) => {
         const instance = await DuckDBInstance.create(path, options);
         const connection = await DuckDBConnection.create(instance);
-        await connection.run(`SET file_search_path=$0`, [context.cwd]);
         const date = new Date();
         let result;
         let rows;

package/dist/src/databases/index.d.ts

@@ -31,9 +31,6 @@ export type PostgresConfig = {
     database?: string;
     ssl?: boolean;
 };
-export type DatabaseContext = {
-    cwd: string;
-};
 export type QueryTemplateFunction = (strings: readonly string[], ...params: QueryParam[]) => Promise<SerializableQueryResult>;
 export type SerializableQueryResult = {
     rows: Record<string, unknown>[];
@@ -42,5 +39,5 @@ export type SerializableQueryResult = {
     date: Date;
 };
 export declare function getDatabaseConfig(sourcePath: string, databaseName: string): Promise<DatabaseConfig>;
-export declare function getDatabase(config: DatabaseConfig, context: DatabaseContext): Promise<QueryTemplateFunction>;
+export declare function getDatabase(config: DatabaseConfig): Promise<QueryTemplateFunction>;
 export declare function getQueryCachePath(sourcePath: string, databaseName: string, strings: readonly string[], ...params: unknown[]): Promise<string>;

package/dist/src/databases/index.js

@@ -33,12 +33,12 @@ export async function getDatabaseConfig(sourcePath, databaseName) {
     }
     return config;
 }
-export async function getDatabase(config, context) {
+export async function getDatabase(config) {
     switch (config.type) {
         case "duckdb":
-            return (await import("./duckdb.js")).default(config, context);
+            return (await import("./duckdb.js")).default(config);
         case "sqlite":
-            return (await import(process.versions.bun ? "./sqlite-bun.js" : "./sqlite-node.js")).default(config, context);
+            return (await import(process.versions.bun ? "./sqlite-bun.js" : "./sqlite-node.js")).default(config);
         case "snowflake":
             return (await import("./snowflake.js")).default(config);
         case "postgres":

package/dist/src/databases/sqlite-bun.d.ts

@@ -1,2 +1,2 @@
-import type { DatabaseContext, SQLiteConfig, QueryTemplateFunction } from "./index.js";
-export default function sqlite({ path }: SQLiteConfig, context: DatabaseContext): QueryTemplateFunction;
+import type { SQLiteConfig, QueryTemplateFunction } from "./index.js";
+export default function sqlite({ path }: SQLiteConfig): QueryTemplateFunction;

package/dist/src/databases/sqlite-bun.js

@@ -1,9 +1,6 @@
-import { join } from "node:path";
 import { Database } from "bun:sqlite";
 import { getColumnType } from "./sqlite.js";
-export default function sqlite({ path = ":memory:" }, context) {
-    if (path !== undefined)
-        path = join(context.cwd, path);
+export default function sqlite({ path = ":memory:" }) {
     return async (strings, ...params) => {
         const date = new Date();
         const database = new Database(path);

package/dist/src/databases/sqlite-node.d.ts

@@ -1,2 +1,2 @@
-import type { DatabaseContext, SQLiteConfig, QueryTemplateFunction } from "./index.js";
-export default function sqlite({ path }: SQLiteConfig, context: DatabaseContext): QueryTemplateFunction;
+import type { SQLiteConfig, QueryTemplateFunction } from "./index.js";
+export default function sqlite({ path }: SQLiteConfig): QueryTemplateFunction;

package/dist/src/databases/sqlite-node.js

@@ -1,9 +1,6 @@
-import { join } from "node:path";
 import { DatabaseSync } from "node:sqlite";
 import { getColumnType } from "./sqlite.js";
-export default function sqlite({ path = ":memory:" }, context) {
-    if (path !== undefined)
-        path = join(context.cwd, path);
+export default function sqlite({ path = ":memory:" }) {
     return async (strings, ...params) => {
         const date = new Date();
         const database = new DatabaseSync(path);

package/dist/src/vite/observable.js

@@ -1,10 +1,11 @@
+import { fork } from "node:child_process";
 import { existsSync } from "node:fs";
-import { mkdir, readFile, writeFile } from "node:fs/promises";
+import { readFile } from "node:fs/promises";
 import { dirname, join, resolve } from "node:path";
 import { relative } from "node:path/posix";
 import { fileURLToPath } from "node:url";
 import { JSDOM } from "jsdom";
-import { getDatabase, getDatabaseConfig, getQueryCachePath } from "../databases/index.js";
+import { getQueryCachePath } from "../databases/index.js";
 import { deserialize } from "../lib/serialize.js";
 import { Sourcemap } from "../javascript/sourcemap.js";
 import { transpile } from "../javascript/transpile.js";
@@ -69,16 +70,12 @@ export function observable({ window = new JSDOM().window, parser = new window.DO
                             const dir = dirname(context.filename);
                             const cachePath = await getQueryCachePath(context.filename, cell.database, [value]);
                             if (!existsSync(cachePath)) {
-                                const config = await getDatabaseConfig(context.filename, cell.database);
-                                try {
-                                    const database = await getDatabase(config, { cwd: dir });
-                                    const results = await database.call(null, [value]);
-                                    await mkdir(dirname(cachePath), { recursive: true });
-                                    await writeFile(cachePath, JSON.stringify(results));
-                                }
-                                catch (error) {
-                                    console.error(error);
-                                }
+                                const args = ["--root", dir, "--database", cell.database, value];
+                                const child = fork(fileURLToPath(import.meta.resolve("../../bin/query.ts")), args);
+                                await new Promise((resolve, reject) => {
+                                    child.on("error", reject);
+                                    child.on("exit", resolve);
+                                });
                             }
                             cell.mode = "js";
                             cell.value = `FileAttachment(${JSON.stringify(relative(dir, cachePath))}).json().then(DatabaseClient.revive)${hidden ? "" : `.then(Inputs.table)${cell.output ? ".then(view)" : ""}`}`;

package/package.json

@@ -5,7 +5,7 @@
     "type": "git",
     "url": "git+https://github.com/observablehq/notebook-kit.git"
   },
-  "version": "1.1.0-rc.14",
+  "version": "1.1.0-rc.16",
   "type": "module",
   "scripts": {
     "test": "vitest",