Add only-labels input

README.md

@@ -35,6 +35,7 @@ Make sure to add the relevant inputs:
 - `repo-token` is your `${{ secrets.GITHUB_TOKEN }}`. You may encounter an error where this token does not have the necessary permissions to access an organization or teams. At EquityBee, we use PATs (Personal Access Token) instead. Create a [personal access token](https://docs.github.com/en/github/authenticating-to-github/keeping-your-account-and-data-secure/creating-a-personal-access-token) with the repo or public_repo scopes enabled, and add the token as an [encrypted secret](https://docs.github.com/en/actions/reference/encrypted-secrets#creating-encrypted-secrets-for-a-repository) for the repository or organization
 - `organization-name` is the name/slug of your Github organization (it comes right after `https://github.com/`)
 - `ignore-labels` is an optional, comma-separated list of labels to ignore
+- `only-labels` is an optional, comma-separated list of the only labels to include
 
 ## Internal use
 

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([]);
+    });
+  });
 });

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,