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: [] });
  }
}