nitro-graphql 1.1.0 → 1.1.2

package/dist/ecosystem/nuxt.d.ts

@@ -1,7 +1,7 @@
-import * as _nuxt_schema5 from "@nuxt/schema";
+import * as _nuxt_schema0 from "@nuxt/schema";
 
 //#region src/ecosystem/nuxt.d.ts
 interface ModuleOptions {}
-declare const _default: _nuxt_schema5.NuxtModule<ModuleOptions, ModuleOptions, false>;
+declare const _default: _nuxt_schema0.NuxtModule<ModuleOptions, ModuleOptions, false>;
 //#endregion
 export { ModuleOptions, _default as default };

package/dist/index.d.ts

@@ -1,8 +1,8 @@
 import { StandardSchemaV1 } from "./types/standard-schema.js";
 import { CodegenClientConfig, CodegenServerConfig, GenImport, GenericSdkConfig, NitroGraphQLOptions } from "./types/index.js";
-import * as nitropack0 from "nitropack";
+import * as nitropack3 from "nitropack";
 
 //#region src/index.d.ts
-declare const _default: nitropack0.NitroModule;
+declare const _default: nitropack3.NitroModule;
 //#endregion
 export { CodegenClientConfig, CodegenServerConfig, GenImport, GenericSdkConfig, NitroGraphQLOptions, StandardSchemaV1, _default as default };

package/dist/index.js

@@ -1,7 +1,8 @@
-import { relativeWithDot, scanDocs, scanResolvers, scanSchemas } from "./utils/index.js";
+import { relativeWithDot, scanDirectives, scanDocs, scanResolvers, scanSchemas } from "./utils/index.js";
 import { clientTypeGeneration, serverTypeGeneration } from "./utils/type-generation.js";
 import { rollupConfig } from "./rollup.js";
-import { existsSync, mkdirSync, writeFileSync } from "node:fs";
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
+import { readFile } from "node:fs/promises";
 import { fileURLToPath } from "node:url";
 import { watch } from "chokidar";
 import consola from "consola";
@@ -62,7 +63,7 @@ var src_default = defineNitroModule({
 		const watcher = watch(watchDirs, {
 			persistent: true,
 			ignoreInitial: true,
-			ignored: nitro.options.ignore
+			ignored: [...nitro.options.ignore, "**/server/graphql/_directives.graphql"]
 		}).on("all", async (event, path) => {
 			if (path.endsWith(".graphql") || path.endsWith(".gql")) await clientTypeGeneration(nitro);
 		});
@@ -78,11 +79,115 @@ var src_default = defineNitroModule({
 		nitro.scanDocuments = docs;
 		const resolvers = await scanResolvers(nitro);
 		nitro.scanResolvers = resolvers;
+		const directives = await scanDirectives(nitro);
+		nitro.scanDirectives = directives;
+		if (directives.length > 0) {
+			const directiveSchemas = [];
+			for (const dir of directives) for (const _imp of dir.imports) {
+				const fileContent = await readFile(dir.specifier, "utf-8");
+				const nameMatch = fileContent.match(/name:\s*['"`](\w+)['"`]/);
+				const locationsMatch = fileContent.match(/locations:\s*\[([\s\S]*?)\]/);
+				const argsMatch = fileContent.match(/args:\s*\{([\s\S]*?)\}\s*,\s*(?:description|transformer)/);
+				if (nameMatch && locationsMatch) {
+					const name = nameMatch[1];
+					const locations = locationsMatch?.[1]?.split(",").map((l) => l.trim().replace(/['"`]/g, "")).filter(Boolean).join(" | ") || "";
+					let args = "";
+					if (argsMatch && argsMatch[1] && argsMatch[1].trim()) {
+						const argDefs = [];
+						const argMatches = argsMatch[1].matchAll(/(\w+):\s*\{([^}]+)\}/g);
+						for (const argMatch of argMatches) {
+							const argName = argMatch[1];
+							const argBody = argMatch[2];
+							const typeMatch = argBody?.match(/type:\s*['"`](\[[\w!]+\]|\w+)['"`]/);
+							const type = typeMatch ? typeMatch[1] : "String";
+							const defaultMatch = argBody?.match(/defaultValue:\s*(['"`]([^'"`]+)['"`]|(\d+)|true|false)/);
+							let defaultValue = "";
+							if (defaultMatch) {
+								const value = defaultMatch[2] || defaultMatch[3] || defaultMatch[1]?.replace(/['"`]/g, "");
+								if (type === "String") defaultValue = ` = "${value}"`;
+								else if (type === "Int" || type === "Float") defaultValue = ` = ${value}`;
+								else if (type === "Boolean") defaultValue = ` = ${value}`;
+							}
+							argDefs.push(`${argName}: ${type}${defaultValue}`);
+						}
+						if (argDefs.length > 0) args = `(${argDefs.join(", ")})`;
+					}
+					directiveSchemas.push(`directive @${name}${args} on ${locations}`);
+				}
+			}
+			if (directiveSchemas.length > 0) {
+				const directivesPath = resolve(nitro.graphql.serverDir, "_directives.graphql");
+				const content = `# WARNING: This file is auto-generated by nitro-graphql
+# Do not modify this file directly. It will be overwritten.
+# To define custom directives, create .directive.ts files using defineDirective()
+
+${directiveSchemas.join("\n\n")}`;
+				let shouldWrite = true;
+				if (existsSync(directivesPath)) {
+					const existingContent = readFileSync(directivesPath, "utf-8");
+					shouldWrite = existingContent !== content;
+				}
+				if (shouldWrite) writeFileSync(directivesPath, content, "utf-8");
+				if (!nitro.scanSchemas.includes(directivesPath)) nitro.scanSchemas.push(directivesPath);
+			}
+		}
 		nitro.hooks.hook("dev:start", async () => {
 			const schemas$1 = await scanSchemas(nitro);
 			nitro.scanSchemas = schemas$1;
 			const resolvers$1 = await scanResolvers(nitro);
 			nitro.scanResolvers = resolvers$1;
+			const directives$1 = await scanDirectives(nitro);
+			nitro.scanDirectives = directives$1;
+			if (directives$1.length > 0) {
+				const directiveSchemas = [];
+				for (const dir of directives$1) for (const _imp of dir.imports) {
+					const fileContent = await readFile(dir.specifier, "utf-8");
+					const nameMatch = fileContent.match(/name:\s*['"`](\w+)['"`]/);
+					const locationsMatch = fileContent.match(/locations:\s*\[([\s\S]*?)\]/);
+					const argsMatch = fileContent.match(/args:\s*\{([\s\S]*?)\}\s*,\s*(?:description|transformer)/);
+					if (nameMatch && locationsMatch) {
+						const name = nameMatch[1];
+						const locations = locationsMatch?.[1]?.split(",").map((l) => l.trim().replace(/['"`]/g, "")).filter(Boolean).join(" | ") || "";
+						let args = "";
+						if (argsMatch && argsMatch[1] && argsMatch[1].trim()) {
+							const argDefs = [];
+							const argMatches = argsMatch[1].matchAll(/(\w+):\s*\{([^}]+)\}/g);
+							for (const argMatch of argMatches) {
+								const argName = argMatch[1];
+								const argBody = argMatch[2];
+								const typeMatch = argBody?.match(/type:\s*['"`](\[?\w+!?\]?)['"`]/);
+								const type = typeMatch ? typeMatch[1] : "String";
+								const defaultMatch = argBody?.match(/defaultValue:\s*(['"`]([^'"`]+)['"`]|(\d+)|true|false)/);
+								let defaultValue = "";
+								if (defaultMatch) {
+									const value = defaultMatch[2] || defaultMatch[3] || defaultMatch[1]?.replace(/['"`]/g, "");
+									if (type === "String") defaultValue = ` = "${value}"`;
+									else if (type === "Int" || type === "Float") defaultValue = ` = ${value}`;
+									else if (type === "Boolean") defaultValue = ` = ${value}`;
+								}
+								argDefs.push(`${argName}: ${type}${defaultValue}`);
+							}
+							if (argDefs.length > 0) args = `(${argDefs.join(", ")})`;
+						}
+						directiveSchemas.push(`directive @${name}${args} on ${locations}`);
+					}
+				}
+				if (directiveSchemas.length > 0) {
+					const directivesPath = resolve(nitro.graphql.serverDir, "_directives.graphql");
+					const content = `# WARNING: This file is auto-generated by nitro-graphql
+# Do not modify this file directly. It will be overwritten.
+# To define custom directives, create .directive.ts files using defineDirective()
+
+${directiveSchemas.join("\n\n")}`;
+					let shouldWrite = true;
+					if (existsSync(directivesPath)) {
+						const existingContent = readFileSync(directivesPath, "utf-8");
+						shouldWrite = existingContent !== content;
+					}
+					if (shouldWrite) writeFileSync(directivesPath, content, "utf-8");
+					if (!nitro.scanSchemas.includes(directivesPath)) nitro.scanSchemas.push(directivesPath);
+				}
+			}
 			const docs$1 = await scanDocs(nitro);
 			nitro.scanDocuments = docs$1;
 		});
@@ -125,7 +230,8 @@ var src_default = defineNitroModule({
 					"defineSubscription",
 					"defineType",
 					"defineGraphQLConfig",
-					"defineSchema"
+					"defineSchema",
+					"defineDirective"
 				]
 			});
 		}
@@ -195,17 +301,26 @@ export default defineGraphQLConfig({
 // },
 })
 `, "utf-8");
-		if (!existsSync(join(nitro.graphql.serverDir, "context.d.ts"))) writeFileSync(join(nitro.graphql.serverDir, "context.d.ts"), `// Example context definition please change it to your needs
-import type { H3EventContext as OriginalH3EventContext } from 'h3'
-
-export interface ExtendedH3EventContext extends OriginalH3EventContext {
-  // useDatabase: () => Database
-  // tables: typeof import('~~/server/drizzle/schema/index')
-}
+		if (!existsSync(join(nitro.graphql.serverDir, "context.ts"))) writeFileSync(join(nitro.graphql.serverDir, "context.ts"), `// Example context definition - please change it to your needs
+// import type { Database } from '../utils/useDb'
 
 declare module 'h3' {
-  interface H3EventContext extends ExtendedH3EventContext {}
+  interface H3EventContext {
+    // Add your custom context properties here
+    // useDatabase: () => Database
+    // tables: typeof import('../drizzle/schema')
+    // auth?: {
+    //   user?: {
+    //     id: string
+    //     role: 'admin' | 'user'
+    //   }
+    // }
+  }
 }`, "utf-8");
+		if (existsSync(join(nitro.graphql.serverDir, "context.d.ts"))) {
+			consola.warn("nitro-graphql: Found context.d.ts file. Please rename it to context.ts for the new structure.");
+			consola.info("The context file should now be context.ts instead of context.d.ts");
+		}
 	}
 });
 

package/dist/rollup.js

@@ -1,7 +1,7 @@
 import { getImportId, scanGraphql } from "./utils/index.js";
 import { clientTypeGeneration, serverTypeGeneration } from "./utils/type-generation.js";
-import { resolve } from "pathe";
 import { readFile } from "node:fs/promises";
+import { resolve } from "pathe";
 import { parse } from "graphql";
 import { genImport } from "knitwork";
 
@@ -9,6 +9,7 @@ import { genImport } from "knitwork";
 async function rollupConfig(app) {
 	virtualSchemas(app);
 	virtualResolvers(app);
+	virtualDirectives(app);
 	getGraphQLConfig(app);
 	app.hooks.hook("rollup:before", (nitro, rollupConfig$1) => {
 		rollupConfig$1.plugins = rollupConfig$1.plugins || [];
@@ -85,6 +86,29 @@ function virtualResolvers(app) {
 		return code;
 	};
 }
+function virtualDirectives(app) {
+	const getDirectives = () => {
+		const directives = [...app.scanDirectives || []];
+		return directives;
+	};
+	app.options.virtual ??= {};
+	app.options.virtual["#nitro-internal-virtual/server-directives"] = () => {
+		const imports = getDirectives();
+		const importsContent = [...imports.map(({ specifier, imports: imports$1, options }) => {
+			return genImport(specifier, imports$1, options);
+		})];
+		const data = imports.map(({ imports: imports$1 }) => imports$1.map((i) => `{ directive: ${i.as} }`).join(",\n")).filter(Boolean).join(",\n");
+		const content = [
+			"export const directives = [",
+			data,
+			"]",
+			""
+		];
+		content.unshift(...importsContent);
+		const code = content.join("\n");
+		return code;
+	};
+}
 function getGraphQLConfig(app) {
 	const configPath = resolve(app.graphql.serverDir, "config.ts");
 	app.options.virtual ??= {};

package/dist/routes/apollo-server.js

@@ -2,21 +2,27 @@ import { startServerAndCreateH3Handler } from "../utils/apollo.js";
 import defu from "defu";
 import { mergeResolvers, mergeTypeDefs } from "@graphql-tools/merge";
 import { importedConfig } from "#nitro-internal-virtual/graphql-config";
+import { directives } from "#nitro-internal-virtual/server-directives";
 import { resolvers } from "#nitro-internal-virtual/server-resolvers";
 import { schemas } from "#nitro-internal-virtual/server-schemas";
 import { ApolloServer } from "@apollo/server";
 import { ApolloServerPluginLandingPageLocalDefault } from "@apollo/server/plugin/landingPage/default";
+import { makeExecutableSchema } from "@graphql-tools/schema";
 
 //#region src/routes/apollo-server.ts
 function createMergedSchema() {
 	try {
-		const mergedSchemas = schemas.map((schema) => schema.def).join("\n\n");
+		const mergedSchemas = schemas.map((schema$1) => schema$1.def).join("\n\n");
 		const typeDefs = mergeTypeDefs([mergedSchemas]);
 		const mergedResolvers = mergeResolvers(resolvers.map((r) => r.resolver));
-		return {
+		let schema = makeExecutableSchema({
 			typeDefs,
 			resolvers: mergedResolvers
-		};
+		});
+		if (directives && directives.length > 0) {
+			for (const { directive } of directives) if (directive.transformer) schema = directive.transformer(schema);
+		}
+		return schema;
 	} catch (error) {
 		console.error("Schema merge error:", error);
 		throw error;
@@ -25,10 +31,9 @@ function createMergedSchema() {
 let apolloServer = null;
 function createApolloServer() {
 	if (!apolloServer) {
-		const { typeDefs, resolvers: mergedResolvers } = createMergedSchema();
+		const schema = createMergedSchema();
 		apolloServer = new ApolloServer(defu({
-			typeDefs,
-			resolvers: mergedResolvers,
+			schema,
 			introspection: true,
 			plugins: [ApolloServerPluginLandingPageLocalDefault({ embed: true })]
 		}, importedConfig));

package/dist/routes/graphql-yoga.d.ts

@@ -1,6 +1,6 @@
-import * as h33 from "h3";
+import * as h31 from "h3";
 
 //#region src/routes/graphql-yoga.d.ts
-declare const _default: h33.EventHandler<h33.EventHandlerRequest, Promise<Response>>;
+declare const _default: h31.EventHandler<h31.EventHandlerRequest, Promise<Response>>;
 //#endregion
 export { _default as default };

package/dist/routes/graphql-yoga.js

@@ -1,10 +1,12 @@
 import defu from "defu";
 import { mergeResolvers, mergeTypeDefs } from "@graphql-tools/merge";
 import { importedConfig } from "#nitro-internal-virtual/graphql-config";
+import { directives } from "#nitro-internal-virtual/server-directives";
 import { resolvers } from "#nitro-internal-virtual/server-resolvers";
 import { schemas } from "#nitro-internal-virtual/server-schemas";
+import { makeExecutableSchema } from "@graphql-tools/schema";
 import { defineEventHandler, toWebRequest } from "h3";
-import { createSchema, createYoga } from "graphql-yoga";
+import { createYoga } from "graphql-yoga";
 
 //#region src/routes/graphql-yoga.ts
 const apolloSandboxHtml = `<!DOCTYPE html>
@@ -26,13 +28,17 @@ new window.EmbeddedSandbox({
 </html>`;
 function createMergedSchema() {
 	try {
-		const mergedSchemas = schemas.map((schema) => schema.def).join("\n\n");
+		const mergedSchemas = schemas.map((schema$1) => schema$1.def).join("\n\n");
 		const typeDefs = mergeTypeDefs([mergedSchemas]);
 		const mergedResolvers = mergeResolvers(resolvers.map((r) => r.resolver));
-		return createSchema({
+		let schema = makeExecutableSchema({
 			typeDefs,
 			resolvers: mergedResolvers
 		});
+		if (directives && directives.length > 0) {
+			for (const { directive } of directives) if (directive.transformer) schema = directive.transformer(schema);
+		}
+		return schema;
 	} catch (error) {
 		console.error("Schema merge error:", error);
 		throw error;

package/dist/routes/health.d.ts

@@ -1,7 +1,7 @@
-import * as h31 from "h3";
+import * as h34 from "h3";
 
 //#region src/routes/health.d.ts
-declare const _default: h31.EventHandler<h31.EventHandlerRequest, Promise<{
+declare const _default: h34.EventHandler<h34.EventHandlerRequest, Promise<{
   status: string;
   message: string;
   timestamp: string;

package/dist/types/index.d.ts

@@ -20,7 +20,7 @@ type CodegenClientConfig = TypeScriptPluginConfig & TypeScriptDocumentsPluginCon
 interface IESMImport {
   name: string;
   as?: string;
-  type: 'resolver' | 'query' | 'mutation' | 'type' | 'subscription';
+  type: 'resolver' | 'query' | 'mutation' | 'type' | 'subscription' | 'directive';
 }
 interface GenImport {
   specifier: string;
@@ -32,6 +32,7 @@ declare module 'nitropack/types' {
     scanSchemas: string[];
     scanDocuments: string[];
     scanResolvers: GenImport[];
+    scanDirectives: GenImport[];
     graphql: {
       buildDir: string;
       watchDirs: string[];

package/dist/utils/define.d.ts

@@ -1,3 +1,4 @@
+import { GraphQLSchema } from "graphql";
 import { ApolloServerOptions } from "@apollo/server";
 import { H3Event } from "h3";
 import { YogaServerOptions } from "graphql-yoga";
@@ -17,5 +18,55 @@ declare function defineSubscription(resolvers?: Resolvers['Subscription']): Reso
 declare function defineType(resolvers: Resolvers): Resolvers;
 type DefineServerConfig<T extends NPMConfig = NPMConfig> = T['framework'] extends 'graphql-yoga' ? Partial<YogaServerOptions<H3Event, Partial<H3Event>>> : T['framework'] extends 'apollo-server' ? Partial<ApolloServerOptions<H3Event>> : Partial<YogaServerOptions<H3Event, Partial<H3Event>>> | Partial<ApolloServerOptions<H3Event>>;
 declare function defineGraphQLConfig<T extends NPMConfig = NPMConfig>(config: Partial<DefineServerConfig<T>>): Partial<DefineServerConfig<T>>;
+type DirectiveLocationName = 'QUERY' | 'MUTATION' | 'SUBSCRIPTION' | 'FIELD' | 'FRAGMENT_DEFINITION' | 'FRAGMENT_SPREAD' | 'INLINE_FRAGMENT' | 'VARIABLE_DEFINITION' | 'SCHEMA' | 'SCALAR' | 'OBJECT' | 'FIELD_DEFINITION' | 'ARGUMENT_DEFINITION' | 'INTERFACE' | 'UNION' | 'ENUM' | 'ENUM_VALUE' | 'INPUT_OBJECT' | 'INPUT_FIELD_DEFINITION';
+type GraphQLScalarType = 'String' | 'Int' | 'Float' | 'Boolean' | 'ID' | 'JSON' | 'DateTime';
+type GraphQLBaseType = GraphQLScalarType | (string & {});
+type GraphQLArgumentType = 'String' | 'Int' | 'Float' | 'Boolean' | 'ID' | 'JSON' | 'DateTime' | 'String!' | 'Int!' | 'Float!' | 'Boolean!' | 'ID!' | 'JSON!' | 'DateTime!' | '[String]' | '[String!]' | '[String]!' | '[String!]!' | '[Int]' | '[Int!]' | '[Int]!' | '[Int!]!' | '[Float]' | '[Float!]' | '[Float]!' | '[Float!]!' | '[Boolean]' | '[Boolean!]' | '[Boolean]!' | '[Boolean!]!' | '[ID]' | '[ID!]' | '[ID]!' | '[ID!]!' | '[JSON]' | '[JSON!]' | '[JSON]!' | '[JSON!]!' | '[DateTime]' | '[DateTime!]' | '[DateTime]!' | '[DateTime!]!' | (string & {});
+interface DirectiveArgument<T extends GraphQLArgumentType = GraphQLArgumentType> {
+  /**
+   * GraphQL type for the argument
+   * @example 'String', 'Int!', '[String!]!', 'DateTime', 'JSON'
+   */
+  type: T;
+  defaultValue?: any;
+  description?: string;
+}
+interface DirectiveArg {
+  type: GraphQLArgumentType;
+  defaultValue?: any;
+  description?: string;
+}
+interface DirectiveDefinition {
+  name: string;
+  locations: DirectiveLocationName[];
+  args?: Record<string, DirectiveArg>;
+  description?: string;
+  isRepeatable?: boolean;
+  transformer?: (schema: GraphQLSchema) => GraphQLSchema;
+}
+interface DefineDirectiveConfig {
+  name: string;
+  locations: ReadonlyArray<DirectiveLocationName>;
+  args?: Record<string, {
+    type: GraphQLArgumentType;
+    defaultValue?: any;
+    description?: string;
+  }>;
+  description?: string;
+  isRepeatable?: boolean;
+  transformer?: (schema: GraphQLSchema) => GraphQLSchema;
+}
+/**
+ * Helper function to create directive arguments with proper type inference
+ * @example
+ * args: {
+ *   myArg: arg('String!', { defaultValue: 'hello' })
+ * }
+ */
+declare function arg<T extends GraphQLArgumentType>(type: T, options?: {
+  defaultValue?: any;
+  description?: string;
+}): DirectiveArgument<T>;
+declare function defineDirective(config: DefineDirectiveConfig): DirectiveDefinition;
 //#endregion
-export { DefineServerConfig, ResolverQuery, defineGraphQLConfig, defineMutation, defineQuery, defineResolver, defineSchema, defineSubscription, defineType };
+export { DefineDirectiveConfig, DefineServerConfig, DirectiveArgument, DirectiveDefinition, GraphQLArgumentType, GraphQLBaseType, GraphQLScalarType, ResolverQuery, arg, defineDirective, defineGraphQLConfig, defineMutation, defineQuery, defineResolver, defineSchema, defineSubscription, defineType };

package/dist/utils/define.js

@@ -20,6 +20,38 @@ function defineType(resolvers) {
 function defineGraphQLConfig(config) {
 	return config;
 }
+/**
+* Helper function to create directive arguments with proper type inference
+* @example
+* args: {
+*   myArg: arg('String!', { defaultValue: 'hello' })
+* }
+*/
+function arg(type, options) {
+	return {
+		type,
+		...options
+	};
+}
+function defineDirective(config) {
+	const args = config.args ? Object.entries(config.args).map(([name, arg$1]) => {
+		const defaultValue = arg$1.defaultValue !== void 0 ? ` = ${JSON.stringify(arg$1.defaultValue)}` : "";
+		return `${name}: ${arg$1.type}${defaultValue}`;
+	}).join(", ") : "";
+	const argsString = args ? `(${args})` : "";
+	const locations = config.locations.join(" | ");
+	const schemaDefinition = `directive @${config.name}${argsString} on ${locations}`;
+	Object.defineProperty(config, "__schema", {
+		value: schemaDefinition,
+		enumerable: false,
+		configurable: false,
+		writable: false
+	});
+	return {
+		...config,
+		locations: [...config.locations]
+	};
+}
 
 //#endregion
-export { defineGraphQLConfig, defineMutation, defineQuery, defineResolver, defineSchema, defineSubscription, defineType };
+export { arg, defineDirective, defineGraphQLConfig, defineMutation, defineQuery, defineResolver, defineSchema, defineSubscription, defineType };

package/dist/utils/index.d.ts

@@ -7,7 +7,7 @@ declare function getImportId(p: string, lazy?: boolean): string;
 declare function relativeWithDot(from: string, to: string): string;
 declare function scanGraphql(nitro: Nitro): Promise<string[]>;
 declare function scanResolvers(nitro: Nitro): Promise<GenImport[]>;
-declare function scanDirectives(nitro: Nitro): Promise<string[]>;
+declare function scanDirectives(nitro: Nitro): Promise<GenImport[]>;
 declare function scanTypeDefs(nitro: Nitro): Promise<string[]>;
 declare function scanSchemas(nitro: Nitro): Promise<string[]>;
 declare function scanDocs(nitro: Nitro): Promise<string[]>;

package/dist/utils/index.js

@@ -1,5 +1,5 @@
-import { join, relative } from "pathe";
 import { readFile } from "node:fs/promises";
+import { join, relative } from "pathe";
 import { hash } from "ohash";
 import { parseAsync } from "oxc-parser";
 import { glob } from "tinyglobby";
@@ -56,6 +56,11 @@ async function scanResolvers(nitro) {
 						type: "subscription",
 						as: `_${hash(decl.id.name + file.path).replace(/-/g, "").slice(0, 6)}`
 					});
+					if (decl.init.callee.type === "Identifier" && decl.init.callee.name === "defineDirective") exports.imports.push({
+						name: decl.id.name,
+						type: "directive",
+						as: `_${hash(decl.id.name + file.path).replace(/-/g, "").slice(0, 6)}`
+					});
 				}
 			}
 		}
@@ -65,7 +70,28 @@ async function scanResolvers(nitro) {
 }
 async function scanDirectives(nitro) {
 	const files = await scanFiles(nitro, "graphql", "**/*.directive.{ts,js}");
-	return files.map((f) => f.fullPath);
+	const exportName = [];
+	for (const file of files) {
+		const fileContent = await readFile(file.fullPath, "utf-8");
+		const parsed = await parseAsync(file.fullPath, fileContent);
+		const exports = {
+			imports: [],
+			specifier: file.fullPath
+		};
+		for (const node of parsed.program.body) if (node.type === "ExportNamedDeclaration" && node.declaration && node.declaration.type === "VariableDeclaration") {
+			for (const decl of node.declaration.declarations) if (decl.type === "VariableDeclarator" && decl.init && decl.id.type === "Identifier") {
+				if (decl.init && decl.init.type === "CallExpression") {
+					if (decl.init.callee.type === "Identifier" && decl.init.callee.name === "defineDirective") exports.imports.push({
+						name: decl.id.name,
+						type: "directive",
+						as: `_${hash(decl.id.name + file.path).replace(/-/g, "").slice(0, 6)}`
+					});
+				}
+			}
+		}
+		if (exports.imports.length > 0) exportName.push(exports);
+	}
+	return exportName;
 }
 async function scanTypeDefs(nitro) {
 	const files = await scanFiles(nitro, "graphql", "**/*.typedef.{ts,js}");

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "nitro-graphql",
   "type": "module",
-  "version": "1.1.0",
+  "version": "1.1.2",
   "description": "GraphQL integration for Nitro",
   "license": "MIT",
   "sideEffects": false,