From 9b775894ddffe69e345f1506f7d4ebd322211e23 Mon Sep 17 00:00:00 2001
From: Benjamin Canac <canacb1@gmail.com>
Date: Fri, 17 Oct 2025 10:58:32 +0200
Subject: [PATCH] feat(module): add `component-meta:schema` hook

---
 playground/pages/index.vue |  2 +-
 src/module.ts              |  7 +++++--
 src/parser/meta-parser.ts  | 12 +++++++++---
 src/types/module.ts        |  1 +
 src/types/parser.ts        |  1 +
 src/utils/unplugin.ts      | 11 ++++++-----
 6 files changed, 23 insertions(+), 11 deletions(-)

diff --git a/playground/pages/index.vue b/playground/pages/index.vue
index 0a8f53e..8fa8ccc 100644
--- a/playground/pages/index.vue
+++ b/playground/pages/index.vue
@@ -1,7 +1,7 @@
 <template>
   <div>
     <div>
-      <TestComponent foo="test" />
+      <TestComponent foo="test" name="test" />
       <TestGlobalComponent />
       <TestTyped :hello="`test`" />
     </div>
diff --git a/src/module.ts b/src/module.ts
index 5376d25..7dda6b8 100644
--- a/src/module.ts
+++ b/src/module.ts
@@ -86,7 +86,7 @@ export default defineNuxtModule<ModuleOptions>({
           const slots = slotNames
             .trim()
             .split(',')
-            .map(s => s.trim().split(':')[0].trim())
+            .map(s => s.trim().split(':')[0]?.trim())
             .map(s => `<slot name="${s}" />`)
           code = code.replace(/<template>/, `<template>\n${slots.join('\n')}\n`)
         }
@@ -137,7 +137,10 @@ export default defineNuxtModule<ModuleOptions>({
       ...options,
       components: [],
       metaSources: {},
-      transformers
+      transformers,
+      beforeWrite: async (schema: NuxtComponentMeta) => {
+        return await nuxt.callHook('component-meta:schema' as any, schema) || schema
+      }
     }
 
     // Resolve loaded components
diff --git a/src/parser/meta-parser.ts b/src/parser/meta-parser.ts
index 82c250d..8e9c8f5 100644
--- a/src/parser/meta-parser.ts
+++ b/src/parser/meta-parser.ts
@@ -9,7 +9,6 @@ import type { ComponentMetaParserOptions, NuxtComponentMeta } from '../types/par
 import { defu } from 'defu'
 import { refineMeta } from './utils'
 
-
 export function useComponentMetaParser (
   {
     outputDir = join(process.cwd(), '.component-meta/'),
@@ -21,7 +20,8 @@ export function useComponentMetaParser (
     transformers = [],
     debug = false,
     metaFields,
-    metaSources = {}
+    metaSources = {},
+    beforeWrite
   }: ComponentMetaParserOptions
 ) {
   /**
@@ -116,8 +116,14 @@ export function useComponentMetaParser (
   /**
    * Write the output file.
    */
-  const updateOutput = (content?: string) => {
+  const updateOutput = async (content?: string) => {
     const path = outputPath + '.mjs'
+
+    // Call beforeWrite hook if provided
+    if (beforeWrite && !content) {
+      components = await beforeWrite(components)
+    }
+
     if (!existsSync(dirname(path))) { fs.mkdirSync(dirname(path), { recursive: true }) }
     if (existsSync(path)) { fs.unlinkSync(path) }
     fs.writeFileSync(
diff --git a/src/types/module.ts b/src/types/module.ts
index e41e181..16d63f2 100644
--- a/src/types/module.ts
+++ b/src/types/module.ts
@@ -72,4 +72,5 @@ export interface ModuleOptions {
 export interface ModuleHooks {
   'component-meta:transformers'(data: TransformersHookData): void
   'component-meta:extend'(data: ExtendHookData): void
+  'component-meta:schema'(schema: NuxtComponentMeta): NuxtComponentMeta | Promise<NuxtComponentMeta>
 }
diff --git a/src/types/parser.ts b/src/types/parser.ts
index 87d1618..936b619 100644
--- a/src/types/parser.ts
+++ b/src/types/parser.ts
@@ -5,6 +5,7 @@ import type { ModuleOptions } from './module'
 export type ComponentMetaParserOptions = Omit<ModuleOptions, 'components' | 'metaSources'> & {
   components: Component[]
   metaSources?: NuxtComponentMeta
+  beforeWrite?: (schema: NuxtComponentMeta) => Promise<NuxtComponentMeta> | NuxtComponentMeta
 }
 export type ComponentData = Omit<Component, 'filePath' | 'shortPath'> & {
   meta: ComponentMeta
diff --git a/src/utils/unplugin.ts b/src/utils/unplugin.ts
index 3b1f2b7..874cb9d 100644
--- a/src/utils/unplugin.ts
+++ b/src/utils/unplugin.ts
@@ -1,6 +1,7 @@
 import { createUnplugin } from 'unplugin'
 import { useComponentMetaParser } from '../parser/meta-parser'
-import type { ComponentMetaParser, ComponentMetaParserOptions } from '../parser/meta-parser';
+import type { ComponentMetaParser } from '../parser/meta-parser'
+import type { ComponentMetaParserOptions } from '../types/parser'
 
 type ComponentMetaUnpluginOptions = { parser?: ComponentMetaParser, parserOptions: ComponentMetaParserOptions }
 
@@ -12,14 +13,14 @@ export const metaPlugin = createUnplugin<ComponentMetaUnpluginOptions>(({ parser
     return {
       name: 'vite-plugin-nuxt-component-meta',
       enforce: 'post',
-      buildStart () {
+      async buildStart () {
         // avoid parsing meta twice in SSR
         if (_configResolved?.build.ssr) {
           return
         }
 
         instance?.fetchComponents()
-        instance?.updateOutput()
+        await instance?.updateOutput()
       },
       buildEnd () {
         if (!_configResolved?.env.DEV && _configResolved?.env.PROD) {
@@ -32,10 +33,10 @@ export const metaPlugin = createUnplugin<ComponentMetaUnpluginOptions>(({ parser
         configResolved (config) {
           _configResolved = config
         },
-        handleHotUpdate ({ file }) {
+        async handleHotUpdate ({ file }) {
           if (instance && Object.entries(instance.components).some(([, comp]: any) => comp.fullPath === file)) {
             instance.fetchComponent(file)
-            instance.updateOutput()
+            await instance.updateOutput()
           }
         }
       }