Add update-all command

src/commands/helpers.js

@@ -1,23 +1,21 @@
-const readSheet = require("./read_sheet");
-
-const getMember = (client, userId) => {
+async function getMember(client, userId) {
   const guild = client.guilds.cache.get(process.env.GUILD_ID);
-  const member = guild.members.cache.get(userId);
+  const member = await guild.members.cache.get(userId);
   return member;
-};
+}
 
-const addRoleToUser = async (client, user_id, role_to_assign) => {
+async function addRoleToUser(client, user_id, role_to_assign) {
   if (!role_to_assign) {
     return null;
   }
-  const member = getMember(client, user_id);
+  const member = await getMember(client, user_id);
   const role = member.guild.roles.cache.find((r) => r.name === role_to_assign);
   member.roles.add(role);
   return role_to_assign;
-};
+}
 
 async function updateUserFromSheet(email, user_id, client) {
-  const members = await readSheet();
+  const members = global.sheetData;
   const user = members.find((value) => value.email === email);
 
   // If we could not find user
@@ -26,15 +24,23 @@ async function updateUserFromSheet(email, user_id, client) {
   }
 
   // Give user respective team role
-  const member = getMember(client, user_id);
+  const member = await getMember(client, user_id);
   const role = member.guild.roles.cache.find((r) => r.name === user.status);
-  member.roles.add(role);
+  if (role) {
+    try {
+      await member.roles.add(role);
+    } catch (err) {
+      console.error(`Failed to set ${user.name}'s role:`, err);
+    }
+  } else {
+    console.error(`Could not find role '${user.status}' for ${user.name}.`);
+  }
 
   // Set user's name
   try {
-    member.setNickname(user.name);
+    await member.setNickname(user.name);
   } catch (err) {
-    console.error(`Failed to set ${user.name}'s nickname: `, err);
+    console.error(`Failed to set ${user.name}'s nickname:`, err);
   }
 
   return user.status;

src/commands/index.js

@@ -4,8 +4,18 @@ const updateRoles = require("./update_roles");
 const verify = require("./verify");
 const whoami = require("./whoami");
 const datalearn = require("./datalearn");
+const updateAll = require("./update_all");
+const updateSheet = require("./update_sheet");
 
-const commands = [iam, updateRoles, verify, whoami, datalearn];
+const commands = [
+  iam,
+  updateRoles,
+  verify,
+  whoami,
+  datalearn,
+  updateAll,
+  updateSheet,
+];
 
 module.exports = (client) => {
   // Collate commands into Collection

src/commands/read_sheet.js

@@ -4,7 +4,7 @@ const API_KEY = process.env.GOOGLE_API_KEY;
 const SHEET_ID = process.env.SHEET_ID;
 const SHEET_GID = process.env.SHEET_GID;
 
-const readSheet = async () => {
+async function updateSheet() {
   const doc = new GoogleSpreadsheet(SHEET_ID);
   doc.useApiKey(API_KEY);
   // https://theoephraim.github.io/node-google-spreadsheet/#/classes/google-spreadsheet-worksheet
@@ -16,7 +16,9 @@ const readSheet = async () => {
   for (const row of rows) {
     data.push({ name: row.Name, email: row.Email, status: row.Status });
   }
-  return data;
-};
+  global.sheetData = data;
+}
 
-module.exports = readSheet;
+updateSheet();
+
+module.exports = { updateSheet };

src/commands/update_all.js

@@ -0,0 +1,29 @@
+const { SlashCommandBuilder, PermissionFlagsBits } = require("discord.js");
+const db = require("../db");
+const { updateUserFromSheet } = require("./helpers");
+
+async function execute(interaction) {
+  const users = db.collection("users").find();
+  const ids_in_guild = (await interaction.guild.members.fetch()).map(
+    (member) => member.user.id
+  );
+
+  users.forEach(async (doc) => {
+    if (doc && ids_in_guild.includes(doc.user_id)) {
+      updateUserFromSheet(doc.email, doc.user_id, interaction.client);
+    }
+  });
+
+  interaction.reply({
+    content: "Updating users...",
+    ephemeral: true,
+  });
+}
+
+module.exports = {
+  data: new SlashCommandBuilder()
+    .setName("update-all")
+    .setDescription("update all users team roles and names (ADMIN ONLY!)")
+    .setDefaultMemberPermissions(PermissionFlagsBits.Administrator),
+  execute,
+};

src/commands/update_sheet.js

@@ -0,0 +1,18 @@
+const { SlashCommandBuilder, PermissionFlagsBits } = require("discord.js");
+const { updateSheet } = require("./read_sheet");
+
+async function execute(interaction) {
+  await updateSheet();
+  interaction.reply({
+    content: "Sheet updated",
+    ephemeral: true,
+  });
+}
+
+module.exports = {
+  data: new SlashCommandBuilder()
+    .setName("update-sheet")
+    .setDescription("update google sheet cached data (ADMIN ONLY!)")
+    .setDefaultMemberPermissions(PermissionFlagsBits.Administrator),
+  execute,
+};

src/index.js

@@ -5,7 +5,9 @@ const registerCommands = require("./commands");
 const SERVER_ID = process.env.SERVER_ID;
 
 // Create a new client instance
-const client = new Client({ intents: [GatewayIntentBits.Guilds] });
+const client = new Client({
+  intents: [GatewayIntentBits.Guilds, GatewayIntentBits.GuildMembers],
+});
 
 // When the client is ready, run this code (only once)
 client.once("ready", async () => {