From b0e12137231bd42b49838e6346b1f9ed8ab600c7 Mon Sep 17 00:00:00 2001
From: Berkmann18 <maxieberkmann@gmail.com>
Date: Sun, 23 Jul 2023 15:46:12 +0100
Subject: [PATCH] feat(cli): include pr based contributions

---
 package.json |  1 +
 src/cli.js   | 57 +++++++++++++++++++++++++++++++++++++++-------------
 2 files changed, 44 insertions(+), 14 deletions(-)

diff --git a/package.json b/package.json
index 0504303..92ca1c4 100644
--- a/package.json
+++ b/package.json
@@ -53,6 +53,7 @@
     "json-fixer": "^1.6.8",
     "lodash": "^4.11.2",
     "name-your-contributors": "^3.10.0",
+    "nclr": "^2.2.5",
     "node-fetch": "^2.6.0",
     "pify": "^5.0.0",
     "yargs": "^15.0.1",
diff --git a/src/cli.js b/src/cli.js
index 33d0ff2..c811f25 100755
--- a/src/cli.js
+++ b/src/cli.js
@@ -6,6 +6,7 @@ const yargs = require('yargs')
 const chalk = require('chalk')
 const inquirer = require('inquirer')
 const didYouMean = require('didyoumean')
+const {info, warn, use} = require('nclr')
 
 const init = require('./init')
 const generate = require('./generate')
@@ -191,7 +192,8 @@ async function fetchContributors(argv) {
   const {
     reviewers,
     commitAuthors,
-    issueCreators /* , prCreators */,
+    issueCreators,
+    prCreators,
   } = await getContributors(argv.projectOwner, argv.projectName, true)
   const args = {...argv, _: []}
   const contributorsToAdd = []
@@ -201,18 +203,21 @@ async function fetchContributors(argv) {
     contributorsToAdd.push({login: usr.login, contributions: ['review']})
 
     console.log(
-      `Adding ${chalk.underline('Reviewer')} ${chalk.blue(usr.login)}`,
+      `Including ${chalk.underline('Reviewer')} ${use('info', usr.login)}`,
     )
   })
 
   const guessCategories = (item, itemType, contributor) => {
     const guessedCategory = learner
       .classify(item)
-      .find(ctr => ctr && ctr !== 'null')
+      .find(ctr => ctr && ctr !== 'null' && ctr !== 'undefined')
 
     if (!guessedCategory) {
-      console.warn(
-        `Oops, I couldn't find any category for the "${item}" ${itemType}`,
+      warn(
+        `Oops, I couldn't find any category for the "${use(
+          'inp',
+          item,
+        )}" ${itemType}`,
       )
 
       return
@@ -222,22 +227,21 @@ async function fetchContributors(argv) {
       contributor.contributions.push(guessedCategory)
 
       console.log(
-        `Adding ${chalk.blue(contributor.login)} for ${chalk.underline(
+        `Including ${use('info', contributor.login)} for ${chalk.underline(
           guessedCategory,
-        )}`,
+        )}, based on "${use('inp', item)}"`,
       )
     }
   }
 
+  info('Looking at issue creators')
   issueCreators.forEach(usr => {
     const contributor = {
       login: usr.login,
       contributions: [],
     }
-    //TODO: Look at the titles field and categories based on that.
 
     usr.labels.forEach(label => guessCategories(label, 'label', contributor))
-
     usr.titles.forEach(title => guessCategories(title, 'title', contributor))
 
     const existingContributor = contributorsToAdd.find(
@@ -251,14 +255,35 @@ async function fetchContributors(argv) {
     }
   })
 
-  //TODO Look at prCreators (including its titles field) and add contributions from there
+  info('Looking at PR creators')
+  prCreators.forEach(usr => {
+    const contributor = {
+      login: usr.login,
+      contributions: [],
+    }
+
+    usr.labels.forEach(label => guessCategories(label, 'PR label', contributor))
+    usr.titles.forEach(title => guessCategories(title, 'PR title', contributor))
+
+    const existingContributor = contributorsToAdd.find(
+      ctr => ctr.login === usr.login,
+    )
+
+    if (existingContributor) {
+      existingContributor.contributions.push(...contributor.contributions)
+    } else {
+      contributorsToAdd.push(contributor)
+    }
+  })
+
+  info('Looking at commit authors')
   commitAuthors.forEach(usr => {
     const existingContributor = contributorsToAdd.find(
       ctr => ctr.login === usr.login,
     )
 
     if (existingContributor) {
-      // There's no label or commit message info so use only code for now
+      // TODO: See how the commit message could be added (this may require the full output) to not just assume it's a code contribution
       if (!existingContributor.contributions.includes('code')) {
         existingContributor.contributions.push('code')
       }
@@ -269,8 +294,12 @@ async function fetchContributors(argv) {
 
   // TODO: Roll onto other contribution categories following https://www.draw.io/#G1uL9saIuZl3rj8sOo9xsLOPByAe28qhwa
 
+  info('Finalising')
   for (const contributor of contributorsToAdd) {
-    if (!contributor.contributions.length) {
+    const isDependabotDuplicates = /dependabot(\[bot\]|-\w+)/.test(
+      contributor.login,
+    )
+    if (!contributor.contributions.length || isDependabotDuplicates) {
       console.log('Skipping', contributor.login)
 
       continue
@@ -280,7 +309,7 @@ async function fetchContributors(argv) {
     const contributions = contributor.contributions.join('/')
 
     console.log(
-      `Adding ${chalk.blue(contributor.login)} for ${chalk.underline(
+      `Adding ${use('info', contributor.login)} for ${chalk.underline(
         contributions,
       )}`,
     )
@@ -292,7 +321,7 @@ async function fetchContributors(argv) {
       await addContribution(args)
     } catch (error) {
       console.error(
-        `Adding ${chalk.blue(contributor.login)} for ${chalk.underline(
+        `Adding ${use('info', contributor.login)} for ${chalk.underline(
           contributions,
         )} Failed: ${JSON.stringify(error)}`,
       )