gear/lib/registry.js

Gear.js - Build System for Node.js and the Browser

/*
 * Copyright (c) 2011-2012, Yahoo! Inc.  All rights reserved.
 * Copyrights licensed under the New BSD License.
 * See the accompanying LICENSE file for terms.
 */
var fs = require('fs'),
    path = require('path');

/*
 * Registry - Container for available tasks.
 */
var Registry = exports.Registry = function Registry(options) {
    var self = this;
    this._tasks = {};

    // Load default tasks
    this.load(process.browser ? {tasks: require('./tasks')} : {dirname: __dirname + '/tasks'});

    if (options) {
        this.load(options);
    }

    Object.defineProperty(this, 'tasks', {get: function() {
        return Object.keys(self._tasks);
    }});
};

Registry.prototype = {
    /*
     * Load tasks from NPM, directory, file, or object.
     */
    load: function(options) {
        options = options || {};

        var module = options.module,
            dirname = options.dirname,
            filename = options.filename,
            tasks = options.tasks;

        if (module) {
            if (!this._loadModule(module)) {
                throw new Error('Module ' + module + ' doesn\'t exist');
            }
        }

        if (dirname) {
            if (!this._loadDir(dirname)) {
                throw new Error('Directory ' + dirname + ' doesn\'t exist');
            }
        }

        if (filename) {
            if (!this._loadFile(filename)) {
                throw new Error('File ' + filename + ' doesn\'t exist');
            }
        }

        if (tasks) {
            if (!this._loadTasks(tasks)) {
                throw new Error('Failed to load tasks');
            }
        }

        return this;
    },

    _loadModule: function(name) {
        if (require) {
            try {
                return this._loadTasks(require(name));
            } catch (err) {}
        }

        return this._loadDir(path.resolve('node_modules', name, 'lib'));
    },

    _loadDir: function(dirname) {
        if (!fs.existsSync(dirname)) {return false;}

        var files = fs.readdirSync(dirname),
            self = this;

        files.forEach(function(filename) {
            var name = path.join(dirname, filename);
            if (path.extname(name) !== '.js') {return;}
            self._loadFile(name);
        });

        return true;
    },

    _loadFile: function(filename) {
        if (!fs.existsSync(filename)) {return false;}

        return this._loadTasks(require(filename));
    },

    _loadTasks: function(tasks) {
        var name;

        for (name in tasks) {
            this._tasks[name] = tasks[name];
        }

        return true;
    },

    task: function(name) {
        if (!(name in this._tasks)) {
            throw new Error('Task ' + name + ' doesn\'t exist');
        }

        return this._tasks[name];
    }
};