raumstatus/node/ip-poll.js

var redis = require("redis");
var util = require('util');
var EventEmitter = require('events').EventEmitter;
var exec = require('child_process').exec;

var redisprefix = "ippoll:";

var IpPoll = function(switchaddr, hostsaddr) {
	var self = this;

    var redisClient = redis.createClient();
	var regexp = /\(([0-9]+) hosts* up\)/;
	var nmap = "nmap -n -sP -T5 ";

    redisClient.on("connect", function () {
        console.log("connected to redis");
        self.emit('ready');
    });
		
	this.pollCount = function() {	
		exec(nmap + "--min-hostgroup 10 " + hostsaddr, function (error, stdout, stderr) {			
			if(error == null) {		
				var matches = regexp.exec(stdout);		
				if(matches != null && matches.length == 2) {
                    var time = Date.now();

                    redisClient.zremrangebyscore('onlinecount', "-inf", time - 7*24*60*1000);

                    var num = matches[1];

                    redisClient.zadd('onlinecount', time, time + "|" + num, function() {
                        self.emit('doneCount', num);
                    });
				}
			}
		});	
	};
	
	this.pollState = function() {
		exec(nmap + switchaddr, function (error, stdout, stderr) {			
			if(error == null) {		
				var matches = regexp.exec(stdout);
				if(matches != null && matches.length == 2) {
					self.emit('doneState', matches[1] == "1");
				}
			} else {
				self.emit('doneState', "unknown");
			}
		});			
	};

    this.getHistory = function(start, end, callback) {

        redisClient.zrangebyscore('onlinecount', "-inf", "+inf", function(err, replies) {
            var data = [];

            replies.forEach(function (reply, i) {
                var line = reply.split('|');
                data.push( { at: moment(parseInt(line[0])).format(), value: line[1] });
            });

            callback(data);
        });
    };
};

util.inherits(IpPoll, EventEmitter);

module.exports = IpPoll;