funky.js

funky.js

'use strict';

import exec from 'ssh-exec';
import express from 'express';
import zt from 'zt';
import routine from 'promise-routine';
import { createConnection } from 'mysql2';
import { CronJob } from 'cron';

const app = express();
app.set('json spaces', 3);
const command = 'netstat -anp |grep 25565 |wc -l';
const servers = [
  {loc: 'New Jersey', host: 'root@**.**.**.***'},
  {loc: 'Atlanta', host: 'root@**.**.***.**'},
  {loc: 'Los Angeles', host: 'root@**.**.**.*'}
];
const db = createConnection({
  user: 'root',
  password: 'root',
  host: '127.0.0.1',
  database: 'xaph'
});

let queryServers = function() {
  let promises = [];
  for(let server of servers) {
    let promise = new Promise((resolve, reject) => {
      exec(command, server.host, (err, stdout, stderror) => {
        if(err) reject(err); if(stderror) reject(stderror);
        resolve({loc: server.loc, players: parseInt(stdout.trim())});
      });
    });
    promises.push(promise);
  }
  return promises;
}

let addDataPoint = function() {
  let date = new Date().getTime();
  Promise.all(queryServers()).then(servers => {
    for(let server of servers) {
      db.execute('INSERT INTO `servers` VALUES(?, ?, ?, ?)', [null, server.loc, server.players, date], (err, rows) => {
        if(err) return console.log(err);
        zt.log(`Successfully created new data point for ${server.loc}`);
      });
    }
  });
}

let getDataPoints = function() {
  let data = [];
  return new Promise((resolve, reject) => resolve(data));
  // return new Promise((resolve, reject) => {
  //   let json = [];
  //   for(let server of servers) {
  //     let data = [];
  //     db.execute('SELECT `server`, `timestamp`, `players` FROM `servers` WHERE `server`=?', [server.loc], (err, rows) => {
  //       rows = rows.sort((given, compared) => {
  //         return given.timestamp - compared.timestamp;
  //       });
  //       for(let row of rows) {
  //         data.push([
  //           parseInt(row.timestamp),
  //           row.players
  //         ]);
  //       }
  //       json.push({
  //         name: server.loc,
  //         data: data
  //       });
  //     });
  //   }
  //   setTimeout(() => resolve(json), 100);
  // });
}

let transact = new CronJob('* * * * * *', () => addDataPoint(), null, true, '');

//getDataPoints().then(data => console.log(data));
app.get('/', (req, res) => {
  routine(server => {
    let json = [];
    let data = [];
    return new Promise((resolve, reject) => {
      db.execute('SELECT `server`, `timestamp`, `players` FROM `servers` WHERE `server`=?', [server.loc], (err, rows) => {
        if(err) reject(err);
        rows = rows.sort((a, b) => { return a.timestamp - b.timestamp });
        for(let row of rows) {
          data.push([
            parseInt(row.timestamp),
            row.players
          ]);
        }
        json.push({
          name: server.loc,
          data: data
        })
        resolve(json[0]);
      });
    });
  }, ...servers).then((data) => {
    res.status(200).jsonp(data)
  }); 
});

app.listen(3333, () => zt.log('Listning on port 3333'));