All files / modules/20-rbac/modals/LinkToSSOProviderModal useLinkToSSOProviderModal.tsx

91.67% Statements 22/24
100% Branches 2/2
75% Functions 6/8
91.3% Lines 21/23

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              10x 10x 10x 10x 10x     10x 10x 10x                       10x     9x 9x   2x                       1x 1x             1x 1x                                   9x   2x 2x         9x 2x        
/*
 * Copyright 2021 Harness Inc. All rights reserved.
 * Use of this source code is governed by the PolyForm Free Trial 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/05/PolyForm-Free-Trial-1.0.0.txt.
 */
 
import React, { useCallback, useState } from 'react'
import { Button } from '@wings-software/uicore'
import { useModalHook } from '@harness/use-modal'
import { Dialog, Classes } from '@blueprintjs/core'
import cx from 'classnames'
 
import type { UserGroupDTO } from 'services/cd-ng'
import LinkToSSOProviderForm from '@rbac/modals/LinkToSSOProviderModal/views/LinkToSSOProviderForm'
import UnlinkSSOProviderForm from '@rbac/modals/LinkToSSOProviderModal/views/UnlinkSSOProviderForm'
import css from './useLinkToSSOProviderModal.module.scss'
 
export interface UseLinkToSSOProviderModalProps {
  onSuccess: () => void
  onCloseModal?: () => void
}
 
export interface UseLinkToSSOProviderModalReturn {
  openLinkToSSOProviderModal: (userGroup: UserGroupDTO) => void
  closeLinkToSSOProviderModal: () => void
}
 
export const useLinkToSSOProviderModal = ({
  onSuccess
}: UseLinkToSSOProviderModalProps): UseLinkToSSOProviderModalReturn => {
  const [userGroupData, setUserGroupData] = useState<UserGroupDTO>({} as UserGroupDTO)
  const [showModal, hideModal] = useModalHook(
    () => (
      <Dialog
        isOpen={true}
        enforceFocus={false}
        onClose={() => {
          hideModal()
        }}
        className={cx(css.dialog, Classes.DIALOG)}
      >
        {userGroupData.ssoLinked ? (
          <UnlinkSSOProviderForm
            userGroupData={userGroupData}
            onSubmit={() => {
              onSuccess()
              hideModal()
            }}
          />
        ) : (
          <LinkToSSOProviderForm
            userGroupData={userGroupData}
            onSubmit={() => {
              onSuccess()
              hideModal()
            }}
          />
        )}
 
        <Button
          minimal
          icon="cross"
          iconProps={{ size: 18 }}
          onClick={() => {
            hideModal()
          }}
          className={css.crossIcon}
        />
      </Dialog>
    ),
    [userGroupData]
  )
  const open = useCallback(
    (_userGroup: UserGroupDTO) => {
      setUserGroupData(_userGroup)
      showModal()
    },
    [showModal]
  )
 
  return {
    openLinkToSSOProviderModal: (userGroup: UserGroupDTO) => open(userGroup),
    closeLinkToSSOProviderModal: hideModal
  }
}