137 lines
3.9 KiB
TypeScript
137 lines
3.9 KiB
TypeScript
import dotenv from 'dotenv';
|
|
dotenv.config();
|
|
import { Database } from 'sqlite'
|
|
|
|
import Discord from 'discord.js';
|
|
import _ from 'lodash';
|
|
import { openDb } from './components/db'
|
|
import logger from './logger';
|
|
|
|
const NOTIF_CHANNEL_ID: string = process.env.NOTIF_CHANNEL_ID || '.';
|
|
const BOT_TOKEN: string = process.env.BOT_TOKEN || '.';
|
|
let db : Database;
|
|
const BOT_PREFIX = '.';
|
|
const client = new Discord.Client();
|
|
|
|
const parseCommand = (message: Discord.Message): { command: string | null; args: string[] } => {
|
|
// extract arguments by splitting by spaces and grouping strings in quotes
|
|
// e.g. .ping 1 "2 3" => ['ping', '1', '2 3']
|
|
let args = message.content.slice(BOT_PREFIX.length).match(/[^\s"']+|"([^"]*)"|'([^']*)'/g);
|
|
args = _.map(args, (arg) => {
|
|
if (arg[0].match(/'|"/g) && arg[arg.length - 1].match(/'|"/g)) {
|
|
return arg.slice(1, arg.length - 1);
|
|
}
|
|
return arg;
|
|
});
|
|
// obtain the first argument after the prefix
|
|
const firstArg = args.shift();
|
|
if (!firstArg) return { command: null, args: [] };
|
|
const command = firstArg.toLowerCase();
|
|
return { command, args };
|
|
};
|
|
|
|
const handleCommand = async (message: Discord.Message, command: string, args: string[]) => {
|
|
// log command and its author info
|
|
logger.info({
|
|
event: 'command',
|
|
messageId: message.id,
|
|
author: message.author.id,
|
|
authorName: message.author.username,
|
|
channel: message.channel.id,
|
|
command,
|
|
args
|
|
});
|
|
|
|
switch (command) {
|
|
case 'ping':
|
|
await message.channel.send('pong');
|
|
|
|
//dev testing commands
|
|
case 'save':
|
|
if (args.length < 1) {
|
|
await message.channel.send('no args')
|
|
return
|
|
}
|
|
await db.run('INSERT INTO saved_data (msg_id,data)' +
|
|
'VALUES(?,?)', [message.id, args[0]]).then((stmt : any, lastID : any, changes : any) => {
|
|
consoel
|
|
})
|
|
break
|
|
case 'dump':
|
|
//TODO: make messages be embeds to test
|
|
let flag: boolean = true
|
|
let outEmbed = new Discord.MessageEmbed().setColor('#0099ff').setTitle('Database Dump').setURL('https://www.youtube.com/watch?v=dQw4w9WgXcQ').addField("test", "test", true)
|
|
db.each('SELECT * FROM saved_data', (err, rows) => {
|
|
if (err) {
|
|
flag = false;
|
|
return;
|
|
}
|
|
console.log(rows['msg_id'], rows['data'])
|
|
outEmbed = outEmbed.addField(rows['msg_id'], rows['data'], true)
|
|
console.log(outEmbed)
|
|
})
|
|
console.log(outEmbed)
|
|
if (flag) {
|
|
if (outEmbed.fields.values.length == 0) {
|
|
await message.channel.send("empty")
|
|
}
|
|
else {
|
|
await message.channel.send(outEmbed)
|
|
}
|
|
}
|
|
else {
|
|
await message.channel.send("error")
|
|
}
|
|
break;
|
|
case 'clear':
|
|
db.run('DELETE FROM saved_data', async (err : any) => {
|
|
if (err) {
|
|
await message.channel.send(err.message)
|
|
return
|
|
}
|
|
await message.channel.send("Cleared.")
|
|
|
|
})
|
|
}
|
|
};
|
|
|
|
const handleMessage = async (message: Discord.Message) => {
|
|
// ignore messages without bot prefix and messages from other bots
|
|
if (!message.content.startsWith(BOT_PREFIX) || message.author.bot) return;
|
|
// obtain command and args from the command message
|
|
const { command, args } = parseCommand(message);
|
|
if (!command) return;
|
|
|
|
try {
|
|
await handleCommand(message, command, args);
|
|
} catch (e) {
|
|
// log error
|
|
logger.error({
|
|
event: 'error',
|
|
messageId: message.id,
|
|
command: command,
|
|
args: args,
|
|
error: e
|
|
});
|
|
}
|
|
};
|
|
|
|
const startBot = async () => {
|
|
client.once('ready', async () => {
|
|
// log bot init event and send system notification
|
|
logger.info({
|
|
event: 'init'
|
|
});
|
|
const notif = (await client.channels.fetch(NOTIF_CHANNEL_ID)) as Discord.TextChannel;
|
|
notif.send('Codey is up!');
|
|
});
|
|
|
|
db = await openDb();
|
|
console.log(db)
|
|
client.on('message', handleMessage);
|
|
|
|
client.login(BOT_TOKEN);
|
|
};
|
|
|
|
startBot();
|