feat: keep custom block type when block has lang

src/index.ts

@@ -478,16 +478,19 @@ export default function vue(opts: Partial<VuePluginOptions> = {}): Plugin {
 
         descriptor.customBlocks.forEach((block, index) => {
           if (!isAllowed(block.type)) return
+          const vuePartRequest = createVuePartRequest(
+            filename,
+            (typeof block.attrs.lang === 'string' && block.attrs.lang) ||
+              createVuePartRequest.defaultLang[block.type],
+            'customBlocks',
+            index,
+            block.type
+          )
           result.code +=
             '\n' +
-            `export * from '${createVuePartRequest(
-              filename,
-              (typeof block.attrs.lang === 'string' && block.attrs.lang) ||
-                createVuePartRequest.defaultLang[block.type] ||
-                block.type,
-              'customBlocks',
-              index
-            )}'`
+            `export * from '${vuePartRequest}'\n` +
+            `import __custom_block_${index}__ from '${vuePartRequest}'\n` +
+            `__custom_block_${index}__(__vue_component__)`
         })
 
         dT(

src/utils.ts

@@ -22,7 +22,13 @@ export interface VuePartRequestMeta {
 }
 
 export interface VuePartRequestCreator {
-  (filename: string, lang: string, type: string, index?: number): string
+  (
+    filename: string,
+    lang: string,
+    type: string,
+    index?: number,
+    blockType?: string
+  ): string
 
   defaultLang: {
     [key: string]: string
@@ -68,15 +74,16 @@ export const createVuePartRequest: VuePartRequestCreator = ((
   filename: string,
   lang: string | undefined,
   type: string,
-  index?: number
+  index?: number,
+  blockType?: string
 ): string => {
   lang = lang || createVuePartRequest.defaultLang[type]
 
   const match = GET_QUERY.exec(filename)
 
   const query = match ? queryString.parse(match[2]) : {}
 
-  query[PARAM_NAME] = [type, index, lang]
+  query[PARAM_NAME] = [type, index, blockType, lang]
     .filter(it => it !== undefined)
     .join('.')