From e35c68b8c098e234fc381271bd9028a68ed3c252 Mon Sep 17 00:00:00 2001
From: Minuk Song <uanid@outlook.com>
Date: Thu, 14 Sep 2023 11:06:36 +0900
Subject: [PATCH] fix: github enterprise login (#533)

---
 src/context/App.tsx           | 3 ++-
 src/hooks/useNotifications.ts | 8 +++-----
 src/utils/auth.ts             | 4 +++-
 3 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/src/context/App.tsx b/src/context/App.tsx
index f3191fd3b..de4eb6243 100644
--- a/src/context/App.tsx
+++ b/src/context/App.tsx
@@ -22,6 +22,7 @@ import { setAutoLaunch } from '../utils/comms';
 import { useInterval } from '../hooks/useInterval';
 import { useNotifications } from '../hooks/useNotifications';
 import Constants from '../utils/constants';
+import { generateGitHubAPIUrl } from '../utils/helpers';
 
 const defaultAccounts: AuthState = {
   token: null,
@@ -134,7 +135,7 @@ export const AppProvider = ({ children }: { children: React.ReactNode }) => {
   const validateToken = useCallback(
     async ({ token, hostname }: AuthTokenOptions) => {
       await apiRequestAuth(
-        `https://api.${hostname}/notifications`,
+        `${generateGitHubAPIUrl(hostname)}notifications`,
         'HEAD',
         token
       );
diff --git a/src/hooks/useNotifications.ts b/src/hooks/useNotifications.ts
index 88e670cb2..82f8236cb 100644
--- a/src/hooks/useNotifications.ts
+++ b/src/hooks/useNotifications.ts
@@ -57,16 +57,14 @@ export const useNotifications = (): NotificationsState => {
         if (!isGitHubLoggedIn) {
           return;
         }
-        const url = `https://api.${Constants.DEFAULT_AUTH_OPTIONS.hostname}/${endpointSuffix}`;
+        const url = `${generateGitHubAPIUrl(Constants.DEFAULT_AUTH_OPTIONS.hostname)}${endpointSuffix}`;
         return apiRequestAuth(url, 'GET', accounts.token);
       }
 
       function getEnterpriseNotifications() {
         return accounts.enterpriseAccounts.map((account) => {
-          const hostname = account.hostname;
-          const token = account.token;
-          const url = `https://${hostname}/api/v3/${endpointSuffix}`;
-          return apiRequestAuth(url, 'GET', token);
+          const url = `${generateGitHubAPIUrl(account.hostname)}${endpointSuffix}`;
+          return apiRequestAuth(url, 'GET', account.token);
         });
       }
 
diff --git a/src/utils/auth.ts b/src/utils/auth.ts
index 2c738d976..506cf64df 100644
--- a/src/utils/auth.ts
+++ b/src/utils/auth.ts
@@ -1,3 +1,5 @@
+import { generateGitHubAPIUrl } from "./helpers";
+
 const { remote } = require('electron');
 const BrowserWindow = remote.BrowserWindow;
 
@@ -78,7 +80,7 @@ export const getUserData = async (
   hostname: string
 ): Promise<User> => {
   const response = await apiRequestAuth(
-    `https://api.${hostname}/user`,
+    `${generateGitHubAPIUrl(hostname)}user`,
     'GET',
     token
   );