Allow Replacement plugin to operate as an output plugin

packages/replace/src/index.js

@@ -47,39 +47,49 @@ export default function replace(options = {}) {
     ? new RegExp(`${escape(delimiters[0])}(${keys.join('|')})${escape(delimiters[1])}`, 'g')
     : new RegExp(`\\b(${keys.join('|')})\\b`, 'g');
 
-  return {
-    name: 'replace',
+  function executeReplacement(code, id) {
+    const magicString = new MagicString(code);
 
-    transform(code, id) {
-      if (!keys.length) return null;
-      if (!filter(id)) return null;
+    let hasReplacements = false;
+    let match;
+    let start;
+    let end;
+    let replacement;
 
-      const magicString = new MagicString(code);
+    // eslint-disable-next-line no-cond-assign
+    while ((match = pattern.exec(code))) {
+      hasReplacements = true;
 
-      let hasReplacements = false;
-      let match;
-      let start;
-      let end;
-      let replacement;
+      start = match.index;
+      end = start + match[0].length;
+      replacement = String(functionValues[match[1]](id));
 
-      // eslint-disable-next-line no-cond-assign
-      while ((match = pattern.exec(code))) {
-        hasReplacements = true;
+      magicString.overwrite(start, end, replacement);
+    }
 
-        start = match.index;
-        end = start + match[0].length;
-        replacement = String(functionValues[match[1]](id));
+    if (!hasReplacements) return null;
 
-        magicString.overwrite(start, end, replacement);
-      }
+    const result = { code: magicString.toString() };
+    if (options.sourceMap !== false && options.sourcemap !== false)
+      result.map = magicString.generateMap({ hires: true });
 
-      if (!hasReplacements) return null;
+    return result;
+  }
 
-      const result = { code: magicString.toString() };
-      if (options.sourceMap !== false && options.sourcemap !== false)
-        result.map = magicString.generateMap({ hires: true });
+  return {
+    name: 'replace',
+
+    renderChunk(code, chunk) {
+      const id = chunk.fileName;
+      if (!keys.length) return null;
+      if (!filter(id)) return null;
+      return executeReplacement(code, id);
+    },
 
-      return result;
+    transform(code, id) {
+      if (!keys.length) return null;
+      if (!filter(id)) return null;
+      return executeReplacement(code, id);
     }
   };
 }

packages/replace/test/misc.js

@@ -111,3 +111,37 @@ test('does not generate sourcemaps if disabled', async (t) => {
   await bundle.generate({ format: 'es', sourcemap: true });
   t.truthy(warned);
 });
+
+test('can be configured with output plugins', async (t) => {
+  const bundle = await rollup({
+    input: 'main.js',
+    plugins: [
+      {
+        resolveId(id) {
+          return id;
+        },
+        load(importee) {
+          if (importee === 'main.js') {
+            return 'log("environment", process.env.NODE_ENV);';
+          }
+        }
+      }
+    ]
+  });
+
+  const { code, map } = getOutputFromGenerated(
+    await bundle.generate({
+      format: 'es',
+      sourcemap: true,
+      plugins: [
+        replace({
+          'process.env.NODE_ENV': JSON.stringify('production'),
+          delimiters: ['', '']
+        })
+      ]
+    })
+  );
+
+  t.is(code.trim(), 'log("environment", "production");');
+  t.truthy(map);
+});