All files / modules/20-rbac/modals/TokenModal/views CreateToken.tsx

100% Statements 30/30
83.33% Branches 5/6
100% Functions 2/2
100% Lines 30/30

Press n or j to go to the next uncovered block, b, p or k for the previous block.

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              11x   11x 11x 11x 11x 11x 11x   11x                       11x 6x 6x 6x 6x 6x 6x 6x 6x   6x 2x 2x 2x 2x   2x 2x 2x 2x 2x 2x             6x                     11x  
/*
 * 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 } from 'react'
import type { ModalErrorHandlerBinding } from '@wings-software/uicore'
import { omit } from 'lodash-es'
import { useParams } from 'react-router-dom'
import { useToaster } from '@common/components'
import { useStrings } from 'framework/strings'
import { TokenDTO, useCreateToken } from 'services/cd-ng'
import useRBACError from '@rbac/utils/useRBACError/useRBACError'
import type { ProjectPathProps } from '@common/interfaces/RouteInterfaces'
import TokenForm, { TokenFormData } from './TokenForm'
 
export interface TokenModalProps {
  data?: TokenDTO
  apiKeyIdentifier: string
  isEdit?: boolean
  apiKeyType?: TokenDTO['apiKeyType']
  parentIdentifier?: string
  onSubmit: () => void
  onClose: () => void
}
 
const CreateTokenForm: React.FC<TokenModalProps> = props => {
  const { onSubmit } = props
  const { accountId } = useParams<ProjectPathProps>()
  const { getRBACErrorMessage } = useRBACError()
  const { getString } = useStrings()
  const { showSuccess } = useToaster()
  const [modalErrorHandler, setModalErrorHandler] = useState<ModalErrorHandlerBinding>()
  const { mutate: createToken, loading: saving } = useCreateToken({ queryParams: { accountIdentifier: accountId } })
  const [token, setToken] = useState<string>()
 
  const handleSubmit = async (values: TokenFormData): Promise<void> => {
    modalErrorHandler?.hide()
    const dataToSubmit = Object.assign({}, omit(values, ['expiryDate']))
    /* istanbul ignore else */ if (values['expiryDate']) {
      dataToSubmit['validTo'] = Date.parse(values['expiryDate'])
    }
    try {
      const created = await createToken(dataToSubmit)
      /* istanbul ignore else */ if (created) {
        showSuccess(getString('rbac.token.form.createSuccess', { name: values.name }))
        onSubmit()
        setToken(created.data)
      }
    } catch (e) {
      /* istanbul ignore next */
      modalErrorHandler?.showDanger(getRBACErrorMessage(e))
    }
  }
  return (
    <TokenForm
      {...props}
      onSubmit={handleSubmit}
      token={token}
      setModalErrorHandler={setModalErrorHandler}
      loading={saving}
    />
  )
}
 
export default CreateTokenForm