Merge pull request #7 from Westixy-FORK/feature_channel-subscribe

Feature channel subscribe

Author: Xstoudi

.gitignore

@@ -59,3 +59,5 @@ typings/
 .vscode
 
 ./config/dev.json
+
+db/**

README.md

@@ -12,6 +12,7 @@ TWITTER_CONSUMER_KEY=
 TWITTER_CONSUMER_SECRET=
 TWITTER_ACCESS_TOKEN=
 TWITTER_ACCESS_TOKEN_SECRET=
+LOKI_DB_PATH=
 ```
 
 ## Commands available

config/prod.json

@@ -118,4 +118,4 @@
     "StrongLoop",
     "JSKongress"
   ]
-}
+}

index.js

@@ -5,12 +5,18 @@ const CommandManager = require('./src/command')
 const TwitterPlugin = require('./src/plugin/twitter')
 const docCmd = require('./src/commands/doc')
 const helpCmd = require('./src/commands/help')
+const subscriberCmd = require('./src/commands/subscriber')
 const welcomeEmbed = require('./src/embeds/welcome')
 const joinLeftEmbed = require('./src/embeds/join-left')
 
 const CM = CommandManager.init()
 CM.addCommand('doc', docCmd)
 CM.addCommand('help', helpCmd)
+CM.addCommand('subscribe', subscriberCmd.subscribe)
+CM.addCommand('sub', subscriberCmd.subscribe)
+CM.addCommand('unsubscribe', subscriberCmd.unsubscribe)
+CM.addCommand('unsub', subscriberCmd.unsubscribe)
+CM.addCommand('alert', subscriberCmd.alert)
 
 const RM = new Event()
 

package-lock.json

@@ -12,6 +12,7 @@
   "dependencies": {
     "@sindresorhus/is": "^0.9.0",
     "discord.js": "^11.3.2",
+    "lokijs": "^1.5.5",
     "twit": "^2.2.9"
   },
   "devDependencies": {

package.json

@@ -0,0 +1,85 @@
+const Loki = require('lokijs')
+
+const dbPath = process.env.LOKI_DB_PATH
+
+const db = new Loki(dbPath, {
+  autoload: true,
+  autosave: true,
+  autoloadCallback: () => {
+    subs = db.getCollection('subscribers')
+    if (subs === null) subs = db.addCollection('subscribers')
+  }
+})
+
+let subs = null
+
+module.exports.subscribe = ({
+  message,
+  args
+}) => {
+  if (args.length === 0) {
+    const subscribedChannel = subs.findOne({
+      userId: message.author.id,
+      channel: message.channel.name
+    })
+    if (subscribedChannel !== null) {
+      message.author.send(`Vous êtes déjà abonné au channel \`#${message.channel.name}\``)
+      return message.delete()
+    }
+    subs.insert({
+      userId: message.author.id,
+      channel: message.channel.name
+    })
+    message.author.send(`Vous vous êtes abonné au channel \`#${message.channel.name}\``)
+    return message.delete()
+  }
+  if (args[0] === 'list') {
+    const subscribedChannels = subs.find({
+      userId: message.author.id
+    })
+    if (subscribedChannels.length === 0) {
+      message.author.send(`Vous n'êtes abonné à aucun channel`)
+      return message.delete()
+    }
+    message.author.send(`Vous êtes abonnés aux channels suivants :\n${subscribedChannels.map(a => `\`#${a.channel}\``).join('\n')}`)
+    return message.delete()
+  }
+}
+
+module.exports.unsubscribe = ({
+  message,
+  args
+}) => {
+  if (args.length === 0) {
+    const subscribedChannel = subs.findOne({
+      userId: message.author.id,
+      channel: message.channel.name
+    })
+    if (subscribedChannel !== null) {
+      subs.remove(subscribedChannel)
+      message.author.send(`Vous vous êtes désabonné au channel \`#${message.channel.name}\``)
+      return message.delete()
+    }
+    message.author.send(`Vous n'êtes pas abonné au channel \`#${message.channel.name}\``)
+    return message.delete()
+  }
+}
+
+module.exports.alert = ({
+  message,
+  args
+}) => {
+  if (args.length === 0) {
+    message.author.send(`Veuillez mettre un message à votre alert : \`!alert <message>\` `)
+    return message.delete()
+  }
+  const subscribersOnChannel = subs.find({
+    channel: message.channel.name
+  })
+  if (subscribersOnChannel.length === 0) {
+    message.channel.send(`${args.join(' ')}\nIl n'y a pas d'abonné à ce channel...`)
+    return message.delete()
+  }
+  message.channel.send(`${args.join(' ')}\n${subscribersOnChannel.map(sub => `<@${sub.userId}>`).join(', ')}`)
+  return message.delete()
+}