From f443f2bbd650ed1a7b1a171f27c191672c957678 Mon Sep 17 00:00:00 2001
From: Randell Dawson <rdawson@onepathtech.com>
Date: Thu, 29 Nov 2018 12:10:57 -0800
Subject: [PATCH] refac: moved labeler.js to pr-tasks

---
 pr-tasks/index.js   |  3 ++-
 pr-tasks/labeler.js | 39 +++++++++++++++++++++++++++++++++++++++
 sweeper.js          |  2 +-
 3 files changed, 42 insertions(+), 2 deletions(-)
 create mode 100644 pr-tasks/labeler.js

diff --git a/pr-tasks/index.js b/pr-tasks/index.js
index 0b7bd6e166..1ccfb250c3 100644
--- a/pr-tasks/index.js
+++ b/pr-tasks/index.js
@@ -1,5 +1,6 @@
 const { addComment } = require('./add-comment');
 const { addLabels } = require('./add-labels');
 const { closeOpen } = require('./close-open');
+const { labeler } = require('./labeler');
 
-module.exports = { addComment, addLabels, closeOpen };
+module.exports = { addComment, addLabels, closeOpen, labeler };
diff --git a/pr-tasks/labeler.js b/pr-tasks/labeler.js
new file mode 100644
index 0000000000..b5e1ea01c8
--- /dev/null
+++ b/pr-tasks/labeler.js
@@ -0,0 +1,39 @@
+const { validLabels } = require('../validation');
+const { addLabels } = require('./add-labels');
+const { rateLimiter } = require('../utils');
+
+const labeler = async (number, prFiles, currentLabels, guideFolderErrorsComment) => {
+  const labelsToAdd = {}; // holds potential labels to add based on file path
+  if (guideFolderErrorsComment) {
+    labelsToAdd['status: needs update'] = 1;
+  }
+  const existingLabels = currentLabels.map(({ name }) => name);
+  prFiles.forEach(({ filename }) => {
+    /* remove '/challenges' from filename so language variable hold the language */
+    const filenameReplacement = filename.replace(/^curriculum\/challenges\//, 'curriculum\/');
+    const regex = /^(docs|curriculum|guide)(?:\/)(arabic|chinese|portuguese|russian|spanish)?\/?/
+    const [ _, articleType, language ] = filenameReplacement.match(regex) || []; // need an array to pass to labelsAdder
+
+    if (articleType && validLabels[articleType]) {
+      labelsToAdd[validLabels[articleType]] = 1
+    }
+    if (language && validLabels[language]) {
+      labelsToAdd[validLabels[language]] = 1
+    }
+    if (articleType === 'curriculum') {
+      labelsToAdd['status: need to test locally'] = 1;
+    }
+  })
+
+  /* this next section only adds needed labels which are NOT currently on the PR. */
+  const newLabels = Object.keys(labelsToAdd).filter(label => !existingLabels.includes(label));
+  if (newLabels.length) {
+    if (process.env.PRODUCTION_RUN === 'true') {
+      addLabels(number, newLabels);
+    }
+    await rateLimiter(+process.env.RATELIMIT_INTERVAL | 1500);
+  }
+  return newLabels;
+};
+
+module.exports = { labeler };
diff --git a/sweeper.js b/sweeper.js
index 0ba6f9569a..0bd2e8e3d1 100644
--- a/sweeper.js
+++ b/sweeper.js
@@ -16,7 +16,7 @@ const octokit = require('@octokit/rest')(octokitConfig);
 const { getPRs, getUserInput } = require('./get-prs');
 const { guideFolderChecks } = require('./validation');
 const { savePrData, ProcessingLog, rateLimiter } = require('./utils');
-const { labeler } = require('./labeler');
+const { labeler } = require('./pr-tasks');
 
 octokit.authenticate(octokitAuth);