nv-pack/lib/mw/images.js

Build tool for packing static distribution packages. Part of the node-front project.

/**
 * Module dependencies.
 */
var utils = require('../utils')
  , async = require('async')
  , path = require('path')
  , _ = require('lodash')
  , fs = require('fs');

// dim tick symbol
var tick = '\033[90m✓\033[39m';
// short cut for the utils.log
var log = utils.log;
// curent working directory
var cwd = process.cwd();
var cfg;

/**
 * Loading config
 */
try {  
  if (program.config) {
    cfg = require(program.config);
  } else {
    cfg = require(path.join(cwd, '/config'));
  }
} catch (e) {
  console.error(" error: cannot find config file");
  process.exit(1);
}

/*
 *
 */
module.exports = function(scope, cb) {
  log('┬ processing images...');

  async.waterfall([
    function(cb) {
      gatherImages(cfg.STATIC_ROOT, scope.tmpDir, function(err, imagesMap) {
        if (err) return cb(err);
        log('├── gathered all images in this project');
        scope.imagesMap = imagesMap;
        cb();
      });
    },
    function(cb) {
      applyImagesMap(scope.assetsMap, scope.imagesMap, function(err) {
        if (err) return cb(err);
        log('├── applyed gathered images to assets');
        cb();
      });
    },
    function(cb) {
      var filePath = cfg.IMAGES_MAP_FILE || path.join(scope.tmpDir, 'images.json');
      
      fs.writeFile(filePath, JSON.stringify(scope.imagesMap), function(err) {
        if (err) return cb(err);
        log('└── created images map file \t\t\t \033[90m%s\033[39m', filePath);
        cb();
      });
    }
  ], cb);
};

/*
 *
 */
function gatherImages(dirPath, outDirPath, cb) {
  findImages(dirPath, function(err, list) {
    if (err) return cb(err);

    var map = {};

    async.forEach(list, function(imgPath, cb) {
      utils.markFile(imgPath, outDirPath, function(err, newPath) {
        if (err) return cb(err);

        var orgImgUrl = imgPath.replace(dirPath + '/', '');
        var newImgUrl = newPath.replace(outDirPath + '/', '');
        map[orgImgUrl] = newImgUrl;
        cb();
      });
    }, function(err) {
      if (err) return cb(err);
      cb(null, map);
    });
  });
}

/*
 *
 */
function applyImagesMap(assetsMap, imagesMap, cb) {
  utils.eachAsset(assetsMap, function(filePath, p, cb) {
    utils.readFile(filePath, function(err, content) {
      if (err) return cb(err);

      _.each(imagesMap, function(newUrl, orgUrl) {
        content = content.split('/' + orgUrl).join(newUrl);
        content = content.split(orgUrl).join(newUrl);
      });
      
      utils.writeFile(filePath, content, cb);
    });
  }, cb);
}

/*
 *
 */
function findImages(dirPath, cb) {
  fs.readdir(dirPath, function(err, list) {
    if (err) return cb(err);

    var images = _.filter(list, function(fileName) {
      return ~fileName.indexOf('.jpg')
          || ~fileName.indexOf('.jpeg')
          || ~fileName.indexOf('.webp') //soon
          || ~fileName.indexOf('.png')
          || ~fileName.indexOf('.gif')
          || ~fileName.indexOf('.svg');
    });

    images = _.map(images, function(fileName) {
      return path.join(dirPath, fileName);
    });

    utils.filterDirs(list, dirPath, function(dirs) {
      dirs = _.map(dirs, function(dirName) {
        return path.join(dirPath, dirName);
      });

      async.concat(dirs, findImages, function(err, other) {
        if (err) return cb(err);
        cb(null, images.concat(other));
      });
    });
  });
}