249 lines
12 KiB
JavaScript
249 lines
12 KiB
JavaScript
const MongoClient = require('mongodb').MongoClient;
|
|
const io = require('socket.io-client');
|
|
const bcrypt = require('bcrypt');
|
|
const Discord = require('discord.js');
|
|
const client = new Discord.Client();
|
|
const config = require('./config');
|
|
|
|
class Bot {
|
|
|
|
constructor(dev, token) {
|
|
this.dev = dev;
|
|
this.token = token;
|
|
this.connectMongo()
|
|
}
|
|
|
|
async connectMongo() {
|
|
this.db = await MongoClient.connect(config.mongoURI,{useUnifiedTopology:true});
|
|
this.dbo = this.db.db(config.dbo);
|
|
}
|
|
|
|
// Updates Prefix
|
|
async newPrefix(message, n) {
|
|
if (n[0]==undefined) return message.channel.send(`You must provide a **new prefix** ${message.author}!`);
|
|
if (n[0].length<1) return message.channel.send('Your new prefix must be \`1\` character!')
|
|
this.dbo.collection("prefixes").updateOne({"server.serverID":message.guild.id},{$set:{"server.prefix":n[0]}},function(err, res) {
|
|
if (err) throw err;
|
|
return message.channel.send(`The new prefix is now **\`${n}\`**`);
|
|
});
|
|
}
|
|
|
|
// Remote Login
|
|
async remoteLogin(message, args) {
|
|
if (args.length==0) return message.author.send(`You must provide a **email** and **password** ${message.author}!`);
|
|
if (!args[0]==null||!args[0]==undefined&&args[1]==null||args[1]==undefined) return message.author.send(`You must provide a **password** ${message.author}!`);
|
|
let user = await this.dbo.collection("users").findOne({"user.email":args[0]}).then(user => user);
|
|
if (user==null||user==undefined) return message.author.send(`Cannot find the email **${args[0]}** ${message.author}!`);
|
|
if (user.user.discord) {
|
|
if (user.user.discord.id==message.author.id) return message.author.send(`You are already logged in ${message.author}!`);
|
|
}
|
|
|
|
// Match Password
|
|
bcrypt.compare(args[1], user.user.password, (err, isMatch) => {
|
|
if (err) throw err;
|
|
if (isMatch) {
|
|
// Modify user to include user.user.discord
|
|
let discord = {
|
|
id: message.author.id,
|
|
username: message.author.username,
|
|
discriminator: message.author.discriminator
|
|
}
|
|
this.dbo.collection("users").updateOne({"user.email":args[0]},{$set:{"user.discord": discord}},function(err,res) {
|
|
if (err) throw err;
|
|
return message.author.send(`Logged in as **${user.user.username}** ${message.author}!`);
|
|
});
|
|
} else {
|
|
return message.author.send(`Incorrect password for **${args[0]}** ${message.author}!`);
|
|
}
|
|
});
|
|
}
|
|
|
|
async remoteLogout(message) {
|
|
let user = await this.dbo.collection("users").findOne({"user.discord.id":message.author.id}).then(user => user);
|
|
if (!user) return message.channel.send(`You cannot logout if your not logged in ${message.author}!`);
|
|
this.dbo.collection("users").updateOne({"user.discord.id":message.author.id},{$unset:{"user.discord":""}},function(err,res) {
|
|
if (err) throw err;
|
|
return message.channel.send(`Succesfully logged out ${message.author}!`);
|
|
});
|
|
}
|
|
|
|
async account(message) {
|
|
let user = await this.dbo.collection("users").findOne({"user.discord.id":message.author.id}).then(user => user);
|
|
if (user==null) return message.channel.send(`You are not logged in ${message.author}!`);
|
|
return message.author.send(`${message.author} Logged in as **${user.user.username}** | **${user.user.email}**`);
|
|
}
|
|
|
|
// Disabled for Production
|
|
|
|
// async nameSearch(message, args) {
|
|
// let user = await this.dbo.collection("users").findOne({"user.discord.id":message.author.id}).then(user => user);
|
|
// if (user.user.activeCommunity=='' || user.user.activeCommunity==null) {
|
|
// if (args.length==0) return message.channel.send(`You must provide a **First Name**, **Last Name** and **DOB**(mm/dd/yyyy) ${message.author}!`);
|
|
// if (args.length==1) return message.channel.send(`You're missing a **Last Name** and **DOB**(mm/dd/yyyy) ${message.author}!`);
|
|
// if (args.length==2) return message.channel.send(`You're missing a **DOB**(mm/dd/yyyy) ${message.author}!`);
|
|
// }
|
|
// if (args.length==0) return message.channel.send(`You must provide a **First Name** and **Last Name** ${message.author}!`);
|
|
// if (args.length==1) return message.channel.send(`You're missing a **Last Name** ${message.author}!`);
|
|
// return message.channel.send('Debug, valid inputs');
|
|
// }
|
|
|
|
async checkStatus(message, args) {
|
|
if (args.length==0) {
|
|
let user = await this.dbo.collection("users").findOne({"user.discord.id":message.author.id}).then(user => user);
|
|
if (user==null) return message.channel.send(`You are not logged in ${message.author}!`);
|
|
return message.channel.send(`${message.author}'s status: ${user.user.dispatchStatus} | Set by: ${user.user.dispatchStatusSetBy}`);
|
|
} else {
|
|
let user = await this.dbo.collection("users").findOne({"user.discord.id":message.author.id}).then(user => user); if (user==null) return message.channel.send(`Cannot find **${args[0]}** ${message.author}!`);
|
|
// This lame line of code to get username without ping on discord
|
|
const User = client.users.cache.get(args[0].replace('<@!', '').replace('>', ''));
|
|
return message.channel.send(`${message.author}, **${User.tag}'s** status: ${user.user.dispatchStatus} | Set by: ${user.user.dispatchStatusSetBy}`);
|
|
}
|
|
}
|
|
|
|
async updateStatus(message, args, prefix) {
|
|
let validStatus=['10-8','10-7','10-6','10-11','10-23','10-97','10-15','10-70','10-80'];
|
|
let user = await this.dbo.collection("users").findOne({"user.discord.id":message.author.id}).then(user => user);
|
|
if (user==null) return message.channel.send(`You are not logged in ${message.author}!`);
|
|
if (args.length==0) return message.channel.send(`You must provide a new status ${message.author} | To see a list of valid statuses, use command \`${prefix}validStatus\`.`);
|
|
if (!validStatus.includes(args[0])) return message.channel.send(`**${args[0]}** is a Invalid Status ${message.author}! To see a list of valid statuses, use command \`${prefix}validStatus\`.`);
|
|
let onDuty=null;
|
|
let updateDuty=false;
|
|
let status = args[0]
|
|
if (args[0]=='10-41') {
|
|
onDuty=true;
|
|
updateDuty=true;
|
|
status='Online';
|
|
}
|
|
if (args[0]=='10-42') {
|
|
onDuty=false;
|
|
updateDuty=true;
|
|
status='Offline';
|
|
}
|
|
req={
|
|
userID: user._id,
|
|
status: status,
|
|
setBy: 'Self',
|
|
onDuty: onDuty,
|
|
updateDuty: updateDuty
|
|
};
|
|
const socket = io.connect(config.socket);
|
|
socket.emit('bot_update_status', req);
|
|
socket.on('bot_updated_status', (res) => {
|
|
return message.channel.send(`Succesfully updated status to **${args[0]}** ${message.author}!`);
|
|
});
|
|
}
|
|
|
|
async getPrefix(message) {
|
|
let prefix;
|
|
if (message.channel.type!="dm") {
|
|
let guild = await this.dbo.collection("prefixes").findOne({"server.serverID":message.guild.id}).then(guild => guild);
|
|
|
|
// If not prefix, generate server default
|
|
if (!guild) {
|
|
let newGuild = {
|
|
server: {
|
|
serverID: message.guild.id,
|
|
prefix: config.defaultPrefix
|
|
}
|
|
}
|
|
this.dbo.collection("prefixes").insertOne(newGuild, function(err, res) {
|
|
if (err) throw err;
|
|
});
|
|
prefix = newGuild.server.serverID;
|
|
} else prefix = guild.server.prefix;
|
|
} else prefix = config.defaultPrefix;
|
|
return prefix
|
|
}
|
|
|
|
main() {
|
|
client.on('ready', () => {
|
|
console.log(`Logged in as ${client.user.tag}`);
|
|
});
|
|
|
|
// Basic Commands
|
|
client.on('message', (message) => {
|
|
this.getPrefix(message).then(prefix => {
|
|
if (!message.content.startsWith(prefix) || message.author.bot) return;
|
|
|
|
const args = message.content.slice(prefix.length).trim().split(' ');
|
|
const command = args.shift().toLowerCase();
|
|
|
|
// Valid Statuses Embed
|
|
let validStatus = new Discord.MessageEmbed()
|
|
.setColor('#0099ff')
|
|
.setTitle('Lines Police CAD')
|
|
.setURL('https://www.linespolice-cad.com/')
|
|
.setAuthor('LPS Website Support', 'https://raw.githubusercontent.com/Linesmerrill/police-cad/master/lines-police-server.png', 'https://discord.gg/jgUW656v2t')
|
|
.setDescription('Valid Statuses')
|
|
.addFields(
|
|
{ name: 'Statuses:', value: `
|
|
10-8 | On Duty
|
|
10-7 | Off Duty
|
|
10-6 | Busy
|
|
10-11 | Traffic Stop
|
|
10-23 | Arrive on Scene
|
|
10-97 | In Route
|
|
10-15 | Subject in Custody
|
|
10-70 | Foot Pursuit
|
|
10-80 | Vehicle Pursiut
|
|
`
|
|
}
|
|
)
|
|
|
|
// Help Embed
|
|
let help = new Discord.MessageEmbed()
|
|
.setColor('#0099ff')
|
|
.setTitle('**Commands:**')
|
|
.setURL('https://discord.gg/jgUW656v2t')
|
|
.setAuthor('LPS Website Support', 'https://raw.githubusercontent.com/Linesmerrill/police-cad/master/lines-police-server.png', 'https://discord.gg/jgUW656v2t')
|
|
.setDescription('Lines Police CAD Bot Commands')
|
|
.addFields(
|
|
{ name: `**${prefix}help**`, value: 'Displays this help page', inline: true },
|
|
{ name: `**${prefix}ping**`, value: 'Responds with Pong to check Bot responce', inline: true },
|
|
{ name: `**${prefix}setPrefix** <new prefix>`, value: 'Sets new prefix (Admin only command)', inline: true },
|
|
{ name: `**${prefix}login** <email> <password>`, value: 'Login to LPS account (DM only command)', inline: true },
|
|
{ name: `**${prefix}logout**`, value: 'Logs out of your current logged in account', inline: true },
|
|
{ name: `**${prefix}validStatus**`, value: 'Shows list of valid statuses to updade to', inline: true },
|
|
{ name: `**${prefix}checkStatus** <user>`, value: 'Leave user blank to check own status', inline: true },
|
|
{ name: `**${prefix}updateStatus** <status>`, value: 'Updates your status', inline: true },
|
|
{ name: `**${prefix}account**`, value: 'returns logged in account', inline: true },
|
|
{ name: `**${prefix}penalCodes**`, value: 'Provides Link to penal codes', inline: true } )
|
|
|
|
if (command == 'ping') message.channel.send('Pong!');
|
|
if (command == 'help') message.channel.send(help);
|
|
if (command == 'setprefix') {
|
|
if (message.channel.type=="dm") return message.author.send(`You cannot set a prefix in a dm ${message.author}!`);
|
|
if(!message.member.hasPermission(["ADMINISTRATOR","MANAGE_GUILD"])) return message.channel.send(`You don't have permission to used this command ${message.author}`);
|
|
this.newPrefix(message, args);
|
|
}
|
|
// Login
|
|
if (command == 'login') {
|
|
if (message.channel.type=="text") return message.channel.send(`You must direct message me to login ${message.author}!`);
|
|
this.remoteLogin(message, args);
|
|
}
|
|
if (command == 'logout') this.remoteLogout(message);
|
|
if (command == 'validstatus') return message.channel.send(validStatus);
|
|
if (command == 'checkstatus') this.checkStatus(message, args);
|
|
if (command == 'updatestatus') this.updateStatus(message, args, prefix);
|
|
if (command == 'account') this.account(message);
|
|
if (command == 'penalcodes') return message.channel.send('https://www.linespolice-cad.com/penal-code');
|
|
if (command == 'namedb') this.nameSearch(message, args);
|
|
|
|
// Disabled for dev
|
|
// if (command == 'platedb') this.plateSearch(message, args);
|
|
// if (command == 'firearmdb') this.weaponSearch(message, args);
|
|
// if (command == 'createbolo') this.createBolo(message, args);
|
|
// if (command == 'panic') this.enablePanic(message);
|
|
|
|
// Dev Commands (not visible in help)
|
|
if (command == 'version') return message.channel.send(`**LPS-BOT Version : ${this.dev}-${config.version}**`)
|
|
if (command == 'whatisthemeaningoflife') message.channel.send('42');
|
|
});
|
|
});
|
|
|
|
client.login(this.token);
|
|
}
|
|
}
|
|
|
|
module.exports = {Bot};
|