index.js

const fs = require('fs');
const path = require('path');

const loaderUtils = require('loader-utils');
// const webpack = require('webpack');
const MT = require('mark-twain');
const R = require('ramda');
const ejs = require('ejs');

const isCode = R.compose(R.contains(R.__, ['js', 'jsx', 'javascript']), R.path(['props', 'lang']));
const isStyle = R.whereEq({ type: 'code', props: { lang: 'css' } });
const isHtml = R.whereEq({ type: 'code', props: { lang: 'html' } });
const getChildren = R.compose(R.prop('children'), R.defaultTo({}));

const util = require('atool-doc-util');

function calculateHtmlPath(cwd, source) {
  const selfPath = path.relative(cwd, source);
  return path.join(path.dirname(selfPath), `${path.basename(selfPath, path.extname(selfPath))}.html`);
}

module.exports = function(content) {

  this.cacheable && this.cacheable();

  const options = this.options;
  const resourcePath = this.resourcePath;
  const resource = new util.Resource(options.cwd, options.demoSource, resourcePath);

  const query = loaderUtils.parseQuery(this.query);

  const fileContentTree = MT(content).content;
  const meta = MT(content).meta;

  const code = getChildren(fileContentTree.find(isCode));
  const style = getChildren(fileContentTree.find(isStyle));
  const html = getChildren(fileContentTree.find(isHtml));

  const tpl = query.template;
  this.addDependency(tpl);

  // const common = options.plugins.some(function(i) {
  //   return i instanceof webpack.optimize.CommonsChunkPlugin;
  // });

  const scripts = [
    path.relative(resourcePath, path.join(resource.demoPath, 'common.js')),
    `${resource.name}.js`,
  ];

  const link = {};
  link['Index'] = path.relative('../', path.relative(resource.path, './index'));
  Object.keys(options.entry).forEach(function(key) {
    link[path.relative(options.demoSource, key)] = path.relative('../', path.relative(resource.path, key));
  });

  const result = ejs.render(fs.readFileSync(tpl, 'utf-8'), {
    file: {
      meta: meta,
      link: link,
      title: meta.title || resource.relativeToCwd + resource.ext,
      resource: resource,
      // script: common ? scripts : scripts.slice(1),
      script: scripts,
      html: html,
      style: style,
      desc: util.marked(fileContentTree),
    },
  });

  this.emitFile(calculateHtmlPath(options.cwd, resourcePath), result);

  return code;
}