1 files changed, 36 insertions, 0 deletions
diff --git a/atlas-plugin.js b/atlas-plugin.js
new file mode 100644
index 0000000..69f1174
--- /dev/null
+++ b/atlas-plugin.js
@@ -0,0 +1,36 @@
+const fs = require("fs")
+const path = require("path")
+const { RawSource } = require("webpack-sources")
+const { generateAtlasViews } = require("./utils.js")
+
+
+class SVGSymbolAtlasViewPlugin {
+	constructor(options = {}) {
+		this.suffix = options.suffix || ".svg"
+		this.symbolId = options.symbolId || null
+	}
+
+	apply(compiler) {
+		compiler.hooks.thisCompilation.tap("SVGSymbolAtlasViewPlugin", (compilation) => {
+			compilation.hooks.processAssets.tap(
+				{
+					name: "SVGSymbolAtlasViewPlugin",
+					stage: compiler.webpack.Compilation.PROCESS_ASSETS_STAGE_OPTIMIZE,
+				},
+				(assets) => {
+					for (const assetName in assets) {
+						if (assetName.endsWith(this.suffix)) {
+							const originalSVG = assets[assetName].source().toString()
+							const atlasSVG = generateAtlasViews(originalSVG, this.symbolId)
+
+							// Replace old SVG asset with new SVG
+							compilation.updateAsset(assetName, new RawSource(atlasSVG))
+						}
+					}
+				}
+			)
+		})
+	}
+}
+
+module.exports = SVGSymbolAtlasViewPlugin