@pothos/plugin-drizzle 0.0.1

package/src/field-builder.ts

@@ -0,0 +1,168 @@
+import { GraphQLResolveInfo } from 'graphql';
+import { FieldKind, ObjectRef, RootFieldBuilder, SchemaTypes } from '@pothos/core';
+import { queryFromInfo } from './utils/map-query';
+import { getRefFromModel } from './utils/refs';
+import { FieldRef } from '@pothos/core';
+import { DrizzleRef } from './interface-ref';
+import { DrizzleConnectionFieldOptions } from './types';
+import { TableRelationalConfig } from 'drizzle-orm';
+import { resolveDrizzleCursorConnection } from './utils/cursors';
+import { getSchemaConfig } from './utils/config';
+
+const fieldBuilderProto = RootFieldBuilder.prototype as PothosSchemaTypes.RootFieldBuilder<
+  SchemaTypes,
+  unknown,
+  FieldKind
+>;
+
+fieldBuilderProto.drizzleField = function drizzleField({ type, resolve, ...options }) {
+  const modelOrRef = Array.isArray(type) ? type[0] : type;
+  const typeRef =
+    typeof modelOrRef === 'string'
+      ? getRefFromModel(modelOrRef, this.builder)
+      : (modelOrRef as ObjectRef<SchemaTypes, unknown>);
+  const typeParam = Array.isArray(type)
+    ? ([typeRef] as [ObjectRef<SchemaTypes, unknown>])
+    : typeRef;
+  return this.field({
+    ...(options as {}),
+    type: typeParam,
+    resolve: (parent: unknown, args: unknown, context: {}, info: GraphQLResolveInfo) => {
+      return resolve(
+        (select) =>
+          queryFromInfo({
+            config: getSchemaConfig(this.builder),
+            context,
+            select,
+            info,
+            // withUsageCheck: !!this.builder.options.drizzle?.onUnusedQuery,
+          }),
+        parent,
+        args as never,
+        context,
+        info,
+      ) as never;
+    },
+  }) as never;
+};
+
+fieldBuilderProto.drizzleFieldWithInput = function drizzleFieldWithInput(
+  this: typeof fieldBuilderProto,
+  {
+    type,
+    resolve,
+    ...options
+  }: { type: ObjectRef<SchemaTypes, unknown> | [string]; resolve: (...args: unknown[]) => unknown },
+) {
+  const modelOrRef = Array.isArray(type) ? type[0] : type;
+  const typeRef =
+    typeof modelOrRef === 'string'
+      ? getRefFromModel(modelOrRef, this.builder)
+      : (modelOrRef as ObjectRef<SchemaTypes, unknown>);
+  const typeParam = Array.isArray(type)
+    ? ([typeRef] as [ObjectRef<SchemaTypes, unknown>])
+    : typeRef;
+
+  return (
+    this as typeof fieldBuilderProto & { fieldWithInput: typeof fieldBuilderProto.field }
+  ).fieldWithInput({
+    ...(options as {}),
+    type: typeParam,
+    resolve: (parent: unknown, args: unknown, context: {}, info: GraphQLResolveInfo) => {
+      const query = queryFromInfo({
+        config: getSchemaConfig(this.builder),
+        context,
+        info,
+        // withUsageCheck: !!this.builder.options.drizzle?.onUnusedQuery,
+      });
+
+      return resolve(query, parent, args as never, context, info) as never;
+    },
+  });
+} as never;
+
+fieldBuilderProto.drizzleConnection = function drizzleConnection<
+  Type extends
+    | DrizzleRef<SchemaTypes, keyof SchemaTypes['DrizzleRelationSchema']>
+    | keyof SchemaTypes['DrizzleRelationSchema'],
+  Nullable extends boolean,
+  ResolveReturnShape,
+>(
+  this: typeof fieldBuilderProto,
+  {
+    type,
+    maxSize = this.builder.options.drizzle?.maxConnectionSize,
+    defaultSize = this.builder.options.drizzle?.defaultConnectionSize,
+    resolve,
+    ...options
+  }: DrizzleConnectionFieldOptions<
+    SchemaTypes,
+    unknown,
+    Type,
+    TableRelationalConfig,
+    ObjectRef<SchemaTypes, {}>,
+    Nullable,
+    {},
+    ResolveReturnShape,
+    FieldKind
+  >,
+  connectionOptions: {} = {},
+  edgeOptions: {} = {},
+) {
+  const ref = typeof type === 'string' ? getRefFromModel(type, this.builder) : type;
+  const typeName = this.builder.configStore.getTypeConfig(ref).name;
+  const fieldRef = (
+    this as typeof fieldBuilderProto & {
+      connection: (...args: unknown[]) => FieldRef<SchemaTypes, unknown>;
+    }
+  ).connection(
+    {
+      ...options,
+      type: ref,
+      resolve: (
+        parent: unknown,
+        args: PothosSchemaTypes.DefaultConnectionArguments,
+        context: {},
+        info: GraphQLResolveInfo,
+      ) => {
+        const drizzleModel =
+          this.builder.options.drizzle.client._.schema?.[
+            typeof type === 'string' ? type : (ref as DrizzleRef<SchemaTypes>).tableName
+          ]!;
+
+        return resolveDrizzleCursorConnection(
+          drizzleModel,
+          info,
+          typeName,
+          getSchemaConfig(this.builder),
+          {
+            ctx: context,
+            maxSize,
+            defaultSize,
+            args,
+          },
+
+          (q) => {
+            // return checkIfQueryIsUsed(
+            //   this.builder,
+            //   query,
+            //   info,
+            return resolve(q as never, parent, args as never, context, info) as never;
+            // );
+          },
+        );
+      },
+    },
+    connectionOptions instanceof ObjectRef
+      ? connectionOptions
+      : {
+          ...connectionOptions,
+          extensions: {
+            ...(connectionOptions as Record<string, {}> | undefined)?.extensions,
+          },
+        },
+    edgeOptions,
+  );
+
+  return fieldRef;
+} as never;

package/src/global-types.ts

@@ -0,0 +1,481 @@
+/* eslint-disable @typescript-eslint/no-unused-vars */
+/* eslint-disable @typescript-eslint/no-empty-interface */
+import type {
+  BuildQueryResult,
+  DBQueryConfig,
+  ExtractTablesWithRelations,
+  SQL,
+  TableRelationalConfig,
+} from 'drizzle-orm';
+import {
+  FieldKind,
+  FieldMap,
+  FieldNullability,
+  FieldRef,
+  InputFieldMap,
+  InterfaceParam,
+  NormalizeArgs,
+  OutputType,
+  PluginName,
+  SchemaTypes,
+  ShapeFromTypeParam,
+  TypeParam,
+} from '@pothos/core';
+import { DrizzleObjectFieldBuilder } from './drizzle-field-builder';
+import { DrizzleInterfaceRef, DrizzleRef } from './interface-ref';
+import { DrizzleObjectRef, drizzleTableKey } from './object-ref';
+import type {
+  AddGraphQLInputTypeOptions,
+  DrizzleConnectionFieldOptions,
+  DrizzleConnectionShape,
+  DrizzleFieldOptions,
+  DrizzleFieldWithInputOptions,
+  DrizzleInterfaceOptions,
+  DrizzleNodeOptions,
+  DrizzleObjectFieldOptions,
+  DrizzleObjectOptions,
+  DrizzlePluginOptions,
+  drizzleTableName,
+  ShapeFromConnection,
+} from './types';
+
+import type { PothosDrizzlePlugin } from '.';
+import { DrizzleNodeRef } from './node-ref';
+import { GraphQLInputObjectType } from 'graphql';
+
+declare global {
+  export namespace PothosSchemaTypes {
+    export interface Plugins<Types extends SchemaTypes> {
+      drizzle: PothosDrizzlePlugin<Types>;
+    }
+
+    export interface SchemaBuilderOptions<Types extends SchemaTypes> {
+      drizzle: DrizzlePluginOptions<Types>;
+    }
+
+    export interface UserSchemaTypes {
+      DrizzleSchema: Record<string, unknown>;
+      DrizzleRelationSchema: Record<string, TableRelationalConfig>;
+    }
+
+    export interface ExtendDefaultTypes<PartialTypes extends Partial<UserSchemaTypes>> {
+      DrizzleSchema: PartialTypes['DrizzleSchema'] & {};
+      DrizzleRelationSchema: ExtractTablesWithRelations<PartialTypes['DrizzleSchema'] & {}>;
+    }
+
+    export interface SchemaBuilder<Types extends SchemaTypes> {
+      drizzleObject: <
+        Interfaces extends InterfaceParam<Types>[],
+        Table extends keyof Types['DrizzleRelationSchema'],
+        Selection extends
+          | DBQueryConfig<
+              'one',
+              false,
+              Types['DrizzleRelationSchema'],
+              Types['DrizzleRelationSchema'][Table]
+            >
+          | true,
+        Shape = BuildQueryResult<
+          Types['DrizzleRelationSchema'],
+          Types['DrizzleRelationSchema'][Table],
+          Selection
+        >,
+      >(
+        table: Table,
+        options: DrizzleObjectOptions<Types, Table, Shape, Selection, Interfaces>,
+      ) => DrizzleObjectRef<Types, Table, Shape>;
+
+      drizzleInterface: <
+        Interfaces extends InterfaceParam<Types>[],
+        Table extends keyof Types['DrizzleRelationSchema'],
+        Selection extends
+          | DBQueryConfig<
+              'one',
+              false,
+              Types['DrizzleRelationSchema'],
+              Types['DrizzleRelationSchema'][Table]
+            >
+          | true,
+        Shape = BuildQueryResult<
+          Types['DrizzleRelationSchema'],
+          Types['DrizzleRelationSchema'][Table],
+          Selection
+        >,
+      >(
+        table: Table,
+        options: DrizzleInterfaceOptions<Types, Table, Shape, Selection, Interfaces>,
+      ) => DrizzleInterfaceRef<Types, Table, Shape>;
+
+      drizzleNode: 'relay' extends PluginName
+        ? <
+            Interfaces extends InterfaceParam<Types>[],
+            Table extends keyof Types['DrizzleRelationSchema'],
+            Selection extends
+              | DBQueryConfig<
+                  'one',
+                  false,
+                  Types['DrizzleRelationSchema'],
+                  Types['DrizzleRelationSchema'][Table]
+                >
+              | true,
+            Shape = BuildQueryResult<
+              Types['DrizzleRelationSchema'],
+              Types['DrizzleRelationSchema'][Table],
+              Selection
+            >,
+          >(
+            table: Table,
+            options: DrizzleNodeOptions<Types, Table, Shape, Selection, Interfaces>,
+          ) => DrizzleNodeRef<Types, Table, Shape>
+        : '@pothos/plugin-relay is required to use this method';
+
+      drizzleObjectField: <
+        Type extends DrizzleObjectRef<Types> | keyof Types['DrizzleRelationSchema'],
+        TableConfig extends TableRelationalConfig = Type extends DrizzleObjectRef<Types>
+          ? Type[typeof drizzleTableKey]
+          : Types['DrizzleRelationSchema'][Type & keyof Types['DrizzleRelationSchema']],
+        Shape extends {} = Type extends DrizzleObjectRef<
+          Types,
+          keyof Types['DrizzleRelationSchema'],
+          infer S
+        >
+          ? S & { [drizzleTableName]?: TableConfig['tsName'] }
+          : BuildQueryResult<Types['DrizzleRelationSchema'], TableConfig, true> & {
+              [drizzleTableName]?: Type;
+            },
+      >(
+        type: Type,
+        fieldName: string,
+        field: (t: DrizzleObjectFieldBuilder<Types, TableConfig, false, Shape>) => FieldRef<Types>,
+      ) => void;
+
+      drizzleInterfaceField: <
+        Type extends DrizzleInterfaceRef<Types> | keyof Types['DrizzleRelationSchema'],
+        TableConfig extends TableRelationalConfig = Type extends DrizzleObjectRef<Types>
+          ? Type[typeof drizzleTableKey]
+          : Types['DrizzleRelationSchema'][Type & keyof Types['DrizzleRelationSchema']],
+        Shape extends {} = Type extends DrizzleInterfaceRef<
+          Types,
+          keyof Types['DrizzleRelationSchema'],
+          infer S
+        >
+          ? S & { [drizzleTableName]?: TableConfig['tsName'] }
+          : BuildQueryResult<Types['DrizzleRelationSchema'], TableConfig, true> & {
+              [drizzleTableName]?: Type;
+            },
+      >(
+        type: Type,
+        fieldName: string,
+        field: (t: DrizzleObjectFieldBuilder<Types, TableConfig, false, Shape>) => FieldRef<Types>,
+      ) => void;
+
+      drizzleObjectFields: <
+        Type extends DrizzleObjectRef<Types> | keyof Types['DrizzleRelationSchema'],
+        TableConfig extends TableRelationalConfig = Type extends DrizzleObjectRef<Types>
+          ? Type[typeof drizzleTableKey]
+          : Types['DrizzleRelationSchema'][Type & keyof Types['DrizzleRelationSchema']],
+        Shape extends {} = Type extends DrizzleObjectRef<
+          Types,
+          keyof Types['DrizzleRelationSchema'],
+          infer S
+        >
+          ? S & { [drizzleTableName]?: TableConfig['tsName'] }
+          : BuildQueryResult<Types['DrizzleRelationSchema'], TableConfig, true> & {
+              [drizzleTableName]?: Type;
+            },
+      >(
+        type: Type,
+        fields: (t: DrizzleObjectFieldBuilder<Types, TableConfig, false, Shape>) => FieldMap,
+      ) => void;
+
+      drizzleInterfaceFields: <
+        Type extends DrizzleInterfaceRef<Types> | keyof Types['DrizzleRelationSchema'],
+        TableConfig extends TableRelationalConfig = Type extends DrizzleObjectRef<Types>
+          ? Type[typeof drizzleTableKey]
+          : Types['DrizzleRelationSchema'][Type & keyof Types['DrizzleRelationSchema']],
+        Shape extends {} = Type extends DrizzleInterfaceRef<
+          Types,
+          keyof Types['DrizzleRelationSchema'],
+          infer S
+        >
+          ? S & { [drizzleTableName]?: TableConfig['tsName'] }
+          : BuildQueryResult<Types['DrizzleRelationSchema'], TableConfig, true> & {
+              [drizzleTableName]?: Type;
+            },
+      >(
+        type: Type,
+        fields: (t: DrizzleObjectFieldBuilder<Types, TableConfig, false, Shape>) => FieldMap,
+      ) => void;
+
+      drizzleGraphQLOrderBy: 'addGraphQL' extends PluginName
+        ? <Table extends keyof Types['DrizzleRelationSchema']>(
+            table: Table,
+            type: GraphQLInputObjectType,
+            ...args: NormalizeArgs<[options: AddGraphQLInputTypeOptions<Types, {}>]>
+          ) => InputObjectRef<Types, SQL | SQL[]>
+        : '@pothos/plugin-add-graphql is required to use this method';
+
+      drizzleGraphQLFilters: 'addGraphQL' extends PluginName
+        ? <Table extends keyof Types['DrizzleRelationSchema']>(
+            table: Table,
+            type: GraphQLInputObjectType,
+            ...args: NormalizeArgs<[options: AddGraphQLInputTypeOptions<Types, {}>]>
+          ) => InputObjectRef<Types, SQL>
+        : '@pothos/plugin-add-graphql is required to use this method';
+
+      drizzleGraphQLInsert: 'addGraphQL' extends PluginName
+        ? <Table extends keyof Types['DrizzleRelationSchema']>(
+            table: Table,
+            type: GraphQLInputObjectType,
+            ...args: NormalizeArgs<[options: AddGraphQLInputTypeOptions<Types, {}>]>
+          ) => InputObjectRef<Types, Record<string, unknown>>
+        : '@pothos/plugin-add-graphql is required to use this method';
+
+      drizzleGraphQLUpdate: 'addGraphQL' extends PluginName
+        ? <Table extends keyof Types['DrizzleRelationSchema']>(
+            table: Table,
+            type: GraphQLInputObjectType,
+            ...args: NormalizeArgs<[options: AddGraphQLInputTypeOptions<Types, {}>]>
+          ) => InputObjectRef<Types, Record<string, unknown>>
+        : '@pothos/plugin-add-graphql is required to use this method';
+    }
+
+    export interface PothosKindToGraphQLType {
+      DrizzleObject: 'Object';
+    }
+
+    export interface FieldOptionsByKind<
+      Types extends SchemaTypes,
+      ParentShape,
+      Type extends TypeParam<Types>,
+      Nullable extends FieldNullability<Type>,
+      Args extends InputFieldMap,
+      ResolveShape,
+      ResolveReturnShape,
+    > {
+      DrizzleObject: DrizzleObjectFieldOptions<
+        Types,
+        ParentShape,
+        Type,
+        Nullable,
+        Args,
+        ResolveShape,
+        ResolveReturnShape
+      >;
+    }
+
+    export interface RootFieldBuilder<
+      Types extends SchemaTypes,
+      ParentShape,
+      Kind extends FieldKind = FieldKind,
+    > {
+      drizzleField: <
+        Args extends InputFieldMap,
+        Param extends
+          | keyof Types['DrizzleRelationSchema']
+          | [keyof Types['DrizzleRelationSchema']]
+          | DrizzleRef<any>
+          | [DrizzleRef<any>],
+        Nullable extends FieldNullability<Type>,
+        ResolveShape,
+        ResolveReturnShape,
+        Table extends
+          keyof Types['DrizzleRelationSchema'] = Param extends keyof Types['DrizzleRelationSchema']
+          ? Param
+          : Param extends [keyof Types['DrizzleRelationSchema']]
+            ? Param[0]
+            : Param extends DrizzleRef<any, infer T>
+              ? T & keyof Types['DrizzleRelationSchema']
+              : Param extends [DrizzleRef<any, infer T>]
+                ? T & keyof Types['DrizzleRelationSchema']
+                : never,
+        Type extends TypeParam<Types> = ObjectRef<
+          Types,
+          BuildQueryResult<
+            Types['DrizzleRelationSchema'],
+            Types['DrizzleRelationSchema'][Table],
+            true
+          >
+        >,
+      >(
+        options: DrizzleFieldOptions<
+          Types,
+          ParentShape,
+          Table,
+          Type,
+          Nullable,
+          Args,
+          Kind,
+          ResolveShape,
+          ResolveReturnShape,
+          Param
+        >,
+      ) => FieldRef<Types, ShapeFromTypeParam<Types, Type, Nullable>>;
+
+      drizzleConnection: 'relay' extends PluginName
+        ? <
+            Type extends
+              | DrizzleRef<any, keyof Types['DrizzleRelationSchema']>
+              | keyof Types['DrizzleRelationSchema'],
+            Nullable extends boolean,
+            ResolveReturnShape,
+            Args extends InputFieldMap = {},
+            Shape = Type extends DrizzleRef<any, keyof Types['DrizzleRelationSchema'], infer S>
+              ? S
+              : BuildQueryResult<
+                  Types['DrizzleRelationSchema'],
+                  Types['DrizzleRelationSchema'][Type & keyof Types['DrizzleRelationSchema']],
+                  true
+                >,
+            ConnectionInterfaces extends InterfaceParam<Types>[] = [],
+            EdgeInterfaces extends InterfaceParam<Types>[] = [],
+          >(
+            options: DrizzleConnectionFieldOptions<
+              Types,
+              ParentShape,
+              Type,
+              Types['DrizzleRelationSchema'][Type extends DrizzleRef<any, infer K>
+                ? K
+                : Type & keyof Types['DrizzleRelationSchema']],
+              ObjectRef<Types, Shape>,
+              Nullable,
+              Args,
+              ResolveReturnShape,
+              Kind
+            >,
+            ...args: NormalizeArgs<
+              [
+                connectionOptions:
+                  | ConnectionObjectOptions<
+                      Types,
+                      ObjectRef<Types, Shape>,
+                      false,
+                      false,
+                      DrizzleConnectionShape<Types, Shape, ParentShape, Args>,
+                      ConnectionInterfaces
+                    >
+                  | ObjectRef<
+                      Types,
+                      ShapeFromConnection<ConnectionShapeHelper<Types, Shape, false>>
+                    >,
+                edgeOptions:
+                  | ConnectionEdgeObjectOptions<
+                      Types,
+                      ObjectRef<Types, Shape>,
+                      false,
+                      DrizzleConnectionShape<Types, Shape, ParentShape, Args>,
+                      EdgeInterfaces
+                    >
+                  | ObjectRef<
+                      Types,
+                      {
+                        cursor: string;
+                        node?: Shape | null | undefined;
+                      }
+                    >,
+              ],
+              0
+            >
+          ) => FieldRef<Types, ShapeFromConnection<ConnectionShapeHelper<Types, Shape, Nullable>>>
+        : '@pothos/plugin-relay is required to use this method';
+
+      drizzleFieldWithInput: 'withInput' extends PluginName
+        ? <
+            Fields extends InputFieldMap,
+            Args extends InputFieldMap,
+            Param extends
+              | keyof Types['DrizzleRelationSchema']
+              | [keyof Types['DrizzleRelationSchema']]
+              | DrizzleRef<any>
+              | [DrizzleRef<any>],
+            Nullable extends FieldNullability<Type>,
+            ResolveShape,
+            ResolveReturnShape,
+            ArgRequired extends boolean,
+            InputName extends string = 'input',
+            Table extends
+              keyof Types['DrizzleRelationSchema'] = Param extends keyof Types['DrizzleRelationSchema']
+              ? Param
+              : Param extends [keyof Types['DrizzleRelationSchema']]
+                ? Param[0]
+                : Param extends DrizzleRef<any, infer T>
+                  ? T & keyof Types['DrizzleRelationSchema']
+                  : Param extends [DrizzleRef<any, infer T>]
+                    ? T & keyof Types['DrizzleRelationSchema']
+                    : never,
+            Type extends TypeParam<Types> = ObjectRef<
+              Types,
+              BuildQueryResult<
+                Types['DrizzleRelationSchema'],
+                Types['DrizzleRelationSchema'][Table],
+                true
+              >
+            >,
+          >(
+            options: DrizzleFieldWithInputOptions<
+              Types,
+              ParentShape,
+              Table,
+              Type,
+              Nullable,
+              Args,
+              Kind,
+              ResolveShape,
+              ResolveReturnShape,
+              Param,
+              InputName,
+              Fields,
+              boolean extends ArgRequired
+                ? (Types & { WithInputArgRequired: boolean })['WithInputArgRequired']
+                : ArgRequired
+            >,
+          ) => FieldRef<Types, ShapeFromTypeParam<Types, Type, Nullable>>
+        : '@pothos/plugin-with-input is required to use this method';
+    }
+
+    export interface ConnectionFieldOptions<
+      Types extends SchemaTypes,
+      ParentShape,
+      Type extends OutputType<Types>,
+      Nullable extends boolean,
+      EdgeNullability extends FieldNullability<[unknown]>,
+      NodeNullability extends boolean,
+      Args extends InputFieldMap,
+      ResolveReturnShape,
+    > {}
+
+    export interface ConnectionObjectOptions<
+      Types extends SchemaTypes,
+      Type extends OutputType<Types>,
+      EdgeNullability extends FieldNullability<[unknown]>,
+      NodeNullability extends boolean,
+      Resolved,
+      Interfaces extends InterfaceParam<Types>[] = [],
+    > {}
+
+    export interface ConnectionEdgeObjectOptions<
+      Types extends SchemaTypes,
+      Type extends OutputType<Types>,
+      NodeNullability extends boolean,
+      Resolved,
+      Interfaces extends InterfaceParam<Types>[] = [],
+    > {}
+
+    export interface DefaultConnectionArguments {
+      first?: number | null | undefined;
+      last?: number | null | undefined;
+      before?: string | null | undefined;
+      after?: string | null | undefined;
+    }
+
+    export interface ConnectionShapeHelper<Types extends SchemaTypes, T, Nullable> {}
+
+    export interface FieldWithInputBaseOptions<
+      Types extends SchemaTypes,
+      Args extends InputFieldMap,
+      Fields extends InputFieldMap,
+      InputName extends string,
+      ArgRequired extends boolean,
+    > {}
+  }
+}