Add privacy controls

.gitignore

@@ -2,3 +2,4 @@ tokens.json
 config.json
 package-lock.json
 node_modules
+user_privacy.json

user-bot.js

@@ -10,6 +10,7 @@ let sshSurfID;  // Variable to store the user ID from the database
 
 // Paths to config and tokens files
 const tokensFile = './tokens.json';
+const privacyFile = './user_privacy.json'; // Privacy file path
 const config = JSON.parse(readFileSync('./config.json', 'utf8'));
 
 // MySQL connection
@@ -38,6 +39,24 @@ function saveTokens(tokens) {
   jsonfile.writeFileSync(tokensFile, tokens, { spaces: 2 });
 }
 
+// Load privacy settings from JSON file
+function loadUserPrivacySettings() {
+  try {
+    return jsonfile.readFileSync(privacyFile);
+  } catch (error) {
+    console.error('Error reading privacy settings file:', error);
+    return {};
+  }
+}
+
+// Save user privacy settings to JSON file
+function saveUserPrivacySettings() {
+  jsonfile.writeFileSync(privacyFile, userPrivacySettings, { spaces: 2 });
+}
+
+// Load user privacy settings on startup
+let userPrivacySettings = loadUserPrivacySettings();
+
 // Automatically request a new token if it doesn't exist or is invalid
 async function fetchAndSaveToken(sshSurfID, interaction) {
   return unirest
@@ -123,6 +142,24 @@ function sendSexyEmbedWithFields(title, fields, interaction, ephemeral = false)
   });
 }
 
+// Toggle privacy for the user
+async function togglePrivacy(interaction) {
+  const userId = interaction.user.id;
+  const currentSetting = userPrivacySettings[userId] || false;
+  userPrivacySettings[userId] = !currentSetting; // Toggle the setting
+  saveUserPrivacySettings();
+
+  const message = userPrivacySettings[userId] 
+    ? 'Your responses are now set to ephemeral (visible only to you).' 
+    : 'Your responses are now standard (visible to everyone).';
+  await interaction.editReply({ content: message, ephemeral: true });
+}
+
+// Check if the user's responses should be ephemeral
+function isEphemeral(userId) {
+  return userPrivacySettings[userId] || false; // Default to false (standard response) if not set
+}
+
 // Slash command definitions
 const commands = [
   new SlashCommandBuilder().setName('hello').setDescription('Say hello via API'),
@@ -140,6 +177,7 @@ const commands = [
     .addStringOption(option => option.setName('command').setDescription('Command to execute').setRequired(true)),
   new SlashCommandBuilder().setName('notify').setDescription('Send a notification to Discord')
     .addStringOption(option => option.setName('message').setDescription('Message to send').setRequired(true)),
+  new SlashCommandBuilder().setName('privacy').setDescription('Toggle ephemeral responses') // New command to toggle privacy
 ];
 
 // Register commands with Discord
@@ -178,9 +216,11 @@ const rest = new REST({ version: '10' }).setToken(config.token);
 // Handle bot interactions
 client.on('interactionCreate', async interaction => {
   if (!interaction.isCommand()) return;
+ // Check if the command is /privacy, always make it ephemeral
+ const isPrivacyCommand = interaction.commandName === 'privacy';
   
-  // Defer the reply to allow time for the command to run
-  await interaction.deferReply();
+ // Defer the reply based on whether it's the privacy command or not
+ await interaction.deferReply({ ephemeral: isPrivacyCommand || isEphemeral(interaction.user.id) });
 
   // First, we fetch the sshSurfID from the database using interaction.user.id
   let sshSurfID = await new Promise((resolve, reject) => {
@@ -205,21 +245,26 @@ client.on('interactionCreate', async interaction => {
     return sendSexyEmbed("Error", "User not found in the database.", interaction);
   }
 
+  // Handle privacy toggle command
+  if (interaction.commandName === 'privacy') {
+    await togglePrivacy(interaction);
+    return;
+  }
+
   // Once sshSurfID is set, we proceed with token fetching and API requests
   const apiToken = await getToken(sshSurfID, interaction);
-
   try {
     switch (interaction.commandName) {
       case 'hello':
         const helloResponse = await makeApiRequest('/hello', apiToken, interaction);
-        sendSexyEmbed("Hello", `Message: ${helloResponse.message}`, interaction);
+        sendSexyEmbed("Hello", `Message: ${helloResponse.message}`, interaction, isEphemeral(interaction.user.id));
         break;
   
       case 'name':
         const nameResponse = await makeApiRequest('/name', apiToken, interaction);
         sendSexyEmbedWithFields('Username', [
           { name: 'Username', value: nameResponse.message }
-        ], interaction);
+        ], interaction, isEphemeral(interaction.user.id));
         break;
   
       case 'start':
@@ -227,7 +272,7 @@ client.on('interactionCreate', async interaction => {
         sendSexyEmbedWithFields('Start Server', [
           { name: 'Status', value: 'Success' },
           { name: 'Message', value: startResponse.message }
-        ], interaction);
+        ], interaction, isEphemeral(interaction.user.id));
         break;
   
       case 'stop':
@@ -235,7 +280,7 @@ client.on('interactionCreate', async interaction => {
         sendSexyEmbedWithFields('Stop Server', [
           { name: 'Status', value: 'Success' },
           { name: 'Message', value: stopResponse.message }
-        ], interaction);
+        ], interaction, isEphemeral(interaction.user.id));
         break;
   
       case 'restart':
@@ -243,13 +288,11 @@ client.on('interactionCreate', async interaction => {
         sendSexyEmbedWithFields('Restart Server', [
           { name: 'Status', value: 'Success' },
           { name: 'Message', value: restartResponse.message }
-        ], interaction);
+        ], interaction, isEphemeral(interaction.user.id));
         break;
   
       case 'info':
         const infoResponse = await makeApiRequest('/info', apiToken, interaction);
-          
-            // Extract and fallback data fields from the response
         const containerName = infoResponse.data?.name || 'N/A';
         const ipAddress = infoResponse.data?.IPAddress || 'N/A';
         const macAddress = infoResponse.data?.MacAddress || 'N/A';
@@ -263,7 +306,6 @@ client.on('interactionCreate', async interaction => {
         const image = infoResponse.data?.image || 'N/A';
         const createdAt = infoResponse.data?.created || 'N/A';
   
-            // Format and send the embed
         sendSexyEmbedWithFields('Container Info', [
           { name: 'Name', value: containerName },
           { name: 'IP Address', value: ipAddress },
@@ -276,7 +318,7 @@ client.on('interactionCreate', async interaction => {
           { name: 'PID', value: `${pid}` },
           { name: 'Started At', value: startedAt },
           { name: 'Created At', value: createdAt }
-            ], interaction);
+        ], interaction, isEphemeral(interaction.user.id));
         break;
   
       case 'stats':
@@ -284,54 +326,45 @@ client.on('interactionCreate', async interaction => {
         sendSexyEmbedWithFields('Container Stats', [
           { name: 'Memory Usage', value: `${statsResponse.data.memory.raw} (${statsResponse.data.memory.percent})` },
           { name: 'CPU Usage', value: statsResponse.data.cpu }
-        ], interaction);
+        ], interaction, isEphemeral(interaction.user.id));
         break;
   
       case 'time':
         const timeResponse = await makeApiRequest('/time', apiToken, interaction);
         sendSexyEmbedWithFields('Container Expire Time', [
           { name: 'Expire Date', value: timeResponse.expireDate }
-        ], interaction);
+        ], interaction, isEphemeral(interaction.user.id));
         break;
   
       case 'root-password':
         const rootPassResponse = await makeApiRequest('/rootpass', apiToken, interaction);
         sendSexyEmbedWithFields('Root Password', [
           { name: 'New Root Password', value: rootPassResponse.newRootPass }
-        ], interaction, true);
+        ], interaction, isEphemeral(interaction.user.id));
         break;
   
       case 'new-api-key':
         const newKeyResponse = await makeApiRequest('/new-key', apiToken, interaction);
         sendSexyEmbedWithFields('New API Key', [
           { name: 'New API Key', value: newKeyResponse.newAPIKey }
-        ], interaction, true);
+        ], interaction, isEphemeral(interaction.user.id));
         break;
   
       case 'key-expire-time':
         const keyTimeResponse = await makeApiRequest('/key-time', apiToken, interaction);
         sendSexyEmbedWithFields('API Key Expire Time', [
           { name: 'Expire Date', value: keyTimeResponse.expireDate }
-        ], interaction, true);
+        ], interaction, isEphemeral(interaction.user.id));
         break;
   
-
-
-        case 'x': {
-
+      case 'x':
         const command = interaction.options.getString('command');
-          
-          // Get the user's current working directory or default to root (/)
         let userPWD = userWorkingDirectories.get(sshSurfID) || '/';
   
-          // Handle 'cd' command logic
         if (command.startsWith('cd')) {
           let argscmd = command.replace('cd ', '').trim();
-            
-            // Handle 'cd ..' for going up one directory
           if (argscmd === '..') {
             if (userPWD !== '/') {
-                // Remove the last part of the current path
               const newPWD = userPWD.split('/').slice(0, -1).join('/') || '/';
               userPWD = newPWD;
               userWorkingDirectories.set(sshSurfID, newPWD);
@@ -342,7 +375,6 @@ client.on('interactionCreate', async interaction => {
             return;
           }
   
-            // Handle '~' for home directory
           if (argscmd === '~') {
             userPWD = '/root';
             userWorkingDirectories.set(sshSurfID, userPWD);
@@ -350,45 +382,32 @@ client.on('interactionCreate', async interaction => {
             return;
           }
   
-            // Handle absolute and relative paths
           let newPWD;
           if (argscmd.startsWith('/')) {
-              // Absolute path
             newPWD = argscmd;
           } else {
-              // Relative path
             newPWD = `${userPWD}/${argscmd}`;
           }
-        
-            // Normalize the path (remove extra slashes)
           newPWD = newPWD.replace(/([^:]\/)\/+/g, "$1");
   
-            // Check if the user is trying to go back multiple directories (e.g., 'cd ../../')
           if (argscmd.includes('../')) {
             const numDirsBack = argscmd.split('../').length - 1;
             newPWD = RemoveLastDirectoryPartOf(userPWD, numDirsBack);
           }
   
-            // Update the working directory
           userWorkingDirectories.set(sshSurfID, newPWD);
           await interaction.editReply(`Directory changed to: ${newPWD}`);
           return;
         }
   
-          // If the command is not 'cd', run the command in the current working directory (or default to '/')
         const execResponse = await makeApiRequest('/exec', apiToken, interaction, 'post', {
           cmd: command,
-            pwd: userPWD // Use the current directory or default to '/'
+          pwd: userPWD
         });
   
-          // Handle the command output
         if (execResponse.stdout.length > 2020) {
-            // If the output is too large, save it to a temporary file and upload to a paste service
-
-            
           const tempFilePath = '/tmp/paste';
           const pasteCommand = `Command: ${command} | Container Owner: ${interaction.user.username}\n${execResponse.stdout}`;
-        
           fs.writeFileSync(tempFilePath, pasteCommand, (err) => {
             if (err) {
               console.error(err);
@@ -398,46 +417,30 @@ client.on('interactionCreate', async interaction => {
   
           const pasteout = await cmd("sleep 2; cat /tmp/paste | dpaste");
   
-            // Create an embed with the paste link
           const mainEmbed = new EmbedBuilder()
             .setColor('#0099ff')
             .setTitle(`Container Owner: ${interaction.user.username}`)
             .setDescription(`The command: ${command} was too large for Discord.`)
-              .addFields(
-                { 
+            .addFields({
               name: 'Please check the below output log:',
               value: pasteout.stdout.replace("Pro tip: you can password protect your paste just by typing a username and password after your paste command.", "")
                 .replace("Paste Saved: ", "")
                 .replace("-------------------------------------------------------", "")
-                }
-              )
+            })
             .setFooter({ text: `Requested by ${interaction.user.username}`, iconURL: `${interaction.user.displayAvatarURL()}` });
   
           await interaction.editReply({ embeds: [mainEmbed] });
   
         } else {
-            // If the output is small enough, format it in a markdown code block
           let replyMessage = `\`\`\`\n${execResponse.stdout || 'No output'}\n\`\`\``;
   
-            // If there is an error, append the error message in another markdown code block
           if (execResponse.stderr && execResponse.stderr.trim()) {
             replyMessage += `\n**Error:**\n\`\`\`\n${execResponse.stderr}\n\`\`\``;
           }
   
-            // Reply with the formatted message
           await interaction.editReply(replyMessage);
         }
         break;
-        }
-        
-        // Helper function to remove directories when using '../'
-        function RemoveLastDirectoryPartOf(the_url, num) {
-          var the_arr = the_url.split('/');
-          the_arr.splice(-num, num);
-          return the_arr.join('/') || '/';
-        }
-        
-
   
       case 'notify':
         const message = interaction.options.getString('message');
@@ -447,16 +450,16 @@ client.on('interactionCreate', async interaction => {
         sendSexyEmbedWithFields('Notification', [
           { name: 'Status', value: 'Success' },
           { name: 'Message', value: notifyResponse.message }
-        ], interaction, true);
+        ], interaction, isEphemeral(interaction.user.id));
         break;
   
       default:
-        interaction.reply('Command not recognized.');
+        interaction.reply({ content: 'Command not recognized.', ephemeral: isEphemeral(interaction.user.id) });
         break;
     }
   } catch (error) {
     console.error('Command error:', error);
-    sendSexyEmbed('Error', 'An error occurred while processing your request.', interaction);
+    sendSexyEmbed('Error', 'An error occurred while processing your request.', interaction, isEphemeral(interaction.user.id));
   }
 });  
 

user_privacy.json

@@ -0,0 +1,3 @@
+{
+  "342128351638585344": false
+}