All files / modules/20-rbac/modals/TokenModal useTokenModal.tsx

100% Statements 27/27
100% Branches 8/8
100% Functions 4/4
100% Lines 26/26

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 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              11x 11x 11x   11x 11x 11x 11x 11x                         11x 39x 39x 39x   39x 4x 1x 3x 1x   2x     39x   4x                                                                             39x   4x 4x 4x 4x         39x          
/*
 * 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, { useCallback, useState } from 'react'
import { useModalHook } from '@harness/use-modal'
import { Dialog } from '@harness/uicore'
import type { TokenDTO } from 'services/cd-ng'
import { StringKeys, String } from 'framework/strings'
import RotateTokenForm from '@rbac/modals/TokenModal/views/RotateTokenForm'
import EditTokenForm from '@rbac/modals/TokenModal/views/EditToken'
import CreateTokenForm from '@rbac/modals/TokenModal/views/CreateToken'
import css from './useTokenModal.module.scss'
export interface useTokenModalProps {
  onSuccess: () => void
  onCloseModal?: () => void
  apiKeyType?: TokenDTO['apiKeyType']
  parentIdentifier?: string
}
 
export interface useTokenModalReturn {
  openTokenModal: (apiKeyIdentifier: string, token?: TokenDTO, _isRotate?: boolean) => void
  closeTokenModal: () => void
}
 
export const useTokenModal = ({ onSuccess, apiKeyType, parentIdentifier }: useTokenModalProps): useTokenModalReturn => {
  const [apiKeyIdentifier, setApiKeyIdentifier] = useState<string>('')
  const [isRotate, setIsRotate] = useState<boolean>()
  const [tokenData, setTokenData] = useState<TokenDTO>()
 
  const getLabelKey = (): StringKeys => {
    if (isRotate) {
      return 'rbac.token.rotateLabel'
    } else if (tokenData) {
      return 'rbac.token.editLabel'
    } else {
      return 'rbac.token.createLabel'
    }
  }
  const [showModal, hideModal] = useModalHook(
    () => (
      <Dialog
        isOpen={true}
        enforceFocus={false}
        onClose={hideModal}
        className={css.modal}
        title={<String stringID={getLabelKey()} />}
      >
        {isRotate ? (
          <RotateTokenForm
            data={tokenData}
            apiKeyIdentifier={apiKeyIdentifier}
            onSubmit={onSuccess}
            onClose={hideModal}
          />
        ) : tokenData ? (
          <EditTokenForm
            data={tokenData}
            apiKeyIdentifier={apiKeyIdentifier}
            isEdit={true}
            apiKeyType={apiKeyType}
            parentIdentifier={parentIdentifier}
            onSubmit={onSuccess}
            onClose={hideModal}
          />
        ) : (
          <CreateTokenForm
            data={tokenData}
            apiKeyIdentifier={apiKeyIdentifier}
            isEdit={false}
            apiKeyType={apiKeyType}
            parentIdentifier={parentIdentifier}
            onSubmit={onSuccess}
            onClose={hideModal}
          />
        )}
      </Dialog>
    ),
    [tokenData, apiKeyIdentifier, isRotate, parentIdentifier]
  )
  const open = useCallback(
    (_apiKeyIdentifier: string, _token?: TokenDTO, _isRotate?: boolean) => {
      setApiKeyIdentifier(_apiKeyIdentifier)
      setTokenData(_token)
      setIsRotate(_isRotate)
      showModal()
    },
    [showModal]
  )
 
  return {
    openTokenModal: open,
    closeTokenModal: hideModal
  }
}