From 151eab1ce9805cb2def491b4485e26903860636f Mon Sep 17 00:00:00 2001
From: Tiago Barbosa <tbarbos@hotmail.com>
Date: Thu, 22 Feb 2024 13:27:20 +0000
Subject: [PATCH] fix: adding support for API base url override which was
 possible with proxy config

Signed-off-by: Tiago Barbosa <tbarbos@hotmail.com>
---
 config.d.ts           |  5 +++++
 src/apis/pagerduty.ts | 21 +++++++++++++--------
 src/service/router.ts |  8 ++++++--
 3 files changed, 24 insertions(+), 10 deletions(-)

diff --git a/config.d.ts b/config.d.ts
index be21500..09bf304 100644
--- a/config.d.ts
+++ b/config.d.ts
@@ -27,6 +27,11 @@ export interface Config {
          * @visibility frontend
          */
         eventsBaseUrl?: string;
+        /**
+         * Optional API Base URL to override the default.
+         * @visibility frontend
+         */
+        apiBaseUrl?: string;
         /**
          * Optional PagerDuty API Token used in API calls from the backend component.
          * @visibility frontend
diff --git a/src/apis/pagerduty.ts b/src/apis/pagerduty.ts
index b62680b..492e04f 100644
--- a/src/apis/pagerduty.ts
+++ b/src/apis/pagerduty.ts
@@ -20,11 +20,16 @@ import {
     HttpError
 } from '@pagerduty/backstage-plugin-common';
 
+let apiBaseUrl = 'https://api.pagerduty.com';
+export function setAPIBaseUrl(url: string): void {
+    apiBaseUrl = url;
+}
+
 // Supporting custom actions
 export async function createService(name: string, description: string, escalationPolicyId: string, alertGrouping?: string): Promise<CreateServiceResponse> {
     let alertGroupingParameters = "null";
     let response: Response;
-    const baseUrl = 'https://api.pagerduty.com/services';
+    const baseUrl = `${apiBaseUrl}/services`;
 
     // Set default body
     let body = JSON.stringify({
@@ -176,7 +181,7 @@ export async function createService(name: string, description: string, escalatio
 
 export async function createServiceIntegration(serviceId: string, vendorId: string): Promise<string> {
     let response: Response;
-    const baseUrl = 'https://api.pagerduty.com/services';
+    const baseUrl = `${apiBaseUrl}/services`;
     const options: RequestInit = {
         method: 'POST',
         body: JSON.stringify({
@@ -242,7 +247,7 @@ async function getEscalationPolicies(offset: number, limit: number): Promise<[Bo
             'Content-Type': 'application/json',
         },
     };
-    const baseUrl = 'https://api.pagerduty.com/escalation_policies';
+    const baseUrl = `${apiBaseUrl}/escalation_policies`;
 
     try {
         response = await fetch(`${baseUrl}?${params}`, options);
@@ -353,7 +358,7 @@ export async function getOncallUsers(escalationPolicy: string): Promise<PagerDut
             'Content-Type': 'application/json',
         },
     };
-    const baseUrl = 'https://api.pagerduty.com/oncalls';
+    const baseUrl = `${apiBaseUrl}/oncalls`;
 
     try {
         response = await fetch(`${baseUrl}?${params}`, options);
@@ -425,7 +430,7 @@ export async function getServiceById(serviceId: string): Promise<PagerDutyServic
             'Content-Type': 'application/json',
         },
     };
-    const baseUrl = 'https://api.pagerduty.com/services';
+    const baseUrl = `${apiBaseUrl}/services`;
 
     try {
         response = await fetch(`${baseUrl}/${serviceId}?${params}`, options);
@@ -467,7 +472,7 @@ export async function getServiceByIntegrationKey(integrationKey: string): Promis
             'Content-Type': 'application/json',
         },
     };
-    const baseUrl = 'https://api.pagerduty.com/services';
+    const baseUrl = `${apiBaseUrl}/services`;
 
     try {
         response = await fetch(`${baseUrl}?${params}`, options);
@@ -513,7 +518,7 @@ export async function getChangeEvents(serviceId: string): Promise<PagerDutyChang
             'Content-Type': 'application/json',
         },
     };
-    const baseUrl = 'https://api.pagerduty.com/services';
+    const baseUrl = `${apiBaseUrl}/services`;
 
     try {
         response = await fetch(`${baseUrl}/${serviceId}/change_events?${params}`, options);
@@ -556,7 +561,7 @@ export async function getIncidents(serviceId: string): Promise<PagerDutyIncident
             'Content-Type': 'application/json',
         },
     };
-    const baseUrl = 'https://api.pagerduty.com/incidents';
+    const baseUrl = `${apiBaseUrl}/incidents`;
 
     try {
         response = await fetch(`${baseUrl}?${params}`, options);
diff --git a/src/service/router.ts b/src/service/router.ts
index 7b6379e..3297037 100644
--- a/src/service/router.ts
+++ b/src/service/router.ts
@@ -3,7 +3,7 @@ import { Config } from '@backstage/config';
 import express from 'express';
 import Router from 'express-promise-router';
 import { Logger } from 'winston';
-import { getAllEscalationPolicies, getChangeEvents, getIncidents, getOncallUsers, getServiceById, getServiceByIntegrationKey } from '../apis/pagerduty';
+import { getAllEscalationPolicies, getChangeEvents, getIncidents, getOncallUsers, getServiceById, getServiceByIntegrationKey, setAPIBaseUrl } from '../apis/pagerduty';
 import { HttpError, PagerDutyChangeEventsResponse, PagerDutyIncidentsResponse, PagerDutyOnCallUsersResponse, PagerDutyServiceResponse } from '@pagerduty/backstage-plugin-common';
 import { loadAuthConfig } from '../auth/auth';
 
@@ -20,8 +20,12 @@ export async function createRouter(
     // Get authentication Config
     await loadAuthConfig(logger, config);
 
+    // Get the PagerDuty API Base URL from config
+    const baseUrl = config.getOptionalString('pagerDuty.apiBaseUrl') !== undefined ? config.getString('pagerDuty.apiBaseUrl') : 'https://api.pagerduty.com';
+    setAPIBaseUrl(baseUrl);
+
     // Create the router
-    const router = Router();
+    const router = Router();    
     router.use(express.json());
 
     // Add routes