feat(api): merge query

Signed-off-by: Adam Setch <[email protected]>

Author: setchy

src/renderer/utils/api/client.ts

@@ -207,9 +207,9 @@ export async function fetchIssueByNumber(
     notification.account.token,
     FetchIssueByNumberDocument,
     {
-      owner: notification.repository.owner.login,
-      name: notification.repository.name,
-      number: number,
+      ownerINDEX: notification.repository.owner.login,
+      nameINDEX: notification.repository.name,
+      numberINDEX: number,
       firstLabels: 100,
       lastComments: 1,
     },
@@ -230,11 +230,11 @@ export async function fetchPullByNumber(
     notification.account.token,
     FetchPullRequestByNumberDocument,
     {
-      owner: notification.repository.owner.login,
-      name: notification.repository.name,
-      number: number,
-      firstLabels: 100,
+      ownerINDEX: notification.repository.owner.login,
+      nameINDEX: notification.repository.name,
+      numberINDEX: number,
       firstClosingIssues: 100,
+      firstLabels: 100,
       lastComments: 1,
       lastReviews: 100,
     },
@@ -255,12 +255,12 @@ export async function fetchDiscussionByNumber(
     notification.account.token,
     FetchDiscussionByNumberDocument,
     {
-      owner: notification.repository.owner.login,
-      name: notification.repository.name,
-      number: number,
+      ownerINDEX: notification.repository.owner.login,
+      nameINDEX: notification.repository.name,
+      numberINDEX: number,
+      firstLabels: 100,
       lastComments: 10,
       lastReplies: 10,
-      firstLabels: 100,
       includeIsAnswered: isAnsweredDiscussionFeatureSupported(
         notification.account,
       ),

src/renderer/utils/api/graphql/discussion.graphql

@@ -1,14 +1,18 @@
 query FetchDiscussionByNumber(
-  $owner: String!
-  $name: String!
-  $number: Int!
+  $ownerINDEX: String!
+  $nameINDEX: String!
+  $numberINDEX: Int!
   $lastComments: Int
   $lastReplies: Int
   $firstLabels: Int
   $includeIsAnswered: Boolean!
 ) {
-  repository(owner: $owner, name: $name) {
-    discussion(number: $number) {
+  ...DiscussionMergeQuery
+}
+
+fragment DiscussionMergeQuery on Query {
+  nodeINDEX: repository(owner: $ownerINDEX, name: $nameINDEX) {
+    discussion(number: $numberINDEX) {
       ...DiscussionDetails
     }
   }

src/renderer/utils/api/graphql/generated/gql.ts

@@ -1,12 +1,15 @@
 query FetchIssueByNumber(
-  $owner: String!
-  $name: String!
-  $number: Int!
+  $ownerINDEX: String!
+  $nameINDEX: String!
+  $numberINDEX: Int!
   $lastComments: Int
   $firstLabels: Int
 ) {
-  repository(owner: $owner, name: $name) {
-    issue(number: $number) {
+  ...IssueMergeQuery
+}
+fragment IssueMergeQuery on Query {
+  nodeINDEX: repository(owner: $ownerINDEX, name: $nameINDEX) {
+    issue(number: $numberINDEX) {
       ...IssueDetails
     }
   }

src/renderer/utils/api/graphql/generated/graphql.ts

@@ -1,14 +1,18 @@
 query FetchPullRequestByNumber(
-  $owner: String!
-  $name: String!
-  $number: Int!
+  $ownerINDEX: String!
+  $nameINDEX: String!
+  $numberINDEX: Int!
   $firstLabels: Int
   $lastComments: Int
   $lastReviews: Int
   $firstClosingIssues: Int
 ) {
-  repository(owner: $owner, name: $name) {
-    pullRequest(number: $number) {
+  ...PullRequestMergeQuery
+}
+
+fragment PullRequestMergeQuery on Query {
+  nodeINDEX: repository(owner: $ownerINDEX, name: $nameINDEX) {
+    pullRequest(number: $numberINDEX) {
       ...PullRequestDetails
     }
   }

src/renderer/utils/api/graphql/issue.graphql

@@ -0,0 +1,22 @@
+export function getQueryFragmentBody(doc: string): string | null {
+  // Find fragment on Query pattern
+  const fragmentMatch = doc.match(/fragment\s+\w+\s+on\s+Query\s+\{/);
+  if (!fragmentMatch) {
+    return null;
+  }
+
+  const start = fragmentMatch.index + fragmentMatch[0].length - 1; // Position of opening brace
+  let depth = 0;
+
+  for (let i = start; i < doc.length; i++) {
+    if (doc[i] === '{') {
+      depth++;
+    } else if (doc[i] === '}') {
+      depth--;
+      if (depth === 0) {
+        return doc.slice(start + 1, i).trim();
+      }
+    }
+  }
+  return null;
+}

src/renderer/utils/api/graphql/pull.graphql

@@ -12,6 +12,10 @@ import { formatForDisplay } from './utils';
 export class DefaultHandler implements NotificationTypeHandler {
   type?: SubjectType;
 
+  mergeQueryConfig() {
+    return undefined;
+  }
+
   query(_notification: Notification) {
     return null;
   }

src/renderer/utils/api/graphql/utils.ts

@@ -19,25 +19,51 @@ import {
 } from '../../../types';
 import type { Notification, Subject } from '../../../typesGitHub';
 import { fetchDiscussionByNumber } from '../../api/client';
-import type {
-  CommentFieldsFragment,
-  DiscussionCommentFieldsFragment,
-  DiscussionDetailsFragment,
+import {
+  type CommentFieldsFragment,
+  type DiscussionCommentFieldsFragment,
+  type DiscussionDetailsFragment,
+  DiscussionDetailsFragmentDoc,
+  DiscussionMergeQueryFragmentDoc,
 } from '../../api/graphql/generated/graphql';
+import { getQueryFragmentBody } from '../../api/graphql/utils';
 import { DefaultHandler, defaultHandler } from './default';
+import type { GraphQLMergedQueryConfig } from './types';
 import { getNotificationAuthor } from './utils';
 
 class DiscussionHandler extends DefaultHandler {
   readonly type = 'Discussion';
 
+  mergeQueryConfig() {
+    return {
+      queryFragment: getQueryFragmentBody(
+        DiscussionMergeQueryFragmentDoc.toString(),
+      ),
+      responseFragment: DiscussionDetailsFragmentDoc.toString(),
+      extras: [
+        { name: 'lastComments', type: 'Int', defaultValue: 100 },
+        { name: 'lastReplies', type: 'Int', defaultValue: 100 },
+        { name: 'firstLabels', type: 'Int', defaultValue: 100 },
+        { name: 'includeIsAnswered', type: 'Boolean!', defaultValue: true },
+      ],
+      selection: (
+        index: number,
+      ) => `node${index}: repository(owner: $owner${index}, name: $name${index}) {
+          discussion(number: $number${index}) {
+            ...DiscussionDetails
+          }
+        }`,
+    } as GraphQLMergedQueryConfig;
+  }
+
   async enrich(
     notification: Notification,
     _settings: SettingsState,
     fetchedData?: DiscussionDetailsFragment,
   ): Promise<GitifySubject> {
     const discussion =
       fetchedData ??
-      (await fetchDiscussionByNumber(notification)).data.repository?.discussion;
+      (await fetchDiscussionByNumber(notification)).data.nodeINDEX?.discussion;
 
     let discussionState: GitifyDiscussionState = 'OPEN';
 

src/renderer/utils/notifications/handlers/default.ts

@@ -16,13 +16,40 @@ import type {
 } from '../../../types';
 import { IconColor } from '../../../types';
 import type { Notification, Subject } from '../../../typesGitHub';
-import type { IssueDetailsFragment } from '../../api/graphql/generated/graphql';
+import {
+  type IssueDetailsFragment,
+  IssueDetailsFragmentDoc,
+  IssueMergeQueryFragmentDoc,
+} from '../../api/graphql/generated/graphql';
+import { getQueryFragmentBody } from '../../api/graphql/utils';
 import { DefaultHandler, defaultHandler } from './default';
+import type { GraphQLMergedQueryConfig } from './types';
 import { getNotificationAuthor } from './utils';
 
 class IssueHandler extends DefaultHandler {
   readonly type = 'Issue';
 
+  mergeQueryConfig() {
+    return {
+      queryFragment: getQueryFragmentBody(
+        IssueMergeQueryFragmentDoc.toString(),
+      ),
+
+      responseFragment: IssueDetailsFragmentDoc.toString(),
+      extras: [
+        { name: 'lastComments', type: 'Int', defaultValue: 100 },
+        { name: 'firstLabels', type: 'Int', defaultValue: 100 },
+      ],
+      selection: (
+        index: number,
+      ) => `node${index}: repository(owner: $owner${index}, name: $name${index}) {
+          issue(number: $number${index}) {
+            ...IssueDetails
+          }
+        }`,
+    } as GraphQLMergedQueryConfig;
+  }
+
   async enrich(
     _notification: Notification,
     _settings: SettingsState,