| import Database from 'better-sqlite3'; |
| import { config } from '../config'; |
| import { logger } from '../logger'; |
|
|
| const log = logger.child({ module: 'sqlite-db' }); |
|
|
| let db: Database.Database; |
|
|
| export function initSQLiteDB(): Database.Database { |
| if (db) { |
| return db; |
| } |
|
|
| const dbPath = config.sqliteUserStorePath; |
| if (!dbPath) { |
| log.error('SQLite user store DB path (SQLITE_USER_STORE_PATH) is not configured.'); |
| throw new Error('SQLite user store DB path is not configured.'); |
| } |
|
|
| log.info({ path: dbPath }, 'Initializing SQLite database for user store...'); |
| db = new Database(dbPath); |
|
|
| |
| db.pragma('journal_mode = WAL'); |
|
|
| |
| |
| |
| db.exec(` |
| CREATE TABLE IF NOT EXISTS users ( |
| token TEXT PRIMARY KEY, |
| ip TEXT, /* JSON string array */ |
| nickname TEXT, |
| type TEXT NOT NULL CHECK(type IN ('normal', 'special', 'temporary')), |
| promptCount INTEGER NOT NULL DEFAULT 0, |
| tokenCounts TEXT, /* JSON string object */ |
| tokenLimits TEXT, /* JSON string object */ |
| tokenRefresh TEXT, /* JSON string object */ |
| createdAt INTEGER NOT NULL, |
| lastUsedAt INTEGER, |
| disabledAt INTEGER, |
| disabledReason TEXT, |
| expiresAt INTEGER, |
| maxIps INTEGER, |
| adminNote TEXT, |
| meta TEXT /* JSON string object */ |
| ); |
| `); |
|
|
| log.info('SQLite database initialized and `users` table created/verified.'); |
| return db; |
| } |
|
|
| export function getDB(): Database.Database { |
| if (!db) { |
| |
| |
| log.warn('SQLite DB instance requested before initialization. Attempting to initialize now.'); |
| return initSQLiteDB(); |
| } |
| return db; |
| } |
|
|
