feat: Make it work

Author: mkg20001

src/alert.js

@@ -6,6 +6,12 @@ class Alert {
   constructor (config) {
     this.config = config
   }
+
+  setID (id) {
+    this.id = id
+    return this
+  }
+
   at (time) {
     this.timeField = moment(time).format('[at] ' + this.config.dateFormat)
     return this
@@ -42,4 +48,4 @@ class Alert {
   }
 }
 
-module.exporrts = Alert
+module.exports = Alert

src/group.js

@@ -1,18 +1,66 @@
 'use strict'
 
 const Alert = require('./alert')
+const crypto = require('crypto')
+const shortHash = (str) => {
+  let hash = crypto.createHash('sha512').update(str).digest('hex')
+  return hash.substr(parseInt(hash.substr(0, 1), 16), 16)
+}
 
 class Group {
-  constructor (main) {
+  constructor (main, config, prevAlerts) {
     this.main = main
+    this.config = config
+    this.alerts = {}
+    this.prevAlerts = prevAlerts || []
   }
   name (name) {
     this.name = name
+    this.id = shortHash(name)
     return this
   }
+  attachAlert (alert) {
+    this.alerts[alert.id] = alert
+  }
   alert (id) {
-    const alert = new Alert(id)
+    const alert = new Alert(this.config).setID(id)
     this.attachAlert(alert)
+    return alert
+  }
+  setAutoClear (v) {
+    this.autoClear = v
+  }
+  process () {
+    let alerts = Object.keys(this.alerts).map(k => this.alerts[k])
+    let prevAlerts = this.prevAlerts
+    let clearable = prevAlerts.filter(pa => !alerts.filter(a => a.id === pa.id).length)
+    let newPrevAlerts = alerts.filter(a => a.type !== 'clear') // don't keep clear events arround
+    let sendNotify = alerts.filter(a => {
+      if (!prevAlerts.filter(pa => pa.id === a.id).length && a.type !== 'clear') { // new non-clear
+        return true
+      }
+
+      if (prevAlerts.filter(pa => pa.id === a.id).length && a.type === 'clear') { // old clear
+        return true
+      }
+
+      return false
+    })
+    if (this.autoClear) {
+      sendNotify = sendNotify.concat(clearable.map(a =>
+        this.alert(a.id).type('clear').title('Cleared alert ' + JSON.stringify(a.title)).body('Alert for group ' + JSON.stringify(this.name) + ' has been cleared')
+      ))
+    } else {
+      newPrevAlerts = newPrevAlerts.concat(clearable) // keep clearable events arround
+    }
+
+    sendNotify.forEach(alert => {
+      this.main.notifies.forEach(notify => {
+        notify.notify(alert)
+      })
+    })
+
+    return newPrevAlerts
   }
 }
 

src/index.js

@@ -1,5 +1,7 @@
 'use strict'
 
+require('colors')
+
 const Notifies = {
   console: require('./notify/console'),
   notifySend: require('./notify/notifySend')
@@ -13,8 +15,9 @@ const Sources = {
 class SysadminNotifier {
   constructor (config) {
     const globalConf = config.global || {}
-    this.notify = []
+    this.notifies = []
     this.sources = []
+    this.prevAlerts = {}
     for (const p in config) { // eslint-disable-line guard-for-in
       switch (true) {
         case p === 'global':
@@ -25,7 +28,7 @@ class SysadminNotifier {
             throw new TypeError('Unknown notification system ' + name)
           }
           const Notify = Notifies[name]
-          this.notify.push(new Notify(this, Object.assign(globalConf, config[p])))
+          this.notifies.push(new Notify(this, Object.assign(Object.assign({}, globalConf), config[p])))
           break
         }
         case p.startsWith('source.'): {

src/notify/console.js

@@ -5,7 +5,7 @@
 const Notify = require('../notify')
 
 const typeColor = {
-  ciritical: 'red',
+  critical: 'red',
   warning: 'yellow',
   clear: 'green'
 }
@@ -15,7 +15,7 @@ class Console extends Notify {
     const color = typeColor[alert.type]
     console.log(`[${alert.timeField ? (alert.timeField.toUpperCase() + '/') : ''}${alert.type.toUpperCase()}] ${alert.title}`[color].bold)
     if (alert.body) {
-      alert.body.split('\n').map(s => ('   ' + s)[color]).forEach(console.log)
+      alert.body.split('\n').map(s => ('   ' + s)[color]).forEach((l) => console.log(l))
     }
   }
 }

src/notify/notifySend.js

@@ -4,7 +4,7 @@ const Notify = require('../notify')
 const cp = require('child_process')
 
 const typeIcon = {
-  ciritical: 'dialog-error',
+  critical: 'dialog-error',
   warning: 'dialog-warning',
   clear: 'flag-green'
 }

src/source.js

@@ -1,15 +1,31 @@
 'use strict'
 
+const Group = require('./group')
+
 class Source {
   constructor (main, config) {
     this.main = main
     this.config = config
+    this.intv = 0
+  }
+  async doCheck () {
+    const groups = await this.check()
+    groups.forEach(g => {
+      this.main.prevAlerts[g.id] = g.process()
+    })
+    this.prevGroups = groups
   }
   async start () {
-    // TODO: add
+    await this.doCheck()
+    this.intv = setInterval(() => this.doCheck(), this.config.interval)
   }
   async stop () {
-    // TODO: add
+    clearInterval(this.intv)
+  }
+  group (name) {
+    const g = new Group(this.main, this.config).name(name)
+    g.prevAlerts = this.main.prevAlerts[g.id] || []
+    return g
   }
 }
 

src/source/netdata.js

@@ -16,14 +16,18 @@ class Netdata extends Source {
 
   }
   async check () {
-    return this.config.hosts.map(host => this.checkHost(host))
+    return Promise.all(this.config.hosts.map(host => this.checkHost(host)))
   }
   async checkHost (host) {
-    const g = this.main.group('netdata host ' + host)
+    const g = this.group('netdata host ' + host)
     try {
+      g.setAutoClear(true)
       let res = await fetch('http://' + host + ':19999/api/v1/alarms?active')
       res = await res.json()
-      console.log(res)
+      for (const alarmID in res.alarms) { // eslint-disable-line guard-for-in
+        const alarm = res.alarams[alarmID]
+        console.log(alarm)
+      }
     } catch (e) {
       g.alert('fetch_error').critical().title('Could not fetch netdata alerts for ' + host).body('Please check network connectivity and host uptime').since(Date.now())
     }

src/source/uptimerobot.js

@@ -12,9 +12,9 @@ const stateMap = {
 }
 
 const stateTypeMap = {
-  0: 'warn',
-  2: 'critical',
-  9: 'clear'
+  0: 'warning',
+  2: 'clear',
+  9: 'critical'
 }
 
 class UptimeRobot extends Source {
@@ -32,7 +32,7 @@ class UptimeRobot extends Source {
       r.last_status_change = Date.parse(r.log[0].datetime)
       return r
     })
-    let g = this.main.group('UptimeRobot')
+    let g = this.group('UptimeRobot')
     res.forEach(monitor => {
       let alert = g.alert(monitor.id).type(monitor.stateType)
         .since(monitor.last_status_change)