angular-cn/tools/broccoli/broccoli-replace.ts

import fs = require('fs');
import fse = require('fs-extra');
import path = require('path');
import {wrapDiffingPlugin, DiffingBroccoliPlugin, DiffResult} from './diffing-broccoli-plugin';

var minimatch = require('minimatch');
var FILE_ENCODING = {encoding: 'utf-8'};

/**
 * Intercepts each changed file and replaces its contents with
 * the associated changes.
 */
class DiffingReplace implements DiffingBroccoliPlugin {
  // TODO: define an interface for options
  constructor(private inputPath: string, private cachePath: string, private options: any) {}

  rebuild(treeDiff: DiffResult) {
    var patterns = this.options.patterns;
    var files = this.options.files;

    treeDiff.addedPaths.concat(treeDiff.changedPaths).forEach((changedFilePath) => {
      var sourceFilePath = path.join(this.inputPath, changedFilePath);
      var destFilePath = path.join(this.cachePath, changedFilePath);
      var destDirPath = path.dirname(destFilePath);

      if (!fs.existsSync(destDirPath)) {
        fse.mkdirpSync(destDirPath);
      }

      var fileMatches = files.some((filePath: string) => minimatch(changedFilePath, filePath));
      if (fileMatches) {
        var content = fs.readFileSync(sourceFilePath, FILE_ENCODING);
        patterns.forEach((pattern: any) => {
          var replacement = pattern.replacement;
          if (typeof replacement === 'function') {
            replacement = function(content: string) {
              return pattern.replacement(content, changedFilePath);
            };
          }
          content = content.replace(pattern.match, replacement);
        });
        fs.writeFileSync(destFilePath, content, FILE_ENCODING);
      } else if (!fs.existsSync(destFilePath)) {
        try {
          fs.symlinkSync(sourceFilePath, destFilePath);
        } catch (e) {
          fs.writeFileSync(destFilePath, fs.readFileSync(sourceFilePath));
        }
      }
    });

    treeDiff.removedPaths.forEach((removedFilePath) => {
      var destFilePath = path.join(this.cachePath, removedFilePath);
      fs.unlinkSync(destFilePath);
    });
  }
}

export default wrapDiffingPlugin(DiffingReplace);