From b9fc5f762d9270ae98a879f778f516256d4d07b0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=89=E5=92=B2=E6=99=BA=E5=AD=90=20Kevin=20Deng?=
 <sxzz@sxzz.moe>
Date: Wed, 30 Aug 2023 21:54:30 +0800
Subject: [PATCH] feat(antdv): customize prefix whatever you want (#616)

Co-authored-by: yuzheng14 <422450117@qq.com>
---
 src/core/resolvers/antdv.ts | 20 +++++++++++++-------
 1 file changed, 13 insertions(+), 7 deletions(-)

diff --git a/src/core/resolvers/antdv.ts b/src/core/resolvers/antdv.ts
index 16f64d37..739b72c9 100644
--- a/src/core/resolvers/antdv.ts
+++ b/src/core/resolvers/antdv.ts
@@ -212,6 +212,11 @@ export interface AntDesignVueResolverOptions {
    * @default 'ant-design-vue'
    */
   packageName?: string
+  /**
+   * customize prefix of component
+   * @default 'A'
+   */
+  prefix?: string
 }
 
 function getStyleDir(compName: string): string {
@@ -251,12 +256,13 @@ function getSideEffects(compName: string, options: AntDesignVueResolverOptions):
   }
 }
 const primitiveNames = ['Affix', 'Anchor', 'AnchorLink', 'AutoComplete', 'AutoCompleteOptGroup', 'AutoCompleteOption', 'Alert', 'Avatar', 'AvatarGroup', 'BackTop', 'Badge', 'BadgeRibbon', 'Breadcrumb', 'BreadcrumbItem', 'BreadcrumbSeparator', 'Button', 'ButtonGroup', 'Calendar', 'Card', 'CardGrid', 'CardMeta', 'Collapse', 'CollapsePanel', 'Carousel', 'Cascader', 'Checkbox', 'CheckboxGroup', 'Col', 'Comment', 'ConfigProvider', 'DatePicker', 'MonthPicker', 'WeekPicker', 'RangePicker', 'QuarterPicker', 'Descriptions', 'DescriptionsItem', 'Divider', 'Dropdown', 'DropdownButton', 'Drawer', 'Empty', 'Form', 'FormItem', 'FormItemRest', 'Grid', 'Input', 'InputGroup', 'InputPassword', 'InputSearch', 'Textarea', 'Image', 'ImagePreviewGroup', 'InputNumber', 'Layout', 'LayoutHeader', 'LayoutSider', 'LayoutFooter', 'LayoutContent', 'List', 'ListItem', 'ListItemMeta', 'Menu', 'MenuDivider', 'MenuItem', 'MenuItemGroup', 'SubMenu', 'Mentions', 'MentionsOption', 'Modal', 'Statistic', 'StatisticCountdown', 'PageHeader', 'Pagination', 'Popconfirm', 'Popover', 'Progress', 'Radio', 'RadioButton', 'RadioGroup', 'Rate', 'Result', 'Row', 'Select', 'SelectOptGroup', 'SelectOption', 'Skeleton', 'SkeletonButton', 'SkeletonAvatar', 'SkeletonInput', 'SkeletonImage', 'Slider', 'Space', 'Spin', 'Steps', 'Step', 'Switch', 'Table', 'TableColumn', 'TableColumnGroup', 'TableSummary', 'TableSummaryRow', 'TableSummaryCell', 'Transfer', 'Tree', 'TreeNode', 'DirectoryTree', 'TreeSelect', 'TreeSelectNode', 'Tabs', 'TabPane', 'Tag', 'CheckableTag', 'TimePicker', 'TimeRangePicker', 'Timeline', 'TimelineItem', 'Tooltip', 'Typography', 'TypographyLink', 'TypographyParagraph', 'TypographyText', 'TypographyTitle', 'Upload', 'UploadDragger', 'LocaleProvider', 'FloatButton', 'FloatButtonGroup', 'Qrcode', 'Watermark', 'Segmented', 'Tour', 'SpaceCompact', 'StyleProvider']
-const prefix = 'A'
 
 let antdvNames: Set<string>
 
 function genAntdNames(primitiveNames: string[]): void {
-  antdvNames = new Set(primitiveNames.map(name => `${prefix}${name}`))
+  // use primitiveNames to construct antdvNames,
+  // in order to make options.resolvePrefix compatible
+  antdvNames = new Set(primitiveNames)
 }
 genAntdNames(primitiveNames)
 
@@ -285,6 +291,7 @@ function getImportName(compName: string): string {
 export function AntDesignVueResolver(options: AntDesignVueResolverOptions = {
 
 }): ComponentResolver {
+  const originPrefix = options.prefix ?? 'A'
   return {
     type: 'component',
     resolve: (name: string) => {
@@ -295,15 +302,14 @@ export function AntDesignVueResolver(options: AntDesignVueResolverOptions = {
         }
       }
 
-      if (isAntdv(name) && !options?.exclude?.includes(name)) {
-        const importName = name.slice(1)
-        // console.log('importName----', importName)
+      const [compName, prefix] = [name.slice(originPrefix.length), name.slice(0, originPrefix.length)]
+      if (prefix === originPrefix && isAntdv(compName) && !options?.exclude?.includes(compName)) {
         const { cjs = false, packageName = 'ant-design-vue' } = options
         const path = `${packageName}/${cjs ? 'lib' : 'es'}`
         return {
-          name: getImportName(importName),
+          name: getImportName(compName),
           from: path,
-          sideEffects: getSideEffects(importName, options),
+          sideEffects: getSideEffects(compName, options),
         }
       }
     },