[RW-13480][risk=no] Workspace Banners for Initial Credits

api/src/main/java/org/pmiops/workbench/utils/mappers/WorkspaceMapper.java

@@ -53,6 +53,10 @@ public interface WorkspaceMapper {
       target = "initialCredits.expirationEpochMillis",
       source = "dbWorkspace.creator",
       qualifiedByName = "getInitialCreditsExpiration")
+  @Mapping(
+      target = "initialCredits.extensionEpochMillis",
+      source = "dbWorkspace.creator",
+      qualifiedByName = "getInitialCreditsExpiration")
   @Mapping(target = "cdrVersionId", source = "dbWorkspace.cdrVersion")
   @Mapping(target = "accessTierShortName", source = "dbWorkspace.cdrVersion.accessTier.shortName")
   @Mapping(target = "googleProject", source = "dbWorkspace.googleProject")
@@ -128,6 +132,10 @@ default List<WorkspaceResponse> toApiWorkspaceResponseList(
       target = "initialCredits.expirationEpochMillis",
       source = "creator",
       qualifiedByName = "getInitialCreditsExpiration")
+  @Mapping(
+      target = "initialCredits.extensionEpochMillis",
+      source = "creator",
+      qualifiedByName = "getInitialCreditsExpiration")
   @Mapping(target = "initialCredits.exhausted", source = "dbWorkspace.initialCreditsExhausted")
   @Mapping(target = "etag", source = "version", qualifiedByName = "versionToEtag")
   @Mapping(

api/src/main/resources/workbench-api.yaml

@@ -13047,6 +13047,9 @@ components:
         expirationEpochMillis:
           type: integer
           format: int64
+        extensionEpochMillis:
+          type: integer
+          format: int64
   parameters:
     userId:
       name: userId

ui/src/app/pages/workspace/invalid-billing-banner.tsx

@@ -4,8 +4,10 @@ import * as fp from 'lodash';
 import { Profile } from 'generated/fetch';
 
 import { Button, LinkButton } from 'app/components/buttons';
+import { AoU } from 'app/components/text-wrappers';
 import { ToastBanner, ToastType } from 'app/components/toast-banner';
 import { withCurrentWorkspace, withUserProfile } from 'app/utils';
+import { plusDays } from 'app/utils/dates';
 import { NavigationProps } from 'app/utils/navigation';
 import { withNavigation } from 'app/utils/with-navigation-hoc';
 import { WorkspaceData } from 'app/utils/workspace-data';
@@ -19,23 +21,119 @@ interface Props extends NavigationProps {
   onClose: Function;
 }
 
+const InitialCreditsArticleLink = () => (
+  <>
+    "
+    <LinkButton
+      onClick={() => window.open(supportUrls.createBillingAccount, '_blank')}
+    >
+      Using <AoU /> Initial Credits
+    </LinkButton>
+    "
+  </>
+);
+
+const BillingAccountArticleLink = () => (
+  <>
+    "
+    <LinkButton
+      onClick={() => window.open(supportUrls.createBillingAccount, '_blank')}
+    >
+      Paying for Your Research
+    </LinkButton>
+    "
+  </>
+);
+
 export const InvalidBillingBanner = fp.flow(
   withCurrentWorkspace(),
   withUserProfile(),
   withNavigation
-)(({ onClose, navigate, workspace }: Props) => {
-  const message = (
-    <div>
-      The initial credits for the creator of this workspace have run out. Please
-      provide a valid billing account.
-      <LinkButton
-        onClick={() => window.open(supportUrls.createBillingAccount, '_blank')}
-      >
-        Learn how to link a billing account.
-      </LinkButton>
-    </div>
-  );
-  const footer = (
+)(({ onClose, navigate, workspace, profileState }: Props) => {
+  const { profile } = profileState;
+  const isCreator =
+    workspace && profile && profile.username === workspace.creator;
+  const isEligibleForExtension = workspace && !workspace.extensionEpochMillis;
+  const isExpired =
+    workspace && workspace.initialCredits.expirationEpochMillis < Date.now();
+  const isExpiringSoon =
+    workspace &&
+    !isExpired &&
+    plusDays(workspace.initialCredits.expirationEpochMillis, 5) < Date.now();
+  let message;
+  let title;
+  if (isExpiringSoon && isEligibleForExtension) {
+    title = 'Workspace credits are expiring soon';
+    if (isCreator) {
+      // Banner 1 in spec
+      message = (
+        <div>
+          Your initial credits are expiring soon. You can request an extension
+          here. For more information, read the <InitialCreditsArticleLink />{' '}
+          article on the User Support Hub.
+        </div>
+      );
+    } else {
+      // Banner 2 in spec
+      message = (
+        <div>
+          This workspace creator’s initial credits are expiring soon. This
+          workspace was created by FIRST NAME LAST NAME. For more information,
+          read the <InitialCreditsArticleLink /> article on the User Support
+          Hub.
+        </div>
+      );
+    }
+  } else if (isExpired) {
+    if (isEligibleForExtension) {
+      title = 'Workspace credits have expired';
+      if (isCreator) {
+        // Banner 3 in spec (changed to trigger modal from here instead of on Profile page)
+        message = (
+          <div>
+            Your initial credits have expired. You can request an extension
+            here. For more information, read the <InitialCreditsArticleLink />{' '}
+            article on the User Support Hub.
+          </div>
+        );
+      } else {
+        // Banner 4 in spec
+        message = (
+          <div>
+            This workspace creator’s initial credits have expired. This
+            workspace was created by FIRST NAME LAST NAME. For more information,
+            read the <InitialCreditsArticleLink /> article on the User Support
+            Hub.
+          </div>
+        );
+      }
+    } else {
+      title = 'This workspace is out of initial credits';
+      if (isCreator) {
+        // Banner 5 in spec
+        message = (
+          <div>
+            Your initial credits have run out. To use the workspace, a valid
+            billing account needs to be provided. To learn more about
+            establishing a billing account, read the{' '}
+            <BillingAccountArticleLink /> article on the User Support Hub.
+          </div>
+        );
+      } else {
+        // Banner 6 in spec
+        message = (
+          <div>
+            This workspace creator’s initial credits have run out. This
+            workspace was created by FIRST NAME LAST NAME. To use the workspace,
+            a valid billing account needs to be provided. To learn more about
+            establishing a billing account, read the{' '}
+            <BillingAccountArticleLink /> article on the User Support Hub.
+          </div>
+        );
+      }
+    }
+  }
+  const footer = isCreator && (
     <Button
       style={{ height: '38px', width: '70%', fontWeight: 400 }}
       onClick={() => {
@@ -53,6 +151,7 @@ export const InvalidBillingBanner = fp.flow(
       Edit Workspace
     </Button>
   );
+
   return (
     <ToastBanner
       {...{ message, footer, onClose }}