31 lines
1 KiB
TypeScript
31 lines
1 KiB
TypeScript
import { Database } from "@db/sqlite";
|
|
|
|
const conn = new Database("./data/video.db");
|
|
conn.exec(`pragma journal_mode = WAL;`);
|
|
|
|
conn.exec(`
|
|
CREATE TABLE IF NOT EXISTS allowlist (
|
|
repo TEXT NOT NULL UNIQUE PRIMARY KEY -- did
|
|
) STRICT;
|
|
`);
|
|
|
|
conn.exec(`
|
|
CREATE TABLE IF NOT EXISTS videos (
|
|
id INTEGER NOT NULL PRIMARY KEY,
|
|
repo TEXT NOT NULL, -- did
|
|
cid TEXT NOT NULL, -- blob cid
|
|
fetched_at INTEGER NOT NULL DEFAULT (unixepoch()), -- datetime
|
|
filename TEXT NOT NULL
|
|
) STRICT;
|
|
`);
|
|
|
|
const selectAllowlist = conn.prepare("SELECT 1 FROM allowlist WHERE repo = ?");
|
|
const selectVideo = conn.prepare("SELECT filename FROM videos WHERE repo = ? AND cid = ?");
|
|
const insertVideo = conn.prepare("INSERT INTO videos (repo, cid, filename) VALUES (?, ?, ?)");
|
|
|
|
export const db = {
|
|
db: conn,
|
|
inAllowlist: (did: string) => (selectAllowlist.value<[boolean]>(did)?.[0] && true) ?? false,
|
|
getVideo: (did: string, cid: string) => selectVideo.value<[string]>(did, cid)?.[0],
|
|
addVideo: (did: string, cid: string, filename: string) => insertVideo.run(did, cid, filename),
|
|
};
|