Add privacy update command and update DB schema

Author: Raymo111

src/commandDetails/coin/privacy.ts

@@ -0,0 +1,64 @@
+import { container } from '@sapphire/framework';
+import { Permissions, User } from 'discord.js';
+import {
+  CodeyCommandDetails,
+  CodeyCommandOptionType,
+  CodeyCommandResponseType,
+  SapphireMessageExecuteType
+} from '../../codeyCommand';
+import { getUserPrivacy, changeUserPrivacy } from '../../components/coin';
+
+// Update coin leaderboard privacy of a user
+const coinUpdateExecuteCommand: SapphireMessageExecuteType = async (client, messageFromUser, args) => {
+  if (!(<Readonly<Permissions>>messageFromUser.member?.permissions).has('ADMINISTRATOR')) {
+    return `You do not have permission to use this command.`;
+  }
+
+  // First mandatory argument is user
+  const user = <User>args['user'];
+  if (!user) {
+    throw new Error('please enter a valid user mention or ID for balance adjustment.');
+  }
+
+  // Second mandatory argument is privacy
+  const privacy = args['privacy'];
+  if (typeof privacy !== 'number' || (privacy !== 0 && privacy !== 1)) {
+    throw new Error('please enter 0/1 for public/private.');
+  }
+
+  // Adjust privacy
+  await changeUserPrivacy(user.id, <number>privacy);
+
+  // Get new privacy
+  const newPrivacy = await getUserPrivacy(user.id);
+  return `${user.username} is now ${newPrivacy ? 'private' : 'not private'}.`;
+};
+
+export const coinPrivacyCommandDetails: CodeyCommandDetails = {
+  name: 'privacy',
+  aliases: ['p'],
+  description: 'Update the leaderboard privacy of a user.',
+  detailedDescription: `**Examples:**
+  \`${container.botPrefix}coin privacy @Codey 1\``,
+
+  isCommandResponseEphemeral: false,
+  messageWhenExecutingCommand: 'Updating leaderboard privacy...',
+  executeCommand: coinUpdateExecuteCommand,
+  codeyCommandResponseType: CodeyCommandResponseType.STRING,
+
+  options: [
+    {
+      name: 'user',
+      description: 'The user to adjust the privacy of,',
+      type: CodeyCommandOptionType.USER,
+      required: true
+    },
+    {
+      name: 'privacy',
+      description: 'The privacy to set the specified user to,',
+      type: CodeyCommandOptionType.NUMBER,
+      required: true
+    }
+  ],
+  subcommandDetails: {}
+};

src/commands/coin/coin.ts

@@ -5,6 +5,7 @@ import { coinCheckCommandDetails } from '../../commandDetails/coin/check';
 import { coinInfoCommandDetails } from '../../commandDetails/coin/info';
 import { coinUpdateCommandDetails } from '../../commandDetails/coin/update';
 import { coinCurrentLeaderboardCommandDetails } from '../../commandDetails/coin/leaderboard';
+import { coinPrivacyCommandDetails } from '../../commandDetails/coin/privacy';
 
 const coinCommandDetails: CodeyCommandDetails = {
   name: 'coin',
@@ -19,13 +20,16 @@ const coinCommandDetails: CodeyCommandDetails = {
 \`${container.botPrefix}coin info\`
 \`${container.botPrefix}coin i\`
 \`${container.botPrefix}coin update @Codey 100\`
-\`${container.botPrefix}coin update @Codey 0 Reset Codey's balance.\``,
+\`${container.botPrefix}coin u @Codey 0 Reset Codey's balance.\`
+\`${container.botPrefix}coin privacy @Codey 1\`
+\`${container.botPrefix}coin p @Codey 0\``,
   options: [],
   subcommandDetails: {
     adjust: coinAdjustCommandDetails,
     check: coinCheckCommandDetails,
     info: coinInfoCommandDetails,
     update: coinUpdateCommandDetails,
+    privacy: coinPrivacyCommandDetails,
     leaderboard: coinCurrentLeaderboardCommandDetails
   },
   defaultSubcommandDetails: coinCheckCommandDetails

src/components/coin.ts

@@ -81,14 +81,30 @@ export const getCoinBalanceByUserId = async (userId: string): Promise<number> =>
   return _.get(res, 'balance', 0);
 };
 
-export const getPrivateUserIdList = async (userId: string): Promise<number> => {
+/*
+  Returns 1 or 0 for whether user is private
+*/
+export const getUserPrivacy = async (userId: string): Promise<number> => {
   const db = await openDB();
   // Query user privacy from DB.
   const res = await db.get('SELECT is_private FROM user_coin WHERE user_id = ?', userId);
   // If user doesn't have a privacy value, default to false (public).
   return _.get(res, 'is_private', 0);
 };
 
+export const changeUserPrivacy = async (userId: string, privacy: number): Promise<void> => {
+  const db = await openDB();
+  await db.run(
+    `
+      INSERT INTO user_coin (user_id, is_private) VALUES (?, ?)
+      ON CONFLICT(user_id)
+      DO UPDATE SET is_private = ?`,
+    userId,
+    privacy,
+    privacy
+  );
+};
+
 /*
   If user doesn't exist, create row with newBalance as the balance.
   Otherwise, update balance to newBalance.

src/components/db.ts

@@ -92,7 +92,8 @@ const initUserCoinTable = async (db: Database): Promise<void> => {
     `
     CREATE TABLE IF NOT EXISTS user_coin (
       user_id VARCHAR(255) PRIMARY KEY NOT NULL,
-      balance INTEGER NOT NULL CHECK(balance>=0)
+      balance INTEGER NOT NULL CHECK(balance>=0),
+      is_private INTEGER NOT NULL CHECK(is_private IN (0, 1))
     )
     `
   );