All files / modules/70-pipeline/components/Notifications useNotificationModal.tsx

90.24% Statements 37/41
50% Branches 12/24
75% Functions 6/8
90% Lines 36/40

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 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119              25x 25x 25x 25x 25x 25x   25x 25x 25x 25x 25x 25x   25x                           25x 25x 25x     25x           9x 9x 9x 9x 9x       9x   2x                                                                   1x 1x 1x               9x   2x 2x 2x 1x 1x 2x         9x   2x        
/*
 * 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 { isNil } from 'lodash-es'
import { StepWizard, Button, MultiSelectOption } from '@wings-software/uicore'
import { useModalHook } from '@harness/use-modal'
import { Dialog, Classes } from '@blueprintjs/core'
import cx from 'classnames'
import type { NotificationRules } from 'services/pipeline-ng'
import { useStrings } from 'framework/strings'
import Overview from './Steps/Overview'
import PipelineEvents from './Steps/PipelineEvents'
import NotificationMethods from './Steps/NotificationMethods'
import { Actions } from './NotificationUtils'
import { NotificationTypeSelectOptions } from './NotificationTypeOptions'
 
import css from './useNotificationModal.module.scss'
 
export interface UseNotificationModalProps {
  onCloseModal?: () => void
  onCreateOrUpdate?: (data?: NotificationRules, index?: number, action?: Actions) => void
  stagesOptions?: MultiSelectOption[]
  getExistingNotificationNames?: (skipIndex?: number) => string[]
}
 
export interface UseNotificationModalReturn {
  openNotificationModal: (NotificationRules?: NotificationRules, index?: number) => void
  closeNotificationModal: () => void
}
 
enum Views {
  CREATE,
  EDIT
}
 
export const useNotificationModal = ({
  onCreateOrUpdate,
  onCloseModal,
  stagesOptions,
  getExistingNotificationNames
}: UseNotificationModalProps): UseNotificationModalReturn => {
  const [view, setView] = useState(Views.CREATE)
  const [index, setIndex] = useState<number>()
  const [notificationRules, setNotificationRules] = useState<NotificationRules>()
  const { getString } = useStrings()
  const wizardCompleteHandler = async (wizardData?: NotificationRules): Promise<void> => {
    onCreateOrUpdate?.(wizardData, index, view === Views.CREATE ? Actions.Added : Actions.Update)
  }
 
  const [showModal, hideModal] = useModalHook(
    () => (
      <Dialog
        isOpen={true}
        enforceFocus={false}
        onClose={() => {
          setView(Views.CREATE)
          hideModal()
          onCloseModal?.()
        }}
        className={cx(Classes.DIALOG, css.dialog)}
      >
        <StepWizard<NotificationRules>
          onCompleteWizard={wizardCompleteHandler}
          icon="new-notification"
          iconProps={{ color: 'white', size: 50 }}
          title={isNil(index) ? getString('newNotification') : getString('editNotification')}
          stepClassName={css.stepWizardContainer}
        >
          <Overview
            name={getString('overview')}
            data={notificationRules}
            existingNotificationNames={getExistingNotificationNames?.(index)}
          />
          <PipelineEvents name={getString('notifications.pipelineEvents')} stagesOptions={stagesOptions} />
          <NotificationMethods
            name={getString('notifications.notificationMethod')}
            typeOptions={NotificationTypeSelectOptions}
          />
        </StepWizard>
        <Button
          minimal
          icon="cross"
          className={css.crossIcon}
          iconProps={{ size: 18 }}
          onClick={() => {
            setView(Views.CREATE)
            hideModal()
            onCloseModal?.()
          }}
        />
      </Dialog>
    ),
    [view, notificationRules, wizardCompleteHandler]
  )
 
  const open = useCallback(
    (_notification?: NotificationRules, _index?: number) => {
      setNotificationRules(_notification)
      setIndex(_index)
      if (_notification) {
        setView(Views.EDIT)
      } else setView(Views.CREATE)
      showModal()
    },
    [showModal]
  )
 
  return {
    openNotificationModal: (_notificationRules?: NotificationRules, _index?: number) =>
      open(_notificationRules, _index),
    closeNotificationModal: hideModal
  }
}