Merge branch 'master' into disable-cache-by-default

Author: eduardoboucas

README.md

@@ -26,7 +26,7 @@
 
 Sourcebit connects to multiple data sources, for example data in a headless CMS like Contentful or Sanity, to a destination, such as a JAMstack site built using Jekyll or Hugo. The easiest way to understand how this works is to see it in action via our [video demo on YouTube](https://www.youtube.com/watch?v=BrZbWMXB4TQ).
 
-[![](https://img.youtube.com/vi/fPvfeP1lzTY/0.jpg)](https://www.youtube.com/watch?v=fPvfeP1lzTY)
+[![](http://i3.ytimg.com/vi/fPvfeP1lzTY/maxresdefault.jpg)](https://www.youtube.com/watch?v=fPvfeP1lzTY)
 
 Sourcebit works through the use of two types of plugins:
 

lib/sourcebit.js

@@ -200,6 +200,7 @@ class Sourcebit {
       objects: []
     };
     const contextSnapshot = cloneDeep(this.context);
+    const onTransformEndCallbacks = [];
     const queue = this.pluginBlocks.reduce((queue, pluginBlock, index) => {
       // If the plugin hasn't been bootstrapped, we don't want to run its
       // transform method just yet.
@@ -215,6 +216,18 @@ class Sourcebit {
           return data;
         }
 
+        if (typeof plugin.onTransformEnd === "function") {
+          onTransformEndCallbacks.push({
+            args: {
+              debug: this.getDebugMethodForPlugin(pluginName),
+              getPluginContext: () => contextSnapshot[pluginName] || {},
+              log: this.logFromPlugin.bind(this),
+              options: this.parsePluginOptions(plugin, pluginBlock.options)
+            },
+            callback: plugin.onTransformEnd
+          });
+        }
+
         return plugin.transform({
           data,
           debug: this.getDebugMethodForPlugin(pluginName),
@@ -244,6 +257,10 @@ class Sourcebit {
         await this.writeFiles(data.files);
       }
 
+      onTransformEndCallbacks.forEach(({ args, callback }) => {
+        callback({ ...args, data });
+      });
+
       if (typeof this.onTransform === "function") {
         this.onTransform(null, data);
       }

lib/sourcebit.test.js

@@ -909,6 +909,71 @@ describe("`transform()`", () => {
     expect(error).toBeInstanceOf(Error);
     expect(callbackData).not.toBeDefined();
   });
+
+  test("calls the `onTransformEndFn` callback of each plugin, if defined, when `transform` has finished running", async () => {
+    const onTransformEndFn = jest.fn();
+    const plugins = [
+      {
+        module: {
+          name: "sourcebit-test1",
+          onTransformEnd: onTransformEndFn,
+          transform: async ({ data }) => {
+            return {
+              ...data,
+              elements: [{ name: "Lead" }]
+            };
+          }
+        }
+      },
+      {
+        module: {
+          name: "sourcebit-test2",
+          onTransformEnd: onTransformEndFn,
+          transform: ({ data }) => {
+            return {
+              ...data,
+              elements: data.elements.concat({
+                name: "Rhenium"
+              })
+            };
+          }
+        }
+      },
+      {
+        module: {
+          name: "sourcebit-test3",
+          onTransformEnd: onTransformEndFn,
+          transform: ({ data }) => {
+            return {
+              ...data,
+              elements: data.elements.concat({
+                name: "Osmium"
+              })
+            };
+          }
+        }
+      }
+    ];
+    const callback = jest.fn();
+    const sourcebit = new Sourcebit();
+
+    sourcebit.onTransform = callback;
+    sourcebit.loadPlugins(plugins);
+
+    await sourcebit.bootstrapAll();
+
+    const data = await sourcebit.transform();
+
+    expect(onTransformEndFn).toHaveBeenCalledTimes(3);
+
+    onTransformEndFn.mock.calls.forEach(call => {
+      expect(call[0].debug).toBeInstanceOf(Function);
+      expect(call[0].getPluginContext).toBeInstanceOf(Function);
+      expect(call[0].log).toBeInstanceOf(Function);
+      expect(call[0].options).toEqual({});
+      expect(call[0].data).toEqual(data);
+    });
+  });
 });
 
 describe("writing files", () => {