import { CheckCircleFilled, CloseCircleFilled, InfoCircleFilled } from '@ant-design/icons-vue'; import { useI18n } from '@eciol/locale'; import { isString } from '@eciol/shared'; import { App, message as Message, Modal as _Modal, notification } from 'ant-design-vue'; import type { ModalStaticFunctions } from 'ant-design-vue/lib/modal/confirm'; import type { ModalFunc, ModalFuncProps } from 'ant-design-vue/lib/modal/Modal'; import { ConfigProps, NotificationArgsProps } from 'ant-design-vue/lib/notification'; export interface NotifyApi { info(config: NotificationArgsProps): void; success(config: NotificationArgsProps): void; error(config: NotificationArgsProps): void; warn(config: NotificationArgsProps): void; warning(config: NotificationArgsProps): void; open(args: NotificationArgsProps): void; close(key: string): void; config(options: ConfigProps): void; destroy(): void; } export declare type NotificationPlacement = 'topLeft' | 'topRight' | 'bottomLeft' | 'bottomRight'; export declare type IconType = 'success' | 'info' | 'error' | 'warning'; export interface ModalOptionsEx extends Omit { iconType: 'warning' | 'success' | 'error' | 'info'; } export type ModalOptionsPartial = Partial & Pick; interface ConfirmOptions { info: ModalFunc; success: ModalFunc; error: ModalFunc; warn: ModalFunc; warning: ModalFunc; } let Modal: Omit = _Modal as any; function getIcon(iconType: string) { if (iconType === 'warning') { return ; } else if (iconType === 'success') { return ; } else if (iconType === 'info') { return ; } else { return ; } } function renderContent({ content }: Pick) { if (isString(content)) { return
${content as string}
`}>; } else { return content; } } /** * @description: Create confirmation box */ function createConfirm(options: ModalOptionsEx): ConfirmOptions { const iconType = options.iconType || 'warning'; Reflect.deleteProperty(options, 'iconType'); const opt: ModalFuncProps = { centered: true, icon: getIcon(iconType), ...options, content: renderContent(options), }; return Modal.confirm(opt) as unknown as ConfirmOptions; } const getBaseOptions = () => { const { t } = useI18n(); return { okText: t('common.okText'), centered: true, }; }; function createModalOptions(options: ModalOptionsPartial, icon: string): ModalOptionsPartial { return { ...getBaseOptions(), ...options, content: renderContent(options), icon: getIcon(icon), }; } function createSuccessModal(options: ModalOptionsPartial) { return Modal.success(createModalOptions(options, 'success')); } function createErrorModal(options: ModalOptionsPartial) { return Modal.error(createModalOptions(options, 'error')); } function createInfoModal(options: ModalOptionsPartial) { return Modal.info(createModalOptions(options, 'info')); } function createWarningModal(options: ModalOptionsPartial) { return Modal.warning(createModalOptions(options, 'warning')); } notification.config({ placement: 'topRight', duration: 3, }); /** * @description: message */ export function useMessage(_App?: any) { const { message = Message, modal = _Modal, notification: notifyApi = notification, } = (_App ?? App).useApp() ?? {}; Modal = modal as any; return { createMessage: message, notification: notifyApi as NotifyApi, createConfirm, createSuccessModal, createErrorModal, createInfoModal, createWarningModal, Modal, }; }