98 lines
3.3 KiB
TypeScript
98 lines
3.3 KiB
TypeScript
import Database from 'better-sqlite3';
|
|
import path from 'path';
|
|
import fs from 'fs';
|
|
|
|
const DATA_DIR = process.env.DATA_DIR ?? '/data';
|
|
const DB_DIR = path.join(DATA_DIR, 'db');
|
|
const DB_PATH = path.join(DB_DIR, 'memer.db');
|
|
|
|
fs.mkdirSync(DB_DIR, { recursive: true });
|
|
|
|
const db = new Database(DB_PATH);
|
|
|
|
db.pragma('journal_mode = WAL');
|
|
db.pragma('foreign_keys = ON');
|
|
|
|
// Core tables
|
|
db.exec(`
|
|
CREATE TABLE IF NOT EXISTS collections (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
name TEXT UNIQUE NOT NULL,
|
|
is_default INTEGER NOT NULL DEFAULT 0,
|
|
created_at TEXT NOT NULL DEFAULT (datetime('now'))
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS memes (
|
|
id TEXT PRIMARY KEY,
|
|
title TEXT NOT NULL,
|
|
description TEXT,
|
|
file_path TEXT NOT NULL,
|
|
file_name TEXT NOT NULL,
|
|
file_size INTEGER NOT NULL,
|
|
mime_type TEXT NOT NULL,
|
|
width INTEGER NOT NULL,
|
|
height INTEGER NOT NULL,
|
|
parent_id TEXT REFERENCES memes(id) ON DELETE CASCADE,
|
|
collection_id INTEGER REFERENCES collections(id) ON DELETE SET NULL,
|
|
ocr_text TEXT,
|
|
created_at TEXT NOT NULL DEFAULT (datetime('now'))
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS tags (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
name TEXT UNIQUE NOT NULL COLLATE NOCASE
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS meme_tags (
|
|
meme_id TEXT NOT NULL REFERENCES memes(id) ON DELETE CASCADE,
|
|
tag_id INTEGER NOT NULL REFERENCES tags(id) ON DELETE CASCADE,
|
|
PRIMARY KEY (meme_id, tag_id)
|
|
);
|
|
|
|
CREATE INDEX IF NOT EXISTS idx_memes_parent_id ON memes(parent_id);
|
|
CREATE INDEX IF NOT EXISTS idx_memes_created_at ON memes(created_at DESC);
|
|
CREATE INDEX IF NOT EXISTS idx_meme_tags_meme_id ON meme_tags(meme_id);
|
|
CREATE INDEX IF NOT EXISTS idx_meme_tags_tag_id ON meme_tags(tag_id);
|
|
`);
|
|
|
|
// Migrations — run after CREATE TABLE IF NOT EXISTS so they only apply to existing DBs
|
|
const memesCols = db.prepare('PRAGMA table_info(memes)').all() as { name: string }[];
|
|
|
|
if (!memesCols.find((c) => c.name === 'collection_id')) {
|
|
db.exec('ALTER TABLE memes ADD COLUMN collection_id INTEGER REFERENCES collections(id) ON DELETE SET NULL');
|
|
}
|
|
|
|
if (!memesCols.find((c) => c.name === 'ocr_text')) {
|
|
db.exec('ALTER TABLE memes ADD COLUMN ocr_text TEXT');
|
|
}
|
|
|
|
if (!memesCols.find((c) => c.name === 'share_count')) {
|
|
db.exec('ALTER TABLE memes ADD COLUMN share_count INTEGER NOT NULL DEFAULT 0');
|
|
}
|
|
|
|
// Indexes that depend on migrated columns — created after columns are guaranteed to exist
|
|
db.exec(`
|
|
CREATE INDEX IF NOT EXISTS idx_memes_collection_id ON memes(collection_id);
|
|
CREATE INDEX IF NOT EXISTS idx_memes_ocr ON memes(ocr_text) WHERE ocr_text IS NOT NULL;
|
|
`);
|
|
|
|
// Seed the default UNSORTED collection
|
|
const defaultCollection = db
|
|
.prepare('SELECT id FROM collections WHERE is_default = 1')
|
|
.get() as { id: number } | undefined;
|
|
|
|
if (!defaultCollection) {
|
|
db.prepare("INSERT INTO collections (name, is_default) VALUES ('Unsorted', 1)").run();
|
|
}
|
|
|
|
const unsorted = db
|
|
.prepare('SELECT id FROM collections WHERE is_default = 1')
|
|
.get() as { id: number };
|
|
|
|
// Assign any existing memes with no collection to UNSORTED
|
|
db.prepare('UPDATE memes SET collection_id = ? WHERE collection_id IS NULL').run(unsorted.id);
|
|
|
|
export const UNSORTED_ID = unsorted.id;
|
|
|
|
export default db;
|