index.js

'use strict';

var fs          = require('fs');
var merge       = require('merge');
var mergeTrees  = require('broccoli-merge-trees');
var flatiron    = require('broccoli-flatiron');
var Funnel      = require('broccoli-funnel');
var SVGOptmizer = require('./svg-optimizer');

module.exports = {
  name: require('./package').name,

  included: function(app) {
    if (app.app) {
      app = app.app;
    }
    this.app = app;
  },

  options: function() {
    return merge(true, {}, {
      paths:   ['public'],
      optimize: { /* svgo defaults */ }
    }, (this.app && this.app.options && this.app.options.svg) || {});
  },

  svgPaths: function() {
    if (this.isDevelopingAddon()) {
      return ['tests/dummy/public'];
    }
    return this.options().paths;
  },

  optimizeSVGs: function(tree) {
    var config = this.options().optimize;
    if (!config) {
      return tree;
    }

    return new SVGOptmizer([tree], {svgoConfig: config});
  },

  treeForApp: function(tree) {
    var existingPaths = this.svgPaths().filter(function(path) {
      return fs.existsSync(path);
    });

    var svgTrees = existingPaths.map(function(path) {
      return new Funnel(path, {
        include: [new RegExp(/\.svg$/)]
      });
    });

    var svgs = mergeTrees(svgTrees, {
      overwrite: true
    });

    var optimized = this.optimizeSVGs(svgs);

    var manifest = flatiron(optimized, {
      outputFile: 'svgs.js',
      trimExtensions: true
    });

    return mergeTrees([tree, manifest]);
  }
};