Store the hostname in the secret store

This commit stores the hostname in the secret store when the user logs in. This ensures that the hostname is available when the user logs in again.

This is necessary because the hostname is used to construct the API URL and the Web URL for a given TFE or HCP Terraform instance.

src/api/terraformCloud/index.ts

@@ -86,7 +86,10 @@ export function earlySetupForHostname(hostname: string) {
   earlyApiClient.use(pluginLogger());
 }
 
-export function apiSetup() {
+export function apiSetup(hostname: string) {
+  TerraformCloudHost = hostname;
+  TerraformCloudAPIUrl = `https://${TerraformCloudHost}/api/v2`;
+  TerraformCloudWebUrl = `https://${TerraformCloudHost}/app`;
   // ApiClient setup
   apiClient = new Zodios(TerraformCloudAPIUrl, [
     ...accountEndpoints,

src/providers/tfc/authenticationProvider.ts

@@ -31,7 +31,11 @@ class TerraformCloudSession implements vscode.AuthenticationSession {
    * @param accessToken The personal access token to use for authentication
    * @param account The user account for the specified token
    */
-  constructor(public readonly accessToken: string, public account: vscode.AuthenticationSessionAccountInformation) {}
+  constructor(
+    public readonly accessToken: string,
+    public readonly hostName: string,
+    public account: vscode.AuthenticationSessionAccountInformation,
+  ) {}
 }
 
 interface TerraformCloudToken {
@@ -69,13 +73,13 @@ class TerraformCloudSessionHandler {
         },
       });
 
-      const session = new TerraformCloudSession(token, {
+      const session = new TerraformCloudSession(token, hostname, {
         label: user.data.attributes.username,
         id: user.data.id,
       });
 
       await this.secretStorage.store(this.sessionKey, JSON.stringify(session));
-      apiSetup();
+      apiSetup(session.hostName);
       return session;
     } catch (error) {
       if (error instanceof ZodiosError) {
@@ -111,7 +115,7 @@ export class TerraformCloudAuthenticationProvider implements vscode.Authenticati
   private sessionHandler: TerraformCloudSessionHandler;
   // this property is used to determine if the session has been changed in another window of VS Code
   // it's a promise, so we can set in the constructor where we can't execute async code
-  private sessionPromise: Promise<vscode.AuthenticationSession | undefined>;
+  private sessionPromise: Promise<TerraformCloudSession | undefined>;
   private disposable: vscode.Disposable | undefined;
 
   private _onDidChangeSessions =
@@ -160,6 +164,8 @@ export class TerraformCloudAuthenticationProvider implements vscode.Authenticati
     try {
       const session = await this.sessionPromise;
       if (session) {
+        earlySetupForHostname(session.hostName);
+        apiSetup(session.hostName);
         this.logger.info('Successfully fetched HCP Terraform session');
         await vscode.commands.executeCommand('setContext', 'terraform.cloud.signed-in', true);
         return [session];