From 094f490357d94a56abe813b4cb8349273c1085ca Mon Sep 17 00:00:00 2001
From: Miikka Virtanen <dragory.up@gmail.com>
Date: Mon, 24 Jul 2017 03:54:03 +0300
Subject: [PATCH] Add !edit_snippet for editing snippets. Add !snippets to list
 all available snippets.

---
 src/index.js    | 51 +++++++++++++++++++++++++++++++++++++++----------
 src/snippets.js |  9 +++++++--
 src/utils.js    | 17 +++++++++++++++++
 3 files changed, 65 insertions(+), 12 deletions(-)

diff --git a/src/index.js b/src/index.js
index f429ee7..46b0394 100644
--- a/src/index.js
+++ b/src/index.js
@@ -405,18 +405,13 @@ bot.registerCommand('logs', (msg, args) => {
         return `\`${formattedDate}\`: <${info.url}>`;
       }).join('\n');
 
-      // Send list of logs in chunks of 15 lines per message
+      // Send the list of logs in chunks of 15 lines per message
       const lines = message.split('\n');
-      const chunks = [];
-      const chunkSize = 15;
-
-      for (let i = 0; i < lines.length; i += chunkSize) {
-        chunks.push(lines.slice(i, i + chunkSize).join('\n'));
-      }
+      const chunks = utils.chunk(lines, 15);
 
       let root = Promise.resolve();
-      chunks.forEach(chunk => {
-        root = root.then(() => msg.channel.createMessage(chunk));
+      chunks.forEach(lines => {
+        root = root.then(() => msg.channel.createMessage(lines.join('\n')));
       });
     });
   }
@@ -465,7 +460,7 @@ bot.registerCommand('snippet', async (msg, args) => {
   if (snippet) {
     if (text) {
       // If the snippet exists and we're trying to create a new one, inform the user the snippet already exists
-      msg.channel.createMessage(`Snippet "${shortcut}" already exists! You can delete it with ${prefix}delete_snippet.`);
+      msg.channel.createMessage(`Snippet "${shortcut}" already exists! You can edit or delete it with ${prefix}edit_snippet and ${prefix}delete_snippet respectively.`);
     } else {
       // If the snippet exists and we're NOT trying to create a new one, show info about the existing snippet
       msg.channel.createMessage(`\`${snippetPrefix}${shortcut}\` replies ${snippet.isAnonymous ? 'anonymously ' : ''}with:\n${snippet.text}`);
@@ -503,6 +498,42 @@ bot.registerCommand('delete_snippet', async (msg, args) => {
 });
 bot.registerCommandAlias('ds', 'delete_snippet');
 
+bot.registerCommand('edit_snippet', async (msg, args) => {
+  if (! msg.channel.guild) return;
+  if (msg.channel.guild.id !== utils.getModmailGuild(bot).id) return;
+  if (! msg.member.permission.has('manageRoles')) return;
+
+  const shortcut = args[0];
+  const text = args.slice(1).join(' ').trim();
+
+  if (! shortcut) return;
+  if (! text) return;
+
+  const snippet = await snippets.get(shortcut);
+  if (! snippet) {
+    msg.channel.createMessage(`Snippet "${shortcut}" doesn't exist!`);
+    return;
+  }
+
+  await snippets.del(shortcut);
+  await snippets.add(shortcut, text, snippet.isAnonymous);
+
+  msg.channel.createMessage(`Snippet "${shortcut}" edited!`);
+});
+bot.registerCommandAlias('es', 'edit_snippet');
+
+bot.registerCommand('snippets', async msg => {
+  if (! msg.channel.guild) return;
+  if (msg.channel.guild.id !== utils.getModmailGuild(bot).id) return;
+  if (! msg.member.permission.has('manageRoles')) return;
+
+  const allSnippets = await snippets.all();
+  const shortcuts = Object.keys(allSnippets);
+  shortcuts.sort();
+
+  msg.channel.createMessage(`Available snippets (prefix ${snippetPrefix}):\n${shortcuts.join(', ')}`);
+});
+
 bot.connect();
 restBot.connect();
 webserver.run();
diff --git a/src/snippets.js b/src/snippets.js
index c93dc02..b638224 100644
--- a/src/snippets.js
+++ b/src/snippets.js
@@ -25,7 +25,7 @@ function getSnippet(shortcut) {
  * @returns {Promise}
  */
 function addSnippet(shortcut, text, isAnonymous = false) {
-  return jsonDb.get('snippets', []).then(snippets => {
+  return jsonDb.get('snippets', {}).then(snippets => {
     snippets[shortcut] = {
       text,
       isAnonymous,
@@ -41,14 +41,19 @@ function addSnippet(shortcut, text, isAnonymous = false) {
  * @returns {Promise}
  */
 function deleteSnippet(shortcut) {
-  return jsonDb.get('snippets', []).then(snippets => {
+  return jsonDb.get('snippets', {}).then(snippets => {
     delete snippets[shortcut];
     jsonDb.save('snippets', snippets);
   });
 }
 
+function getAllSnippets() {
+  return jsonDb.get('snippets', {});
+}
+
 module.exports = {
   get: getSnippet,
   add: addSnippet,
   del: deleteSnippet,
+  all: getAllSnippets,
 };
diff --git a/src/utils.js b/src/utils.js
index acfb00f..00594e6 100644
--- a/src/utils.js
+++ b/src/utils.js
@@ -83,6 +83,22 @@ function getMainRole(member) {
   return roles.find(r => r.hoist);
 }
 
+/**
+ * Splits array items into chunks of the specified size
+ * @param {Array} items
+ * @param {Number} chunkSize
+ * @returns {Array}
+ */
+function chunk(items, chunkSize) {
+  const result = [];
+
+  for (let i = 0; i < items.length; i += chunkSize) {
+    result.push(items.slice(i, i + chunkSize));
+  }
+
+  return result;
+}
+
 module.exports = {
   getModmailGuild,
   getMainGuild,
@@ -91,4 +107,5 @@ module.exports = {
   disableLinkPreviews,
   getSelfUrl,
   getMainRole,
+  chunk,
 };