index.js

'use strict';

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

function WebpackArchivePlugin(options) {
	options = options || {};
	if(typeof options === 'string') {
		this.options = {output: options};
	} else {
		this.options = options;
	}
}

WebpackArchivePlugin.prototype.apply = function(compiler) {
	const options = this.options;
	compiler.plugin('done', function(compiler) {
		// Set output location
		const output = options.output?
			options.output:compiler.options.output.path;

		// Create archive streams
		let streams = [];
		let zip = true;
		let tar = true;
		if(options.format) {
			if(typeof options.format === 'string') {
				zip = (options.format === 'zip');
				tar = (options.format === 'tar');
			} else if(Array.isArray(options.format)) {
				zip = (options.format.indexOf('zip') != -1);
				tar = (options.format.indexOf('tar') != -1);
			}
		}
		if(zip) {
			const ext = options.ext || 'zip'
			let stream = archiver('zip');
			stream.pipe(fs.createWriteStream(`${output}.${ext}`));
			if(options.directory) stream.directory(options.directory, options.directoryRoot || false)
			streams.push(stream);
		}
		if(tar) {
			const ext = options.ext || 'tar.gz'
			let stream = archiver('tar', {
				gzip: true,
				gzipOptions: {
					level: 1
				}
			});
			stream.pipe(fs.createWriteStream(`${output}.${ext}`));
			if(options.directory) stream.directory(options.directory, options.directoryRoot || false)
			streams.push(stream);
		}

		// Add assets
		if(!options.directory) {
			for(let asset in compiler.assets) {
				if(compiler.assets.hasOwnProperty(asset)) {
					for(let stream of streams) {
						stream.append(fs.createReadStream(compiler.assets[asset].existsAt), {name: asset});
					}
				}
			}	
		}

		// Finalize streams
		for(let stream of streams) {
			stream.finalize();
		}
	});
}

module.exports = WebpackArchivePlugin;