Spaces:
Running
Running
File size: 3,922 Bytes
5c2ed06 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 |
"use strict";
var __defProp = Object.defineProperty;
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
var __getOwnPropNames = Object.getOwnPropertyNames;
var __hasOwnProp = Object.prototype.hasOwnProperty;
var __export = (target, all) => {
for (var name in all)
__defProp(target, name, { get: all[name], enumerable: true });
};
var __copyProps = (to, from, except, desc) => {
if (from && typeof from === "object" || typeof from === "function") {
for (let key of __getOwnPropNames(from))
if (!__hasOwnProp.call(to, key) && key !== except)
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
}
return to;
};
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
var database_exports = {};
__export(database_exports, {
onDatabaseStart: () => onDatabaseStart,
statements: () => statements,
transactions: () => transactions
});
module.exports = __toCommonJS(database_exports);
var import_lib = require("../../lib");
var import__ = require(".");
const statements = {
send: "INSERT INTO offline_pms (sender, receiver, message, time) VALUES (?, ?, ?, ?)",
clear: "DELETE FROM offline_pms WHERE receiver = ?",
fetch: "SELECT * FROM offline_pms WHERE receiver = ?",
fetchNew: "SELECT * FROM offline_pms WHERE receiver = ? AND seen IS NULL",
clearDated: "DELETE FROM offline_pms WHERE ? - time >= ?",
checkSentCount: "SELECT count(*) as count FROM offline_pms WHERE sender = ? AND receiver = ?",
setSeen: "UPDATE offline_pms SET seen = ? WHERE receiver = ? AND seen IS NULL",
clearSeen: "DELETE FROM offline_pms WHERE ? - seen >= ?",
getSettings: "SELECT * FROM pm_settings WHERE userid = ?",
setBlock: "REPLACE INTO pm_settings (userid, view_only) VALUES (?, ?)",
deleteSettings: "DELETE FROM pm_settings WHERE userid = ?"
};
class StatementMap {
constructor(env) {
this.env = env;
}
run(name, args) {
return this.getStatement(name).run(args);
}
all(name, args) {
return this.getStatement(name).all(args);
}
get(name, args) {
return this.getStatement(name).get(args);
}
getStatement(name) {
const source = statements[name];
return this.env.statements.get(source);
}
}
const transactions = {
send: (args, env) => {
const statementList = new StatementMap(env);
const [sender, receiver, message] = args;
const count = statementList.get("checkSentCount", [sender, receiver])?.count;
if (count && count > import__.MAX_PENDING) {
return { error: `You have already sent the maximum ${import__.MAX_PENDING} offline PMs to that user.` };
}
return statementList.run("send", [sender, receiver, message, Date.now()]);
},
listNew: (args, env) => {
const list = new StatementMap(env);
const [receiver] = args;
const pms = list.all("fetchNew", [receiver]);
list.run("setSeen", [Date.now(), receiver]);
return pms;
}
};
function onDatabaseStart(database) {
let version;
try {
version = database.prepare("SELECT * FROM db_info").get().version;
} catch {
const schemaContent = (0, import_lib.FS)("databases/schemas/pms.sql").readSync();
database.exec(schemaContent);
}
const migrations = (0, import_lib.FS)("databases/migrations/pms").readdirIfExistsSync();
if (version !== migrations.length) {
for (const migration of migrations) {
const num = /(\d+)\.sql$/.exec(migration)?.[1];
if (!num || version >= num)
continue;
database.exec("BEGIN TRANSACTION");
try {
database.exec((0, import_lib.FS)(`databases/migrations/pms/${migration}`).readSync());
} catch (e) {
console.log(`Error in PM migration ${migration} - ${e.message}`);
console.log(e.stack);
database.exec("ROLLBACK");
continue;
}
database.exec("COMMIT");
}
}
}
//# sourceMappingURL=database.js.map
|