add optional email notification using mailgun

rngadam · rngadam · commit 1786c5aed470 · 2017-02-06T16:17:21.000+08:00
diff --git a/README.md b/README.md
@@ -8,7 +8,6 @@ including per-group support tickets management.
 # installation
 
 ```
-brew install chromedriver
 npm install
 ```
 
@@ -35,4 +34,4 @@ Run and re-run tests anytime files are updated:
 
 ```
 npm run watch
-```
+```
diff --git a/bot.js b/bot.js
@@ -5,20 +5,46 @@
 const {Wechaty} = require('wechaty')
 
 const reload = require('./reloader')
+const tmp = require('tmp')
 
 const data = {}
+const urls = {}
 
-Wechaty.instance()
-.on('scan', (url, code) => {
+function handleScan(url, code) {
+    if(urls[url]) {
+        console.log(`already notified user for ${url}`)
+        return
+    }
+    urls[url] = true
+    var config = reload('./config.json')
+    if(config.mailgun) {
+        var mg = reload('./mailgun.js')
+        if(mg.isValidMailgunConfig(config.mailgun)) {
+            /* eslint-disable no-sync */
+            var filename = tmp.tmpNameSync({template: '/tmp/qrcode-XXXXXX.jpg'});
+            mg.getHttpFile(url, filename, () => mg.email(
+                mg.createMailgunClient(config.mailgun),
+                mg.createEmailData(config.mailgun, filename, url))
+                .then((body) => console.log(body))
+                .catch((err) => console.log(err)))
+        } else {
+            console.log(`invalid configuration ${JSON.stringify(config.mailgun)}`)
+        }
+    } else {
+        console.log('warning: mailgun not configured')
+    }
     console.log(`Scan QR Code to login: ${code}\n${url}`)
-})
-.on('login', (user) => {
-    console.log(`User ${user} logined`)
-})
-.on('message', (message) => {
-    message.ready().then(() => {
-        var config = reload('./config.json')
-        reload('./handler.js')(config, data, message)
+}
+
+Wechaty.instance()
+    .on('scan', handleScan)
+    .on('login', (user) => {
+        console.log(`User ${user} logined`)
+    })
+    .on('message', (message) => {
+        message.ready().then(() => {
+            var config = reload('./config.json')
+            reload('./handler.js')(config, data, message)
+        })
     })
-})
-.init()
+    .init()
diff --git a/mailgun.js b/mailgun.js
@@ -0,0 +1,63 @@
+'use strict'
+
+const fs = require('fs')
+const os = require('os')
+const request = require('request');
+
+function email(mailgun, emailData) {
+    return new Promise((resolve, reject) => {
+        mailgun.messages().send(emailData, function (error, body) {
+            if(error) {
+                reject(error)
+            } else {
+                resolve(body);
+            }
+        });
+    })
+}
+
+function getHttpFile(url, filename, cb) {
+    request(url)
+        .on('error', function(err) {
+            cb(err)
+        })
+        .on('end', function() {
+            cb()
+        })
+        .pipe(fs.createWriteStream(filename))
+}
+
+function isValidMailgunConfig(config) {
+    return Boolean(config.apiKey &&
+        config.domain &&
+        config.from &&
+        config.to &&
+        config.subject)
+}
+
+function createMailgunClient(config) {
+    var mailgun = require('mailgun-js')({
+        apiKey: config.apiKey,
+        domain: config.domain
+    });
+    return mailgun
+}
+
+function createEmailData(config, filename, text) {
+    var emailData = {
+        from: config.from,
+        inline: filename,
+        subject: `${new Date().getTime()} ${config.subject}`,
+        text: `${text}\n\n${new Date()}\n${os.hostname()}`,
+        to: config.to,
+    };
+    return emailData
+}
+
+module.exports = {
+    createEmailData,
+    createMailgunClient,
+    email,
+    getHttpFile,
+    isValidMailgunConfig
+}
diff --git a/package.json b/package.json
@@ -5,6 +5,8 @@
   "main": "bot.js",
   "dependencies": {
     "candobot-data": "git+ssh://git@github.com/coderbunker/candobot-data.git",
+    "mailgun-js": "^0.8.1",
+    "request": "^2.79.0",
     "require-reload": "^0.2.2",
     "wechaty": "^0.7.3"
   },
diff --git a/test/mailgun_test.js b/test/mailgun_test.js
@@ -0,0 +1,16 @@
+'use strict'
+
+const assert = require('chai').assert
+const mg = require('../mailgun')
+
+describe('mailgun', () => {
+    it('validates config', () => {
+        assert.isTrue(mg.isValidMailgunConfig({
+            'apiKey': 'key-testkey',
+            'domain': 'mg.coderbunker.com',
+            'from': 'Coderbunker IT <it@coderbunker.com>',
+            'subject': 'Please scan WeChat QRcode with Coderbunker Phone',
+            'to': 'wechaty@coderbunker.com',
+        }))
+    })
+})
