cloud/utils/ratelimits.js

import { Database } from "bun:sqlite";
const ratelimitsDB = new Database("./data/ratelimits.sqlite");

export default (type, value, limit, seconds) => {
    ratelimitsDB.run(`create table if not exists ${type} (
        subject text not null primary key,
        countLeft integer default ${limit},
        timestamp datetime default current_timestamp
    );`)

    const statement = ratelimitsDB.prepare(`
        INSERT INTO ${type} (subject)
        VALUES ($value)
        ON CONFLICT(subject) DO UPDATE SET
            countLeft = CASE 
                WHEN unixepoch('now') > unixepoch(timestamp) + $seconds THEN $limit
                ELSE MAX(0, countLeft - 1)
            END,
            timestamp = CASE 
                WHEN unixepoch('now') > unixepoch(timestamp) + $seconds THEN CURRENT_TIMESTAMP
                ELSE timestamp
            END
        RETURNING countLeft, timestamp, (unixepoch(timestamp) + $seconds - unixepoch('now')) AS duration;
    `);

    const result = statement.get({
        $value: value,
        $seconds: seconds,
        $limit: limit
    });
    
    if (result.countLeft <= 0) {
        return { status: true, ...result };
    }

    return { status: false, ...result };
}