From d9a222ae7f8bf9976d9ae0169a2e46b040dee0a5 Mon Sep 17 00:00:00 2001
From: Malik Parvez <84777619+malikparvez@users.noreply.github.com>
Date: Wed, 12 Jul 2023 01:06:27 +0530
Subject: [PATCH] CONT-1219 : fail ci for puppetlabs members if no label

---
 README.md                |  1 +
 __tests__/github.test.ts | 45 +++++++++++++++++++++++++++++++++++-----
 action.yml               |  4 ++++
 src/github.ts            |  8 +++++--
 src/main.ts              | 20 +++++++++++-------
 5 files changed, 64 insertions(+), 14 deletions(-)

diff --git a/README.md b/README.md
index 71141a3..c1c3b8a 100644
--- a/README.md
+++ b/README.md
@@ -10,6 +10,7 @@ This is a basic action that will label issues and pull request with a given labe
 | label_color | false | The color of the label. If the label already exists in the repository, this setting will have no effect. | 5319E7 |
 | org_membership | false | Contributions from users that are not members of the specified organisations will be labeled with the configured label. The value can be a single organisation or a comma-separated list of organisations. | puppetlabs |
 | logins_to_ignore | false | Contributions from the specified users will not be labeled by this action. The value can be a single login or a comma-separated list of logins. | `N/A` |
+| fail_if_member | false | Pipeline will fail, if the user is member of specified organisations and no label has been added manually. | false |
 | token | true | A token with enough privilege to view org memberships and repo content. | `N/A` |
 
 ## Security
diff --git a/__tests__/github.test.ts b/__tests__/github.test.ts
index 04d748f..de5c25b 100644
--- a/__tests__/github.test.ts
+++ b/__tests__/github.test.ts
@@ -160,13 +160,48 @@ describe('With a new github client', () => {
     )
   })
 
-  test('hasLabel returns true if the label already exists on the issue or pr', () => {
-    const result = client.hasLabel('community')
-    expect(result).toBe(true)
+  test('getLabels returns true if the label already exists on the issue or pr', () => {
+    const result = client.getLabels()
+    expect(result).toEqual(['community'])
+  })
+})
+
+describe('With a new github client without label', () => {
+  let client: GitHubClient
+
+  beforeAll(() => {
+    process.env['GITHUB_PATH'] = ''
+
+    Object.defineProperty(github, 'context', {
+      value: {
+        eventName: 'issues',
+        repo: {
+          owner: 'puppetlabs',
+          repo: 'iac'
+        },
+        issue: {
+          number: 331
+        },
+        payload: {
+          action: 'opened',
+          sender: {
+            login: 'dependabot[bot]'
+          },
+          issue: {
+            labels: [
+            ]
+          }
+        }
+      }
+    })
+
+    client = new GitHubClient('1234')
+    console.log('::stop-commands::stoptoken')
+    process.stdout.write = jest.fn()
   })
 
-  test('hasLabel returns false if the label does not exist on the issue or pr', () => {
-    const result = client.hasLabel('test')
+  test('getLabels returns false if the label does not exist on the issue or pr', () => {
+    const result = client.getLabels()
     expect(result).toBe(false)
   })
 })
diff --git a/action.yml b/action.yml
index 555c96a..ba3b1fb 100644
--- a/action.yml
+++ b/action.yml
@@ -17,6 +17,10 @@ inputs:
   logins_to_ignore:
     description: 'Contributions from the specified users will not be labeled by this action. The value can be a single login or a comma-separated list of logins.'
     required: false
+  fail_if_member:
+    description: 'Pipeline will fail, if the user is member of specified organisations and no label has been added manually.'
+    default: 'false'
+    required: false
   token:
     description: 'The GitHub token to use for authentication.'
     required: true
diff --git a/src/github.ts b/src/github.ts
index e4a8d45..d08c021 100644
--- a/src/github.ts
+++ b/src/github.ts
@@ -88,7 +88,7 @@ export class GitHubClient {
   }
 
   // Checks if the issue already has the label
-  hasLabel(name: string): boolean {
+  getLabels(): string[] | false {
     let labels!: Label[]
 
     if (github.context.eventName === 'issues') {
@@ -103,7 +103,11 @@ export class GitHubClient {
       return false
     }
 
-    return labels.some((l) => l.name === name)
+    if (labels.length > 0) {
+      return labels.map((label) => label.name)
+    }
+
+    return false
   }
 
   /* Adds the label to the issue/pull request. It will attempt to create the label
diff --git a/src/main.ts b/src/main.ts
index d4d060a..d593d10 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -11,21 +11,27 @@ export async function run(): Promise<void> {
     const labelName = core.getInput('label_name', {required: true})
     const labelColor = core.getInput('label_color', {required: true})
     const loginsToIgnore = core.getInput('logins_to_ignore', {required: false})
+    const failIfMember = core.getInput('fail_if_member', {required: false})
     const orgs = getOrgMembershipList()
     const token = core.getInput('token', {required: true})
-
     const client = new GitHubClient(token)
-
-    if (client.hasLabel(labelName)) {
-      core.info(`The '${labelName}' label is already present on this issue! 🙌`)
-      return
-    }
+    const labels = client.getLabels()
+    const missingCommunityLabel = labels && !labels.includes(labelName)
 
     if (
       (await client.checkOrgMembership(orgs)) ||
       client.isExcludedLogin(loginsToIgnore)
     ) {
-      core.info("Looks like this issue doesn't need labeling! 👍")
+      if (failIfMember === 'true' && !labels) {
+        core.setFailed('The PR is missing a label!')
+      } else {
+        core.info("Looks like this issue doesn't need labeling! 👍")
+      }
+      return
+    }
+
+    if (missingCommunityLabel) {
+      core.setFailed('Community PRs must be labelled')
       return
     }