All files / modules/20-rbac/modals/UserGroupModal useUserGroupModal.tsx

92.31% Statements 24/26
60% Branches 6/10
100% Functions 6/6
92% Lines 23/25

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              11x 11x 11x     11x 11x                       11x 17x 17x 17x 17x 1x     1x     1x     17x   1x           1x 1x               17x   1x 1x 1x         17x 1x        
/*
 * 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 { Dialog } from '@wings-software/uicore'
import { useModalHook } from '@harness/use-modal'
 
import type { UserGroupDTO } from 'services/cd-ng'
import UserGroupForm from '@rbac/modals/UserGroupModal/views/UserGroupForm'
import { useStrings } from 'framework/strings'
 
export interface UseUserGroupModalProps {
  onSuccess: () => void
  onCloseModal?: () => void
}
 
export interface UseUserGroupModalReturn {
  openUserGroupModal: (userGroup?: UserGroupDTO, _isAddMember?: boolean) => void
  closeUserGroupModal: () => void
}
 
export const useUserGroupModal = ({ onSuccess }: UseUserGroupModalProps): UseUserGroupModalReturn => {
  const [userGroupData, setUserGroupData] = useState<UserGroupDTO>()
  const [isAddMember, setIsAddMember] = useState<boolean>(false)
  const { getString } = useStrings()
  const getTitle = (): string => {
    Iif (!!userGroupData && !isAddMember) {
      return getString('rbac.userGroupPage.editUserGroup')
    }
    Iif (isAddMember) {
      return getString('rbac.userGroupPage.addMembers')
    }
    return getString('rbac.userGroupPage.newUserGroup')
  }
 
  const [showModal, hideModal] = useModalHook(
    () => (
      <Dialog isOpen={true} enforceFocus={false} title={getTitle()} onClose={hideModal}>
        <UserGroupForm
          data={userGroupData}
          isEdit={!!userGroupData && !isAddMember}
          isAddMember={isAddMember}
          onSubmit={() => {
            onSuccess()
            hideModal()
          }}
          onCancel={hideModal}
        />
      </Dialog>
    ),
    [userGroupData, isAddMember]
  )
  const open = useCallback(
    (_userGroup?: UserGroupDTO, _isAddMember?: boolean) => {
      setUserGroupData(_userGroup)
      setIsAddMember(_isAddMember || false)
      showModal()
    },
    [showModal]
  )
 
  return {
    openUserGroupModal: (userGroup?: UserGroupDTO, _isAddMember?: boolean) => open(userGroup, _isAddMember),
    closeUserGroupModal: hideModal
  }
}