homu.js

var request = require("request");

function currentBuildCount(cb) {
    retrieveBuildbotBuilders(function(builders) {
        let count = 0;
        for (var builder in builders) {
            var builderData = builders[builder];
            if ('currentBuilds' in builderData) {
                count += builderData['currentBuilds'].length;
            }
        }
        cb(count);
    })
}

function anyBuildersBuilding(builders) {
    for (var builder in builders) {
        var builderData = builders[builder];
        if ('currentBuilds' in builderData && builderData['currentBuilds'].length) {
            return true;
        }
    }
    return false;
}

function checkHomuQueue(cb) {
    // First check if any buildbot builders report an ongoing build.
    retrieveBuildbotBuilders(function(builders) {
        if (anyBuildersBuilding(builders)) {
            return;
        }
        // Next see if homu thinks that any PRs are ready to be built.
        queueLength(function(numPending, numApproved) {
            if (numApproved) {
                cb(numPending + numApproved);
            }
        });
    })
}

function queueLength(cb) {
    retrieveHomuQueue(function(body) {
        var numPending = (body.match(/<td class="pending">/g) || []).length;
        var numApproved = (body.match(/<td class="approved">/g) || []).length;
        cb(numPending, numApproved);
    });
}

function retrieveHomuQueue(cb) {
    request("http://build.servo.org/homu/queue/servo", function(err, response, body) {
        if (!err && response.statusCode >= 200 && response.statusCode < 300) {
            cb(body);
        }
    });
}

function retrieveBuildbotBuilders(cb) {
    request("http://build.servo.org/json/builders/", function(err, response, body) {
        if (!err && response.statusCode >= 200 && response.statusCode < 300) {
            cb(JSON.parse(body));
        }
    });
}

function retrieveSlaves(cb) {
    request("http://build.servo.org/json/slaves/", function(err, response, body) {
        if (!err && response.statusCode >= 200 && response.statusCode < 300) {
            cb(JSON.parse(body));
        }
    });
}

exports.checkHomuQueue = checkHomuQueue;
exports.queueLength = queueLength;
exports._anyBuildersBuilding = anyBuildersBuilding;
exports.retrieveSlaves = retrieveSlaves;
exports.retrieveBuildbotBuilders = retrieveBuildbotBuilders;
exports.currentBuildCount = currentBuildCount;