githubmirror/hexo

lib/extend/renderer.js

63 строки1 KB

segayuu

Destructuring assignment of functions in path built-in module

25 окт 2018, 05:27

25 окт 2018, 05:2752eec7e

'use strict';

const { extname } = require('path');
const Promise = require('bluebird');

function getExtname(str) {
  if (typeof str !== 'string') return '';

  const ext = extname(str) || str;
  return ext[0] === '.' ? ext.slice(1) : ext;
}

function Renderer() {
  this.store = {};
  this.storeSync = {};
}

Renderer.prototype.list = function(sync) {
  return sync ? this.storeSync : this.store;
};

Renderer.prototype.get = function(name, sync) {
  const store = this[sync ? 'storeSync' : 'store'];

  return store[getExtname(name)] || store[name];
};

Renderer.prototype.isRenderable = function(path) {
  return Boolean(this.get(path));
};

Renderer.prototype.isRenderableSync = function(path) {
  return Boolean(this.get(path, true));
};

Renderer.prototype.getOutput = function(path) {
  const renderer = this.get(path);
  return renderer ? renderer.output : '';
};

Renderer.prototype.register = function(name, output, fn, sync) {
  if (!name) throw new TypeError('name is required');
  if (!output) throw new TypeError('output is required');
  if (typeof fn !== 'function') throw new TypeError('fn must be a function');

  name = getExtname(name);
  output = getExtname(output);

  if (sync) {
    this.storeSync[name] = fn;
    this.storeSync[name].output = output;

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

  this.store[name].output = output;
  this.store[name].compile = fn.compile;
};

module.exports = Renderer;