Quote-Generator/db.js

const Database = require('better-sqlite3');
const path = require('path');
const fs = require('fs');
const crypto = require('crypto');

const DATA_DIR = path.join(__dirname, 'data');
const DB_PATH = path.join(DATA_DIR, 'sessions.db');

// make sure data dir exists
if (!fs.existsSync(DATA_DIR)) {
    fs.mkdirSync(DATA_DIR);
}

const db = new Database(DB_PATH);

// setup tables
db.exec(`
    CREATE TABLE IF NOT EXISTS sessions (
        id TEXT PRIMARY KEY,
        display_name TEXT DEFAULT 'Anonymous',
        username TEXT DEFAULT '@anonymous',
        text TEXT DEFAULT 'No text provided',
        avatar_url TEXT,
        image_url TEXT,
        timestamp INTEGER,
        verified INTEGER DEFAULT 0,
        engagement_likes INTEGER,
        engagement_retweets INTEGER,
        engagement_replies INTEGER,
        engagement_views INTEGER,
        created_at INTEGER NOT NULL,
        updated_at INTEGER NOT NULL
    );

    CREATE INDEX IF NOT EXISTS idx_sessions_updated_at ON sessions(updated_at);
`);

// migration: add verified column if it doesn't exist
try {
    db.exec(`ALTER TABLE sessions ADD COLUMN verified INTEGER DEFAULT 0`);
} catch (err) {
    // column already exists, ignore
}


function generateId() {
    return crypto.randomUUID();
}

function nowEpoch() {
    return Math.floor(Date.now() / 1000);
}

// convert db row to api resposne format
function rowToSession(row) {
    if (!row) return null;

    let engagement = null;
    if (row.engagement_likes !== null &&
        row.engagement_retweets !== null &&
        row.engagement_replies !== null &&
        row.engagement_views !== null) {
        engagement = {
            likes: row.engagement_likes,
            retweets: row.engagement_retweets,
            replies: row.engagement_replies,
            views: row.engagement_views
        };
    }

    return {
        id: row.id,
        displayName: row.display_name,
        username: row.username,
        text: row.text,
        avatarUrl: row.avatar_url,
        imageUrl: row.image_url,
        timestamp: row.timestamp,
        verified: Boolean(row.verified),
        engagement: engagement,
        createdAt: row.created_at,
        updatedAt: row.updated_at
    };
}

function createSession(data = {}) {
    const id = generateId();
    const now = nowEpoch();

    const stmt = db.prepare(`
        INSERT INTO sessions (
            id, display_name, username, text, avatar_url, image_url, timestamp, verified,
            engagement_likes, engagement_retweets, engagement_replies, engagement_views,
            created_at, updated_at
        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
    `);

    stmt.run(
        id,
        data.displayName || 'Anonymous',
        data.username || '@anonymous',
        data.text || 'No text provided',
        data.avatarUrl || null,
        data.imageUrl || null,
        data.timestamp || null,
        data.verified ? 1 : 0,
        data.engagement?.likes ?? null,
        data.engagement?.retweets ?? null,
        data.engagement?.replies ?? null,
        data.engagement?.views ?? null,
        now,
        now
    );

    return getSession(id);
}

function getSession(id) {
    const stmt = db.prepare('SELECT * FROM sessions WHERE id = ?');
    const row = stmt.get(id);
    return rowToSession(row);
}

function updateSession(id, data) {
    const session = getSession(id);
    if (!session) return null;

    const updates = [];
    const values = [];

    if (data.displayName !== undefined) {
        updates.push('display_name = ?');
        values.push(data.displayName);
    }
    if (data.username !== undefined) {
        updates.push('username = ?');
        values.push(data.username);
    }
    if (data.text !== undefined) {
        updates.push('text = ?');
        values.push(data.text);
    }
    if (data.avatarUrl !== undefined) {
        updates.push('avatar_url = ?');
        values.push(data.avatarUrl);
    }
    if (data.imageUrl !== undefined) {
        updates.push('image_url = ?');
        values.push(data.imageUrl);
    }
    if (data.timestamp !== undefined) {
        updates.push('timestamp = ?');
        values.push(data.timestamp);
    }

    // engagement updates (atomic clear or partial updates)
    if (data.engagement !== undefined) {
        if (data.engagement === null) {
            // explicitly clear all engagement fields
            updates.push('engagement_likes = ?');
            updates.push('engagement_retweets = ?');
            updates.push('engagement_replies = ?');
            updates.push('engagement_views = ?');
            values.push(null, null, null, null);
        } else {
            // partial engagement field updates
            if (data.engagement.likes !== undefined) {
                updates.push('engagement_likes = ?');
                values.push(data.engagement.likes);
            }
            if (data.engagement.retweets !== undefined) {
                updates.push('engagement_retweets = ?');
                values.push(data.engagement.retweets);
            }
            if (data.engagement.replies !== undefined) {
                updates.push('engagement_replies = ?');
                values.push(data.engagement.replies);
            }
            if (data.engagement.views !== undefined) {
                updates.push('engagement_views = ?');
                values.push(data.engagement.views);
            }
        }
    }

    if (updates.length === 0) {
        return session;
    }

    updates.push('updated_at = ?');
    values.push(nowEpoch());
    values.push(id);

    const stmt = db.prepare(`UPDATE sessions SET ${updates.join(', ')} WHERE id = ?`);
    stmt.run(...values);

    return getSession(id);
}

function deleteSession(id) {
    const stmt = db.prepare('DELETE FROM sessions WHERE id = ?');
    const result = stmt.run(id);
    return result.changes > 0;
}


function cleanupExpiredSessions() {
    const ONE_DAY = 24 * 60 * 60;
    const cutoff = nowEpoch() - ONE_DAY;

    const stmt = db.prepare('DELETE FROM sessions WHERE updated_at < ?');
    const result = stmt.run(cutoff);

    if (result.changes > 0) {
        console.log(`Cleaned up ${result.changes} expired session(s)`);
    }
}

module.exports = {
    createSession,
    getSession,
    updateSession,
    deleteSession,
    cleanupExpiredSessions
};