Daggerbot-TS/src/commands/mp.ts

import Discord from 'discord.js';
import TClient from '../client.js';
import path from 'node:path';
import canvas from 'canvas';
import {readFileSync} from 'node:fs';
import {FSData, TServer} from 'src/typings/interfaces.js';

const serverChoices = [
  {name: 'Main Server', value: 'mainServer'},
  {name: 'Second Server', value: 'secondServer'}
]

export default {
  async run(client: TClient, interaction: Discord.ChatInputCommandInteraction<'cached'>){
    if (client.uptime < 30000) return interaction.reply('I have just restarted, please wait for MPLoop to finish initializing.')
    const serverSelector = interaction.options.getString('server', true);
    if (interaction.channelId === '468835769092669461' && !client.isStaff(interaction.member) && ['status', 'players'].includes(interaction.options.getSubcommand())) return interaction.reply('Please use <#739084625862852715> for `/mp status/players` commands to prevent clutter in this channel.').then(()=>setTimeout(()=>interaction.deleteReply(), 6000));

    const database = await client.MPServer._content.findById(interaction.guildId);
    const endpoint = await fetch(database[serverSelector].ip+'/feed/dedicated-server-stats.json?code='+database[serverSelector].code, {signal: AbortSignal.timeout(7500),headers:{'User-Agent':`Daggerbot - MPdata/fetch`}}).then(r=>r.json() as Promise<FSData>);
    const embed = new client.embed();
    ({
      players: async()=>{
        const data = JSON.parse(readFileSync(path.join(`src/database/${client.MPServerCache[serverSelector].name}PlayerData.json`), {encoding: 'utf8'})).slice(client.statsGraph);
        // handle negative days
        for (const [i, change] of data.entries()) if (change < 0) data[i] = data[i - 1] || data[i + 1] || 0;

        const first_graph_top = 16;
        const second_graph_top = 16;
        const textSize = 40;
        const img = canvas.createCanvas(1500, 750);
        const ctx = img.getContext('2d');
        const graphOrigin = [15, 65];
        const graphSize = [1300, 630];
        const nodeWidth = graphSize[0] / (data.length - 1);
        ctx.fillStyle = '#36393f';
        ctx.fillRect(0, 0, img.width, img.height);

        // grey horizontal lines
        ctx.lineWidth = 5;

        const interval_candidates: [number, number, number][] = [];
        for (let i = 4; i < 10; i++) {
          const interval = first_graph_top / i;
          if (Number.isInteger(interval)) {
            let intervalString = interval.toString();
            const reference_number = i * Math.max(intervalString.split('').filter(x => x === '0').length / intervalString.length, 0.3) * (['1', '2', '4', '5', '6', '8'].includes(intervalString[0]) ? 1.5 : 0.67)
            interval_candidates.push([interval, i, reference_number]);
          }
        }
        const chosen_interval = interval_candidates.sort((a, b) => b[2] - a[2])[0];
        const previousY: number[] = [];
        ctx.strokeStyle = '#202225';

        for (let i = 0; i <= chosen_interval[1]; i++) {
          const y = graphOrigin[1] + graphSize[1] - (i * (chosen_interval[0] / second_graph_top) * graphSize[1]);
          if (y < graphOrigin[1]) continue;
          const even = ((i + 1) % 2) === 0;
          if (even) ctx.strokeStyle = '#2c2f33';
          ctx.beginPath();
          ctx.lineTo(graphOrigin[0], y);
          ctx.lineTo(graphOrigin[0] + graphSize[0], y);
          ctx.stroke();
          ctx.closePath();
          if (even) ctx.strokeStyle = '#202225';
          previousY.push(y, i * chosen_interval[0]);
        }

        // 30d mark
        ctx.setLineDash([8, 16]);
        ctx.beginPath();
        const lastMonthStart = graphOrigin[0] + (nodeWidth * (data.length - 60));
        ctx.lineTo(lastMonthStart, graphOrigin[1]);
        ctx.lineTo(lastMonthStart, graphOrigin[1] + graphSize[1]);
        ctx.stroke();
        ctx.closePath();
        ctx.setLineDash([]);

        // draw points
        ctx.lineWidth = 5;

        const gradient = ctx.createLinearGradient(0, graphOrigin[1], 0, graphOrigin[1] + graphSize[1]);
        gradient.addColorStop(1 / 16, '#e62c3b'); // Red
        gradient.addColorStop(5 / 16, '#ffea00'); // Yellow
        gradient.addColorStop(12 / 16, '#57f287'); // Green

        let lastCoords: number[] = [];

        for (let [i, curPC /* current player count */] of data.entries()) {
          if (curPC < 0) curPC = 0;
          const x = i * nodeWidth + graphOrigin[0];
          const y = ((1 - (curPC / second_graph_top)) * graphSize[1]) + graphOrigin[1];
          const nexPC /* next player count */ = data[i + 1];
          const prvPC /* previous player count */ = data[i - 1];
          ctx.strokeStyle = gradient;
          ctx.beginPath();
          if (lastCoords.length) ctx.moveTo(lastCoords[0], lastCoords[1]);
          // if the line being drawn is horizontal, make it go until it has to go down
          if (y === lastCoords[1]) {
            let newX = x;
            for (let j = i + 1; j <= data.length; j++) {
              if (data[j] === curPC) newX += nodeWidth;
              else break;
            }
            ctx.lineTo(newX, y);
          } else ctx.lineTo(x, y);
            lastCoords = [x, y];
            ctx.stroke();
            ctx.closePath();

          if (curPC !== prvPC || curPC !== nexPC) { // Ball if vertical different to next or prev point
            // ball
            ctx.fillStyle = gradient;
            ctx.beginPath();
            ctx.arc(x, y, ctx.lineWidth * 1.3, 0, 2 * Math.PI)
            ctx.closePath();
            ctx.fill();
          };
        }

        // draw text
        ctx.font = '400 ' + textSize + 'px sans-serif';
        ctx.fillStyle = 'white';

        // highest value
        if (!isNaN(previousY.at(-2) as number)) {
          const maxx = graphOrigin[0] + graphSize[0] + textSize / 2;
          const maxy = (previousY.at(-2) as number) + (textSize / 3);
          ctx.fillText((previousY.at(-1) as number).toLocaleString('en-US'), maxx, maxy);
        }

        // lowest value
        const lowx = graphOrigin[0] + graphSize[0] + textSize / 2;
        const lowy = graphOrigin[1] + graphSize[1] + (textSize / 3);
        ctx.fillText('0 players', lowx, lowy);

        // 30d
        ctx.fillText('30 min ago', lastMonthStart, graphOrigin[1] - (textSize / 2));

        // time ->
        const tx = graphOrigin[0] + (textSize / 2);
        const ty = graphOrigin[1] + graphSize[1] + (textSize);
        ctx.fillText('time ->', tx, ty);

        const playerData: string[] = [];
        let Color = client.config.embedColor;
        if (endpoint.slots.used === endpoint.slots.capacity) Color = client.config.embedColorRed;
        else if (endpoint.slots.used > 8) Color = client.config.embedColorYellow;
        else Color = client.config.embedColorGreen;

        for (const player of endpoint.slots.players.filter(x=>x.isUsed)){
          let decorator = player.isAdmin ? ':detective:' : '';
          decorator += player.name.includes('Toast') ? '<:toastv2:1132681026662056079>' : '';
          playerData.push(`**${player.name}${decorator}**\nFarming for ${client.formatPlayerUptime(player.uptime)}`)
        }

        const slot = `${endpoint.slots.used}/${endpoint.slots.capacity}`;
        const ingameTime = `${('0'+Math.floor((endpoint.server.dayTime/3600/1000))).slice(-2)}:${('0'+Math.floor((endpoint.server.dayTime/60/1000)%60)).slice(-2)}`;
        interaction.reply({embeds:[new client.embed().setColor(Color).setTitle(endpoint.server.name.length > 0 ? endpoint.server.name : 'Offline').setDescription(endpoint.slots.used < 1 ? '*No players online*' : playerData.join('\n\n')).setImage('attachment://FSStats.png').setAuthor({name:slot}).setFooter({text: 'Current time: '+ingameTime})], files: [new client.attachmentBuilder(img.toBuffer(),{name:'FSStats.png'})]})
      },
      status: async()=>{
        if (!endpoint) return console.log('Endpoint failed - status');
        try {
          if (endpoint.server.name.length > 1){
            interaction.reply({embeds: [embed.setTitle('Status/Details').setColor(client.config.embedColor).addFields(
              {name: 'Server name', value: `${endpoint?.server.name.length === 0 ? '\u200b' : `\`${endpoint?.server.name}\``}`, inline: true},
              {name: 'Players', value: `${endpoint.slots.used} out of ${endpoint.slots.capacity}`, inline: true},
              {name: 'Current map', value: `${endpoint?.server.mapName.length === 0 ? '\u200b' : endpoint.server.mapName}`, inline: true},
              {name: 'Version', value: `${endpoint?.server.version.length === 0 ? '\u200b' : endpoint.server.version}`, inline: true},
              {name: 'In-game Time', value: `${('0' + Math.floor((endpoint.server.dayTime/3600/1000))).slice(-2)}:${('0' + Math.floor((endpoint.server.dayTime/60/1000)%60)).slice(-2)}`, inline: true}
            )]})
          } else if (endpoint.server.name.length === 0) interaction.reply('Server is currently offline.')
        } catch (err){
          console.log(err)
          interaction.reply('Ah, you caught a rare one... Please notify <@&'+client.config.mainServer.roles.bottech+'>')
        }
      },
      info: async()=>{
        if (!endpoint) return console.log('Endpoint failed - info')
        if (endpoint.server.name.length < 1) embed.setFooter({text: 'Server is currently offline.'})
        interaction.reply({embeds: [embed.setColor(client.config.embedColor).setDescription([
          `**Server name**: \`${endpoint?.server.name.length === 0 ? '\u200b' : endpoint.server.name}\``,
          '**Password:** `mf4700`',
          '**Crossplay server**',
          `**Map:** ${endpoint.server.mapName.length < 1 ? 'Null Island' : endpoint.server.mapName}`,
          `**Mods:** [Click here](${database[serverSelector].ip}/mods.html) **|** [Direct Download](${database[serverSelector].ip}/all_mods_download?onlyActive=true)`,
          '**Filters:** [Click here](https://discord.com/channels/468835415093411861/468835769092669461/926581585938120724)',
          'Please see <#543494084363288637> for additional information.'
        ].join('\n'))]});
      },
      url: async()=>{
        if (client.config.mainServer.id == interaction.guildId) {
          if (!interaction.member.roles.cache.has(client.config.mainServer.roles.mpmanager) && !interaction.member.roles.cache.has(client.config.mainServer.roles.bottech) && !interaction.member.roles.cache.has(client.config.mainServer.roles.admin)) return client.youNeedRole(interaction, 'mpmanager');
        }
        const address = interaction.options.getString('address');
        if (!address){
          try {
            const Url = await client.MPServer._content.findById(interaction.guildId);
            if (interaction.options.getString('server').includes('mainServer')) {
              if (Url.mainServer.ip && Url.mainServer.code) return interaction.reply(Url.mainServer.ip+'/feed/dedicated-server-stats.json?code='+Url.mainServer.code)
            } else {
              if (Url.secondServer.ip && Url.secondServer.code) return interaction.reply(Url.secondServer.ip+'/feed/dedicated-server-stats.json?code='+Url.secondServer.code)
            }
          } catch(err){
            console.log(`MPDB :: ${err}`);
            interaction.reply(`\`\`\`${err}\`\`\``)
          }
        } else {
          if (!address.match(/dedicated-server-stats/)) return interaction.reply('The URL does not match `dedicated-server-stats.xml`');
          const newURL = address.replace('xml','json').split('/feed/dedicated-server-stats.json?code=');
          try {
            if (interaction.options.getString('server').includes('mainServer')) {
              console.log(`MPDB :: Main Server\'s URL for ${interaction.guild.name} has been modified by ${interaction.member.displayName} (${interaction.member.id})`);
              await client.MPServer._content.create({_id: interaction.guildId, mainServer: { ip: newURL[0], code: newURL[1] }, secondServer: { ip: 'unknown', code: 'unknown' }})
              .then(()=>interaction.reply('This guild is not found in database, therefore I have created it for you.'))
              .catch((err:Error)=>interaction.reply(`I ran into a brick wall while creating the server data:\n${err.message}`))
            } else {
              console.log(`MPDB :: Second Server\'s URL for ${interaction.guild.name} has been modified by ${interaction.member.displayName} (${interaction.member.id})`)
              await client.MPServer._content.findOneAndUpdate({_id: interaction.guildId},{$set: {secondServer: {ip: newURL[0], code: newURL[1]}}})
              .then(()=>interaction.reply('URL for second server successfully updated.'))
              .catch((err:Error)=>interaction.reply(`I got hit by a flying fish while updating the server data:\n${err.message}`))
            }
          } catch(err) {
            if (interaction.options.getString('server').includes('mainServer')) {
              const affected = await client.MPServer._content.findByIdAndUpdate({_id: interaction.guildId}, {$set: {mainServer:{ip: newURL[0], code: newURL[1]}}})
              if (affected) return interaction.reply('URL for Main Server successfully updated.')
            } else {
              const affected = await client.MPServer._content.findByIdAndUpdate({_id: interaction.guildId}, {$set: {secondServer:{ip: newURL[0], code: newURL[1]}}})
              if (affected) return interaction.reply('URL for Second Server successfully updated.')
            }
          }
        }
      },
      maintenance: ()=>{
        if (client.config.mainServer.id == interaction.guildId) {
          if (!interaction.member.roles.cache.has(client.config.mainServer.roles.mpmanager) && !interaction.member.roles.cache.has(client.config.mainServer.roles.bottech) && !interaction.member.roles.cache.has(client.config.mainServer.roles.admin)) return client.youNeedRole(interaction, 'mpmanager');
        }
        const maintenanceMessage = interaction.options.getString('message');
        const activePlayersChannel = '739084625862852715';
        const channel = (client.channels.cache.get(activePlayersChannel) as Discord.TextChannel);
        if (channel.permissionOverwrites.cache.get(interaction.guildId).deny.has('SendMessages')) {
          channel.permissionOverwrites.edit(interaction.guildId, {SendMessages: true}, {type: 0, reason: `Unlocked by ${interaction.member.displayName}`});
          channel.send({embeds: [new client.embed().setColor(client.config.embedColor).setAuthor({name: interaction.member.displayName, iconURL: interaction.member.displayAvatarURL({size:1024})}).setTitle('🔓 Channel unlocked').setDescription(`**Reason:**\n${maintenanceMessage}`).setTimestamp()]});
          interaction.reply({content: `<#${activePlayersChannel}> has been unlocked!`, ephemeral: true});
        } else if (channel.permissionOverwrites.cache.get(interaction.guildId).allow.has('SendMessages')) {
          channel.permissionOverwrites.edit(interaction.guildId, {SendMessages: false}, {type: 0, reason: `Locked by ${interaction.member.displayName}`});
          channel.send({embeds: [new client.embed().setColor(client.config.embedColor).setAuthor({name: interaction.member.displayName, iconURL: interaction.member.displayAvatarURL({size:1024})}).setTitle('🔒 Channel locked').setDescription(`**Reason:**\n${maintenanceMessage}`).setTimestamp()]});
          interaction.reply({content: `<#${activePlayersChannel}> has been locked!`, ephemeral: true});
        }
      }
    })[interaction.options.getSubcommand()]();
  },
  data: new Discord.SlashCommandBuilder()
  .setName('mp')
  .setDescription('Display MP status and other things')
  .addSubcommand(x=>x
    .setName('status')
    .setDescription('Display server status')
    .addStringOption(x=>x
      .setName('server')
      .setDescription('The server to update')
      .setRequired(true)
      .setChoices(serverChoices[0])))
  .addSubcommand(x=>x
    .setName('players')
    .setDescription('Display players on server')
    .addStringOption(x=>x
      .setName('server')
      .setDescription('The server to display players for')
      .setRequired(true)
      .setChoices(serverChoices[0])))
  /* .addSubcommand(x=>x
    .setName('url')
    .setDescription('View or update the server URL')
    .addStringOption(x=>x
      .setName('server')
      .setDescription('The server to update')
      .setRequired(true)
      .setChoices(
        {name: 'Main Server', value: 'mainServer'},
        {name: 'Second Server', value: 'secondServer'}
      ))
    .addStringOption(x=>x
      .setName('address')
      .setDescription('The URL to the dedicated-server-stats.json file')
      .setRequired(false))) */
  .addSubcommand(x=>x
    .setName('info')
    .setDescription('Display server information')
    .addStringOption(x=>x
      .setName('server')
      .setDescription('The server to display information for')
      .setRequired(true)
      .setChoices(serverChoices[0])))
  .addSubcommand(x=>x
    .setName('maintenance')
    .setDescription('Toggle maintenance mode for #mp-active-players')
    .addStringOption(x=>x
      .setName('message')
      .setDescription('The message to display in the channel')))
}