You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
103 lines
3.1 KiB
JavaScript
103 lines
3.1 KiB
JavaScript
const music = require('./music'),
|
|
utils = require('./utils'),
|
|
config = require('../config.json'),
|
|
sqliteAsync = require('./utils/sqliteAsync'),
|
|
logging = require('./utils/logging'),
|
|
fs = require('fs-extra'),
|
|
dataDir = config.dataPath || './data';
|
|
|
|
/**
|
|
* The Guild Handler handles guild settings and data.
|
|
* @type {GuildHandler}
|
|
*/
|
|
class GuildHandler {
|
|
|
|
constructor(guild) {
|
|
this.guild = guild;
|
|
this._logger = new logging.Logger(`${this.constructor.name}@${this.guild}`);
|
|
this.musicPlayer = new music.MusicPlayer(null);
|
|
this._logger.silly('Initialized Guild Handler');
|
|
this._votes = {};
|
|
}
|
|
|
|
/**
|
|
* Initializes the database
|
|
* @returns {Promise<void>}
|
|
*/
|
|
async initDatabase() {
|
|
this._logger.silly('Initializing Database');
|
|
await fs.ensureDir(dataDir + '/gdb');
|
|
this.db = new sqliteAsync.Database(`${dataDir}/gdb/${this.guild}.db`);
|
|
await this.db.init();
|
|
this._logger.debug(`Connected to the database for ${this.guild}`);
|
|
this._logger.debug('Creating Databases');
|
|
await this._createTables();
|
|
}
|
|
|
|
/**
|
|
* Destroys the guild handler
|
|
*/
|
|
destroy() {
|
|
this._logger.debug('Ending musicPlayer');
|
|
this.musicPlayer.stop();
|
|
this._logger.debug('Ending Database');
|
|
this.db.close();
|
|
}
|
|
|
|
/**
|
|
* Creates all tables needed in the Database.
|
|
* These are at the moment:
|
|
* messages - logs all messages send on the server
|
|
* playlists - save playlists to play them later
|
|
*/
|
|
async _createTables() {
|
|
await this.db.run(`${utils.sql.tableExistCreate} messages (
|
|
${utils.sql.pkIdSerial},
|
|
creation_timestamp DATETIME NOT NULL,
|
|
author VARCHAR(128) NOT NULL,
|
|
author_name VARCHAR(128),
|
|
content TEXT NOT NULL
|
|
)`);
|
|
this._logger.silly('Created Table messages');
|
|
await this.db.run(`${utils.sql.tableExistCreate} playlists (
|
|
${utils.sql.pkIdSerial},
|
|
name VARCHAR(32) UNIQUE NOT NULL,
|
|
url VARCHAR(255) NOT NULL
|
|
)`);
|
|
this._logger.silly('Created Table playlists');
|
|
await this.db.run(`${utils.sql.tableExistCreate} commands (
|
|
${utils.sql.pkIdSerial},
|
|
name VARCHAR(32) UNIQUE NOT NULL,
|
|
command VARCHAR(255) NOT NULL
|
|
)`);
|
|
this._logger.silly('Created Table commands');
|
|
}
|
|
|
|
/**
|
|
* Sets the vote counter for a command up and adds the user.
|
|
* @param command {String}
|
|
* @param user {String}
|
|
*/
|
|
updateCommandVote(command, user) {
|
|
if (!this._votes[command])
|
|
this._votes[command] = {count: 0, users: []};
|
|
if (!this._votes[command].users.includes(user)) {
|
|
this._votes[command].count++;
|
|
this._votes[command].users.push(user);
|
|
}
|
|
return this._votes[command];
|
|
}
|
|
|
|
/**
|
|
* Resets the vote counter and voted users for a command.
|
|
* @param command {String}
|
|
*/
|
|
resetCommandVote(command) {
|
|
this._votes[command] = {count: 0, users: []};
|
|
}
|
|
}
|
|
|
|
Object.assign(exports, {
|
|
GuildHandler: GuildHandler
|
|
});
|