mirror of
https://github.com/arthur-pbty/selfbot-discord.git
synced 2026-06-22 18:19:40 +02:00
update
This commit is contained in:
@@ -0,0 +1,18 @@
|
||||
import sqlite3 from 'sqlite3';
|
||||
require('dotenv').config();
|
||||
|
||||
module.exports = function addBaseInDB(client: any) {
|
||||
const dbName = process.env.DB_NAME || 'db.db';
|
||||
let db = new sqlite3.Database(dbName, sqlite3.OPEN_READWRITE | sqlite3.OPEN_CREATE, (err) => {
|
||||
if (err) {
|
||||
console.error(err.message);
|
||||
}
|
||||
});
|
||||
|
||||
const prefix = process.env.DEFAULT_PREFIX || '!!';
|
||||
db.run('INSERT OR IGNORE INTO prefix(prefix) VALUES(?)', [prefix], (err) => {
|
||||
if (err) {
|
||||
console.error(err.message);
|
||||
}
|
||||
});
|
||||
}
|
||||
@@ -0,0 +1,31 @@
|
||||
import sqlite3 from 'sqlite3';
|
||||
require('dotenv').config();
|
||||
|
||||
function getPrefix() {
|
||||
return new Promise((resolve, reject) => {
|
||||
const dbName = process.env.DB_NAME || 'db.db';
|
||||
|
||||
let db = new sqlite3.Database(dbName, sqlite3.OPEN_READWRITE | sqlite3.OPEN_CREATE, (err: Error | null) => {
|
||||
if (err) {
|
||||
console.error(err.message);
|
||||
}
|
||||
});
|
||||
|
||||
db.get('SELECT prefix FROM prefix', (err: Error | null, row: any) => {
|
||||
if (err) {
|
||||
console.error(err.message);
|
||||
reject(err);
|
||||
}
|
||||
const prefix = row ? row.prefix : process.env.DEFAULT_PREFIX;
|
||||
resolve(prefix);
|
||||
});
|
||||
|
||||
db.close((err) => {
|
||||
if (err) {
|
||||
console.error(err.message);
|
||||
}
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
module.exports = getPrefix;
|
||||
@@ -0,0 +1,19 @@
|
||||
const sqlite3 = require('sqlite3').verbose();
|
||||
require('dotenv').config();
|
||||
|
||||
module.exports = function initDB() {
|
||||
let db = new sqlite3.Database(process.env.DB_NAME, sqlite3.OPEN_READWRITE | sqlite3.OPEN_CREATE, (err: Error) => {
|
||||
if (err) {
|
||||
console.error(err.message);
|
||||
}
|
||||
});
|
||||
|
||||
|
||||
db.run(`CREATE TABLE IF NOT EXISTS prefix(
|
||||
prefix TEXT NOT NULL DEFAULT '${process.env.DEFAULT_PREFIX}'
|
||||
)`, (err: Error) => {
|
||||
if (err) {
|
||||
console.error(err.message);
|
||||
}
|
||||
});
|
||||
}
|
||||
Reference in New Issue
Block a user