All files / modules/75-ce/components/Connectors/AWSCOConnector/steps TestConnectionStep.tsx

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
/*
 * Copyright 2021 Harness Inc. All rights reserved.
 * Use of this source code is governed by the PolyForm Shield 1.0.0 license
 * that can be found in the licenses directory at the root of this repository, also available at
 * https://polyformproject.org/wp-content/uploads/2020/06/PolyForm-Shield-1.0.0.txt.
 */
 
import React, { useState, useEffect } from 'react'
import { useParams } from 'react-router-dom'
import {
  Layout,
  StepProps,
  Heading,
  Container,
  StepsProgress,
  ModalErrorHandler,
  ModalErrorHandlerBinding,
  Button
} from '@wings-software/uicore'
import { Intent } from '@harness/design-system'
import type { ConnectorInfoDTO, ConnectorConfigDTO } from 'services/cd-ng'
import { useGetTestConnectionResult } from 'services/cd-ng'
import { useStrings } from 'framework/strings'
import css from './Steps.module.scss'
 
export enum Status {
  // WAIT = 'WAIT',
  PROCESS = 'PROCESS',
  DONE = 'DONE',
  ERROR = 'ERROR'
}
 
interface TestConnectionStepProps extends StepProps<ConnectorInfoDTO> {
  onSuccess: (data: ConnectorInfoDTO) => void
  onFailure: () => void
}
 
const TestConnectionStep: React.FC<StepProps<ConnectorConfigDTO> & TestConnectionStepProps> = props => {
  const { accountId } = useParams<{ accountId: string; orgIdentifier: string }>()
  const { prevStepData, onFailure } = props
  const [steps, setSteps] = useState<string[]>([])
  const [currentStep] = useState(1)
  const [currentStatus, setCurrentStatus] = useState<Status>(Status.ERROR)
  const [currentIntent, setCurrentIntent] = useState<Intent>(Intent.NONE)
  const connectorDetails = prevStepData as ConnectorInfoDTO
  const { getString } = useStrings()
 
  const [modalErrorHandler, setModalErrorHandler] = useState<ModalErrorHandlerBinding | undefined>()
 
  const { mutate: reloadTestConnection } = useGetTestConnectionResult({
    identifier: connectorDetails.identifier,
    queryParams: {
      accountIdentifier: accountId
      // orgIdentifier: orgIdentifier,
      // projectIdentifier: connectorDetails.projectIdentifier
    },
    requestOptions: {
      headers: {
        'content-type': 'application/json'
      }
    }
  })
 
  const verifyOptimizationPermissions = async (): Promise<void> => {
    const newSteps = [...steps]
    newSteps.push(getString('ce.connector.AWS.testConnection.crossARN.valid'))
    setSteps(newSteps)
 
    try {
      setCurrentStatus(Status.PROCESS)
      const result = await reloadTestConnection()
      if (result.data?.status === 'SUCCESS') {
        setCurrentIntent(Intent.SUCCESS)
        setCurrentStatus(Status.DONE)
      } else {
        throw new Error('Could not verify connectivity')
      }
    } catch (e) {
      modalErrorHandler?.showDanger(e.data?.errorSummary || e.message)
      setCurrentStatus(Status.ERROR)
      setCurrentIntent(Intent.DANGER)
    }
  }
  useEffect(() => {
    verifyOptimizationPermissions()
  }, [])
 
  const handleSuccess = () => {
    props.onSuccess(connectorDetails)
  }
  return (
    <Layout.Vertical>
      <ModalErrorHandler bind={setModalErrorHandler} />
      <Container padding="medium">
        <Heading level={2} font={{ weight: 'bold' }}>
          {'Test Connection'}
        </Heading>
      </Container>
      <StepsProgress steps={steps} intent={currentIntent} current={currentStep} currentStatus={currentStatus} />
      {currentIntent == Intent.SUCCESS && currentStatus === Status.DONE && (
        <Button
          intent="primary"
          text={getString('ce.connector.AWS.testConnection.finish')}
          onClick={handleSuccess}
          className={css.nextButton}
        />
      )}
      {currentStatus === Status.ERROR && (
        <Button
          intent="primary"
          text={getString('ce.connector.AWS.crossAccountRole.submitText')}
          onClick={() => onFailure()}
          className={css.nextButton}
        />
      )}
    </Layout.Vertical>
  )
}
 
export default TestConnectionStep