maclennann/hubot-chartroom

// Description:
//   Fetch a graph from graphite via render URL, upload to a
//   designated room, then share the URL to whichever room/user requested it.
//
// Dependencies:
//   "request": "~2.48.0",
//   "node-promise": "*"
//
// Configuration:
//   GRAPHITE_SERVER: the hostname/ip of your graphite server,
//   GRAPH_ROOM_ID: the API ID of a HipChat room all graphs will first be uploaded to,
//   HIPCHAT_TOKEN: an API token for your hubot can use for uploading to hipchat
//
// Commands:
//   hubot graph me <graphname or render query> (from <time code>) - fetches a graph by saved name or render api querystrings
//   hubot save grah <name> as <render query> - save a graph to hubot with the desired render api querystrings
//   hubot list graphs - show all saved graphs
//   hubot forget all graphs - deletes all saved graphs
//   hubot forget graph <name> - delete the named graph
//
// Notes:
//   If your Graphite is accessible by the outside world (e.g. HipChat can get to it
//   to generate thumbnails, and users can get to the graphs wherever they are),
//   you're probably better off using hubot-graphite.
//
//  This script's utility is in uploading the image to HipChat (which uploads it to S3)
//  so HipChat can generate a thumbnail and all users can view it.
//
// Author:
//   maclennann

/*jslint node:true*/
'use strict';

// Graphite and HipChat configuration
var GRAPHITE_SERVER = process.env.GRAPHITE_SERVER;
var GRAPH_ROOM_ID = process.env.GRAPH_ROOM_ID;
var HIPCHAT_TOKEN = process.env.HIPCHAT_TOKEN;

var Graph = require('./graphite.js'),
    brain = require('./brainOperations.js'),
    messages = require('./messages.js'),
    util = require('util');

module.exports = function (robot) {
    // Filter out graphs named <name> and save the results
    robot.respond(/forget graph (\w*)/i, function (msg) {
        var name = msg.match[1].trim(),
            graphs = brain.getGraphs(robot),
            newGraphs = graphs.filter(function (e) { return e.name !== name; });

        brain.setGraphs(robot, newGraphs);
        msg.send(messages.success);
    });

    // Empty the graphs array
    robot.respond(/forget all graphs/i, function (msg) {
        brain.setGraphs(robot, []);
        msg.send(messages.success);
    });

    // Take a render URL or a saved graph name and fetch it
    robot.respond(/graph me (\S*)( from )?([\-\d\w]*)$/i, function (msg) {
        var target = msg.match[1].trim(),
            from = msg.match[3],
            graph;

        // Figure out our intended render URL QSPs
        target = brain.getIntendedTarget(target, from, robot);

        // Construct our image repository
        graph = new Graph({
            server: target.server || GRAPHITE_SERVER,
            roomId: GRAPH_ROOM_ID,
            apiToken: HIPCHAT_TOKEN
        });

        msg.send(util.format(messages.loading, messages.success));

        // Fetch the graph from Graphite
        // Upload it to the Chart Room
        // Find the S3 URL
        // Return that to our requestor
        graph.fetch(target.target)
            .then(function (buffer) {
                return graph.upload(buffer);
            })
            .then(function (guid) {
                return graph.getLink(guid);
            })
            .then(function (link) {
                msg.send(link);
            });

    });

    // Save a render URL with a friendly name
    robot.respond(/save graph (\w*) as ([\S]*)( on )?([\-.\w]*)/i, function (msg) {
        var name = msg.match[1].trim(),
            target = msg.match[2].trim(),
            server = msg.match[4];

        if (server !== undefined) {
            server = server.trim();
        }

        if (brain.maybeGetSavedTarget(name, robot).length !== 0) {
            msg.send(util.format(messages.alreadyExists, name));
            return;
        }

        brain.saveNewTarget(name, target, robot, server);
        msg.send(util.format(messages.savedGraph, name));
    });

    // List all saved graphs
    robot.respond(/list graphs/i, function (msg) {
        var graphs = brain.getGraphs(robot),
            reply = util.format(messages.listHeader, graphs.length);

        graphs.forEach(function (e) {
            reply += util.format(messages.listItem, e.name, e.target);
        });

        msg.send(reply);
    });
};