feat: add modal to auto import

wiidede · Nov 20, 2023 · d2902d2 · d2902d2
1 parent d624154
commit d2902d2
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 3 deletions.
diff --git a/playground/pages/index.vue b/playground/pages/index.vue
@@ -10,6 +10,14 @@ function handleMessage() {
 function handleNotification() {
   ANotification.success('This is a success message!')
 }
+function handleModal() {
+  AModal.info({
+    title: 'Info',
+    content: 'This is a info message',
+  })
+}
+
+const visible = ref(false)
 </script>
 
 <template>
@@ -23,6 +31,12 @@ function handleNotification() {
     <a-button status="success" @click="handleNotification">
       Success Notification
     </a-button>
+    <a-button @click="handleModal">
+      Modal Function
+    </a-button>
+    <a-button @click="visible = true">
+      Modal Template
+    </a-button>
   </div>
   <a-divider />
   <a-row :gutter="24">
@@ -53,4 +67,10 @@ function handleNotification() {
       </a-card>
     </a-col>
   </a-row>
+  <AModal v-model:visible="visible">
+    <template #title>
+      h
+    </template>
+    modal template
+  </AModal>
 </template>
diff --git a/src/config.ts b/src/config.ts
@@ -20,12 +20,13 @@ export const allHooks: Options['hooks'] = [
 export const allImportsWithStyle: Options['imports'] = [
   'Notification',
   'Message',
+  'Modal',
 ]
 
 export const allComponents = Object.keys(AllComponents).filter(name =>
   !allLocales.includes(name)
   && !allHooks.includes(name as any)
-  && !allImportsWithStyle.includes(name as any)
+  && !['Notification', 'Message'].includes(name as any)
   && !['install', 'default'].includes(name),
 )
 

diff --git a/src/types.ts b/src/types.ts
@@ -54,10 +54,10 @@ export interface Options extends TransformOptions {
    *
    * @default
    * ```ts
-   *  ['Notification', 'Message']
+   *  ['Notification', 'Message', 'Modal']
    * ```
    */
-  imports: false | ('Notification' | 'Message')[]
+  imports: false | ('Notification' | 'Message' | 'Modal')[]
 
   /**
    * Prefix of the import name.