38 lines
1.6 KiB
TypeScript
38 lines
1.6 KiB
TypeScript
|
import { Message, TextChannel } from 'eris';
|
||
|
|
import { Client, Command } from '../class';
|
||
|
|
|
||
|
|
export default class Train extends Command {
|
||
|
|
constructor(client: Client) {
|
||
|
|
super(client);
|
||
|
|
this.name = 'train';
|
||
|
|
this.description = 'Trains a neural network.';
|
||
|
|
this.usage = `${this.client.config.prefix}train <channel> <message id> <0: good | 1: bad>`;
|
||
|
|
this.permissions = 1;
|
||
|
|
this.guildOnly = false;
|
||
|
|
this.enabled = true;
|
||
|
|
}
|
||
|
|
|
||
|
|
public async run(message: Message, args: string[]) {
|
||
|
|
try {
|
||
|
|
if (args?.length < 3) return this.client.commands.get('help').run(message, [this.name]);
|
||
|
|
if (args[2] !== '0' && args[2] !== '1') return this.error(message.channel, 'Result must be either 0 or 1.');
|
||
|
|
const channel = <TextChannel> this.client.util.resolveGuildChannel(args[0], this.mainGuild);
|
||
|
|
if (!channel) return this.error(message.channel, 'Channel could not be found.');
|
||
|
|
if (channel.type !== 0) return this.error(message.channel, 'Invalid channel type.');
|
||
|
|
let msg: Message;
|
||
|
|
try {
|
||
|
|
msg = await channel.getMessage(args[1]);
|
||
|
|
} catch {
|
||
|
|
return this.error(message.channel, 'Could not find message.');
|
||
|
|
}
|
||
|
|
if (!msg) return this.error(message.channel, 'Message could not be found.');
|
||
|
|
|
||
|
|
await this.client.db.NNTrainingData.updateOne({ name: 'tc' }, { $addToSet: { data: { input: this.client.util.encode(msg.content), output: { res: Number(args[2]) } } } });
|
||
|
|
await message.delete();
|
||
|
|
return this.success(message.channel, 'Neural Network trained successfully.');
|
||
|
|
} catch (err) {
|
||
|
|
return this.client.util.handleError(err, message, this);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|