index.js

'use strict';

const fs = require('fs');
const path = require('path');
const {SourceNode, SourceMapConsumer} = require('source-map');
const loaderUtils = require('loader-utils');
const makeIdentitySourceMap = require('./makeIdentitySourceMap');
const patch = require('./patch');

let tagCommonJSExportsSource = null;

function transform(source, map) {
  const callback = this.async();
  const resourcePath = this.resourcePath;

  if (process.env.NODE_ENV === 'production') {
    return callback(null, source, map);
  }
  if (source && source.types && source.types.IfStatement) {
    throw new Error(
      'React Hot Loader: You are erroneously trying to use a Webpack loader ' +
      'as a Babel plugin. Replace "react-hot-loader/webpack" with ' +
      '"react-hot-loader/babel" in the "plugins" section of your .babelrc file. ' +
      'While we recommend the above, if you prefer not to use Babel, ' +
      'you may remove "react-hot-loader/webpack" from the "plugins" section of ' +
      'your .babelrc file altogether, and instead add "react-hot-loader/webpack" ' +
      'to the "loaders" section of your Webpack configuration.',
    );
  }

  if (this.cacheable) {
    this.cacheable();
  }

  const options = Object.assign({withPatch: true}, loaderUtils.getOptions(this));
  if (options.withPatch) {
    source = patch(source);
  }
  if (source.indexOf('reactHotLoader.register') > 0) {
    return callback(null, source, map);
  }
  // This is a Webpack loader, but the user put it in the Babel config.

  // Read the helper once.
  if (!tagCommonJSExportsSource) {
    tagCommonJSExportsSource = fs
      .readFileSync(path.join(__dirname, 'webpackTagCommonJSExports.js'), 'utf8')
      // Babel inserts these.
      // Ideally we'd opt out for one file but this is simpler.
      .replace(/['"]use strict['"];/, '')
      // eslint comments don't need to end up in the output
      .replace(/\/\* (.*) \*\//, '')
      .replace(/\/\/ eslint-disable-line .*\n/g, '\n')
      .replace(/\/\* global.*\*\//, '')
      .split(/\n\s*/)
      .join(' ');
  }

  // Parameterize the helper with the current filename.
  const separator = '\n';
  const appendText = tagCommonJSExportsSource.replace(/__FILENAME__/g, JSON.stringify(resourcePath));

  if (this.sourceMap === false) {
    return callback(null, [source, appendText].join(separator));
  }

  if (!map) {
    map = makeIdentitySourceMap(source, resourcePath); // eslint-disable-line no-param-reassign
  }
  const sourceMapConsumer = new SourceMapConsumer(map);

  const onSourceMapReady = consumedMap => {
    const node = new SourceNode(null, null, null, [
      SourceNode.fromStringWithSourceMap(source, consumedMap),
      new SourceNode(null, null, resourcePath, appendText),
    ]).join(separator);
    const result = node.toStringWithSourceMap();
    callback(null, result.code, result.map.toJSON() || undefined);
  };

  if (sourceMapConsumer.then) {
    sourceMapConsumer.then(onSourceMapReady);
  } else {
    onSourceMapReady(sourceMapConsumer);
  }
}

transform.patch = patch;

module.exports = transform;