From b8d537df76ae8e5db8997e7ba90057d77f5b3360 Mon Sep 17 00:00:00 2001
From: David Malchin <malchin459@gmail.com>
Date: Sat, 19 Oct 2024 02:47:23 +0300
Subject: [PATCH] Add `playerlist` command

---
 package-lock.json                             |  13 ++
 package.json                                  |   1 +
 server/components/loader.ts                   |   2 +
 server/components/standalone/index.ts         |   8 ++
 .../standalone/playerlist/action-row.ts       | 114 ++++++++++++++++++
 .../standalone/playerlist/command.ts          |  27 +++++
 .../components/standalone/playerlist/embed.ts |  49 ++++++++
 7 files changed, 214 insertions(+)
 create mode 100644 server/components/standalone/index.ts
 create mode 100644 server/components/standalone/playerlist/action-row.ts
 create mode 100644 server/components/standalone/playerlist/command.ts
 create mode 100644 server/components/standalone/playerlist/embed.ts

diff --git a/package-lock.json b/package-lock.json
index 56d8238..56d6976 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -8,6 +8,7 @@
       "dependencies": {
         "@discordjs/collection": "1.5.2",
         "@discordjs/core": "1.0.0",
+        "@discordjs/formatters": "0.3.1",
         "@discordjs/rest": "2.0.0",
         "@discordjs/util": "1.0.0",
         "@discordjs/ws": "1.0.0"
@@ -55,6 +56,18 @@
         "node": ">=16.9.0"
       }
     },
+    "node_modules/@discordjs/formatters": {
+      "version": "0.3.1",
+      "resolved": "https://registry.npmjs.org/@discordjs/formatters/-/formatters-0.3.1.tgz",
+      "integrity": "sha512-M7X4IGiSeh4znwcRGcs+49B5tBkNDn4k5bmhxJDAUhRxRHTiFAOTVUNQ6yAKySu5jZTnCbSvTYHW3w0rAzV1MA==",
+      "license": "Apache-2.0",
+      "dependencies": {
+        "discord-api-types": "^0.37.41"
+      },
+      "engines": {
+        "node": ">=16.9.0"
+      }
+    },
     "node_modules/@discordjs/rest": {
       "version": "2.0.0",
       "resolved": "https://registry.npmjs.org/@discordjs/rest/-/rest-2.0.0.tgz",
diff --git a/package.json b/package.json
index f1089ee..4d717f9 100644
--- a/package.json
+++ b/package.json
@@ -12,6 +12,7 @@
   "dependencies": {
     "@discordjs/collection": "1.5.2",
     "@discordjs/core": "1.0.0",
+    "@discordjs/formatters": "0.3.1",
     "@discordjs/rest": "2.0.0",
     "@discordjs/util": "1.0.0",
     "@discordjs/ws": "1.0.0"
diff --git a/server/components/loader.ts b/server/components/loader.ts
index 7c5193f..d27cef2 100644
--- a/server/components/loader.ts
+++ b/server/components/loader.ts
@@ -9,6 +9,7 @@ import { client, gateway, rest, zdiscordBridge } from '../utils/env.js';
 import { isStatefulInteraction } from '../utils/stateful.js';
 import core from './core/index.js';
 import ping from './ping/index.js';
+import standalone from './standalone/index.js';
 import {
     ApplicationCommand,
     Component,
@@ -86,6 +87,7 @@ function loadComponent({
 export function loadComponents() {
     loadComponent(core);
     loadComponent(ping);
+    loadComponent(standalone);
     if (zdiscordBridge) {
         loadComponent(zdiscord);
         console.log('zdiscord bridge enabled');
diff --git a/server/components/standalone/index.ts b/server/components/standalone/index.ts
new file mode 100644
index 0000000..093834a
--- /dev/null
+++ b/server/components/standalone/index.ts
@@ -0,0 +1,8 @@
+import { Component } from '../types.js';
+import { playerlistButtonBase } from './playerlist/action-row.js';
+import { playerlistCommand } from './playerlist/command.js';
+
+export default {
+    commands: [playerlistCommand],
+    messageComponents: [playerlistButtonBase],
+} satisfies Component;
diff --git a/server/components/standalone/playerlist/action-row.ts b/server/components/standalone/playerlist/action-row.ts
new file mode 100644
index 0000000..af39363
--- /dev/null
+++ b/server/components/standalone/playerlist/action-row.ts
@@ -0,0 +1,114 @@
+import {
+    APIActionRowComponent,
+    APIButtonComponentWithCustomId,
+    APIMessageActionRowComponent,
+    ButtonStyle,
+    ComponentType,
+} from '@discordjs/core';
+import { createStatefulInteraction } from '../../../utils/stateful.js';
+import { Button } from '../../types.js';
+import { playerlistEmbed, playerlistPageCount } from './embed.js';
+
+export enum PlayerlistButtonType {
+    First,
+    Previous,
+    Reload,
+    Next,
+    Last,
+}
+
+const typeData: Record<
+    PlayerlistButtonType,
+    {
+        emoji: string;
+        state: (page: number) => string;
+    }
+> = {
+    [PlayerlistButtonType.First]: {
+        emoji: '⏪',
+        state: () => 'first',
+    },
+    [PlayerlistButtonType.Previous]: {
+        emoji: '⬅️',
+        state: (page) => `${page - 1}`,
+    },
+    [PlayerlistButtonType.Reload]: {
+        emoji: '🔃',
+        state: (page) => `${page}`,
+    },
+    [PlayerlistButtonType.Next]: {
+        emoji: '➡️',
+        state: (page) => `${page + 1}`,
+    },
+    [PlayerlistButtonType.Last]: {
+        emoji: '⏩',
+        state: () => 'last',
+    },
+};
+
+export const playerlistButtonBase = createStatefulInteraction<Button>({
+    data: {
+        type: ComponentType.Button,
+        style: ButtonStyle.Secondary,
+        custom_id: 'playerlist_button',
+    },
+    async execute({ data: interaction, api, state }) {
+        const pageCount = playerlistPageCount();
+        let page = Math.min(
+            {
+                first: 1,
+                last: pageCount,
+            }[state] ?? parseInt(state),
+            pageCount,
+        );
+
+        await api.interactions.updateMessage(
+            interaction.id,
+            interaction.token,
+            {
+                embeds: [playerlistEmbed(page)],
+                components: [playerlistActionRow(page)],
+            },
+        );
+    },
+});
+
+function playerlistButton(
+    type: PlayerlistButtonType,
+    page: number = 1,
+): APIButtonComponentWithCustomId {
+    const data = typeData[type];
+    const buttonData = playerlistButtonBase.stateful(data.state(page));
+    buttonData.emoji = { name: data.emoji };
+
+    if (
+        (page === 1 &&
+            [
+                PlayerlistButtonType.First,
+                PlayerlistButtonType.Previous,
+            ].includes(type)) ||
+        (page === playerlistPageCount() &&
+            [PlayerlistButtonType.Next, PlayerlistButtonType.Last].includes(
+                type,
+            ))
+    ) {
+        buttonData.disabled = true;
+    }
+
+    return buttonData;
+}
+
+export function playerlistActionRow(
+    page: number = 1,
+): APIActionRowComponent<APIMessageActionRowComponent> {
+    return {
+        type: ComponentType.ActionRow,
+        components: [
+            playerlistButton(PlayerlistButtonType.First, page),
+            playerlistButton(PlayerlistButtonType.Previous, page),
+            playerlistButton(PlayerlistButtonType.Reload, page),
+            playerlistButton(PlayerlistButtonType.Next, page),
+            playerlistButton(PlayerlistButtonType.Last, page),
+        ],
+    };
+}
diff --git a/server/components/standalone/playerlist/command.ts b/server/components/standalone/playerlist/command.ts
new file mode 100644
index 0000000..a4249e8
--- /dev/null
+++ b/server/components/standalone/playerlist/command.ts
@@ -0,0 +1,27 @@
+import {
+    ApplicationCommandType,
+    ApplicationIntegrationType,
+    InteractionContextType,
+    MessageFlags,
+} from '@discordjs/core';
+import { ChatInputCommand } from '../../types.js';
+import { playerlistActionRow } from './action-row.js';
+import { playerlistEmbed } from './embed.js';
+
+export const playerlistCommand = {
+    data: {
+        type: ApplicationCommandType.ChatInput,
+        name: 'playerlist',
+        description: 'Player list',
+        default_member_permissions: '0',
+        integration_types: [ApplicationIntegrationType.GuildInstall],
+        contexts: [InteractionContextType.Guild],
+    },
+    async execute({ data: interaction, api }) {
+        await api.interactions.reply(interaction.id, interaction.token, {
+            embeds: [playerlistEmbed()],
+            components: [playerlistActionRow()],
+            flags: MessageFlags.Ephemeral,
+        });
+    },
+} satisfies ChatInputCommand;
diff --git a/server/components/standalone/playerlist/embed.ts b/server/components/standalone/playerlist/embed.ts
new file mode 100644
index 0000000..8174f14
--- /dev/null
+++ b/server/components/standalone/playerlist/embed.ts
@@ -0,0 +1,49 @@
+import { APIEmbed } from '@discordjs/core';
+import { escapeMarkdown } from '@discordjs/formatters';
+
+const NAME_MAX_LENGTH = 32 + 5 + 5; // max steam name length + max id length + extra chars
+const PLAYERS_PER_PAGE = Math.floor(4096 / NAME_MAX_LENGTH) || 1; // max embed length
+export function playerlistPageCount() {
+    return Math.ceil(GetNumPlayerIndices() / PLAYERS_PER_PAGE) || 1;
+}
+
+export function playerlistEmbed(page: number = 1) {
+    const playerCount = GetNumPlayerIndices();
+    const maxPlayers = GetConvarInt('sv_maxClients', 32);
+    const embed: APIEmbed = {
+        title: `Online Players (${playerCount}/${maxPlayers})`,
+        footer: {
+            text: `Page ${page}/${playerlistPageCount()}`,
+        },
+        timestamp: new Date().toISOString(),
+    };
+
+    if (playerCount === 0) {
+        embed.description = 'There are no players online.';
+        return embed;
+    }
+
+    let startingIndex;
+    do {
+        startingIndex = PLAYERS_PER_PAGE * --page;
+    } while (startingIndex > playerCount);
+
+    const lines: string[] = [];
+    for (
+        let i = startingIndex;
+        i < playerCount && i < startingIndex + PLAYERS_PER_PAGE;
+        i++
+    ) {
+        const playerId = GetPlayerFromIndex(i);
+        const playerName = GetPlayerName(playerId);
+        lines.push(
+            `\`${playerId}\` - ${escapeMarkdown(playerName)}`.substring(
+                0,
+                NAME_MAX_LENGTH + 1,
+            ),
+        );
+    }
+
+    embed.description = lines.join('\n');
+    return embed;
+}