myblog/node_modules/hexo/lib/extend/generator.js

'use strict';

const Promise = require('bluebird');

function Generator() {
  this.id = 0;
  this.store = {};
}

Generator.prototype.list = function() {
  return this.store;
};

Generator.prototype.get = function(name) {
  return this.store[name];
};

Generator.prototype.register = function(name, fn) {
  if (!fn) {
    if (typeof name === 'function') {
      fn = name;
      name = `generator-${this.id++}`;
    } else {
      throw new TypeError('fn must be a function');
    }
  }

  if (fn.length > 1) fn = Promise.promisify(fn);
  this.store[name] = Promise.method(fn);
};

module.exports = Generator;