feat: improve hooks

src/runtime/server/routes/llms.txt.get.ts

@@ -8,9 +8,7 @@ export default eventHandler(async (event) => {
 
   const llms = JSON.parse(JSON.stringify(options))
 
-  const sections = llms.sections || []
-
-  await llmsHooks.callHook('llms', event, llms, sections)
+  await llmsHooks.callHook('generate', event, llms)
 
   const document = [
     `# ${llms.title || 'Documentation'}`,
@@ -20,7 +18,7 @@ export default eventHandler(async (event) => {
     document.push(`> ${options.description}`)
   }
 
-  for (const section of sections) {
+  for (const section of llms.sections) {
     document.push(`## ${section.title}`)
     if (section.description) {
       document.push(section.description)

src/runtime/server/routes/llms_full.txt.get.ts

@@ -9,7 +9,7 @@ export default eventHandler(async (event) => {
   const contents = [] as string[]
   const llms = JSON.parse(JSON.stringify(options))
 
-  await llmsHooks.callHook('llms:full', event, llms, contents)
+  await llmsHooks.callHook('generate:full', event, llms, contents)
 
   const document = [
     `# ${llms.llmsFull?.title || 'Documentation'}`,

src/runtime/server/utils/hooks.ts

@@ -3,8 +3,22 @@ import type { H3Event } from 'h3'
 import type { ModuleOptions } from '@nuxtjs/llms'
 
 export interface LLMSHooks {
-  'llms': (event: H3Event, options: ModuleOptions, sections: ModuleOptions['sections']) => void
-  'llms:full': (event: H3Event, options: ModuleOptions, contents: string[]) => void
+  'generate': (event: H3Event, options: ModuleOptions) => void
+  'generate:full': (event: H3Event, options: ModuleOptions, contents: string[]) => void
 }
 
 export const llmsHooks = createHooks<LLMSHooks>()
+
+/**
+ * Run a callback when LLMs is being generated.
+ */
+export function onLLMsGenerate(cb: LLMSHooks['generate']) {
+  return llmsHooks.hook('generate', cb)
+}
+
+/**
+ * Run a callback when Full LLMs is being generated.
+ */
+export function onLLMsGenerateFull(cb: LLMSHooks['generate:full']) {
+  return llmsHooks.hook('generate:full', cb)
+}