From c3116f0169db7580afd339312298bb7640a3962b Mon Sep 17 00:00:00 2001
From: Lisa Kim <lisa@goteleport.com>
Date: Fri, 9 Aug 2024 08:51:48 -0700
Subject: [PATCH] Address CRs

---
 .../CreateDatabase/CreateDatabaseDialog.tsx   | 109 +++++++++---------
 .../DeployService/AutoDeploy/AutoDeploy.tsx   |   7 +-
 .../src/services/integrations/types.ts        |   1 -
 3 files changed, 57 insertions(+), 60 deletions(-)

diff --git a/web/packages/teleport/src/Discover/Database/CreateDatabase/CreateDatabaseDialog.tsx b/web/packages/teleport/src/Discover/Database/CreateDatabase/CreateDatabaseDialog.tsx
index 40670573b31da..3eea195d8c083 100644
--- a/web/packages/teleport/src/Discover/Database/CreateDatabase/CreateDatabaseDialog.tsx
+++ b/web/packages/teleport/src/Discover/Database/CreateDatabase/CreateDatabaseDialog.tsx
@@ -32,10 +32,7 @@ import Dialog, { DialogContent } from 'design/DialogConfirmation';
 import { Timeout } from 'teleport/Discover/Shared/Timeout';
 import { TextIcon } from 'teleport/Discover/Shared';
 
-import {
-  dbWithoutDbServerExistsErrorMsg as dbExistsWithoutDbServerErrorMsg,
-  timeoutErrorMsg,
-} from './const';
+import { dbWithoutDbServerExistsErrorMsg, timeoutErrorMsg } from './const';
 
 import type { Attempt } from 'shared/hooks/useAttemptNext';
 
@@ -60,46 +57,47 @@ export function CreateDatabaseDialog({
   onOverwrite,
   onTimeout,
 }: CreateDatabaseDialogProps) {
-  /**
-   * Most likely cause of timeout is when we found a matching db_service
-   * but no db_server heartbeats. Most likely cause is because db_service
-   * has been stopped but is not removed from teleport yet (there is some
-   * minutes delay on expiry)
-   */
-  const skipOnTimeout =
-    attempt.status === 'failed' && attempt.statusText === timeoutErrorMsg;
-
   let content: JSX.Element;
-  if (attempt.status === 'failed' && !skipOnTimeout) {
-    // Only allow overwriting if the database error
-    // states that it's a existing database without a db_server.
-    const canOverwriteDb = attempt.statusText.includes(
-      dbExistsWithoutDbServerErrorMsg
-    );
+  if (attempt.status === 'failed') {
+    /**
+     * Most likely cause of timeout is when we found a matching db_service
+     * but no db_server heartbeats. Most likely cause is because db_service
+     * has been stopped but is not removed from teleport yet (there is some
+     * minutes delay on expiry)
+     */
+    if (attempt.statusText === timeoutErrorMsg) {
+      content = <SuccessContent dbName={dbName} onClick={onTimeout} />;
+    } else {
+      // Only allow overwriting if the database error
+      // states that it's a existing database without a db_server.
+      const canOverwriteDb = attempt.statusText.includes(
+        dbWithoutDbServerExistsErrorMsg
+      );
 
-    // TODO(bl-nero): Migrate this to alert boxes.
-    content = (
-      <>
-        <Flex mb={5} alignItems="center">
-          <Icons.Warning size="large" ml={1} mr={2} color="error.main" />
-          <Text>{attempt.statusText}</Text>
-        </Flex>
-        <Flex gap={3}>
-          <ButtonPrimary width="50%" onClick={retry}>
-            Retry
-          </ButtonPrimary>
-          {canOverwriteDb && (
-            <ButtonWarning width="50%" onClick={onOverwrite}>
-              Overwrite
-            </ButtonWarning>
-          )}
-          <ButtonSecondary width="50%" onClick={close}>
-            Close
-          </ButtonSecondary>
-        </Flex>
-      </>
-    );
-  } else if (attempt.status === 'processing') {
+      // TODO(bl-nero): Migrate this to alert boxes.
+      content = (
+        <>
+          <Flex mb={5} alignItems="center">
+            <Icons.Warning size="large" ml={1} mr={2} color="error.main" />
+            <Text>{attempt.statusText}</Text>
+          </Flex>
+          <Flex gap={3} width="100%">
+            <ButtonPrimary onClick={retry} style={{ flex: 1 }}>
+              Retry
+            </ButtonPrimary>
+            {canOverwriteDb && (
+              <ButtonWarning onClick={onOverwrite} style={{ flex: 1 }}>
+                Overwrite
+              </ButtonWarning>
+            )}
+            <ButtonSecondary onClick={close} style={{ flex: 1 }}>
+              Close
+            </ButtonSecondary>
+          </Flex>
+        </>
+      );
+    }
+  } else if (attempt.status === 'processing' || attempt.status === '') {
     content = (
       <>
         <AnimatedProgressBar mb={1} />
@@ -121,19 +119,8 @@ export function CreateDatabaseDialog({
         </ButtonPrimary>
       </>
     );
-  } else {
-    // success
-    content = (
-      <>
-        <Text mb={5}>
-          <Icons.Check size="small" ml={1} mr={2} color="success.main" />
-          Database "{dbName}" successfully registered
-        </Text>
-        <ButtonPrimary width="100%" onClick={skipOnTimeout ? onTimeout : next}>
-          Next
-        </ButtonPrimary>
-      </>
-    );
+  } else if (attempt.status === 'success') {
+    content = <SuccessContent dbName={dbName} onClick={next} />;
   }
 
   return (
@@ -150,3 +137,15 @@ export function CreateDatabaseDialog({
     </Dialog>
   );
 }
+
+const SuccessContent = ({ dbName, onClick }) => (
+  <>
+    <Text mb={5}>
+      <Icons.Check size="small" ml={1} mr={2} color="success.main" />
+      Database "{dbName}" successfully registered
+    </Text>
+    <ButtonPrimary width="100%" onClick={onClick}>
+      Next
+    </ButtonPrimary>
+  </>
+);
diff --git a/web/packages/teleport/src/Discover/Database/DeployService/AutoDeploy/AutoDeploy.tsx b/web/packages/teleport/src/Discover/Database/DeployService/AutoDeploy/AutoDeploy.tsx
index 5251f9248f808..3006b0233c948 100644
--- a/web/packages/teleport/src/Discover/Database/DeployService/AutoDeploy/AutoDeploy.tsx
+++ b/web/packages/teleport/src/Discover/Database/DeployService/AutoDeploy/AutoDeploy.tsx
@@ -110,6 +110,9 @@ export function AutoDeploy({ toggleDeployMethod }: DeployServiceProp) {
   }
 
   function handleDeploy(validator) {
+    setSvcDeployedAwsUrl('');
+    setDeployFinished(false);
+
     if (!validator.validate()) {
       return;
     }
@@ -125,8 +128,6 @@ export function AutoDeploy({ toggleDeployMethod }: DeployServiceProp) {
 
     if (wantAutoDiscover) {
       setAttempt({ status: 'processing' });
-      setSvcDeployedAwsUrl('');
-      setDeployFinished(false);
 
       integrationService
         .deployDatabaseServices(integrationName, {
@@ -153,8 +154,6 @@ export function AutoDeploy({ toggleDeployMethod }: DeployServiceProp) {
         });
     } else {
       setAttempt({ status: 'processing' });
-      setSvcDeployedAwsUrl('');
-      setDeployFinished(false);
       integrationService
         .deployAwsOidcService(integrationName, {
           deploymentMode: 'database-service',
diff --git a/web/packages/teleport/src/services/integrations/types.ts b/web/packages/teleport/src/services/integrations/types.ts
index bbd6e30ba8801..80dd390f5ec9a 100644
--- a/web/packages/teleport/src/services/integrations/types.ts
+++ b/web/packages/teleport/src/services/integrations/types.ts
@@ -325,7 +325,6 @@ export type AwsOidcDeployServiceRequest = {
   region: Regions;
   subnetIds: string[];
   taskRoleArn: string;
-  databaseAgentMatcherLabels?: Label[];
   securityGroups?: string[];
   vpcId: string;
   accountId: string;