feat: 初始化项目
This commit is contained in:
159
server/db.js
Normal file
159
server/db.js
Normal file
@@ -0,0 +1,159 @@
|
||||
/**
|
||||
* SQLite:统计、留言、后端配置(弹幕开关与位置)
|
||||
*/
|
||||
const Database = require('better-sqlite3');
|
||||
const path = require('path');
|
||||
|
||||
const DB_PATH = path.join(__dirname, 'data', 'pano.db');
|
||||
|
||||
function getDb() {
|
||||
const db = new Database(DB_PATH);
|
||||
db.pragma('journal_mode = WAL');
|
||||
return db;
|
||||
}
|
||||
|
||||
function initDb() {
|
||||
const fs = require('fs');
|
||||
const dir = path.join(__dirname, 'data');
|
||||
if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
|
||||
const db = getDb();
|
||||
db.exec(`
|
||||
CREATE TABLE IF NOT EXISTS stats (
|
||||
id INTEGER PRIMARY KEY CHECK (id = 1),
|
||||
view_count INTEGER NOT NULL DEFAULT 0,
|
||||
like_count INTEGER NOT NULL DEFAULT 0,
|
||||
share_count INTEGER NOT NULL DEFAULT 0,
|
||||
watching_now INTEGER NOT NULL DEFAULT 0
|
||||
);
|
||||
INSERT OR IGNORE INTO stats (id, view_count, like_count, share_count, watching_now) VALUES (1, 0, 0, 0, 0);
|
||||
|
||||
CREATE TABLE IF NOT EXISTS viewers (
|
||||
viewer_id TEXT PRIMARY KEY,
|
||||
updated_at INTEGER NOT NULL
|
||||
);
|
||||
|
||||
CREATE TABLE IF NOT EXISTS comments (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
content TEXT NOT NULL,
|
||||
nickname TEXT,
|
||||
created_at INTEGER NOT NULL
|
||||
);
|
||||
|
||||
CREATE TABLE IF NOT EXISTS settings (
|
||||
key TEXT PRIMARY KEY,
|
||||
value TEXT NOT NULL
|
||||
);
|
||||
INSERT OR IGNORE INTO settings (key, value) VALUES ('danmaku_enabled', '0');
|
||||
INSERT OR IGNORE INTO settings (key, value) VALUES ('danmaku_position', 'top');
|
||||
`);
|
||||
db.close();
|
||||
}
|
||||
|
||||
function getConfig() {
|
||||
const db = getDb();
|
||||
const rows = db.prepare('SELECT key, value FROM settings').all();
|
||||
db.close();
|
||||
const map = {};
|
||||
rows.forEach((r) => { map[r.key] = r.value; });
|
||||
return {
|
||||
danmakuEnabled: map.danmaku_enabled === '1',
|
||||
danmakuPosition: (map.danmaku_position || 'top').toLowerCase(),
|
||||
};
|
||||
}
|
||||
|
||||
function getStats() {
|
||||
const db = getDb();
|
||||
const row = db.prepare('SELECT view_count, like_count, share_count, watching_now FROM stats WHERE id = 1').get();
|
||||
const commentRow = db.prepare('SELECT COUNT(*) as n FROM comments').get();
|
||||
db.close();
|
||||
return {
|
||||
viewCount: row.view_count,
|
||||
commentCount: commentRow.n,
|
||||
likeCount: row.like_count,
|
||||
shareCount: row.share_count,
|
||||
watchingNow: row.watching_now,
|
||||
};
|
||||
}
|
||||
|
||||
function incView() {
|
||||
const db = getDb();
|
||||
db.prepare('UPDATE stats SET view_count = view_count + 1 WHERE id = 1').run();
|
||||
const out = getStats();
|
||||
db.close();
|
||||
return out;
|
||||
}
|
||||
|
||||
function incLike() {
|
||||
const db = getDb();
|
||||
db.prepare('UPDATE stats SET like_count = like_count + 1 WHERE id = 1').run();
|
||||
const row = db.prepare('SELECT like_count FROM stats WHERE id = 1').get();
|
||||
db.close();
|
||||
return { likeCount: row.like_count };
|
||||
}
|
||||
|
||||
function incShare() {
|
||||
const db = getDb();
|
||||
db.prepare('UPDATE stats SET share_count = share_count + 1 WHERE id = 1').run();
|
||||
const row = db.prepare('SELECT share_count FROM stats WHERE id = 1').get();
|
||||
db.close();
|
||||
return { shareCount: row.share_count };
|
||||
}
|
||||
|
||||
function joinViewer(viewerId) {
|
||||
const db = getDb();
|
||||
const now = Date.now();
|
||||
db.prepare('INSERT OR REPLACE INTO viewers (viewer_id, updated_at) VALUES (?, ?)').run(viewerId, now);
|
||||
db.prepare('DELETE FROM viewers WHERE updated_at < ?').run(now - 120000);
|
||||
const row = db.prepare('SELECT COUNT(*) as n FROM viewers').get();
|
||||
db.prepare('UPDATE stats SET watching_now = ? WHERE id = 1').run(row.n);
|
||||
db.close();
|
||||
return row.n;
|
||||
}
|
||||
|
||||
function leaveViewer(viewerId) {
|
||||
const db = getDb();
|
||||
db.prepare('DELETE FROM viewers WHERE viewer_id = ?').run(viewerId);
|
||||
const row = db.prepare('SELECT COUNT(*) as n FROM viewers').get();
|
||||
db.prepare('UPDATE stats SET watching_now = ? WHERE id = 1').run(row.n);
|
||||
db.close();
|
||||
return row.n;
|
||||
}
|
||||
|
||||
function getComments(limit = 100) {
|
||||
const db = getDb();
|
||||
const rows = db.prepare(
|
||||
'SELECT id, content, nickname, created_at FROM comments ORDER BY id DESC LIMIT ?'
|
||||
).all(limit);
|
||||
db.close();
|
||||
return rows.reverse().map((r) => ({
|
||||
id: r.id,
|
||||
content: r.content,
|
||||
nickname: r.nickname || '游客',
|
||||
createdAt: r.created_at,
|
||||
}));
|
||||
}
|
||||
|
||||
function addComment(content, nickname) {
|
||||
const db = getDb();
|
||||
const now = Date.now();
|
||||
const r = db.prepare('INSERT INTO comments (content, nickname, created_at) VALUES (?, ?, ?)').run(
|
||||
String(content).trim().slice(0, 200) || '(空)',
|
||||
nickname ? String(nickname).trim().slice(0, 32) : null,
|
||||
now
|
||||
);
|
||||
db.close();
|
||||
return { id: r.lastInsertRowid, content: content.trim().slice(0, 200), nickname: nickname || '游客', createdAt: now };
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
initDb,
|
||||
getConfig,
|
||||
getStats,
|
||||
incView,
|
||||
incLike,
|
||||
incShare,
|
||||
joinViewer,
|
||||
leaveViewer,
|
||||
getComments,
|
||||
addComment,
|
||||
};
|
||||
Reference in New Issue
Block a user