Allow selectively including polyfills

index.d.ts

@@ -39,6 +39,7 @@ declare namespace NodePolyfillPlugin {
 		By default, the modules that were polyfilled in Webpack 4 are mirrored over. However, if you don't want a module like console to be polyfilled you can specify alises to be skipped here.
 		*/
 		excludeAliases?: readonly Alias[]
+                includeAliases?: readonly Alias[]
 	}
 }
 

index.js

@@ -2,25 +2,30 @@
 const { ProvidePlugin } = require("webpack")
 const filterObject = require("filter-obj")
 
-const excludeObjectKeys = (object, excludeKeys) => filterObject(object, key => !excludeKeys.includes(key))
+const filterObjectKeys = (object, keys, isInclude) => filterObject(object, key => keys.includes(key) === isInclude)
 
 module.exports = class NodePolyfillPlugin {
-	constructor(options = {}) {
-		this.options = {
-			excludeAliases: [],
-			...options
+	constructor({ excludeAliases = [], includeAliases = [] } = {}) {
+		if (includeAliases.length > 0 && excludeAliases.length > 0) {
+			throw new Error("Unable to use excludeAliases and includeAliases options!")
 		}
+
+		this.options = { excludeAliases, includeAliases }
 	}
 
 	apply(compiler) {
-		compiler.options.plugins.push(new ProvidePlugin(excludeObjectKeys({
+		const isInclude = this.options.includeAliases.length > 0
+		const keys = isInclude ? this.options.includeAliases : this.options.excludeAliases
+		const filterFunc = object => filterObjectKeys(object, keys, isInclude)
+
+		compiler.options.plugins.push(new ProvidePlugin(filterFunc({
 			Buffer: [require.resolve("buffer/"), "Buffer"],
 			console: require.resolve("console-browserify"),
 			process: require.resolve("process/browser")
-		}, this.options.excludeAliases)))
+		})))
 
 		compiler.options.resolve.fallback = {
-			...excludeObjectKeys({
+			...filterFunc({
 				assert: require.resolve("assert/"),
 				buffer: require.resolve("buffer/"),
 				console: require.resolve("console-browserify"),
@@ -51,7 +56,7 @@ module.exports = class NodePolyfillPlugin {
 				util: require.resolve("util/"),
 				vm: require.resolve("vm-browserify"),
 				zlib: require.resolve("browserify-zlib")
-			}, this.options.excludeAliases),
+			}),
 			...compiler.options.resolve.fallback
 		}
 	}

readme.md

@@ -52,6 +52,23 @@ module.exports = {
 }
 ```
 
+#### includeAliases
+
+Alternatively, you can chose to only include certain aliases. In this example, only `console` wil be polyfilled.
+
+```js
+const NodePolyfillPlugin = require("node-polyfill-webpack-plugin")
+
+module.exports = {
+	// Other rules...
+	plugins: [
+		new NodePolyfillPlugin({
+			includeAliases: ["console"]
+		})
+	]
+}
+```
+
 ## Aliases
 
 ### Globals

test.js

@@ -3,7 +3,7 @@ const test = require("ava")
 const webpack = require("p-webpack")
 const NodePolyfillPlugin = require(".")
 
-test("main", async t => {
+test.serial("main", async t => {
 	await webpack({
 		entry: "./fixture",
 		output: {
@@ -22,4 +22,38 @@ test("main", async t => {
 
 	// https://github.com/browserify/console-browserify/blob/f7eefc7c908c29d2e94954e5c6c1098e8c1028b4/index.js#L63
 	t.false(fs.readFileSync("./dist/main.js").toString().includes("No such label: "))
+
+	// https://github.com/feross/buffer/blob/master/index.js#L80
+	t.true(fs.readFileSync("./dist/main.js").toString().includes("is invalid for option \"size\""))
+})
+
+test.serial("Test Include", async t => {
+	await webpack({
+		entry: "./fixture",
+		output: {
+			library: {
+				type: "commonjs-module"
+			}
+		},
+		plugins: [
+			new NodePolyfillPlugin({
+				includeAliases: ["console"]
+			})
+		]
+	})
+
+	t.is(require("./dist/main.js"), "Hello World")
+
+	// https://github.com/browserify/console-browserify/blob/f7eefc7c908c29d2e94954e5c6c1098e8c1028b4/index.js#L63
+	t.true(fs.readFileSync("./dist/main.js").toString().includes("No such label: "))
+
+	// https://github.com/feross/buffer/blob/master/index.js#L80
+	t.false(fs.readFileSync("./dist/main.js").toString().includes("is invalid for option \"size\""))
+})
+
+test("Test Error", t => {
+	t.throws(() => new NodePolyfillPlugin({
+		includeAliases: ["console"],
+		excludeAliases: ["crypto"]
+	}), { instanceOf: Error })
 })