DataCenterCDC-DS/dolphinscheduler-ui-next/src/views/security/worker-group-manage/components/use-modal.ts

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { reactive, ref, SetupContext } from 'vue'
import { useI18n } from 'vue-i18n'
import { useAsyncState } from '@vueuse/core'
import {
  queryWorkerAddressList,
  saveWorkerGroup
} from '@/service/modules/worker-groups'

export function useModal(
  props: any,
  ctx: SetupContext<('cancelModal' | 'confirmModal')[]>
) {
  const { t } = useI18n()

  const variables = reactive({
    workerGroupFormRef: ref(),
    model: {
      id: ref<number>(-1),
      name: ref(''),
      addrList: ref<Array<number>>([]),
      generalOptions: []
    },
    rules: {
      name: {
        required: true,
        trigger: ['input', 'blur'],
        validator() {
          if (variables.model.name === '') {
            return new Error(t('security.worker_group.group_name_tips'))
          }
        }
      },
      addrList: {
        required: true,
        trigger: ['input', 'blur'],
        validator() {
          if (variables.model.addrList.length < 1) {
            return new Error(t('security.worker_group.worker_addresses_tips'))
          }
        }
      }
    }
  })

  const getListData = () => {
    const { state } = useAsyncState(
      queryWorkerAddressList().then((res: Array<string>) => {
        variables.model.generalOptions = res.map(
          (item): { label: string; value: string } => {
            return {
              label: item,
              value: item
            }
          }
        ) as any
      }),
      {}
    )

    return state
  }

  const handleValidate = (statusRef: number) => {
    variables.workerGroupFormRef.validate((errors: any) => {
      if (!errors) {
        statusRef === 0 ? submitWorkerGroupModal() : updateWorkerGroupModal()
      } else {
        return
      }
    })
  }

  const submitWorkerGroupModal = () => {
    const data = {
      name: variables.model.name,
      addrList: variables.model.addrList.toString()
    }

    saveWorkerGroup(data).then(() => {
      variables.model.name = ''
      variables.model.addrList = []
      ctx.emit('confirmModal', props.showModalRef)
    })
  }

  const updateWorkerGroupModal = () => {
    const data = {
      id: variables.model.id,
      name: variables.model.name,
      addrList: variables.model.addrList.toString()
    }

    saveWorkerGroup(data).then(() => {
      ctx.emit('confirmModal', props.showModalRef)
    })
  }

  return {
    variables,
    handleValidate,
    getListData
  }
}