const { TOKEN, OWNER, PREFIX, INVITE } = process.env;
const path = require('path');
const { FriendlyError } = require('discord.js-commando');
const CommandoClient = require('./structures/CommandoClient');
const client = new CommandoClient({
    commandPrefix: PREFIX,
    owner: OWNER,
    disableEveryone: true,
    invite: INVITE,
    unknownCommandResponse: false
});
const { RichEmbed } = require('discord.js');
const { carbon, dBots } = require('./structures/Stats');
const SequelizeProvider = require('./providers/Sequelize');

client.setProvider(new SequelizeProvider(client.database));

client.registry
    .registerDefaultTypes()
    .registerGroups([
        ['userinfo', 'User Info'],
        ['guildinfo', 'Server Info'],
        ['moderation', 'Moderation'],
        ['settings', 'Server Settings'],
        ['response', 'Random Response'],
        ['randomimg', 'Random Image'],
        ['avataredit', 'Avatar Manipulation'],
        ['textedit', 'Text Manipulation'],
        ['numedit', 'Number Manipulation'],
        ['search', 'Search'],
        ['games', 'Games'],
        ['random', 'Random/Other'],
        ['roleplay', 'Roleplay']
    ])
    .registerDefaultGroups()
    .registerDefaultCommands({ help: false })
    .registerCommandsIn(path.join(__dirname, 'commands'));

client.on('ready', () => {
    console.log(`[Ready] Shard ${client.shard.id} Logged in!`);
    client.user.setGame(`x;help | Shard ${client.shard.id}`);
});

client.on('disconnect', (event) => {
    console.log(`[Disconnect] Shard ${client.shard.id} disconnected with Code ${event.code}.`);
    process.exit(0);
});

client.on('error', console.error);

client.on('warn', console.warn);

client.on('commandError', (command, err) => {
    if (err instanceof FriendlyError) return;
    console.error(command.name, err);
});

client.dispatcher.addInhibitor(msg => {
    if (msg.channel.type === 'dm') return false;
    const role = msg.guild.settings.get('singleRole');
    if (!role) return false;
    if (!msg.guild.roles.has(role)) return false;
    if (client.isOwner(msg.author)) return false;
    if (msg.member.hasPermission('ADMINISTRATOR')) return false;
    if (!msg.member.roles.has(role))
        return ['singleRole', msg.reply(`Only the ${msg.guild.roles.get(role).name} role may use commands.`)];
});

client.on('message', async (msg) => {
    if (msg.guild && msg.guild.settings.get('inviteGuard') && /(discord(\.gg\/|app\.com\/invite\/|\.me\/))/gi.test(msg.content)) {
        if (msg.author.bot ||
            msg.member.hasPermission('ADMINISTRATOR') ||
            msg.author.id === msg.guild.ownerID ||
            msg.member.roles.has(msg.guild.settings.get('staffRole')) ||
            !msg.channel.permissionsFor(client.user).has('SEND_MESSAGES')) return;
        if (msg.channel.permissionsFor(client.user).has('MANAGE_MESSAGES')) msg.delete();
        else msg.channel.send('Message could not be deleted, missing the `Manage Messages` permission.');
        return msg.reply('Invites are prohibited from being posted here.');
    } else return;
});

client.on('messageReactionAdd', (reaction, user) => {
    if (reaction.emoji.name !== '⭐') return;
    if (reaction.count > 1) return;
    if (user.bot) return;
    const msg = reaction.message;
    const channel = msg.guild.channels.get(msg.guild.settings.get('starboard'));
    if (!channel) return;
    if (!channel.permissionsFor(client.user).has(['SEND_MESSAGES', 'EMBED_LINKS'])) return;
    if (user.id === msg.author.id) {
        if (msg.channel.permissionsFor(client.user).has('MANAGE_MESSAGES'))
            reaction.remove(user);
        return msg.reply('You cannot star your own messages, idiot.');
    }
    const embed = new RichEmbed()
        .setColor(0xFFFF00)
        .setAuthor(msg.author.tag, msg.author.displayAvatarURL)
        .setDescription(msg.content)
        .setImage(msg.attachments.first() ? msg.attachments.first().url : null)
        .setTimestamp();
    return channel.send({ embed });
});

client.on('guildMemberAdd', (member) => {
    const role = member.guild.roles.get(member.guild.settings.get('joinRole'));
    if (member.guild.me.hasPermission('MANAGE_ROLES') && role) 
        member.addRole(role).catch(() => null);
    const channel = member.guild.channels.get(member.guild.settings.get('memberLog'));
    if (!channel) return;
    if (!channel.permissionsFor(client.user).has('SEND_MESSAGES')) return;
    const msg = member.guild.settings.get('joinMsg', 'Welcome <user>!')
        .replace(/(<user>)/gi, member.user.username)
        .replace(/(<server>)/gi, member.guild.name)
        .replace(/(<mention>)/gi, member);
    return channel.send(msg);
});

client.on('guildMemberRemove', (member) => {
    const channel = member.guild.channels.get(member.guild.settings.get('memberLog'));
    if (!channel) return;
    if (!channel.permissionsFor(client.user).has('SEND_MESSAGES')) return;
    const msg = member.guild.settings.get('leaveMsg', 'Bye <user>...')
        .replace(/(<user>)/gi, member.user.username)
        .replace(/(<server>)/gi, member.guild.name)
        .replace(/(<mention>)/gi, member);
    return channel.send(msg);
});

client.on('guildCreate', async (guild) => {
    console.log(`[Guild] I have joined ${guild.name}! (${guild.id})`);
    const guilds = await client.shard.fetchClientValues('guilds.size');
    const count = guilds.reduce((prev, val) => prev + val, 0);
    carbon(count);
    dBots(count, client.user.id);
});

client.on('guildDelete', async (guild) => {
    console.log(`[Guild] I have left ${guild.name}... (${guild.id})`);
    const guilds = await client.shard.fetchClientValues('guilds.size');
    const count = guilds.reduce((prev, val) => prev + val, 0);
    carbon(count);
    dBots(count, client.user.id);
});

client.setTimeout(() => {
    console.log(`[Restart] Shard ${client.shard.id} Restarted.`);
    process.exit(0);
}, 14400000);

client.login(TOKEN);

process.on('unhandledRejection', console.error);