@effect-gql/core 1.3.4 → 1.4.1

package/index.js

@@ -1,7 +1,7 @@
 import { GraphQLDirective, GraphQLEnumType, GraphQLInputObjectType, GraphQLInterfaceType, GraphQLObjectType, GraphQLUnionType, GraphQLSchema, GraphQLNonNull, GraphQLString, GraphQLFloat, GraphQLBoolean, GraphQLInt, GraphQLList, parse, GraphQLError, validate, execute as execute$1, Kind, specifiedRules, NoSchemaIntrospectionCustomRule, subscribe, GraphQLScalarType } from 'graphql';
 export { DirectiveLocation, GraphQLBoolean, GraphQLEnumType, GraphQLFloat, GraphQLID, GraphQLInputObjectType, GraphQLInt, GraphQLInterfaceType, GraphQLList, GraphQLNonNull, GraphQLObjectType, GraphQLScalarType, GraphQLSchema, GraphQLString, GraphQLUnionType, Kind, graphql, lexicographicSortSchema, printSchema } from 'graphql';
 import { Pipeable, Context, Data, Layer, Effect, Ref, HashMap, Config, Option, Schema, Runtime, Queue, Stream, Fiber, Deferred } from 'effect';
-import * as S2 from 'effect/Schema';
+import * as S from 'effect/Schema';
 import * as AST from 'effect/SchemaAST';
 import DataLoader from 'dataloader';
 import { HttpIncomingMessage, HttpServerResponse, HttpServerRequest, HttpRouter } from '@effect/platform';
@@ -21,6 +21,34 @@ var isIntegerType = (ast) => {
   }
   return false;
 };
+var isOptionDeclaration = (ast) => {
+  if (ast._tag === "Declaration") {
+    const annotations = ast.annotations;
+    if (annotations) {
+      const TypeConstructorSymbol = /* @__PURE__ */ Symbol.for("effect/annotation/TypeConstructor");
+      const typeConstructor = annotations[TypeConstructorSymbol];
+      if (typeConstructor && typeConstructor._tag === "effect/Option") {
+        return true;
+      }
+    }
+  }
+  return false;
+};
+var isOptionTransformation = (ast) => {
+  if (ast._tag === "Transformation") {
+    return isOptionDeclaration(ast.to);
+  }
+  return false;
+};
+var getOptionInnerType = (ast) => {
+  if (ast._tag === "Declaration") {
+    const typeParams = ast.typeParameters;
+    if (typeParams && typeParams.length > 0) {
+      return typeParams[0];
+    }
+  }
+  return void 0;
+};
 var toGraphQLType = (schema) => {
   const ast = schema.ast;
   if (ast._tag === "StringKeyword") return GraphQLString;
@@ -30,7 +58,7 @@ var toGraphQLType = (schema) => {
     if (isIntegerType(ast)) {
       return GraphQLInt;
     }
-    return toGraphQLType(S2.make(ast.from));
+    return toGraphQLType(S.make(ast.from));
   }
   if (ast._tag === "Literal") {
     if (typeof ast.literal === "string") return GraphQLString;
@@ -42,7 +70,7 @@ var toGraphQLType = (schema) => {
   if (ast._tag === "TupleType") {
     const elements = ast.elements;
     if (elements.length > 0) {
-      const elementSchema = S2.make(elements[0].type);
+      const elementSchema = S.make(elements[0].type);
       return new GraphQLList(toGraphQLType(elementSchema));
     }
   }
@@ -50,7 +78,8 @@ var toGraphQLType = (schema) => {
     const fields = {};
     for (const field2 of ast.propertySignatures) {
       const fieldName = String(field2.name);
-      const fieldSchema = S2.make(field2.type);
+      if (fieldName === "_tag") continue;
+      const fieldSchema = S.make(field2.type);
       let fieldType = toGraphQLType(fieldSchema);
       if (!field2.isOptional) {
         fieldType = new GraphQLNonNull(fieldType);
@@ -64,17 +93,35 @@ var toGraphQLType = (schema) => {
     });
   }
   if (ast._tag === "Transformation") {
-    return toGraphQLType(S2.make(ast.to));
+    if (isOptionTransformation(ast)) {
+      const innerType = getOptionInnerType(ast.to);
+      if (innerType) {
+        return toGraphQLType(S.make(innerType));
+      }
+    }
+    return toGraphQLType(S.make(ast.to));
+  }
+  if (ast._tag === "Declaration") {
+    if (isOptionDeclaration(ast)) {
+      const innerType = getOptionInnerType(ast);
+      if (innerType) {
+        return toGraphQLType(S.make(innerType));
+      }
+    }
+    const typeParams = ast.typeParameters;
+    if (typeParams && typeParams.length > 0) {
+      return toGraphQLType(S.make(typeParams[0]));
+    }
   }
   if (ast._tag === "Union") {
     const types = ast.types;
     if (types.length > 0) {
-      return toGraphQLType(S2.make(types[0]));
+      return toGraphQLType(S.make(types[0]));
     }
   }
   if (ast._tag === "Suspend") {
     const innerAst = ast.f();
-    return toGraphQLType(S2.make(innerAst));
+    return toGraphQLType(S.make(innerAst));
   }
   return GraphQLString;
 };
@@ -87,7 +134,7 @@ var toGraphQLInputType = (schema) => {
     if (isIntegerType(ast)) {
       return GraphQLInt;
     }
-    return toGraphQLInputType(S2.make(ast.from));
+    return toGraphQLInputType(S.make(ast.from));
   }
   if (ast._tag === "Literal") {
     if (typeof ast.literal === "string") return GraphQLString;
@@ -99,7 +146,7 @@ var toGraphQLInputType = (schema) => {
   if (ast._tag === "TupleType") {
     const elements = ast.elements;
     if (elements.length > 0) {
-      const elementSchema = S2.make(elements[0].type);
+      const elementSchema = S.make(elements[0].type);
       return new GraphQLList(toGraphQLInputType(elementSchema));
     }
   }
@@ -107,7 +154,8 @@ var toGraphQLInputType = (schema) => {
     const fields = {};
     for (const field2 of ast.propertySignatures) {
       const fieldName = String(field2.name);
-      const fieldSchema = S2.make(field2.type);
+      if (fieldName === "_tag") continue;
+      const fieldSchema = S.make(field2.type);
       let fieldType = toGraphQLInputType(fieldSchema);
       if (!field2.isOptional) {
         fieldType = new GraphQLNonNull(fieldType);
@@ -121,27 +169,56 @@ var toGraphQLInputType = (schema) => {
     });
   }
   if (ast._tag === "Transformation") {
-    return toGraphQLInputType(S2.make(ast.from));
+    return toGraphQLInputType(S.make(ast.from));
+  }
+  if (ast._tag === "Declaration") {
+    if (isOptionDeclaration(ast)) {
+      const innerType = getOptionInnerType(ast);
+      if (innerType) {
+        return toGraphQLInputType(S.make(innerType));
+      }
+    }
+    const typeParams = ast.typeParameters;
+    if (typeParams && typeParams.length > 0) {
+      return toGraphQLInputType(S.make(typeParams[0]));
+    }
   }
   if (ast._tag === "Union") {
     const types = ast.types;
+    const nonNullTypes = types.filter((t) => t._tag !== "Literal" || t.literal !== null).filter((t) => t._tag !== "UndefinedKeyword");
+    if (nonNullTypes.length > 0) {
+      return toGraphQLInputType(S.make(nonNullTypes[0]));
+    }
     if (types.length > 0) {
-      return toGraphQLInputType(S2.make(types[0]));
+      return toGraphQLInputType(S.make(types[0]));
     }
   }
   if (ast._tag === "Suspend") {
     const innerAst = ast.f();
-    return toGraphQLInputType(S2.make(innerAst));
+    return toGraphQLInputType(S.make(innerAst));
   }
   return GraphQLString;
 };
 var toGraphQLObjectType = (name, schema, additionalFields) => {
-  const ast = schema.ast;
+  let ast = schema.ast;
+  while (ast._tag === "Transformation") {
+    ast = ast.to;
+  }
+  if (ast._tag === "Declaration") {
+    const typeParams = ast.typeParameters;
+    if (typeParams && typeParams.length > 0) {
+      ast = typeParams[0];
+      while (ast._tag === "Transformation") {
+        ast = ast.to;
+      }
+    }
+  }
   if (ast._tag === "TypeLiteral") {
     const fields = {};
     for (const field2 of ast.propertySignatures) {
       const fieldName = String(field2.name);
-      const fieldSchema = S2.make(field2.type);
+      if (fieldName === "_tag") continue;
+      const fieldSchema = S.make(field2.type);
       let fieldType = toGraphQLType(fieldSchema);
       if (!field2.isOptional) {
         fieldType = new GraphQLNonNull(fieldType);
@@ -172,7 +249,8 @@ var toGraphQLArgs = (schema) => {
     const args = {};
     for (const field2 of ast.propertySignatures) {
       const fieldName = String(field2.name);
-      const fieldSchema = S2.make(field2.type);
+      if (fieldName === "_tag") continue;
+      const fieldSchema = S.make(field2.type);
       let fieldType = toGraphQLInputType(fieldSchema);
       if (!field2.isOptional) {
         fieldType = new GraphQLNonNull(fieldType);
@@ -272,9 +350,21 @@ function toGraphQLTypeWithRegistry(schema, ctx) {
   if (ast._tag === "TupleType") {
     return handleTupleTypeAST(ast, ctx);
   }
+  if (ast._tag === "Declaration") {
+    if (isOptionDeclaration2(ast)) {
+      const innerType = getOptionInnerType2(ast);
+      if (innerType) {
+        return toGraphQLTypeWithRegistry(S.make(innerType), ctx);
+      }
+    }
+    const typeParams = ast.typeParameters;
+    if (typeParams && typeParams.length > 0) {
+      return toGraphQLTypeWithRegistry(S.make(typeParams[0]), ctx);
+    }
+  }
   if (ast._tag === "Suspend") {
     const innerAst = ast.f();
-    return toGraphQLTypeWithRegistry(S2.make(innerAst), ctx);
+    return toGraphQLTypeWithRegistry(S.make(innerAst), ctx);
   }
   return toGraphQLType(schema);
 }
@@ -292,20 +382,48 @@ function findRegisteredInterface(schema, ast, ctx) {
   }
   return void 0;
 }
+function isOptionDeclaration2(ast) {
+  if (ast._tag === "Declaration") {
+    const annotations = ast.annotations;
+    if (annotations) {
+      const TypeConstructorSymbol = /* @__PURE__ */ Symbol.for("effect/annotation/TypeConstructor");
+      const typeConstructor = annotations[TypeConstructorSymbol];
+      if (typeConstructor && typeConstructor._tag === "effect/Option") {
+        return true;
+      }
+    }
+  }
+  return false;
+}
+function getOptionInnerType2(ast) {
+  if (ast._tag === "Declaration") {
+    const typeParams = ast.typeParameters;
+    if (typeParams && typeParams.length > 0) {
+      return typeParams[0];
+    }
+  }
+  return void 0;
+}
 function handleTransformationAST(ast, ctx) {
   const toAst = ast.to;
+  if (isOptionDeclaration2(toAst)) {
+    const innerType = getOptionInnerType2(toAst);
+    if (innerType) {
+      return toGraphQLTypeWithRegistry(S.make(innerType), ctx);
+    }
+  }
   if (toAst._tag === "TupleType") {
     if (toAst.rest && toAst.rest.length > 0) {
-      const elementSchema = S2.make(toAst.rest[0].type);
+      const elementSchema = S.make(toAst.rest[0].type);
       const elementType = toGraphQLTypeWithRegistry(elementSchema, ctx);
       return new GraphQLList(elementType);
     } else if (toAst.elements.length > 0) {
-      const elementSchema = S2.make(toAst.elements[0].type);
+      const elementSchema = S.make(toAst.elements[0].type);
       const elementType = toGraphQLTypeWithRegistry(elementSchema, ctx);
       return new GraphQLList(elementType);
     }
   }
-  return toGraphQLTypeWithRegistry(S2.make(ast.to), ctx);
+  return toGraphQLTypeWithRegistry(S.make(ast.to), ctx);
 }
 function handleUnionAST(ast, ctx) {
   const allLiterals = ast.types.every((t) => t._tag === "Literal");
@@ -317,9 +435,9 @@ function handleUnionAST(ast, ctx) {
     if (unionType2) return unionType2;
   }
   if (ast.types.length > 0) {
-    return toGraphQLTypeWithRegistry(S2.make(ast.types[0]), ctx);
+    return toGraphQLTypeWithRegistry(S.make(ast.types[0]), ctx);
   }
-  return toGraphQLType(S2.make(ast));
+  return toGraphQLType(S.make(ast));
 }
 function findEnumForLiteralUnion(types, ctx) {
   const literalValues = types.map((t) => String(t.literal)).sort();
@@ -362,15 +480,15 @@ function findEnumForLiteral(ast, ctx) {
 }
 function handleTupleTypeAST(ast, ctx) {
   if (ast.rest && ast.rest.length > 0) {
-    const elementSchema = S2.make(ast.rest[0].type);
+    const elementSchema = S.make(ast.rest[0].type);
     const elementType = toGraphQLTypeWithRegistry(elementSchema, ctx);
     return new GraphQLList(elementType);
   } else if (ast.elements && ast.elements.length > 0) {
-    const elementSchema = S2.make(ast.elements[0].type);
+    const elementSchema = S.make(ast.elements[0].type);
     const elementType = toGraphQLTypeWithRegistry(elementSchema, ctx);
     return new GraphQLList(elementType);
   }
-  return toGraphQLType(S2.make(ast));
+  return toGraphQLType(S.make(ast));
 }
 function schemaToFields(schema, ctx) {
   let ast = schema.ast;
@@ -379,15 +497,22 @@ function schemaToFields(schema, ctx) {
   }
   if (ast._tag === "Declaration") {
     const typeParams = ast.typeParameters;
-    if (typeParams && typeParams.length > 0 && typeParams[0]._tag === "TypeLiteral") {
-      ast = typeParams[0];
+    if (typeParams && typeParams.length > 0) {
+      let innerAst = typeParams[0];
+      while (innerAst._tag === "Transformation") {
+        innerAst = innerAst.to;
+      }
+      if (innerAst._tag === "TypeLiteral") {
+        ast = innerAst;
+      }
     }
   }
   if (ast._tag === "TypeLiteral") {
     const fields = {};
     for (const field2 of ast.propertySignatures) {
       const fieldName = String(field2.name);
-      const fieldSchema = S2.make(field2.type);
+      if (fieldName === "_tag") continue;
+      const fieldSchema = S.make(field2.type);
       let fieldType = toGraphQLTypeWithRegistry(fieldSchema, ctx);
       if (!field2.isOptional) {
         fieldType = getNonNull(fieldType);
@@ -399,12 +524,16 @@ function schemaToFields(schema, ctx) {
   return {};
 }
 function schemaToInputFields(schema, enumRegistry, inputRegistry, inputs, enums, cache) {
-  const ast = schema.ast;
+  let ast = schema.ast;
+  while (ast._tag === "Transformation") {
+    ast = ast.to;
+  }
   if (ast._tag === "TypeLiteral") {
     const fields = {};
     for (const field2 of ast.propertySignatures) {
       const fieldName = String(field2.name);
-      const fieldSchema = S2.make(field2.type);
+      if (fieldName === "_tag") continue;
+      const fieldSchema = S.make(field2.type);
       let fieldType = toGraphQLInputTypeWithRegistry(
         fieldSchema,
         enumRegistry,
@@ -443,17 +572,6 @@ function buildInputTypeLookupCache(inputs, enums) {
 }
 function toGraphQLInputTypeWithRegistry(schema, enumRegistry, inputRegistry, inputs, enums, cache) {
   const ast = schema.ast;
-  if (ast._tag === "Transformation") {
-    const toAst = ast.to;
-    return toGraphQLInputTypeWithRegistry(
-      S2.make(toAst),
-      enumRegistry,
-      inputRegistry,
-      inputs,
-      enums,
-      cache
-    );
-  }
   if (cache?.schemaToInputName || cache?.astToInputName) {
     const inputName = cache.schemaToInputName?.get(schema) ?? cache.astToInputName?.get(ast);
     if (inputName) {
@@ -468,12 +586,23 @@ function toGraphQLInputTypeWithRegistry(schema, enumRegistry, inputRegistry, inp
       }
     }
   }
+  if (ast._tag === "Transformation") {
+    const toAst = ast.to;
+    return toGraphQLInputTypeWithRegistry(
+      S.make(toAst),
+      enumRegistry,
+      inputRegistry,
+      inputs,
+      enums,
+      cache
+    );
+  }
   if (ast._tag === "Union") {
     const unionAst = ast;
     const nonUndefinedTypes = unionAst.types.filter((t) => t._tag !== "UndefinedKeyword");
     if (nonUndefinedTypes.length === 1 && nonUndefinedTypes[0]._tag === "Union") {
       return toGraphQLInputTypeWithRegistry(
-        S2.make(nonUndefinedTypes[0]),
+        S.make(nonUndefinedTypes[0]),
         enumRegistry,
         inputRegistry,
         inputs,
@@ -483,7 +612,7 @@ function toGraphQLInputTypeWithRegistry(schema, enumRegistry, inputRegistry, inp
     }
     if (nonUndefinedTypes.length === 1 && nonUndefinedTypes[0]._tag === "TypeLiteral") {
       return toGraphQLInputTypeWithRegistry(
-        S2.make(nonUndefinedTypes[0]),
+        S.make(nonUndefinedTypes[0]),
         enumRegistry,
         inputRegistry,
         inputs,
@@ -523,7 +652,7 @@ function toGraphQLInputTypeWithRegistry(schema, enumRegistry, inputRegistry, inp
   if (ast._tag === "Suspend") {
     const innerAst = ast.f();
     return toGraphQLInputTypeWithRegistry(
-      S2.make(innerAst),
+      S.make(innerAst),
       enumRegistry,
       inputRegistry,
       inputs,
@@ -539,7 +668,8 @@ function toGraphQLArgsWithRegistry(schema, enumRegistry, inputRegistry, inputs,
     const args = {};
     for (const field2 of ast.propertySignatures) {
       const fieldName = String(field2.name);
-      const fieldSchema = S2.make(field2.type);
+      if (fieldName === "_tag") continue;
+      const fieldSchema = S.make(field2.type);
       let fieldType = toGraphQLInputTypeWithRegistry(
         fieldSchema,
         enumRegistry,
@@ -557,6 +687,29 @@ function toGraphQLArgsWithRegistry(schema, enumRegistry, inputRegistry, inputs,
   }
   return toGraphQLArgs(schema);
 }
+function isOptionSchema(schema) {
+  const ast = schema.ast;
+  if (ast._tag === "Transformation") {
+    const toAst = ast.to;
+    if (toAst._tag === "Declaration") {
+      const annotations = toAst.annotations;
+      if (annotations) {
+        const TypeConstructorSymbol = /* @__PURE__ */ Symbol.for("effect/annotation/TypeConstructor");
+        const typeConstructor = annotations[TypeConstructorSymbol];
+        if (typeConstructor && typeConstructor._tag === "effect/Option") {
+          return true;
+        }
+      }
+    }
+  }
+  return false;
+}
+function encodeResolverOutput(schema, value) {
+  if (isOptionSchema(schema)) {
+    return Effect.orDie(S.encode(schema)(value));
+  }
+  return Effect.succeed(value);
+}
 function applyDirectives(effect, directives, directiveRegistrations) {
   if (!directives) return effect;
   let wrapped = effect;
@@ -591,7 +744,8 @@ function buildField(config, ctx) {
       );
       const middlewareContext = { parent: _parent, args, info };
       effect = applyMiddleware(effect, middlewareContext, ctx.middlewares);
-      return await Runtime.runPromise(context.runtime)(effect);
+      const result = await Runtime.runPromise(context.runtime)(effect);
+      return await Runtime.runPromise(context.runtime)(encodeResolverOutput(config.type, result));
     }
   };
   if (config.args) {
@@ -620,7 +774,8 @@ function buildObjectField(config, ctx) {
       );
       const middlewareContext = { parent, args, info };
       effect = applyMiddleware(effect, middlewareContext, ctx.middlewares);
-      return await Runtime.runPromise(context.runtime)(effect);
+      const result = await Runtime.runPromise(context.runtime)(effect);
+      return await Runtime.runPromise(context.runtime)(encodeResolverOutput(config.type, result));
     }
   };
   if (config.args) {
@@ -655,13 +810,18 @@ function buildSubscriptionField(config, ctx) {
       return streamToAsyncIterator(stream, context.runtime);
     },
     // The resolve function transforms each yielded value
-    // If no custom resolve is provided, return the payload directly
+    // If no custom resolve is provided, encode and return the payload directly
     resolve: config.resolve ? async (value, args, context, info) => {
       let effect = config.resolve(value, args);
       const middlewareContext = { parent: value, args, info };
       effect = applyMiddleware(effect, middlewareContext, ctx.middlewares);
-      return await Runtime.runPromise(context.runtime)(effect);
-    } : (value) => value
+      const result = await Runtime.runPromise(context.runtime)(effect);
+      return await Runtime.runPromise(context.runtime)(
+        encodeResolverOutput(config.type, result)
+      );
+    } : async (value, _args, context) => {
+      return await Runtime.runPromise(context.runtime)(encodeResolverOutput(config.type, value));
+    }
   };
   if (config.args) {
     fieldConfig.args = toGraphQLArgsWithRegistry(