@pothos/plugin-drizzle 0.0.1

package/esm/schema-builder.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=schema-builder.d.ts.map

package/esm/schema-builder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema-builder.d.ts","sourceRoot":"","sources":["../src/schema-builder.ts"],"names":[],"mappings":""}

package/esm/src/drizzle-field-builder.js

@@ -0,0 +1,215 @@
+function _define_property(obj, key, value) {
+    if (key in obj) {
+        Object.defineProperty(obj, key, {
+            value: value,
+            enumerable: true,
+            configurable: true,
+            writable: true
+        });
+    }
+    else {
+        obj[key] = value;
+    }
+    return obj;
+}
+import { getOperators, Many } from 'drizzle-orm';
+import { isThenable, ObjectRef, PothosSchemaError, RootFieldBuilder } from '@pothos/core';
+import { getRefFromModel } from './utils/refs.js';
+import { drizzleCursorConnectionQuery, getCursorFormatter, wrapConnectionResult } from './utils/cursors.js';
+// Workaround for FieldKind not being extended on Builder classes
+const RootBuilder = RootFieldBuilder;
+const ops = getOperators();
+export class DrizzleObjectFieldBuilder extends RootBuilder {
+    variant(variant, ...allArgs) {
+        const [{ isNull, nullable, ...options } = {}] = allArgs;
+        const ref = typeof variant === "string" ? getRefFromModel(variant, this.builder) : variant;
+        const selfSelect = (args, context, nestedQuery) => nestedQuery({});
+        return this.field({
+            ...options,
+            type: ref,
+            extensions: {
+                ...options === null || options === void 0 ? void 0 : options.extensions,
+                pothosDrizzleSelect: selfSelect
+            },
+            nullable: nullable !== null && nullable !== void 0 ? nullable : !!isNull,
+            resolve: isNull ? (parent, args, context, info) => {
+                const parentIsNull = isNull(parent, args, context, info);
+                if (parentIsNull) {
+                    if (isThenable(parentIsNull)) {
+                        return parentIsNull.then((resolved) => resolved ? null : parent);
+                    }
+                    return null;
+                }
+                return parent;
+            } : (parent) => parent
+        });
+    }
+    relation(name, ...allArgs) {
+        var _this_builder_options_drizzle_client___schema;
+        const [options = {}] = allArgs;
+        const relationField = (_this_builder_options_drizzle_client___schema = this.builder.options.drizzle.client._.schema) === null || _this_builder_options_drizzle_client___schema === void 0 ? void 0 : _this_builder_options_drizzle_client___schema[this.table].relations[name];
+        if (!relationField) {
+            throw new PothosSchemaError(`Could not find relation ${name} on table ${this.table}`);
+        }
+        var _options_type;
+        const ref = (_options_type = options.type) !== null && _options_type !== void 0 ? _options_type : getRefFromModel(relationField.referencedTableName, this.builder);
+        const { query = {}, extensions, ...rest } = options;
+        const relationSelect = (args, context, nestedQuery) => {
+            const relQuery = {
+                columns: {},
+                with: {
+                    [name]: {
+                        ...nestedQuery(query),
+                        ...typeof query === "function" ? query(args, context) : query
+                    }
+                }
+            };
+            return relQuery;
+        };
+        return this.field({
+            ...rest,
+            type: relationField instanceof Many ? [
+                ref
+            ] : ref,
+            extensions: {
+                ...extensions,
+                pothosDrizzleSelect: relationSelect
+            },
+            resolve: (parent) => parent[name]
+        });
+    }
+    expose(...args) {
+        var _typeConfig_extensions;
+        const [name, options = {}] = args;
+        const typeConfig = this.builder.configStore.getTypeConfig(this.typename, "Object");
+        const usingSelect = !!((_typeConfig_extensions = typeConfig.extensions) === null || _typeConfig_extensions === void 0 ? void 0 : _typeConfig_extensions.pothosDrizzleSelect);
+        return this.exposeField(name, {
+            ...options,
+            extensions: {
+                ...options.extensions,
+                pothosDrizzleVariant: name,
+                pothosDrizzleSelect: usingSelect && {
+                    columns: {
+                        [name]: true
+                    }
+                }
+            }
+        });
+    }
+    createExpose(type) {
+        return (...args) => {
+            const [name, { description, ...options } = {}] = args;
+            return this.expose(name, {
+                ...options,
+                type
+            });
+        };
+    }
+    constructor(typename, builder, table, graphqlKind = "Object") {
+        var _this_builder_options_drizzle, _this_builder_options_drizzle1;
+        super(builder, "DrizzleObject", graphqlKind);
+        _define_property(this, "exposeBoolean", this.createExpose("Boolean"));
+        _define_property(this, "exposeFloat", this.createExpose("Float"));
+        _define_property(this, "exposeInt", this.createExpose("Int"));
+        _define_property(this, "exposeID", this.createExpose("ID"));
+        _define_property(this, "exposeString", this.createExpose("String"));
+        _define_property(this, "exposeBooleanList", this.createExpose([
+            "Boolean"
+        ]));
+        _define_property(this, "exposeFloatList", this.createExpose([
+            "Float"
+        ]));
+        _define_property(this, "exposeIntList", this.createExpose([
+            "Int"
+        ]));
+        _define_property(this, "exposeIDList", this.createExpose([
+            "ID"
+        ]));
+        _define_property(this, "exposeStringList", this.createExpose([
+            "String"
+        ]));
+        _define_property(this, "table", void 0);
+        _define_property(this, "typename", void 0);
+        _define_property(this, "relatedConnection", function relatedConnection(name, { maxSize = (_this_builder_options_drizzle = this.builder.options.drizzle) === null || _this_builder_options_drizzle === void 0 ? void 0 : _this_builder_options_drizzle.maxConnectionSize, defaultSize = (_this_builder_options_drizzle1 = this.builder.options.drizzle) === null || _this_builder_options_drizzle1 === void 0 ? void 0 : _this_builder_options_drizzle1.defaultConnectionSize, query, resolve, extensions, description, ...options }, connectionOptions = {}, edgeOptions = {}) {
+            var _this_builder_options_drizzle_client___schema, _this_builder_options_drizzle_client___schema1;
+            const relationField = (_this_builder_options_drizzle_client___schema = this.builder.options.drizzle.client._.schema) === null || _this_builder_options_drizzle_client___schema === void 0 ? void 0 : _this_builder_options_drizzle_client___schema[this.table].relations[name];
+            const relatedModel = (_this_builder_options_drizzle_client___schema1 = this.builder.options.drizzle.client._.schema) === null || _this_builder_options_drizzle_client___schema1 === void 0 ? void 0 : _this_builder_options_drizzle_client___schema1[relationField === null || relationField === void 0 ? void 0 : relationField.referencedTableName];
+            if (!relationField) {
+                throw new PothosSchemaError(`Could not find relation ${name} on table ${this.table}`);
+            }
+            var _options_type;
+            const ref = (_options_type = options.type) !== null && _options_type !== void 0 ? _options_type : getRefFromModel(relationField.referencedTableName, this.builder);
+            let typeName;
+            const getQuery = (args, ctx) => {
+                var _ref;
+                const { limit, offset, orderBy, where, ...fieldQuery } = (_ref = typeof query === "function" ? query(args, ctx) : query) !== null && _ref !== void 0 ? _ref : {};
+                const { cursorColumns, columns, ...connectionQuery } = drizzleCursorConnectionQuery({
+                    ctx,
+                    maxSize,
+                    defaultSize,
+                    args,
+                    orderBy: orderBy ? orderBy(relatedModel.columns) : relatedModel.primaryKey,
+                    where: where ? where(relatedModel.columns, ops) : undefined
+                });
+                return {
+                    select: {
+                        ...fieldQuery,
+                        ...connectionQuery,
+                        columns: {
+                            ...fieldQuery.columns,
+                            ...columns
+                        },
+                        limit: Math.abs(limit !== null && limit !== void 0 ? limit : connectionQuery.limit),
+                        offset: offset !== null && offset !== void 0 ? offset : connectionQuery.offset
+                    },
+                    cursorColumns
+                };
+            };
+            const relationSelect = (args, context, nestedQuery) => {
+                typeName !== null && typeName !== void 0 ? typeName : typeName = this.builder.configStore.getTypeConfig(ref).name;
+                const nested = nestedQuery(getQuery(args, context).select, {
+                    getType: () => typeName,
+                    paths: [
+                        [
+                            {
+                                name: "nodes"
+                            }
+                        ],
+                        [
+                            {
+                                name: "edges"
+                            },
+                            {
+                                name: "node"
+                            }
+                        ]
+                    ]
+                });
+                return {
+                    columns: {},
+                    with: {
+                        [name]: nested
+                    }
+                };
+            };
+            const fieldRef = this.connection({
+                ...options,
+                extensions: {
+                    ...extensions,
+                    pothosDrizzleSelect: relationSelect
+                },
+                type: ref,
+                resolve: (parent, args, context) => {
+                    const { select, cursorColumns } = getQuery(args, context);
+                    return wrapConnectionResult(parent[name], args, select.limit, getCursorFormatter(cursorColumns));
+                }
+            }, connectionOptions instanceof ObjectRef ? connectionOptions : {
+                ...connectionOptions
+            }, edgeOptions);
+            return fieldRef;
+        });
+        this.table = table;
+        this.typename = typename;
+    }
+}
+//# sourceMappingURL=drizzle-field-builder.js.map

package/esm/src/drizzle-field-builder.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/drizzle-field-builder.ts"],"names":["getOperators","Many","isThenable","ObjectRef","PothosSchemaError","RootFieldBuilder","getRefFromModel","drizzleCursorConnectionQuery","getCursorFormatter","wrapConnectionResult","RootBuilder","ops","DrizzleObjectFieldBuilder","variant","allArgs","isNull","nullable","options","ref","builder","selfSelect","args","context","nestedQuery","field","type","extensions","pothosDrizzleSelect","resolve","parent","info","parentIsNull","then","resolved","relation","name","relationField","drizzle","client","_","schema","table","relations","referencedTableName","query","rest","relationSelect","relQuery","columns","with","expose","typeConfig","configStore","getTypeConfig","typename","usingSelect","exposeField","pothosDrizzleVariant","createExpose","description","constructor","graphqlKind","exposeBoolean","exposeFloat","exposeInt","exposeID","exposeString","exposeBooleanList","exposeFloatList","exposeIntList","exposeIDList","exposeStringList","relatedConnection","maxSize","maxConnectionSize","defaultSize","defaultConnectionSize","connectionOptions","edgeOptions","relatedModel","typeName","getQuery","ctx","limit","offset","orderBy","where","fieldQuery","cursorColumns","connectionQuery","primaryKey","undefined","select","Math","abs","nested","getType","paths","fieldRef","connection"],"mappings":";;;;;;;;;;;;;AAAA,SAASA,YAAY,EAAyBC,IAAI,QAA+B,cAAc;AAC/F,SAOEC,UAAU,EAGVC,SAAS,EAETC,iBAAiB,EACjBC,gBAAgB,QAIX,eAAe;AAUtB,SAASC,eAAe,QAAQ,eAAe;AAC/C,SACEC,4BAA4B,EAC5BC,kBAAkB,EAClBC,oBAAoB,QACf,kBAAkB;AAIzB,iEAAiE;AACjE,MAAMC,cAOFL;AAEJ,MAAMM,MAAMX;AAEZ,OAAO,MAAMY,kCAKHF;IAuPRG,QAOEA,OAAgB,EAChB,GAAGC,OAaF,EACyC;QAC1C,MAAM,CAAC,EAAEC,MAAM,EAAEC,QAAQ,EAAE,GAAGC,SAAS,GAAG,CAAC,CAAU,CAAC,GAAGH;QACzD,MAAMI,MACJ,OAAOL,YAAY,WAAWP,gBAAgBO,SAAS,IAAI,CAACM,OAAO,IAAIN;QAEzE,MAAMO,aAAa,CAACC,MAAcC,SAAiBC,cACjDA,YAAY,CAAC;QAEf,OAAO,IAAI,CAACC,KAAK,CAAC;YAChB,GAAIP,OAAO;YACXQ,MAAMP;YACNQ,YAAY;mBACPT,oBAAAA,8BAAAA,QAASS,UAAU,AAAtB;gBACAC,qBAAqBP;YACvB;YACAJ,UAAUA,qBAAAA,sBAAAA,WAAY,CAAC,CAACD;YACxBa,SAASb,SACL,CAACc,QAAQR,MAAMC,SAASQ;gBACtB,MAAMC,eAAehB,OAAOc,QAAiBR,MAAeC,SAASQ;gBACrE,IAAIC,cAAc;oBAChB,IAAI7B,WAAW6B,eAAe;wBAC5B,OAAOA,aAAaC,IAAI,CAAC,CAACC,WAAcA,WAAW,OAAOJ;oBAC5D;oBACA,OAAO;gBACT;gBACA,OAAOA;YACT,IACA,CAACA,SAAWA;QAClB;IACF;IAEAK,SAMEC,IAAW,EACX,GAAGrB,OAYF,EACoF;YAGnF;QAFF,MAAM,CAACG,UAAU,CAAC,CAAU,CAAC,GAAGH;QAChC,MAAMsB,iBACJ,gDAAA,IAAI,CAACjB,OAAO,CAACF,OAAO,CAACoB,OAAO,CAACC,MAAM,CAACC,CAAC,CAACC,MAAM,cAA5C,oEAAA,6CAA8C,CAAC,IAAI,CAACC,KAAK,CAAC,CAACC,SAAS,CAACP,KAAe;QAEtF,IAAI,CAACC,eAAe;YAClB,MAAM,IAAIhC,kBACR,CAAC,wBAAwB,EAAE+B,KAAe,UAAU,EAAE,IAAI,CAACM,KAAK,CAAC,CAAC;QAEtE;YAEYxB;QAAZ,MAAMC,MAAMD,CAAAA,gBAAAA,QAAQQ,IAAI,cAAZR,2BAAAA,gBAAgBX,gBAAgB8B,cAAcO,mBAAmB,EAAE,IAAI,CAACxB,OAAO;QAE3F,MAAM,EAAEyB,QAAQ,CAAC,CAAC,EAAElB,UAAU,EAAE,GAAGmB,MAAM,GAAG5B;QAE5C,MAAM6B,iBAAiB,CAACzB,MAAcC,SAAiBC;YACrD,MAAMwB,WAAW;gBACfC,SAAS,CAAC;gBACVC,MAAM;oBACJ,CAACd,KAAK,EAAE;wBACN,GAAGZ,YAAYqB,MAAM;wBACrB,GAAK,OAAOA,UAAU,aAClB,AAACA,MAAwCvB,MAAMC,WAC/CsB,KAAK;oBACX;gBACF;YACF;YAEA,OAAOG;QACT;QAEA,OAAO,IAAI,CAACvB,KAAK,CAAC;YAChB,GAAIqB,IAAI;YACRpB,MAAMW,yBAAyBnC,OAAO;gBAACiB;aAAI,GAAGA;YAC9CQ,YAAY;gBACV,GAAGA,UAAU;gBACbC,qBAAqBmB;YACvB;YACAlB,SAAS,CAACC,SAAkCA,MAAM,CAACM,KAAe;QACpE;IACF;IAEAe,OAME,GAAG7B,IAgBF,EACD;YAIsB8B;QAHtB,MAAM,CAAChB,MAAMlB,UAAU,CAAC,CAAU,CAAC,GAAGI;QAEtC,MAAM8B,aAAa,IAAI,CAAChC,OAAO,CAACiC,WAAW,CAACC,aAAa,CAAC,IAAI,CAACC,QAAQ,EAAE;QACzE,MAAMC,cAAc,CAAC,GAACJ,yBAAAA,WAAWzB,UAAU,cAArByB,6CAAAA,uBAAuBxB,mBAAmB;QAEhE,OAAO,IAAI,CAAC6B,WAAW,CAACrB,MAAe;YACrC,GAAGlB,OAAO;YACVS,YAAY;gBACV,GAAGT,QAAQS,UAAU;gBACrB+B,sBAAsBtB;gBACtBR,qBAAqB4B,eAAe;oBAClCP,SAAS;wBAAE,CAACb,KAAe,EAAE;oBAAK;gBACpC;YACF;QACF;IACF;IAEQuB,aAA4CjC,IAAU,EAAE;QAC9D,OAAO,CAUL,GAAGJ;YAoBH,MAAM,CAACc,MAAM,EAAEwB,WAAW,EAAE,GAAG1C,SAAS,GAAG,CAAC,CAAU,CAAC,GAAGI;YAE1D,OAAO,IAAI,CAAC6B,MAAM,CAChBf,MACA;gBACE,GAAGlB,OAAO;gBACVQ;YACF;QAEJ;IACF;IAjNAmC,YACEN,QAAgB,EAChBnC,OAA+C,EAC/CsB,KAAa,EACboB,cAAoE,QAAQ,CAC5E;YA9HY,+BACI;QA8HhB,KAAK,CAAC1C,SAAS,iBAAiB0C;QAhPlCC,uBAAAA,iBAAgB,IAAI,CAACJ,YAAY,CAAC;QAElCK,uBAAAA,eAAc,IAAI,CAACL,YAAY,CAAC;QAEhCM,uBAAAA,aAAY,IAAI,CAACN,YAAY,CAAC;QAE9BO,uBAAAA,YAAW,IAAI,CAACP,YAAY,CAAC;QAE7BQ,uBAAAA,gBAAe,IAAI,CAACR,YAAY,CAAC;QAEjCS,uBAAAA,qBAAoB,IAAI,CAACT,YAAY,CAAC;YAAC;SAAU;QAEjDU,uBAAAA,mBAAkB,IAAI,CAACV,YAAY,CAAC;YAAC;SAAQ;QAE7CW,uBAAAA,iBAAgB,IAAI,CAACX,YAAY,CAAC;YAAC;SAAM;QAEzCY,uBAAAA,gBAAe,IAAI,CAACZ,YAAY,CAAC;YAAC;SAAK;QAEvCa,uBAAAA,oBAAmB,IAAI,CAACb,YAAY,CAAC;YAAC;SAAS;QAE/CjB,uBAAAA,SAAAA,KAAAA;QAEAa,uBAAAA,YAAAA,KAAAA;QAEAkB,uBAAAA,qBAqF4D,SAASA,kBAEnErC,IAAY,EACZ,EACEsC,WAAU,gCAAA,IAAI,CAACtD,OAAO,CAACF,OAAO,CAACoB,OAAO,cAA5B,oDAAA,8BAA8BqC,iBAAiB,EACzDC,eAAc,iCAAA,IAAI,CAACxD,OAAO,CAACF,OAAO,CAACoB,OAAO,cAA5B,qDAAA,+BAA8BuC,qBAAqB,EACjEhC,KAAK,EACLhB,OAAO,EACPF,UAAU,EACViC,WAAW,EACX,GAAG1C,SAeJ,EACD4D,oBAAoB,CAAC,CAAC,EACtBC,cAAc,CAAC,CAAC;gBAGd,+CAEA;YAHF,MAAM1C,iBACJ,gDAAA,IAAI,CAACjB,OAAO,CAACF,OAAO,CAACoB,OAAO,CAACC,MAAM,CAACC,CAAC,CAACC,MAAM,cAA5C,oEAAA,6CAA8C,CAAC,IAAI,CAACC,KAAK,CAAC,CAACC,SAAS,CAACP,KAAe;YACtF,MAAM4C,gBACJ,iDAAA,IAAI,CAAC5D,OAAO,CAACF,OAAO,CAACoB,OAAO,CAACC,MAAM,CAACC,CAAC,CAACC,MAAM,cAA5C,qEAAA,8CAA8C,CAACJ,0BAAAA,oCAAAA,cAAeO,mBAAmB,CAAE;YAErF,IAAI,CAACP,eAAe;gBAClB,MAAM,IAAIhC,kBACR,CAAC,wBAAwB,EAAE+B,KAAe,UAAU,EAAE,IAAI,CAACM,KAAK,CAAC,CAAC;YAEtE;gBAEYxB;YAAZ,MAAMC,MAAMD,CAAAA,gBAAAA,QAAQQ,IAAI,cAAZR,2BAAAA,gBAAgBX,gBAAgB8B,cAAcO,mBAAmB,EAAE,IAAI,CAACxB,OAAO;YAC3F,IAAI6D;YAEJ,MAAMC,WAAW,CAAC5D,MAAoD6D;oBAEjE;gBADH,MAAM,EAAEC,KAAK,EAAEC,MAAM,EAAEC,OAAO,EAAEC,KAAK,EAAE,GAAGC,YAAY,GACpD,CAAC,OAAA,OAAO3C,UAAU,aAAaA,MAAMvB,MAAM6D,OAAOtC,mBAAjD,kBAAA,OAA2D,CAAC;gBAE/D,MAAM,EAAE4C,aAAa,EAAExC,OAAO,EAAE,GAAGyC,iBAAiB,GAAGlF,6BAA6B;oBAClF2E;oBACAT;oBACAE;oBACAtD;oBACAgE,SAASA,UAAUA,QAAQN,aAAa/B,OAAO,IAAI+B,aAAaW,UAAU;oBAC1EJ,OAAOA,QAAQA,MAAMP,aAAa/B,OAAO,EAAErC,OAAOgF;gBACpD;gBAEA,OAAO;oBACLC,QAAQ;wBACN,GAAGL,UAAU;wBACb,GAAGE,eAAe;wBAClBzC,SAAS;4BACP,GAAGuC,WAAWvC,OAAO;4BACrB,GAAGA,OAAO;wBACZ;wBACAmC,OAAOU,KAAKC,GAAG,CAACX,kBAAAA,mBAAAA,QAASM,gBAAgBN,KAAK;wBAC9CC,QAAQA,mBAAAA,oBAAAA,SAAUK,gBAAgBL,MAAM;oBAC1C;oBACAI;gBACF;YACF;YAEA,MAAM1C,iBAAiB,CACrBzB,MACAC,SACAC;gBAEAyD,qBAAAA,sBAAAA,WAAAA,WAAa,IAAI,CAAC7D,OAAO,CAACiC,WAAW,CAACC,aAAa,CAACnC,KAAKiB,IAAI;gBAC7D,MAAM4D,SAASxE,YAAY0D,SAAS5D,MAAMC,SAASsE,MAAM,EAAE;oBACzDI,SAAS,IAAMhB;oBACfiB,OAAO;wBAAC;4BAAC;gCAAE9D,MAAM;4BAAQ;yBAAE;wBAAE;4BAAC;gCAAEA,MAAM;4BAAQ;4BAAG;gCAAEA,MAAM;4BAAO;yBAAE;qBAAC;gBACrE;gBAEA,OAAO;oBACLa,SAAS,CAAC;oBACVC,MAAM;wBACJ,CAACd,KAAK,EAAE4D;oBACV;gBACF;YACF;YACA,MAAMG,WAAW,AACf,IAAI,CAGJC,UAAU,CACV;gBACE,GAAGlF,OAAO;gBACVS,YAAY;oBACV,GAAGA,UAAU;oBACbC,qBAAqBmB;gBACvB;gBACArB,MAAMP;gBACNU,SAAS,CACPC,QACAR,MACAC;oBAEA,MAAM,EAAEsE,MAAM,EAAEJ,aAAa,EAAE,GAAGP,SAAS5D,MAAMC;oBACjD,OAAOb,qBACL,AAACoB,MAAgC,CAACM,KAAK,EACvCd,MACAuE,OAAOT,KAAK,EACZ3E,mBAAmBgF;gBAEvB;YACF,GACAX,6BAA6B1E,YACzB0E,oBACA;gBACE,GAAGA,iBAAiB;YACtB,GACJC;YAEF,OAAOoB;QACT;QAUE,IAAI,CAACzD,KAAK,GAAGA;QACb,IAAI,CAACa,QAAQ,GAAGA;IAClB;AAwMF"}

package/esm/src/field-builder.js

@@ -0,0 +1,77 @@
+var _this_builder_options_drizzle, _this_builder_options_drizzle1;
+import { ObjectRef, RootFieldBuilder } from '@pothos/core';
+import { queryFromInfo } from './utils/map-query.js';
+import { getRefFromModel } from './utils/refs.js';
+import { resolveDrizzleCursorConnection } from './utils/cursors.js';
+import { getSchemaConfig } from './utils/config.js';
+const fieldBuilderProto = RootFieldBuilder.prototype;
+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;
+    const typeParam = Array.isArray(type) ? [
+        typeRef
+    ] : typeRef;
+    return this.field({
+        ...options,
+        type: typeParam,
+        resolve: (parent, args, context, info) => {
+            return resolve((select) => queryFromInfo({
+                config: getSchemaConfig(this.builder),
+                context,
+                select,
+                info
+            }), parent, args, context, info);
+        }
+    });
+};
+fieldBuilderProto.drizzleFieldWithInput = function drizzleFieldWithInput({ type, resolve, ...options }) {
+    const modelOrRef = Array.isArray(type) ? type[0] : type;
+    const typeRef = typeof modelOrRef === "string" ? getRefFromModel(modelOrRef, this.builder) : modelOrRef;
+    const typeParam = Array.isArray(type) ? [
+        typeRef
+    ] : typeRef;
+    return this.fieldWithInput({
+        ...options,
+        type: typeParam,
+        resolve: (parent, args, context, info) => {
+            const query = queryFromInfo({
+                config: getSchemaConfig(this.builder),
+                context,
+                info
+            });
+            return resolve(query, parent, args, context, info);
+        }
+    });
+};
+fieldBuilderProto.drizzleConnection = function drizzleConnection({ type, maxSize = (_this_builder_options_drizzle = this.builder.options.drizzle) === null || _this_builder_options_drizzle === void 0 ? void 0 : _this_builder_options_drizzle.maxConnectionSize, defaultSize = (_this_builder_options_drizzle1 = this.builder.options.drizzle) === null || _this_builder_options_drizzle1 === void 0 ? void 0 : _this_builder_options_drizzle1.defaultConnectionSize, resolve, ...options }, connectionOptions = {}, edgeOptions = {}) {
+    const ref = typeof type === "string" ? getRefFromModel(type, this.builder) : type;
+    const typeName = this.builder.configStore.getTypeConfig(ref).name;
+    const fieldRef = this.connection({
+        ...options,
+        type: ref,
+        resolve: (parent, args, context, info) => {
+            var _this_builder_options_drizzle_client___schema;
+            const drizzleModel = (_this_builder_options_drizzle_client___schema = this.builder.options.drizzle.client._.schema) === null || _this_builder_options_drizzle_client___schema === void 0 ? void 0 : _this_builder_options_drizzle_client___schema[typeof type === "string" ? type : ref.tableName];
+            return resolveDrizzleCursorConnection(drizzleModel, info, typeName, getSchemaConfig(this.builder), {
+                ctx: context,
+                maxSize,
+                defaultSize,
+                args
+            }, (q) => {
+                // return checkIfQueryIsUsed(
+                //   this.builder,
+                //   query,
+                //   info,
+                return resolve(q, parent, args, context, info);
+                // );
+            });
+        }
+    }, connectionOptions instanceof ObjectRef ? connectionOptions : {
+        ...connectionOptions,
+        extensions: {
+            ...connectionOptions === null || connectionOptions === void 0 ? void 0 : connectionOptions.extensions
+        }
+    }, edgeOptions);
+    return fieldRef;
+};
+//# sourceMappingURL=field-builder.js.map

package/esm/src/field-builder.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/field-builder.ts"],"names":["ObjectRef","RootFieldBuilder","queryFromInfo","getRefFromModel","resolveDrizzleCursorConnection","getSchemaConfig","fieldBuilderProto","prototype","drizzleField","type","resolve","options","modelOrRef","Array","isArray","typeRef","builder","typeParam","field","parent","args","context","info","select","config","drizzleFieldWithInput","fieldWithInput","query","drizzleConnection","maxSize","drizzle","maxConnectionSize","defaultSize","defaultConnectionSize","connectionOptions","edgeOptions","ref","typeName","configStore","getTypeConfig","name","fieldRef","connection","drizzleModel","client","_","schema","tableName","ctx","q","extensions"],"mappings":"IA6Fc,+BACI;AA7FlB,SAAoBA,SAAS,EAAEC,gBAAgB,QAAqB,eAAe;AACnF,SAASC,aAAa,QAAQ,oBAAoB;AAClD,SAASC,eAAe,QAAQ,eAAe;AAK/C,SAASC,8BAA8B,QAAQ,kBAAkB;AACjE,SAASC,eAAe,QAAQ,iBAAiB;AAEjD,MAAMC,oBAAoBL,iBAAiBM,SAAS;AAMpDD,kBAAkBE,YAAY,GAAG,SAASA,aAAa,EAAEC,IAAI,EAAEC,OAAO,EAAE,GAAGC,SAAS;IAClF,MAAMC,aAAaC,MAAMC,OAAO,CAACL,QAAQA,IAAI,CAAC,EAAE,GAAGA;IACnD,MAAMM,UACJ,OAAOH,eAAe,WAClBT,gBAAgBS,YAAY,IAAI,CAACI,OAAO,IACvCJ;IACP,MAAMK,YAAYJ,MAAMC,OAAO,CAACL,QAC3B;QAACM;KAAQ,GACVA;IACJ,OAAO,IAAI,CAACG,KAAK,CAAC;QAChB,GAAIP,OAAO;QACXF,MAAMQ;QACNP,SAAS,CAACS,QAAiBC,MAAeC,SAAaC;YACrD,OAAOZ,QACL,CAACa,SACCrB,cAAc;oBACZsB,QAAQnB,gBAAgB,IAAI,CAACW,OAAO;oBACpCK;oBACAE;oBACAD;gBAEF,IACFH,QACAC,MACAC,SACAC;QAEJ;IACF;AACF;AAEAhB,kBAAkBmB,qBAAqB,GAAG,SAASA,sBAEjD,EACEhB,IAAI,EACJC,OAAO,EACP,GAAGC,SAC4F;IAEjG,MAAMC,aAAaC,MAAMC,OAAO,CAACL,QAAQA,IAAI,CAAC,EAAE,GAAGA;IACnD,MAAMM,UACJ,OAAOH,eAAe,WAClBT,gBAAgBS,YAAY,IAAI,CAACI,OAAO,IACvCJ;IACP,MAAMK,YAAYJ,MAAMC,OAAO,CAACL,QAC3B;QAACM;KAAQ,GACVA;IAEJ,OAAO,AACL,IAAI,CACJW,cAAc,CAAC;QACf,GAAIf,OAAO;QACXF,MAAMQ;QACNP,SAAS,CAACS,QAAiBC,MAAeC,SAAaC;YACrD,MAAMK,QAAQzB,cAAc;gBAC1BsB,QAAQnB,gBAAgB,IAAI,CAACW,OAAO;gBACpCK;gBACAC;YAEF;YAEA,OAAOZ,QAAQiB,OAAOR,QAAQC,MAAeC,SAASC;QACxD;IACF;AACF;AAEAhB,kBAAkBsB,iBAAiB,GAAG,SAASA,kBAQ7C,EACEnB,IAAI,EACJoB,WAAU,gCAAA,IAAI,CAACb,OAAO,CAACL,OAAO,CAACmB,OAAO,cAA5B,oDAAA,8BAA8BC,iBAAiB,EACzDC,eAAc,iCAAA,IAAI,CAAChB,OAAO,CAACL,OAAO,CAACmB,OAAO,cAA5B,qDAAA,+BAA8BG,qBAAqB,EACjEvB,OAAO,EACP,GAAGC,SAWJ,EACDuB,oBAAwB,CAAC,CAAC,EAC1BC,cAAkB,CAAC,CAAC;IAEpB,MAAMC,MAAM,OAAO3B,SAAS,WAAWN,gBAAgBM,MAAM,IAAI,CAACO,OAAO,IAAIP;IAC7E,MAAM4B,WAAW,IAAI,CAACrB,OAAO,CAACsB,WAAW,CAACC,aAAa,CAACH,KAAKI,IAAI;IACjE,MAAMC,WAAW,AACf,IAAI,CAGJC,UAAU,CACV;QACE,GAAG/B,OAAO;QACVF,MAAM2B;QACN1B,SAAS,CACPS,QACAC,MACAC,SACAC;gBAGE;YADF,MAAMqB,gBACJ,gDAAA,IAAI,CAAC3B,OAAO,CAACL,OAAO,CAACmB,OAAO,CAACc,MAAM,CAACC,CAAC,CAACC,MAAM,cAA5C,oEAAA,6CAA8C,CAC5C,OAAOrC,SAAS,WAAWA,OAAO,AAAC2B,IAAgCW,SAAS,CAC7E;YAEH,OAAO3C,+BACLuC,cACArB,MACAe,UACAhC,gBAAgB,IAAI,CAACW,OAAO,GAC5B;gBACEgC,KAAK3B;gBACLQ;gBACAG;gBACAZ;YACF,GAEA,CAAC6B;gBACC,6BAA6B;gBAC7B,kBAAkB;gBAClB,WAAW;gBACX,UAAU;gBACV,OAAOvC,QAAQuC,GAAY9B,QAAQC,MAAeC,SAASC;YAC3D,KAAK;YACP;QAEJ;IACF,GACAY,6BAA6BlC,YACzBkC,oBACA;QACE,GAAGA,iBAAiB;QACpBgB,YAAY;eACNhB,8BAAAA,wCAAD,AAACA,kBAAsDgB,UAAU,AAApE;QACF;IACF,GACJf;IAGF,OAAOM;AACT"}

package/esm/src/global-types.js

@@ -0,0 +1,2 @@
+/* eslint-disable @typescript-eslint/no-unused-vars */ /* eslint-disable @typescript-eslint/no-empty-interface */ export {};
+//# sourceMappingURL=global-types.js.map

package/esm/src/global-types.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/global-types.ts"],"names":[],"mappings":"AAAA,oDAAoD,GACpD,wDAAwD,GA0CxD,WAAiD"}

package/esm/src/index.js

@@ -0,0 +1,107 @@
+import './global-types.js';
+import './field-builder.js';
+import './schema-builder.js';
+import SchemaBuilder, { BasePlugin, createInputValueMapper, mapInputFields } from '@pothos/core';
+import { extractFilters, extractOrderBy, remapFromGraphQLSingleInput } from './utils/drizzle-graphql.js';
+import { getLoaderMapping, setLoaderMappings } from './utils/loader-map.js';
+export * from './types.js';
+const pluginName = "drizzle";
+export default pluginName;
+export class PothosDrizzlePlugin extends BasePlugin {
+    onOutputFieldConfig(fieldConfig) {
+        const argMappings = mapInputFields(fieldConfig.args, this.buildCache, (inputField) => {
+            if (inputField.type.kind === "InputObject") {
+                var _config_extensions;
+                const config = this.buildCache.getTypeConfig(inputField.type.ref);
+                var _config_extensions_drizzleGraphQL;
+                return (_config_extensions_drizzleGraphQL = (_config_extensions = config.extensions) === null || _config_extensions === void 0 ? void 0 : _config_extensions.drizzleGraphQL) !== null && _config_extensions_drizzleGraphQL !== void 0 ? _config_extensions_drizzleGraphQL : null;
+            }
+            return null;
+        });
+        const argMapper = argMappings ? createInputValueMapper(argMappings, (input, mappings, ctx, info) => {
+            if (!mappings.value) {
+                return input;
+            }
+            const { table, tableConfig, inputType } = mappings.value;
+            switch (inputType) {
+                case "orderBy":
+                    return extractOrderBy(tableConfig, input);
+                case "filters":
+                    return extractFilters(tableConfig, table, input);
+                case "insert":
+                    return remapFromGraphQLSingleInput(input, tableConfig);
+                case "update":
+                    return remapFromGraphQLSingleInput(input, tableConfig);
+                default:
+                    throw new Error(`Unknown drizzle input type: ${inputType}`);
+            }
+        }) : null;
+        var _fieldConfig_argMappers;
+        const argMappers = argMapper ? [
+            ...(_fieldConfig_argMappers = fieldConfig.argMappers) !== null && _fieldConfig_argMappers !== void 0 ? _fieldConfig_argMappers : [],
+            (args, context, info) => argMapper(args, undefined, context, info)
+        ] : fieldConfig.argMappers;
+        if (fieldConfig.kind === "DrizzleObject" && fieldConfig.pothosOptions.select) {
+            const { select } = fieldConfig.pothosOptions;
+            return {
+                ...fieldConfig,
+                argMappers,
+                extensions: {
+                    ...fieldConfig.extensions,
+                    pothosDrizzleSelect: typeof select === "function" ? (args, ctx, nestedQuery) => ({
+                        select: select(args, ctx, nestedQuery)
+                    }) : select
+                }
+            };
+        }
+        return {
+            ...fieldConfig,
+            argMappers
+        };
+    }
+    wrapResolve(resolver, fieldConfig) {
+        var _fieldConfig_extensions, _parentConfig_extensions;
+        if (fieldConfig.kind !== "DrizzleObject" || !((_fieldConfig_extensions = fieldConfig.extensions) === null || _fieldConfig_extensions === void 0 ? void 0 : _fieldConfig_extensions.pothosDrizzleSelect)) {
+            return resolver;
+        }
+        const parentConfig = this.buildCache.getTypeConfig(fieldConfig.parentType);
+        const parentTypes = new Set([
+            fieldConfig.parentType
+        ]);
+        if (parentConfig.kind === "Interface" || parentConfig.kind === "Object") {
+            parentConfig.interfaces.forEach((iface) => {
+                var _interfaceConfig_extensions;
+                const interfaceConfig = this.buildCache.getTypeConfig(iface, "Interface");
+                if ((_interfaceConfig_extensions = interfaceConfig.extensions) === null || _interfaceConfig_extensions === void 0 ? void 0 : _interfaceConfig_extensions.pothosDrizzleModel) {
+                    parentTypes.add(interfaceConfig.name);
+                }
+            });
+        }
+        const modelLoader = (_parentConfig_extensions = parentConfig.extensions) === null || _parentConfig_extensions === void 0 ? void 0 : _parentConfig_extensions.pothosDrizzleLoader;
+        if (!modelLoader) {
+            throw new Error(`ModelLoader not found for type ${parentConfig.name}`);
+        }
+        return (parent, args, context, info) => {
+            let mapping = getLoaderMapping(context, info.path, info.parentType.name);
+            if (!mapping) {
+                for (const parentType of parentTypes) {
+                    mapping = getLoaderMapping(context, info.path, parentType);
+                    if (mapping) {
+                        break;
+                    }
+                }
+            }
+            if (mapping) {
+                setLoaderMappings(context, info, mapping);
+                return resolver(parent, args, context, info);
+            }
+            const primaryKeys = parentConfig.extensions.pothosDrizzleTable.primaryKey;
+            if (!primaryKeys || primaryKeys.length === 0) {
+                throw new Error(`Field ${fieldConfig.name} not resolved in initial query and no primary key found for type ${parentConfig.name}`);
+            }
+            return modelLoader(context).loadSelection(info, parent).then((result) => resolver(result, args, context, info));
+        };
+    }
+}
+SchemaBuilder.registerPlugin(pluginName, PothosDrizzlePlugin, {});
+//# sourceMappingURL=index.js.map

package/esm/src/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/index.ts"],"names":["SchemaBuilder","BasePlugin","createInputValueMapper","mapInputFields","extractFilters","extractOrderBy","remapFromGraphQLSingleInput","getLoaderMapping","setLoaderMappings","pluginName","PothosDrizzlePlugin","onOutputFieldConfig","fieldConfig","argMappings","args","buildCache","inputField","type","kind","config","getTypeConfig","ref","extensions","drizzleGraphQL","argMapper","input","mappings","ctx","info","value","table","tableConfig","inputType","Error","argMappers","context","undefined","pothosOptions","select","pothosDrizzleSelect","nestedQuery","wrapResolve","resolver","parentConfig","parentType","parentTypes","Set","interfaces","forEach","iface","interfaceConfig","pothosDrizzleModel","add","name","modelLoader","pothosDrizzleLoader","parent","mapping","path","primaryKeys","pothosDrizzleTable","primaryKey","length","loadSelection","then","result","registerPlugin"],"mappings":"AAAA,OAAO,iBAAiB;AACxB,OAAO,kBAAkB;AACzB,OAAO,mBAAmB;AAE1B,OAAOA,iBACLC,UAAU,EACVC,sBAAsB,EACtBC,cAAc,QAIT,eAAe;AAGtB,SACEC,cAAc,EACdC,cAAc,EACdC,2BAA2B,QACtB,0BAA0B;AACjC,SAASC,gBAAgB,EAAEC,iBAAiB,QAAQ,qBAAqB;AAEzE,cAAc,UAAU;AAExB,MAAMC,aAAa;AAEnB,eAAeA,WAAW;AAE1B,OAAO,MAAMC,4BAAuDT;IACzDU,oBACPC,WAA2C,EACJ;QACvC,MAAMC,cAAcV,eAClBS,YAAYE,IAAI,EAChB,IAAI,CAACC,UAAU,EACf,CAACC;YACC,IAAIA,WAAWC,IAAI,CAACC,IAAI,KAAK,eAAe;oBAGlCC;gBAFR,MAAMA,SAAS,IAAI,CAACJ,UAAU,CAACK,aAAa,CAACJ,WAAWC,IAAI,CAACI,GAAG;oBAExDF;gBAAR,OAAO,CAACA,qCAAAA,qBAAAA,OAAOG,UAAU,cAAjBH,yCAAAA,mBAAmBI,cAAc,cAAjCJ,+CAAAA,oCAAuE;YACjF;YAEA,OAAO;QACT;QAGF,MAAMK,YAAYX,cACdX,uBACEW,aACA,CAACY,OAAOC,UAAUC,KAAuBC;YACvC,IAAI,CAACF,SAASG,KAAK,EAAE;gBACnB,OAAOJ;YACT;YAEA,MAAM,EAAEK,KAAK,EAAEC,WAAW,EAAEC,SAAS,EAAE,GAAGN,SAASG,KAAK;YAExD,OAAQG;gBACN,KAAK;oBACH,OAAO3B,eAAe0B,aAAaN;gBACrC,KAAK;oBACH,OAAOrB,eAAe2B,aAAaD,OAAOL;gBAC5C,KAAK;oBACH,OAAOnB,4BAA4BmB,OAAgBM;gBACrD,KAAK;oBACH,OAAOzB,4BAA4BmB,OAAgBM;gBAErD;oBACE,MAAM,IAAIE,MAAM,CAAC,4BAA4B,EAAED,UAAU,CAAC;YAC9D;QACF,KAEF;YAIMpB;QAFV,MAAMsB,aAA4CV,YAC9C;eACMZ,CAAAA,0BAAAA,YAAYsB,UAAU,cAAtBtB,qCAAAA,0BAA0B,EAAE;YAChC,CAACE,MAAMqB,SAASP,OAASJ,UAAUV,MAAMsB,WAAWD,SAASP;SAC9D,GACDhB,YAAYsB,UAAU;QAE1B,IAAItB,YAAYM,IAAI,KAAK,mBAAmBN,YAAYyB,aAAa,CAACC,MAAM,EAAE;YAC5E,MAAM,EAAEA,MAAM,EAAE,GAAG1B,YAAYyB,aAAa;YAC5C,OAAO;gBACL,GAAGzB,WAAW;gBACdsB;gBACAZ,YAAY;oBACV,GAAGV,YAAYU,UAAU;oBACzBiB,qBACE,OAAOD,WAAW,aACd,CACExB,MACAa,KACAa,cACI,CAAA;4BACJF,QAAQ,AAACA,OACPxB,MACAa,KACAa;wBAEJ,CAAA,IACAF;gBACR;YACF;QACF;QAEA,OAAO;YACL,GAAG1B,WAAW;YACdsB;QACF;IACF;IAESO,YACPC,QAA0E,EAC1E9B,WAA2C,EACc;YACZA,yBAiBzB+B;QAjBpB,IAAI/B,YAAYM,IAAI,KAAK,mBAAmB,GAACN,0BAAAA,YAAYU,UAAU,cAAtBV,8CAAAA,wBAAwB2B,mBAAmB,GAAE;YACxF,OAAOG;QACT;QAEA,MAAMC,eAAe,IAAI,CAAC5B,UAAU,CAACK,aAAa,CAACR,YAAYgC,UAAU;QAEzE,MAAMC,cAAc,IAAIC,IAAI;YAAClC,YAAYgC,UAAU;SAAC;QAEpD,IAAID,aAAazB,IAAI,KAAK,eAAeyB,aAAazB,IAAI,KAAK,UAAU;YACvEyB,aAAaI,UAAU,CAACC,OAAO,CAAC,CAACC;oBAE3BC;gBADJ,MAAMA,kBAAkB,IAAI,CAACnC,UAAU,CAACK,aAAa,CAAC6B,OAAO;gBAC7D,KAAIC,8BAAAA,gBAAgB5B,UAAU,cAA1B4B,kDAAAA,4BAA4BC,kBAAkB,EAAE;oBAClDN,YAAYO,GAAG,CAACF,gBAAgBG,IAAI;gBACtC;YACF;QACF;QAEA,MAAMC,eAAcX,2BAAAA,aAAarB,UAAU,cAAvBqB,+CAAAA,yBAAyBY,mBAAmB;QAIhE,IAAI,CAACD,aAAa;YAChB,MAAM,IAAIrB,MAAM,CAAC,+BAA+B,EAAEU,aAAaU,IAAI,CAAC,CAAC;QACvE;QAEA,OAAO,CAACG,QAAQ1C,MAAMqB,SAASP;YAC7B,IAAI6B,UAAUlD,iBAAiB4B,SAASP,KAAK8B,IAAI,EAAE9B,KAAKgB,UAAU,CAACS,IAAI;YAEvE,IAAI,CAACI,SAAS;gBACZ,KAAK,MAAMb,cAAcC,YAAa;oBACpCY,UAAUlD,iBAAiB4B,SAASP,KAAK8B,IAAI,EAAEd;oBAC/C,IAAIa,SAAS;wBACX;oBACF;gBACF;YACF;YAEA,IAAIA,SAAS;gBACXjD,kBAAkB2B,SAASP,MAAM6B;gBAEjC,OAAOf,SAASc,QAAQ1C,MAAMqB,SAASP;YACzC;YAEA,MAAM+B,cAAc,AAClBhB,aAAarB,UAAU,CAAEsC,kBAAkB,CAC3CC,UAAU;YAEZ,IAAI,CAACF,eAAeA,YAAYG,MAAM,KAAK,GAAG;gBAC5C,MAAM,IAAI7B,MACR,CAAC,MAAM,EAAErB,YAAYyC,IAAI,CAAC,iEAAiE,EAAEV,aAAaU,IAAI,CAAC,CAAC;YAEpH;YAEA,OAAOC,YAAYnB,SAChB4B,aAAa,CAACnC,MAAM4B,QACpBQ,IAAI,CAAC,CAACC,SAAWvB,SAASuB,QAAQnD,MAAMqB,SAASP;QACtD;IACF;AACF;AAEA5B,cAAckE,cAAc,CAACzD,YAAYC,qBAAqB,CAAC"}

package/esm/src/interface-ref.js

@@ -0,0 +1,26 @@
+function _define_property(obj, key, value) {
+    if (key in obj) {
+        Object.defineProperty(obj, key, {
+            value: value,
+            enumerable: true,
+            configurable: true,
+            writable: true
+        });
+    }
+    else {
+        obj[key] = value;
+    }
+    return obj;
+}
+import { InterfaceRef } from '@pothos/core';
+import { drizzleTableKey } from './object-ref.js';
+let _drizzleTableKey = drizzleTableKey;
+export class DrizzleInterfaceRef extends InterfaceRef {
+    constructor(name, tableName) {
+        super(name);
+        _define_property(this, _drizzleTableKey, void 0);
+        _define_property(this, "tableName", void 0);
+        this.tableName = tableName;
+    }
+}
+//# sourceMappingURL=interface-ref.js.map

package/esm/src/interface-ref.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/interface-ref.ts"],"names":["InterfaceRef","drizzleTableKey","DrizzleInterfaceRef","constructor","name","tableName"],"mappings":";;;;;;;;;;;;;AAAA,SAASA,YAAY,QAAqB,eAAe;AACzD,SAA2BC,eAAe,QAAQ,eAAe;IAa9DA,mBAAAA;AALH,OAAO,MAAMC,4BAIHF;IAKRG,YAAYC,IAAY,EAAEC,SAAiB,CAAE;QAC3C,KAAK,CAACD;QALR,uBAACH,kBAAD,KAAA;QAEAI,uBAAAA,aAAAA,KAAAA;QAKE,IAAI,CAACA,SAAS,GAAGA;IACnB;AACF"}

package/esm/src/model-loader.js

@@ -0,0 +1,168 @@
+function _define_property(obj, key, value) {
+    if (key in obj) {
+        Object.defineProperty(obj, key, {
+            value: value,
+            enumerable: true,
+            configurable: true,
+            writable: true
+        });
+    }
+    else {
+        obj[key] = value;
+    }
+    return obj;
+}
+import { inArray, sql } from 'drizzle-orm';
+import { createContextCache } from '@pothos/core';
+import { cacheKey, setLoaderMappings } from './utils/loader-map.js';
+import { selectionStateFromInfo, stateFromInfo } from './utils/map-query.js';
+import { mergeSelection, selectionCompatible, selectionToQuery } from './utils/selections.js';
+import { getSchemaConfig } from './utils/config.js';
+export class ModelLoader {
+    static forModel(modelName, builder, columns) {
+        return createContextCache((model) => new ModelLoader(model, builder, modelName, columns));
+    }
+    sqlForModel(model) {
+        const values = this.columns.map((key) => {
+            const columnName = key.name;
+            if (columnName in model) {
+                return model[columnName];
+            }
+            throw new Error(`Primary key column ${columnName} not found on ${this.modelName}`);
+        });
+        return this.columns.length > 1 ? values : values[0];
+    }
+    getSelection(info) {
+        const key = cacheKey(info.parentType.name, info.path);
+        if (!this.queryCache.has(key)) {
+            const selection = selectionStateFromInfo(this.config, this.context, info);
+            this.queryCache.set(key, {
+                selection,
+                query: selectionToQuery(selection)
+            });
+        }
+        return this.queryCache.get(key);
+    }
+    getSelectionForField(info, typeName) {
+        const key = cacheKey(typeName, info.path);
+        if (!this.queryCache.has(key)) {
+            const selection = stateFromInfo({
+                config: this.config,
+                context: this.context,
+                info,
+                typeName
+            });
+            this.queryCache.set(key, {
+                selection,
+                query: selectionToQuery(selection)
+            });
+        }
+        return this.queryCache.get(key);
+    }
+    async loadSelection(info, model) {
+        const { selection, query } = this.getSelection(info);
+        const result = await this.stageQuery(selection, query, model);
+        if (result) {
+            const mappings = selection.mappings[info.path.key];
+            if (mappings) {
+                setLoaderMappings(this.context, info, mappings.mappings);
+            }
+        }
+        return result;
+    }
+    async loadSelectionForField(info, model, returnType) {
+        const { selection, query } = this.getSelectionForField(info, returnType);
+        const result = await this.stageQuery(selection, query, model);
+        if (result) {
+            setLoaderMappings(this.context, info, selection.mappings);
+        }
+        return result;
+    }
+    async stageQuery(selection, query, model) {
+        for (const entry of this.staged) {
+            if (selectionCompatible(entry.state, query)) {
+                mergeSelection(this.config, entry.state, query);
+                if (!entry.models.has(model)) {
+                    entry.models.set(model, createResolvablePromise());
+                }
+                return entry.models.get(model).promise;
+            }
+        }
+        return this.initLoad(selection, model);
+    }
+    initLoad(selection, model) {
+        const promise = createResolvablePromise();
+        const entry = {
+            state: selection,
+            models: new Map([
+                [
+                    model,
+                    promise
+                ]
+            ])
+        };
+        this.staged.add(entry);
+        const nextTick = createResolvablePromise();
+        nextTick.promise.then(() => {
+            const api = this.builder.options.drizzle.client.query[this.modelName];
+            const query = api.findMany({
+                ...selectionToQuery(selection),
+                where: inArray(this.selectSQL, [
+                    entry.models.keys()
+                ].map(([model]) => this.sqlForModel(model)))
+            });
+            query.then((results) => {
+                for (const [model, promise] of entry.models.entries()) {
+                    const result = results.find((row) => this.table.primaryKey.every((key) => row[key.name] === model[key.name]));
+                    if (result) {
+                        promise.resolve(result !== null && result !== void 0 ? result : null);
+                    }
+                    else
+                        promise.reject(new Error(`Model ${this.modelName}(${this.sqlForModel(model)}) not found`));
+                }
+            }, (err) => {
+                for (const promise of entry.models.values()) {
+                    promise.reject(err);
+                }
+            });
+        }).catch((err) => {
+            for (const promise of entry.models.values()) {
+                promise.reject(err);
+            }
+        });
+        setTimeout(() => void nextTick.resolve(), 0);
+        return promise.promise;
+    }
+    constructor(context, builder, modelName, columns) {
+        _define_property(this, "context", void 0);
+        _define_property(this, "builder", void 0);
+        _define_property(this, "modelName", void 0);
+        _define_property(this, "queryCache", new Map());
+        _define_property(this, "staged", new Set());
+        _define_property(this, "config", void 0);
+        _define_property(this, "table", void 0);
+        _define_property(this, "columns", void 0);
+        _define_property(this, "selectSQL", void 0);
+        this.context = context;
+        this.builder = builder;
+        this.modelName = modelName;
+        this.config = getSchemaConfig(builder);
+        this.table = this.config.schema[modelName];
+        this.columns = columns !== null && columns !== void 0 ? columns : this.table.primaryKey;
+        this.selectSQL = this.columns.length > 1 ? sql `(${sql.join(this.columns, sql `, `)})` : this.columns[0].getSQL();
+    }
+}
+function createResolvablePromise() {
+    let resolveFn;
+    let rejectFn;
+    const promise = new Promise((resolve, reject) => {
+        resolveFn = resolve;
+        rejectFn = reject;
+    });
+    return {
+        promise,
+        resolve: resolveFn,
+        reject: rejectFn
+    };
+}
+//# sourceMappingURL=model-loader.js.map