wip falcon cloud autodiscover

CrowdStrike · Jan 13, 2022 · 0601b92 · 0601b92
1 parent 898faad
commit 0601b92
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/src/FalconCloud.ts b/src/FalconCloud.ts
@@ -5,6 +5,9 @@ type CloudUsGov1 = "us-gov-1";
 
 export type FalconCloud = CloudUs1 | CloudUs2 | CloudEu1 | CloudUsGov1;
 
+export type CloudAutoDiscover = "autodiscover";
+export type FalconCloudInput = FalconCloud | CloudAutoDiscover;
+
 const CloudHost = {
     "us-1": "api.crowdstrike.com",
     "us-2": "api.us-2.crowdstrike.com",

diff --git a/src/middleware/oauth2.ts b/src/middleware/oauth2.ts
@@ -1,6 +1,6 @@
 import { Oauth2Api } from "../apis";
 import { Configuration, FetchAPI } from "../runtime";
-import { FalconCloud, CloudBasePath } from "../FalconCloud";
+import { FalconCloud, FalconCloudInput, CloudBasePath } from "../FalconCloud";
 
 type Token = {
     accessToken: string;
@@ -9,7 +9,7 @@ type Token = {
 
 type OAuth2Options = {
     fetchApi?: FetchAPI;
-    cloud: FalconCloud;
+    cloud?: FalconCloudInput;
     clientId: string;
     clientSecret: string;
 };
@@ -54,6 +54,15 @@ export class OAuth2 {
     }
 
     private async cloud(): Promise<FalconCloud> {
+        if (this.options.cloud instanceof FalconCloud) {
+            return this.options.cloud;
+        } else {
+            // User has not provided FalconCloud. Let's try to autodiscover.
+            token = OAuth2.refreshTokenInternal(FalconCloudUs1, this.options.clientId, this.options.clientSecret);
+            // TODO token.XCSRegion header missing
+            // TODO patch openapi-generator to support response headers
+        }
+
         return this.options.cloud;
     }