fix: replace github search api with graphql in success lifecycle method

lib/success.js

@@ -1,4 +1,4 @@
-import { isNil, uniqBy, template, flatten, isEmpty } from "lodash-es";
+import { isNil, uniqBy, template, flatten, isEmpty, toArray } from "lodash-es";
 import pFilter from "p-filter";
 import AggregateError from "aggregate-error";
 import issueParser from "issue-parser";
@@ -65,21 +65,62 @@ export default async function success(pluginConfig, context, { Octokit }) {
     const releaseInfos = releases.filter((release) => Boolean(release.name));
     const shas = commits.map(({ hash }) => hash);
 
-    const searchQueries = getSearchQueries(
-      `repo:${owner}/${repo}+type:pr+is:merged`,
-      shas,
-    ).map(
-      async (q) =>
-        (await octokit.request("GET /search/issues", { q })).data.items,
-    );
+    // MOCK ONLY
+    // const mockReponse = {
+    //   "data": {
+    //     "repository": {
+    //       "com1": {
+    //         "associatedPullRequests": {
+    //           "nodes": [
+    //             {
+    //               "number": 2
+    //             }
+    //           ]
+    //         }
+    //       },
+    //       "com2": {
+    //         "associatedPullRequests": {
+    //           "nodes": [
+    //             {
+    //               "number": 1
+    //             }
+    //           ]
+    //         }
+    //       }
+    //     }
+    //   }
+    // };
+    // console.log(somedata)
+
+    const { data } = await octokit.graphql(loadAssociatedPRsQuery(shas), { owner, repo });
+
+    const associatedPRs = toArray(mockReponse.data.repository).map((item) => item.associatedPullRequests.nodes);
+    // console.log("norm: ", mockPRNodes);
+
+    // const searchQueries = getSearchQueries(
+    //   `repo:${owner}/${repo}+type:pr+is:merged`,
+    //   shas,
+    // ).map(
+    //   async (q) =>
+    //     (await octokit.request("GET /search/issues", { q })).data.items,
+    // );
+
+    // const searchQueriesResults = await Promise.all(searchQueries);
+    // console.log("searchQueriesResults: ", searchQueriesResults);
+    // const uniqueSearchQueriesResults = uniqBy(
+    //   flatten(searchQueriesResults),
+    //   "number",
+    // );
+    // console.log("uniqueSearchQueriesResults: ", uniqueSearchQueriesResults);
 
-    const searchQueriesResults = await Promise.all(searchQueries);
-    const uniqueSearchQueriesResults = uniqBy(
-      flatten(searchQueriesResults),
+    const uniqueAssociatedPRs = uniqBy(
+      flatten(associatedPRs),
       "number",
     );
+    // console.log(uniquePRs);
+
     const prs = await pFilter(
-      uniqueSearchQueriesResults,
+      uniqueAssociatedPRs,
       async ({ number }) => {
         const commits = await octokit.paginate(
           "GET /repos/{owner}/{repo}/pulls/{pull_number}/commits",
@@ -250,3 +291,24 @@ export default async function success(pluginConfig, context, { Octokit }) {
     throw new AggregateError(errors);
   }
 }
+
+export function loadAssociatedPRsQuery (shas) {
+  return `#graphql
+    query getAssociatedPRs($owner: String!, $repo: String!) {
+      repository(owner: $owner, name: $repo) {
+        ${shas.map((sha) => {
+          return `"${sha.slice(0, 6)}": object(oid: ${sha}) {
+            ...on Commit {
+              associatedPullRequests(first: 100) {
+                nodes {
+                  url
+                  number
+                }
+              }
+            }
+          }`
+        }).join("")}
+      }
+    }
+  `;
+}

package.json

@@ -40,6 +40,7 @@
     "url-join": "^5.0.0"
   },
   "devDependencies": {
+    "assert": "^2.1.0",
     "ava": "6.1.3",
     "c8": "10.1.2",
     "cpy": "11.0.1",

test/success.test.js

@@ -2,14 +2,16 @@ import { repeat } from "lodash-es";
 import sinon from "sinon";
 import test from "ava";
 import fetchMock from "fetch-mock";
+import assert from "assert";
+import { graphql } from "@octokit/graphql";
 
 import { ISSUE_ID } from "../lib/definitions/constants.js";
 import getReleaseLinks from "../lib/get-release-links.js";
 import { TestOctokit } from "./helpers/test-octokit.js";
 
 /* eslint camelcase: ["error", {properties: "never"}] */
 
-import success from "../lib/success.js";
+import success, { loadAssociatedPRsQuery } from "../lib/success.js";
 
 test.beforeEach((t) => {
   // Mock logger
@@ -57,14 +59,37 @@ test("Add comment and labels to PRs associated with release commits and issues s
     .getOnce(`https://api.github.local/repos/${owner}/${repo}`, {
       full_name: `${redirectedOwner}/${redirectedRepo}`,
     })
-    .getOnce(
-      `https://api.github.local/search/issues?q=${encodeURIComponent(
-        `repo:${redirectedOwner}/${redirectedRepo}`,
-      )}+${encodeURIComponent("type:pr")}+${encodeURIComponent(
-        "is:merged",
-      )}+${commits.map((commit) => commit.hash).join("+")}`,
-      { items: prs },
-    )
+    .postOnce("https://api.github.local/graphql", (url, options) => {
+      assert.strictEqual(options.headers.authorization, "token github_token");
+      assert.strictEqual(
+        options.body,
+        `${loadAssociatedPRsQuery(commits.map(c => c.hash))}`,
+      );
+      return {
+        data: {
+          repository: {
+            com1: {
+              associatedPullRequests: {
+                nodes: [
+                  {
+                    "number": 2
+                  }
+                ]
+              }
+            },
+            com2: {
+              associatedPullRequests: {
+                nodes: [
+                  {
+                    "number": 2
+                  }
+                ]
+              }
+            }
+          }
+        }
+      }
+    })
     .getOnce(
       `https://api.github.local/repos/${redirectedOwner}/${redirectedRepo}/pulls/1/commits`,
       [{ sha: commits[0].hash }],