Initial commit

Author: leighton

.gitignore

@@ -0,0 +1,3 @@
+node_modules
+logs
+.DS_Store

activity.js

@@ -0,0 +1,71 @@
+var client = require('./lib/client.js');
+
+var values = (function(){
+  var args=process.argv.splice(2);
+
+  if(args.length==0){
+    console.log('Usage: ./activity.js -t <timeout> -m "<message>"');
+    process.exit(1);
+  }
+  var arg_parser = {
+    "-t" : function(val){
+      var unit = val.charAt(val.length-1);
+      var nval = parseInt(val.substring(0,val.length-1));
+      switch(unit){
+        case 's':
+          return nval*1000;
+        case 'm':
+          return nval*1000*60;
+        case 'h':
+          return nval*1000*60*60;
+        default:
+          throw Error('Could not parse timeout! '+val);
+      }
+    },
+    "-m" : function(val){
+      return val;
+    }
+  };
+
+  var vals = {
+    "-t" : 0,
+    "-m" : "?"
+  };
+
+  args.forEach(function(v,i){
+    if(v.charAt(0)=='-'){
+      if(arg_parser[v]!==undefined){
+        if(args.length>=i+2){
+          vals[v]=arg_parser[v](args[i+1]);
+        }else{
+          new Error('No argument for '+v);
+        }
+      }else{
+        throw Error('Could not parse arguments '+v);
+      }
+    }	
+  });
+  return vals;
+}());
+
+process.on('uncaughtException', function (err) {
+  console.log('Caught exception: ' + err);
+});
+
+var updater = new client.UpdateClient({
+  hmmac : require('./etc/hmmac-config.json'),
+  pusher_cred : require('./etc/pusher-key.json'),
+  api_cred : require('./etc/api-key.json'),
+  host : "localhost",
+  port : "3000"
+});
+
+updater.activity(values['-m'],values['-t'], function(res) {
+  console.log('STATUS: ' + res.statusCode);
+  res.setEncoding('utf8');
+  res.on('data', function (chunk) {
+    console.log(chunk);
+  });
+});
+
+

bin/activity

@@ -0,0 +1,3 @@
+#!/bin/sh
+
+node ../activity.js $*

bin/cpu-daemon

@@ -0,0 +1,3 @@
+#!/bin/sh
+
+node ../cpu-daemon.js &> ../logs/cpu-daemon.log &

cpu-daemon.js

@@ -0,0 +1,15 @@
+var monitor = require('./lib/iostat'), 
+    Pusher = require('node-pusher'), 
+    fs = require('fs');
+
+process.on('uncaughtException', function (err) {
+  console.log('Caught uncaught exception: ' + err);
+});
+
+var cred = require('./etc/pusher-key.json');
+var pusher = new Pusher(cred);
+
+monitor.cpu(3)
+       .on('update',function(val){
+	        pusher.trigger('macbook','cpu',val);
+        });

etc/.gitignore

@@ -0,0 +1,2 @@
+api-key.json
+pusher-key.json

etc/hmmac-config.json

@@ -0,0 +1,5 @@
+{
+  "headerPrefix": "Hmmac",
+  "hash": "sha1",
+  "serviceLabel": "HMAC"
+}

lib/client.js

@@ -0,0 +1,68 @@
+var http = require("http"),
+    Hmmac = require('hmmac'),
+    crypto = require('crypto'),
+    Pusher = require('node-pusher');
+
+
+var default_config = {
+  host : "leightonturner.org",
+  port : 80,
+  hmmac : {
+    headerPrefix:'Hmmac', 
+    hash: 'sha1', 
+    serviceLabel: 'HMAC' 
+  },
+  pusher_cred : {},
+  api_cred : {}
+};
+
+var UpdateClient = exports.UpdateClient = function(config){
+  var cfg = {};
+  for(var m in default_config){
+    config[m] ? cfg[m] = config[m] : cfg[m] = default_config[m];
+  }
+  this.config = cfg;
+  this.hmmac = new Hmmac(this.config.hmmac);
+}
+
+UpdateClient.prototype.activity = function(msg, timeout,fn){
+
+  var data =  JSON.stringify({
+    msg : msg,
+    timeout : timeout
+  });
+
+  var hmac_cred = {
+    accessKeyId : this.config.api_cred.user,
+    accessKeySecret : this.config.api_cred.secret
+  };
+
+  var opts = {
+    host : this.config.host,
+    port : this.config.port,
+    path : "/activity",
+    method : 'PUT',
+    data : data,
+    headers : {
+      'Content-Type' : 'application/json',
+      'Content-MD5' : crypto.createHash('md5').update(data || '').digest('hex')
+    }
+  };
+
+  var signed_request = this.hmmac.signHttpRequest(hmac_cred, opts);
+  //console.log(signed_request); 
+  
+  console.log("Attempting to update server activity cache");
+  var req = http.request(signed_request,fn);
+  req.write(data);
+  req.end();
+
+  var pusher = new Pusher(this.config.pusher_cred);
+  console.log('Pushing message to connected clients');
+  pusher.trigger('macbook','activity',{
+    msg : msg,
+    timeout : timeout
+  });
+
+}
+

lib/iostat.js

@@ -0,0 +1,52 @@
+
+var EventEmitter = require('events').EventEmitter, 
+    spawn = require('child_process').spawn;
+
+var CPU = exports.CPU = function(interval){
+  
+  var self = this;
+
+  this.proc = spawn('iostat',['-w',interval]);
+
+  this.proc.stdout.setEncoding('utf8');
+  this.proc.stdout.on('data',function(data){
+    try {
+      self.update(data);
+    } catch (err) {
+      self.kill();
+      throw err;
+    }
+  });
+
+  this.proc.stderr.setEncoding('utf8');
+  this.proc.stderr.on('data', function (data) {
+    self.kill();
+    self.emit('error',data);
+  });
+
+  this.proc.on('exit', function (code) {
+    console.log('CPU monitoring process exited: ' + code);
+  });
+
+  process.on('exit',function(){
+    console.log("CPU has recognised that the parent process is exiting");
+    self.kill();
+  });
+}
+
+CPU.prototype.__proto__ = EventEmitter.prototype;
+
+CPU.prototype.update = function(data){
+  var tokens = data.split(/[ ]+/);
+  //Subtract the idle cpu percent from 100 and add two random decimal places for effect. 
+  this.emit('update',100-tokens[tokens.length-4]+(Math.floor((Math.random()*100)+1)/100));
+};
+
+CPU.prototype.kill = function(){
+  this.proc.kill();
+}
+
+var cpu = exports.cpu = function(interval){
+  return new CPU(interval);
+}
+

package.json

@@ -0,0 +1,9 @@
+{
+    "name": "ltorg-client"
+  , "version": "0.0.1"
+  , "private": true
+  , "dependencies": {
+      "node-pusher" : "*",
+      "hmmac" : "0.1.1"
+  }
+}