feat(message): tusimple theme danger type

src/message/index.ts

@@ -1,2 +1,7 @@
 export { default as NMessagePrivider } from './src/MessageProvider'
 export { useMessage } from './src/use-message'
+export type {
+  MessageOptions,
+  MessageApiInjection,
+  MessageReactive
+} from './src/MessageProvider'

src/message/src/MessageProvider.ts

@@ -23,11 +23,11 @@ export interface MessageOptions {
 }
 
 export interface MessageApiInjection {
-  info: (content: string, options: MessageOptions) => void
-  success: (content: string, options: MessageOptions) => void
-  warning: (content: string, options: MessageOptions) => void
-  error: (content: string, options: MessageOptions) => void
-  loading: (content: string, options: MessageOptions) => void
+  info: (content: string, options: MessageOptions) => MessageReactive
+  success: (content: string, options: MessageOptions) => MessageReactive
+  warning: (content: string, options: MessageOptions) => MessageReactive
+  error: (content: string, options: MessageOptions) => MessageReactive
+  loading: (content: string, options: MessageOptions) => MessageReactive
 }
 
 export interface MessageReactive {

src/message/src/use-message.ts

@@ -1,6 +1,9 @@
 import { inject } from 'vue'
+import { throwError } from '../../_utils'
 import type { MessageApiInjection } from './MessageProvider'
 
-export function useMessage (): MessageApiInjection | undefined {
-  return inject<MessageApiInjection>('message')
+export function useMessage (): MessageApiInjection {
+  const api = inject<MessageApiInjection>('message')
+  if (api === undefined) { throwError('use-message', 'No out <n-message-provider /> founded.') }
+  return api
 }

themes/tusimple/src/index.ts

@@ -1,2 +1,3 @@
 export { default as TsConfigProvider } from './ts-config-provider'
 export { useDialog } from './use-ts-dialog'
+export { useMessage } from './use-ts-message'

themes/tusimple/src/use-ts-message.ts

@@ -0,0 +1,28 @@
+import {
+  useMessage as _useMessage,
+  MessageOptions,
+  MessageReactive,
+  MessageApiInjection
+} from 'naive-ui'
+import { icons } from './icons'
+
+export interface ExtendedApi {
+  danger: (content: string, options: MessageOptions) => MessageReactive
+}
+
+export type TsMessageApi = MessageApiInjection & ExtendedApi
+
+function useMessage (): TsMessageApi {
+  const messageApi = _useMessage()
+  const extendedApi: ExtendedApi = {
+    danger: (content: string, options: MessageOptions) => {
+      return messageApi.error(content, {
+        ...options,
+        icon: icons.warning
+      })
+    }
+  }
+  return Object.assign(extendedApi, messageApi)
+}
+
+export { useMessage }