From 31915d9ebefb2acff66ecc332bd38ad0ae99f3d7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gon=C3=A9ri=20Le=20Bouder?= <goneri@lebouder.net>
Date: Mon, 4 Dec 2023 14:36:29 -0500
Subject: [PATCH] WIP - AAP-18386

---
 src/features/lightspeed/api.ts                |  8 +++++
 .../lightspeed/lightSpeedOAuthProvider.ts     | 29 ++++++++++++++++++-
 src/features/lightspeed/statusBar.ts          |  8 ++++-
 src/features/lightspeed/utils/webUtils.ts     |  2 ++
 src/interfaces/lightspeed.ts                  |  2 +-
 5 files changed, 46 insertions(+), 3 deletions(-)

diff --git a/src/features/lightspeed/api.ts b/src/features/lightspeed/api.ts
index d1d1666bf..1c0aa39df 100644
--- a/src/features/lightspeed/api.ts
+++ b/src/features/lightspeed/api.ts
@@ -150,6 +150,14 @@ export class LightSpeedAPI {
             vscode.window.showErrorMessage(
               `Model ID "${this.settingsManager.settings.lightSpeedService.model}" is invalid. Please contact your administrator.`
             );
+          } else if (
+            responseErrorData &&
+            responseErrorData.hasOwnProperty("code") &&
+            responseErrorData.code === "permission_denied_wca_api_key_is_missing"
+          ) {
+            vscode.window.showErrorMessage(
+              `You do not have a licensed seat for Ansible Lightspeed and your\norganization is using the paid commercial service. Contact your Red Hat Organization's administrator for more information on how to get a licensed seat.`
+            );
           } else {
             vscode.window.showErrorMessage(
               `User not authorized to access Ansible Lightspeed.`
diff --git a/src/features/lightspeed/lightSpeedOAuthProvider.ts b/src/features/lightspeed/lightSpeedOAuthProvider.ts
index e9073781f..a8bdcc5bf 100644
--- a/src/features/lightspeed/lightSpeedOAuthProvider.ts
+++ b/src/features/lightspeed/lightSpeedOAuthProvider.ts
@@ -132,6 +132,9 @@ export class LightSpeedAuthenticationProvider
 
       const identifier = uuid();
       const userName = userinfo.external_username || userinfo.username || "";
+      const rhOrgHasSubscription = userinfo.rh_org_has_subscription
+        ? userinfo.rh_org_has_subscription
+        : false;
       const rhUserHasSeat = userinfo.rh_user_has_seat
         ? userinfo.rh_user_has_seat
         : false;
@@ -139,6 +142,8 @@ export class LightSpeedAuthenticationProvider
       let label = userName;
       if (rhUserHasSeat) {
         label += " (licensed)";
+      } else if (rhOrgHasSubscription) {
+        label += " (no seat assigned)";
       } else {
         label += " (Tech Preview)";
       }
@@ -172,7 +177,8 @@ export class LightSpeedAuthenticationProvider
 
       lightSpeedManager.statusBarProvider.statusBar.text =
         await lightSpeedManager.statusBarProvider.getLightSpeedStatusBarText(
-          rhUserHasSeat
+          rhUserHasSeat,
+          rhOrgHasSubscription
         );
 
       lightSpeedManager.statusBarProvider.setLightSpeedStatusBarTooltip(
@@ -630,4 +636,25 @@ export class LightSpeedAuthenticationProvider
       return false;
     }
   }
+
+  public async rhOrgHasSubscription(): Promise<boolean | undefined> {
+    const authSession = await this.getLightSpeedAuthSession();
+    if (authSession === undefined) {
+      console.log(
+        "[ansible-lightspeed-oauth] User authentication session not found."
+      );
+      return undefined;
+    } else if (authSession?.rhOrgHasSubscription) {
+      console.log(
+        `[ansible-lightspeed-oauth] User "${authSession?.account?.label}" has an Org with a subscription.`
+      );
+      return true;
+    } else {
+      console.log(
+        `[ansible-lightspeed-oauth] User "${authSession?.account?.label}" does have an Org with a subscription.`
+      );
+      return false;
+    }
+  }
+
 }
diff --git a/src/features/lightspeed/statusBar.ts b/src/features/lightspeed/statusBar.ts
index 514468caa..61bae6717 100644
--- a/src/features/lightspeed/statusBar.ts
+++ b/src/features/lightspeed/statusBar.ts
@@ -52,14 +52,20 @@ export class LightspeedStatusBar {
   }
 
   public async getLightSpeedStatusBarText(
-    rhUserHasSeat?: boolean
+    rhUserHasSeat?: boolean,
+    rhOrgHasSubscription?: boolean
   ): Promise<string> {
     let lightSpeedStatusbarText;
     if (rhUserHasSeat === undefined) {
       rhUserHasSeat = await this.lightSpeedAuthProvider.rhUserHasSeat();
     }
+    if (rhOrgHasSubscription === undefined) {
+      rhOrgHasSubscription = await this.lightSpeedAuthProvider.rhOrgHasSubscription();
+    }
     if (rhUserHasSeat === true) {
       lightSpeedStatusbarText = "Lightspeed (licensed)";
+    } else if (rhOrgHasSubscription === true) {
+      lightSpeedStatusbarText = "Lightspeed (no seat assigned)";
     } else if (rhUserHasSeat === false) {
       lightSpeedStatusbarText = "Lightspeed (Tech Preview)";
     } else {
diff --git a/src/features/lightspeed/utils/webUtils.ts b/src/features/lightspeed/utils/webUtils.ts
index 8d3acf3f3..6fdf7c980 100644
--- a/src/features/lightspeed/utils/webUtils.ts
+++ b/src/features/lightspeed/utils/webUtils.ts
@@ -77,6 +77,8 @@ export function getLoggedInSessionDetails(
   const modelInfo: LightspeedSessionModelInfo = {};
   if (sessionData.rhUserHasSeat) {
     userInfo.userType = "Licensed";
+  } else if (sessionData.rhOrgHasSubscription) {
+    userInfo.userType = "no seat assigned";
   } else {
     userInfo.userType = "Tech Preview";
   }
diff --git a/src/interfaces/lightspeed.ts b/src/interfaces/lightspeed.ts
index 1167eee0b..af1811fb9 100644
--- a/src/interfaces/lightspeed.ts
+++ b/src/interfaces/lightspeed.ts
@@ -166,7 +166,7 @@ export interface IAdditionalContext {
 }
 
 export interface LightspeedSessionUserInfo {
-  userType?: "Licensed" | "Tech Preview";
+  userType?: "Licensed" | "no seat assigned" | "Tech Preview";
   role?: string;
   subscribed?: boolean;
 }