drb-server/discordBot/modules/wrappers.mjs

import { DebugBuilder } from "../../modules/debugger.mjs";
const log = new DebugBuilder("server", "discordBot.modules.wrappers");
import {
  checkIfNodeIsConnectedToVC,
  getNodeDiscordID,
  getNodeDiscordUsername,
  checkIfNodeHasOpenDiscordClient,
  getNodeCurrentListeningSystem,
  requestNodeJoinSystem,
} from "../../modules/socketServerWrappers.mjs";
import { getAllDiscordIDs } from "../../modules/mongo-wrappers/mongoDiscordIDWrappers.mjs";
import { ActionRowBuilder, ButtonBuilder, ButtonStyle } from "discord.js";

export const checkOnlineBotsInGuild = async (nodeIo, guildId) => {
  let onlineBots = [];
  const openSockets = [...(await nodeIo.allSockets())];
  await Promise.all(
    openSockets.map(async (openSocket) => {
      openSocket = await nodeIo.sockets.sockets.get(openSocket);
      const connected = await checkIfNodeIsConnectedToVC(
        nodeIo,
        guildId,
        openSocket.node.nuid,
      );
      log.INFO("Connected:", connected);
      if (connected) {
        const username = await getNodeDiscordUsername(openSocket, guildId);
        const discordID = await getNodeDiscordID(openSocket);
        onlineBots.push({
          name: username,
          discord_id: discordID,
          nuid: openSocket.node.nuid,
        });
      }
    }),
  );

  return onlineBots;
};

export const getAvailableTokensInGuild = async (nodeIo, guildId) => {
  try {
    // Execute both asynchronous functions concurrently
    const [discordIDs, onlineBots] = await Promise.all([
      getAllDiscordIDs(), // Fetch all Discord IDs
      checkOnlineBotsInGuild(nodeIo, guildId), // Check online bots in the guild
    ]);

    // Use the results of both promises here
    log.INFO("Available Discord IDs:", discordIDs);
    log.INFO("Online bots in the guild:", onlineBots);

    // Filter any discordIDs that are not active
    const availableDiscordIDs = discordIDs
      .filter((discordID) => discordID.active == true)
      .filter(
        (discordID) =>
          !onlineBots.some(
            (bot) => Number(bot.discord_id) == discordID.discord_id,
          ),
      );

    // Return the unavailable discordIDs
    return availableDiscordIDs;
  } catch (error) {
    console.error("Error getting available tokens in guild:", error);
    throw error;
  }
};

/**
 * Get the nodes with given system that are available to be used within a given server
 * @param {any} nodeIo The nodeIO object contained in the discord server object
 * @param {any} guildId The guild ID to search in
 * @param {any} system The system to filter the nodes by
 * @returns {any}
 */
export const getAvailableNodes = async (nodeIo, guildId, system) => {
  // Get all open socket nodes
  const openSockets = [...(await nodeIo.allSockets())]; // TODO - Filter the returned nodes to only nodes that have the radio capability
  log.DEBUG("All open sockets: ", openSockets);

  var availableNodes = [];
  // Check each open socket to see if the node has the requested system
  await Promise.all(
    openSockets.map(async (openSocket) => {
      openSocket = await nodeIo.sockets.sockets.get(openSocket);
      // Check if the node has an existing open client (meaning the radio is already being listened to)
      const hasOpenClient = await checkIfNodeHasOpenDiscordClient(openSocket);
      if (hasOpenClient) {
        let currentSystem = await getNodeCurrentListeningSystem(openSocket);
        if (currentSystem != system.name) {
          log.INFO(
            "Node is listening to a different system than requested",
            openSocket.node.name,
          );
          return;
        }
      }

      // Check if the bot has an open voice connection in the requested server already
      const connected = await checkIfNodeIsConnectedToVC(
        nodeIo,
        guildId,
        openSocket.node.nuid,
      );
      log.INFO("Connected:", connected);
      if (!connected) {
        // Check if this node has the requested system, if so add it to the availble array
        if (system.nodes.includes(openSocket.node.nuid)) {
          availableNodes.push(openSocket);
        }
      }
    }),
  );

  log.DEBUG(
    "Availble nodes:",
    availableNodes.map((socket) => socket.node.name),
  );

  return availableNodes;
};

/**
 * Gets the voice channel the user is currently in.
 * @param {any} interaction - The interaction object.
 * @returns {any} - The voice channel object, or null if the user is not in a voice channel.
 */
export const getUserVoiceChannel = (interaction) => {
  if (!interaction.member.voice.channel) {
    interaction.editReply({
      content: `<@${interaction.member.id}>, you need to enter a voice channel before using this command`,
      ephemeral: true,
    });
    return null;
  }
  return interaction.member.voice.channel;
};

/**
 * Joins a node to a specified system and voice channel.
 * @param {any} nodeIo - The nodeIO server for manipulation of sockets.
 * @param {any} interaction - The interaction object.
 * @param {string} nodeId - The ID of the node to join.
 * @param {any} system - The system object to join.
 * @param {any} channel - The voice channel to join.
 */
export const joinNode = async (
  nodeIo,
  interaction,
  nodeId,
  system,
  channel,
) => {
  try {
    const openSocket = await nodeIo.sockets.sockets.get(nodeId);
    const discordTokens = await getAvailableTokensInGuild(
      nodeIo,
      interaction.guild.id,
    );

    if (discordTokens.length === 0) {
      await interaction.editReply({
        content: `<@${interaction.member.id}>, there are no free bots available.`,
        ephemeral: true,
      });
      return;
    }

    log.INFO(
      "Joining node:",
      nodeId,
      system.name,
      channel.id,
      openSocket.node.name,
      discordTokens[0].token,
    );
    await requestNodeJoinSystem(
      openSocket,
      system.name,
      channel.id,
      discordTokens[0].token,
    );

    await interaction.editReply({
      content: `<@${interaction.member.id}>, a bot will join your channel listening to '${system.name}' shortly.`,
      ephemeral: true,
    });
  } catch (err) {
    log.ERROR("Failed to join node:", err);
    await interaction.editReply({
      content: `<@${interaction.member.id}>, an error occurred while joining the node: ${err.message}`,
      ephemeral: true,
    });
  }
};

/**
 * Prompts the user to select a node from available nodes.
 * @param {any} interaction - The interaction object.
 * @param {Array} availableNodes - The list of available nodes.
 * @param {Function} onNodeSelected - Callback function to handle the selected node.
 */
export const promptNodeSelection = async (
  interaction,
  availableNodes,
  onNodeSelected,
) => {
  const nodeSelectionButtons = availableNodes.map((node) =>
    new ButtonBuilder()
      .setCustomId(node.id)
      .setLabel(node.node.name)
      .setStyle(ButtonStyle.Primary),
  );

  const actionRow = new ActionRowBuilder().addComponents(nodeSelectionButtons);

  const response = await interaction.editReply({
    content: `<@${interaction.member.id}>, please select the Node you would like to join with this system:`,
    components: [actionRow],
    ephemeral: true,
  });

  const collectorFilter = (i) => i.user.id === interaction.user.id;
  try {
    const selectedNode = await response.awaitMessageComponent({
      filter: collectorFilter,
      time: 60_000,
    });
    await onNodeSelected(selectedNode.customId);
  } catch (e) {
    log.ERROR("Node selection timeout:", e);
    await interaction.editReply({
      content: "Confirmation not received within 1 minute, cancelling.",
      components: [],
    });
  }
};