MTP2.0_NEWMANAGE_WEB/src/hooks/enum/index.ts

import { reactive, toRefs, onUnmounted } from 'vue'
import { enumType } from '@/services/api/common'

export function useEnum(code: string) {
    const state = reactive({
        loading: true,
        enums: <Model.EnumTypeRsp[]>[]
    })

    const controller = new AbortController()
    const readyCallbackMap = new Map<symbol, () => void>()

    // 获取枚举名称
    const getEnumTypeName = (value: number) => {
        const item = state.enums.find((e) => e.enumitemname === value)
        const { enumdicname = value?.toString() } = item ?? {}
        return enumdicname
    }

    // 获取枚举列表
    const getEnumOptions = (filtered: number | number[] = [], reverse = false) => {
        const valuesSet = new Set(Array.isArray(filtered) ? filtered : [filtered])
        const result = state.enums.map((e) => ({
            label: e.enumdicname,
            value: e.enumitemname
        }))
        if (valuesSet.size > 0) {
            if (reverse) {
                // 排除枚举
                return result.filter((e) => !valuesSet.has(e.value))
            } else {
                // 匹配枚举
                return result.filter((e) => valuesSet.has(e.value))
            }
        }
        return result
    }

    // 注册成功回调
    const registerEnumReadyCallback = (callback: () => void) => {
        if (state.loading) {
            const key = Symbol()
            readyCallbackMap.set(key, callback)
        } else {
            callback()
        }
    }

    // 获取枚举列表
    enumType({
        controller,
        data: {
            typeCode: code
        }
    }).then((res) => {
        state.enums = res.data
        readyCallbackMap.forEach((callback) => callback())
        readyCallbackMap.clear()
    })

    onUnmounted(() => {
        controller.abort()
        readyCallbackMap.clear()
    })

    return {
        ...toRefs(state),
        getEnumOptions,
        getEnumTypeName,
        registerEnumReadyCallback
    }
}