fix: [UIE-10253] - IAM Delegate: company name for users with no permissions

Author: mpolotsk-akamai

packages/manager/src/features/Account/SwitchAccountButton.tsx

@@ -19,10 +19,15 @@ export const SwitchAccountButton = (props: ButtonProps) => {
         },
         font: theme.tokens.alias.Typography.Label.Semibold.S,
         marginTop: theme.tokens.spacing.S4,
+        ...(isDelegateUserType && {
+          '&.MuiButton-root': {
+            textTransform: 'none',
+          },
+        }),
       })}
       {...props}
     >
-      {isDelegateUserType ? 'Switch back to your account' : 'Switch Account'}
+      {isDelegateUserType ? 'Switch Back to Your Account' : 'Switch Account'}
     </Button>
   );
 };

packages/manager/src/features/Account/SwitchAccountDrawer.tsx

@@ -20,21 +20,24 @@ import { useSwitchToParentAccount } from 'src/features/Account/SwitchAccounts/us
 import { setTokenInLocalStorage } from 'src/features/Account/SwitchAccounts/utils';
 import { useIsIAMDelegationEnabled } from 'src/features/IAM/hooks/useIsIAMEnabled';
 import { sendSwitchToParentAccountEvent } from 'src/utilities/analytics/customEventAnalytics';
-import { getStorage, storage } from 'src/utilities/storage';
+import { getStorage, setStorage, storage } from 'src/utilities/storage';
 
 import { ChildAccountList } from './SwitchAccounts/ChildAccountList';
 import { ChildAccountsTable } from './SwitchAccounts/ChildAccountsTable';
 import { updateParentTokenInLocalStorage } from './SwitchAccounts/utils';
 
 import type { APIError, Filter, UserType } from '@linode/api-v4';
 
+const SWITCH_ACCOUNT_COMPANY_KEY = 'switch_account/company_name';
+
 interface Props {
   onClose: () => void;
   open: boolean;
   userType: undefined | UserType;
 }
 
 interface HandleSwitchToChildAccountProps {
+  company?: string;
   currentTokenWithBearer?: string;
   euuid: string;
   event: React.MouseEvent<HTMLElement>;
@@ -128,11 +131,12 @@ export const SwitchAccountDrawer = (props: Props) => {
       event,
       onClose,
       userType,
+      company,
     }: HandleSwitchToChildAccountProps) => {
       const isProxyOrDelegateUserType =
         userType === 'proxy' || userType === 'delegate';
-
       try {
+        setStorage(SWITCH_ACCOUNT_COMPANY_KEY, company ?? '');
         if (isProxyOrDelegateUserType) {
           // Revoke proxy token before switching accounts.
           await revokeToken().catch(() => {
@@ -162,6 +166,7 @@ export const SwitchAccountDrawer = (props: Props) => {
         location.replace('/linodes');
       } catch {
         // Error is handled by createTokenError.
+        setStorage(SWITCH_ACCOUNT_COMPANY_KEY, '');
       }
     },
     [createToken, isProxyUserType, updateCurrentToken, revokeToken]

packages/manager/src/features/Account/SwitchAccounts/ChildAccountsTable.tsx

@@ -26,8 +26,10 @@ export interface ChildAccountsTableProps {
     euuid,
     event,
     onClose,
+    company,
     userType,
   }: {
+    company?: string;
     currentTokenWithBearer?: string;
     euuid: string;
     event: React.MouseEvent<HTMLElement>;
@@ -107,6 +109,7 @@ export const ChildAccountsTable = (props: ChildAccountsTableProps) => {
                     onSwitchAccount({
                       currentTokenWithBearer,
                       euuid: childAccount.euuid,
+                      company: childAccount.company,
                       event,
                       onClose,
                       userType,

packages/manager/src/features/TopMenu/UserMenu/UserMenu.tsx

@@ -43,8 +43,10 @@ export const UserMenu = React.memo(() => {
   const open = Boolean(anchorEl);
   const id = open ? 'user-menu-popover' : undefined;
 
+  const switchAccountCompanyName = getStorage('switch_account/company_name');
+
   const companyNameOrEmail = getCompanyNameOrEmail({
-    company: account?.company,
+    company: account?.company ? account?.company : switchAccountCompanyName,
     profile,
   });