hexo/lib/plugins/generator/asset.js

'use strict';

const fs = require('hexo-fs');
const Promise = require('bluebird');
const pathFn = require('path');
const chalk = require('chalk');

function assetGenerator(locals) {
  const self = this;

  function process(name) {
    return Promise.filter(self.model(name).toArray(), asset => fs.exists(asset.source).then(exist => {
      if (exist) return exist;
      return asset.remove().thenReturn(exist);
    })).map(asset => {
      const source = asset.source;
      let path = asset.path;
      const data = {
        modified: asset.modified
      };

      if (asset.renderable && self.render.isRenderable(path)) {
        // Replace extension name if the asset is renderable
        const extname = pathFn.extname(path);
        const filename = path.substring(0, path.length - extname.length);

        path = `${filename}.${self.render.getOutput(path)}`;

        data.data = () => self.render.render({
          path: source,
          toString: true
        }).catch(err => {
          self.log.error({err}, 'Asset render failed: %s', chalk.magenta(path));
        });
      } else {
        data.data = () => fs.createReadStream(source);
      }

      return {
        path,
        data
      };
    });
  }

  return Promise.all([
    process('Asset'),
    process('PostAsset')
  ]).then(data => Array.prototype.concat.apply([], data));
}

module.exports = assetGenerator;