From cf1136cb1e6e49f197c48b419894067f71006a70 Mon Sep 17 00:00:00 2001
From: toby <toby@canva.com>
Date: Sat, 8 May 2021 22:11:52 +1000
Subject: [PATCH] feat: addBuildDependency support

---
 src/index.js       | 1 +
 src/worker.js      | 5 +++++
 test/pitch.test.js | 1 +
 3 files changed, 7 insertions(+)

diff --git a/src/index.js b/src/index.js
index 59fee3e..db1bfa9 100644
--- a/src/index.js
+++ b/src/index.js
@@ -36,6 +36,7 @@ function pitch() {
     },
     (err, r) => {
       if (r) {
+        r.buildDependencies.forEach((d) => this.addBuildDependency(d));
         r.fileDependencies.forEach((d) => this.addDependency(d));
         r.contextDependencies.forEach((d) => this.addContextDependency(d));
       }
diff --git a/src/worker.js b/src/worker.js
index 0377538..73faa3f 100644
--- a/src/worker.js
+++ b/src/worker.js
@@ -119,6 +119,7 @@ const queue = asyncQueue(({ id, data }, taskCallback) => {
       });
       nextQuestionId += 1;
     };
+    const buildDependencies = [];
     loaderRunner.runLoaders(
       {
         loaders: data.loaders,
@@ -230,6 +231,9 @@ const queue = asyncQueue(({ id, data }, taskCallback) => {
             module._compile(code, filename); // eslint-disable-line no-underscore-dangle
             return module.exports;
           },
+          addBuildDependency: (filename) => {
+            buildDependencies.push(filename);
+          },
           options: {
             context: data.optionsContext,
           },
@@ -279,6 +283,7 @@ const queue = asyncQueue(({ id, data }, taskCallback) => {
           result: {
             result: convertedResult,
             cacheable,
+            buildDependencies,
             fileDependencies,
             contextDependencies,
           },
diff --git a/test/pitch.test.js b/test/pitch.test.js
index be20866..4fb1171 100644
--- a/test/pitch.test.js
+++ b/test/pitch.test.js
@@ -15,6 +15,7 @@ const runGetPoolMock = (error) => {
       isAbleToRun: () => true,
       run: jest.fn((opts, cb) => {
         cb(error, {
+          buildDependencies: [],
           fileDependencies: [],
           contextDependencies: [],
           result: {},