Spaces:

reversely
/

lifekit

Sleeping

Docker Space: web app and API over a shipped database, online retrieval off

377fc96 verified about 2 months ago

2.49 kB

// Opens library.sqlite in the browser through the official SQLite wasm build,
// which carries FTS5, so the browser runs the same search as the Python side.

import sqlite3InitModule, { type Database, type SqlValue } from '@sqlite.org/sqlite-wasm';
import { headers } from './api';

export type { Database, SqlValue };

let ready: Promise<Database> | null = null;
let generation = 0;

/** Drop the cached database and fetch the file again (after a sync or a verdict). */
export function reload(): Promise<Database> {
  ready = null;
  generation += 1;
  return database();
}

export function database(): Promise<Database> {
  if (!ready) {
    ready = (async () => {
      const sqlite3 = await sqlite3InitModule();
      const bytes = new Uint8Array(await (await fetch(`/library.sqlite?v=${generation}`, { cache: 'no-store', headers: headers() })).arrayBuffer());
      const pointer = sqlite3.wasm.allocFromTypedArray(bytes);
      const db = new sqlite3.oo1.DB();
      const rc = sqlite3.capi.sqlite3_deserialize(
        db.pointer!,
        'main',
        pointer,
        bytes.length,
        bytes.length,
        sqlite3.capi.SQLITE_DESERIALIZE_FREEONCLOSE,
      );
      if (rc !== 0) throw new Error(`sqlite3_deserialize failed with ${rc}`);
      return db;
    })();
  }
  return ready;
}

/** Run a query and return rows as objects keyed by column name. */
export function rows<T = Record<string, SqlValue>>(db: Database, sql: string, params: SqlValue[] = []): T[] {
  return db.selectObjects(sql, params as any) as T[];
}

/** Mirror a verdict the API applied into the in-memory copy, so the screens change
 *  at once instead of waiting for the file to download again. */
export function applyVerdict(db: Database, volumeId: string, admission: string, section: string | null): void {
  db.exec({ sql: 'UPDATE volume SET admission = ? WHERE id = ?', bind: [admission, volumeId] });
  if (section) {
    db.exec({ sql: 'DELETE FROM shelf_volume WHERE volume_id = ?', bind: [volumeId] });
    db.exec({ sql: "INSERT INTO shelf_volume VALUES (?, ?, (SELECT coalesce(max(\"order\"), 0) + 1 FROM shelf_volume WHERE shelf_id = ?), 'reference')", bind: [section, volumeId, section] });
    db.exec({ sql: "DELETE FROM volume_tag WHERE volume_id = ? AND tag LIKE 'section:%'", bind: [volumeId] });
    db.exec({ sql: "INSERT OR REPLACE INTO volume_tag VALUES (?, ?, 'hand', '')", bind: [volumeId, `section:${section}`] });
  }
  dispatchEvent(new CustomEvent('lifekit:changed'));
}

	// Opens library.sqlite in the browser through the official SQLite wasm build,
	// which carries FTS5, so the browser runs the same search as the Python side.

	import sqlite3InitModule, { type Database, type SqlValue } from '@sqlite.org/sqlite-wasm';
	import { headers } from './api';

	export type { Database, SqlValue };

	let ready: Promise<Database> \| null = null;
	let generation = 0;

	/** Drop the cached database and fetch the file again (after a sync or a verdict). */
	export function reload(): Promise<Database> {
	ready = null;
	generation += 1;
	return database();
	}

	export function database(): Promise<Database> {
	if (!ready) {
	ready = (async () => {
	const sqlite3 = await sqlite3InitModule();
	const bytes = new Uint8Array(await (await fetch(`/library.sqlite?v=${generation}`, { cache: 'no-store', headers: headers() })).arrayBuffer());
	const pointer = sqlite3.wasm.allocFromTypedArray(bytes);
	const db = new sqlite3.oo1.DB();
	const rc = sqlite3.capi.sqlite3_deserialize(
	db.pointer!,
	'main',
	pointer,
	bytes.length,
	bytes.length,
	sqlite3.capi.SQLITE_DESERIALIZE_FREEONCLOSE,
	);
	if (rc !== 0) throw new Error(`sqlite3_deserialize failed with ${rc}`);
	return db;
	})();
	}
	return ready;
	}

	/** Run a query and return rows as objects keyed by column name. */
	export function rows<T = Record<string, SqlValue>>(db: Database, sql: string, params: SqlValue[] = []): T[] {
	return db.selectObjects(sql, params as any) as T[];
	}

	/** Mirror a verdict the API applied into the in-memory copy, so the screens change
	* at once instead of waiting for the file to download again. */
	export function applyVerdict(db: Database, volumeId: string, admission: string, section: string \| null): void {
	db.exec({ sql: 'UPDATE volume SET admission = ? WHERE id = ?', bind: [admission, volumeId] });
	if (section) {
	db.exec({ sql: 'DELETE FROM shelf_volume WHERE volume_id = ?', bind: [volumeId] });
	db.exec({ sql: "INSERT INTO shelf_volume VALUES (?, ?, (SELECT coalesce(max(\"order\"), 0) + 1 FROM shelf_volume WHERE shelf_id = ?), 'reference')", bind: [section, volumeId, section] });
	db.exec({ sql: "DELETE FROM volume_tag WHERE volume_id = ? AND tag LIKE 'section:%'", bind: [volumeId] });
	db.exec({ sql: "INSERT OR REPLACE INTO volume_tag VALUES (?, ?, 'hand', '')", bind: [volumeId, `section:${section}`] });
	}
	dispatchEvent(new CustomEvent('lifekit:changed'));
	}