feat(core): add headers as a new argument of REST

spinnaker · Jun 4, 2021 · b702b5b · b702b5b
1 parent 23e60f6
commit b702b5b
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/app/scripts/modules/core/src/api/ApiService.ts b/app/scripts/modules/core/src/api/ApiService.ts
@@ -256,14 +256,17 @@ class DeprecatedRequestBuilderRoot extends DeprecatedRequestBuilder {
 
 export const invalidContentMessage = 'API response was neither JSON nor zero-length html or text';
 
-export function makeRequestBuilderConfig(pathPrefix?: string): IRequestBuilderConfig {
+export function makeRequestBuilderConfig(
+  pathPrefix?: string,
+  headers?: IRequestBuilderConfig['headers'],
+): IRequestBuilderConfig {
   return {
     url: joinPaths(pathPrefix),
     cache: false,
     data: undefined,
     params: {},
     timeout: (SETTINGS.pollSchedule || 30000) * 2 + 5000,
-    headers: { 'X-RateLimit-App': 'deck' },
+    headers: { 'X-RateLimit-App': 'deck', ...headers },
   };
 }
 
@@ -275,6 +278,6 @@ export const API: IDeprecatedRequestBuilder = new DeprecatedRequestBuilderRoot(m
  * @param staticPathPrefix a static string, i.e., '/proxies/foo/endpoint' --
  *        avoid dynamic strings like `/entity/${id}`, use .path('entity', id) instead
  */
-export const REST = (staticPathPrefix?: string): IRequestBuilder => {
-  return new RequestBuilder(makeRequestBuilderConfig(staticPathPrefix));
+export const REST = (staticPathPrefix?: string, headers?: IRequestBuilderConfig['headers']): IRequestBuilder => {
+  return new RequestBuilder(makeRequestBuilderConfig(staticPathPrefix, headers));
 };