feat: add optional only-labels input

equitybee · May 28, 2024 · 1bcf355 · 1bcf355
1 parent 732147c
commit 1bcf355
Show file tree

Hide file tree

Showing 4 changed files with 36 additions and 2 deletions.
diff --git a/dist/index.js b/dist/index.js
diff --git a/dist/index.js.map b/dist/index.js.map
diff --git a/src/__tests__/octokit-queries.test.ts b/src/__tests__/octokit-queries.test.ts
@@ -58,4 +58,30 @@ describe('getTeamSlugsForAuthor', () => {
       expect(teamSlugs).toStrictEqual(['team-active']);
     });
   });
+
+  describe('with only teams', () => {
+    beforeEach(async () => {
+      teamSlugs = await getTeamSlugsForAuthor(octokit, 'ORG', 'USER', [], ['team-active']);
+    });
+
+    it('should return filtered teams', () => {
+      expect(teamSlugs).toStrictEqual(['team-active']);
+    });
+  });
+
+  describe('with only/ignored teams', () => {
+    beforeEach(async () => {
+      teamSlugs = await getTeamSlugsForAuthor(
+        octokit,
+        'ORG',
+        'USER',
+        ['team-active'],
+        ['team-active'],
+      );
+    });
+
+    it('applies ignored teams', () => {
+      expect(teamSlugs).toStrictEqual([]);
+    });
+  });
 });
diff --git a/src/octokit-queries.ts b/src/octokit-queries.ts
@@ -6,6 +6,7 @@ export const getTeamSlugsForAuthor = async (
   org: string,
   username: string,
   ignoreSlugs: string[] = [],
+  onlySlugs: string[] = [],
 ): Promise<string[]> => {
   const { data: allTeams } = await octokit.rest.teams.list({
     org,
@@ -18,6 +19,10 @@ export const getTeamSlugsForAuthor = async (
       continue;
     }
 
+    if (onlySlugs.length > 0 && !onlySlugs.includes(slug)) {
+      continue;
+    }
+
     try {
       const { data: membership } = await octokit.rest.teams.getMembershipForUserInOrg({
         org,