@pothos/plugin-drizzle 0.1.0 → 0.2.1

package/lib/src/schema-builder.js

@@ -8,6 +8,7 @@ const _interfaceref = require("./interface-ref");
 const _modelloader = require("./model-loader");
 const _noderef = require("./node-ref");
 const _objectref = require("./object-ref");
+const _config = require("./utils/config");
 const _cursors = require("./utils/cursors");
 const _refs = require("./utils/refs");
 function _getRequireWildcardCache(nodeInterop) {
@@ -53,7 +54,7 @@ function _interop_require_wildcard(obj, nodeInterop) {
 }
 const schemaBuilderProto = _core.default.prototype;
 schemaBuilderProto.drizzleObject = function drizzleObject(table, { select, fields, ...options }) {
-    var _this_options_drizzle_client___schema;
+    var _getSchemaConfig_schema;
     var _options_variant, _ref;
     const name = (_ref = (_options_variant = options.variant) !== null && _options_variant !== void 0 ? _options_variant : options.name) !== null && _ref !== void 0 ? _ref : table;
     const ref = options.variant ? new _objectref.DrizzleObjectRef(options.variant, name) : (0, _refs.getRefFromModel)(table, this, 'object');
@@ -63,7 +64,7 @@ schemaBuilderProto.drizzleObject = function drizzleObject(table, { select, field
         extensions: {
             ...options.extensions,
             pothosDrizzleModel: table,
-            pothosDrizzleTable: (_this_options_drizzle_client___schema = this.options.drizzle.client._.schema) === null || _this_options_drizzle_client___schema === void 0 ? void 0 : _this_options_drizzle_client___schema[table],
+            pothosDrizzleTable: (_getSchemaConfig_schema = (0, _config.getSchemaConfig)(this).schema) === null || _getSchemaConfig_schema === void 0 ? void 0 : _getSchemaConfig_schema[table],
             pothosDrizzleSelect: select !== null && select !== void 0 ? select : {},
             pothosDrizzleLoader: _modelloader.ModelLoader.forModel(table, this)
         },
@@ -73,7 +74,7 @@ schemaBuilderProto.drizzleObject = function drizzleObject(table, { select, field
     return ref;
 };
 schemaBuilderProto.drizzleInterface = function drizzleInterface(table, { select, fields, ...options }) {
-    var _this_options_drizzle_client___schema;
+    var _getSchemaConfig_schema;
     var _options_name;
     const name = (_options_name = options.name) !== null && _options_name !== void 0 ? _options_name : table;
     const ref = options.variant ? new _interfaceref.DrizzleInterfaceRef(options.variant, name) : (0, _refs.getRefFromModel)(table, this, 'interface');
@@ -83,7 +84,7 @@ schemaBuilderProto.drizzleInterface = function drizzleInterface(table, { select,
         extensions: {
             ...options.extensions,
             pothosDrizzleModel: table,
-            pothosDrizzleTable: (_this_options_drizzle_client___schema = this.options.drizzle.client._.schema) === null || _this_options_drizzle_client___schema === void 0 ? void 0 : _this_options_drizzle_client___schema[table],
+            pothosDrizzleTable: (_getSchemaConfig_schema = (0, _config.getSchemaConfig)(this).schema) === null || _getSchemaConfig_schema === void 0 ? void 0 : _getSchemaConfig_schema[table],
             pothosDrizzleSelect: select !== null && select !== void 0 ? select : {},
             pothosDrizzleLoader: _modelloader.ModelLoader.forModel(table, this)
         },
@@ -94,7 +95,7 @@ schemaBuilderProto.drizzleInterface = function drizzleInterface(table, { select,
 };
 schemaBuilderProto.drizzleNode = function drizzleNode(table, { id: { column, ...idOptions }, name, variant, ...options }) {
     var _this_nodeInterfaceRef, _this;
-    const tableConfig = this.options.drizzle.client._.schema[table];
+    const tableConfig = (0, _config.getSchemaConfig)(this).schema[table];
     const idColumn = typeof column === 'function' ? column(tableConfig.columns) : column;
     const idColumns = Array.isArray(idColumn) ? idColumn : [
         idColumn
@@ -177,7 +178,7 @@ schemaBuilderProto.drizzleInterfaceFields = function drizzleInterfaceFields(type
     });
 };
 schemaBuilderProto.drizzleGraphQLOrderBy = function drizzleGraphQLOrderBy(table, type, options) {
-    var _this_options_drizzle_client___schema;
+    var _getSchemaConfig_schema;
     return this.addGraphQLInput(type, {
         ...options,
         extensions: {
@@ -185,13 +186,13 @@ schemaBuilderProto.drizzleGraphQLOrderBy = function drizzleGraphQLOrderBy(table,
             drizzleGraphQL: {
                 inputType: 'orderBy',
                 table,
-                tableConfig: (_this_options_drizzle_client___schema = this.options.drizzle.client._.schema) === null || _this_options_drizzle_client___schema === void 0 ? void 0 : _this_options_drizzle_client___schema[table]
+                tableConfig: (_getSchemaConfig_schema = (0, _config.getSchemaConfig)(this).schema) === null || _getSchemaConfig_schema === void 0 ? void 0 : _getSchemaConfig_schema[table]
             }
         }
     });
 };
 schemaBuilderProto.drizzleGraphQLFilters = function drizzleGraphQLFilters(table, type, options) {
-    var _this_options_drizzle_client___schema;
+    var _getSchemaConfig_schema;
     return this.addGraphQLInput(type, {
         ...options,
         extensions: {
@@ -199,13 +200,13 @@ schemaBuilderProto.drizzleGraphQLFilters = function drizzleGraphQLFilters(table,
             drizzleGraphQL: {
                 inputType: 'filters',
                 table,
-                tableConfig: (_this_options_drizzle_client___schema = this.options.drizzle.client._.schema) === null || _this_options_drizzle_client___schema === void 0 ? void 0 : _this_options_drizzle_client___schema[table]
+                tableConfig: (_getSchemaConfig_schema = (0, _config.getSchemaConfig)(this).schema) === null || _getSchemaConfig_schema === void 0 ? void 0 : _getSchemaConfig_schema[table]
             }
         }
     });
 };
 schemaBuilderProto.drizzleGraphQLInsert = function drizzleGraphQLInsert(table, type, options) {
-    var _this_options_drizzle_client___schema;
+    var _getSchemaConfig_schema;
     return this.addGraphQLInput(type, {
         ...options,
         extensions: {
@@ -213,13 +214,13 @@ schemaBuilderProto.drizzleGraphQLInsert = function drizzleGraphQLInsert(table, t
             drizzleGraphQL: {
                 inputType: 'insert',
                 table,
-                tableConfig: (_this_options_drizzle_client___schema = this.options.drizzle.client._.schema) === null || _this_options_drizzle_client___schema === void 0 ? void 0 : _this_options_drizzle_client___schema[table]
+                tableConfig: (_getSchemaConfig_schema = (0, _config.getSchemaConfig)(this).schema) === null || _getSchemaConfig_schema === void 0 ? void 0 : _getSchemaConfig_schema[table]
             }
         }
     });
 };
 schemaBuilderProto.drizzleGraphQLUpdate = function drizzleGraphQLUpdate(table, type, options) {
-    var _this_options_drizzle_client___schema;
+    var _getSchemaConfig_schema;
     return this.addGraphQLInput(type, {
         ...options,
         extensions: {
@@ -227,7 +228,7 @@ schemaBuilderProto.drizzleGraphQLUpdate = function drizzleGraphQLUpdate(table, t
             drizzleGraphQL: {
                 inputType: 'update',
                 table,
-                tableConfig: (_this_options_drizzle_client___schema = this.options.drizzle.client._.schema) === null || _this_options_drizzle_client___schema === void 0 ? void 0 : _this_options_drizzle_client___schema[table]
+                tableConfig: (_getSchemaConfig_schema = (0, _config.getSchemaConfig)(this).schema) === null || _getSchemaConfig_schema === void 0 ? void 0 : _getSchemaConfig_schema[table]
             }
         }
     });

package/lib/src/schema-builder.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/schema-builder.ts"],"names":["schemaBuilderProto","SchemaBuilder","prototype","drizzleObject","table","select","fields","options","name","variant","ref","DrizzleObjectRef","getRefFromModel","objectType","extensions","pothosDrizzleModel","pothosDrizzleTable","drizzle","client","_","schema","pothosDrizzleSelect","pothosDrizzleLoader","ModelLoader","forModel","DrizzleObjectFieldBuilder","undefined","drizzleInterface","DrizzleInterfaceRef","interfaceType","drizzleNode","id","column","idOptions","tableConfig","idColumn","columns","idColumns","Array","isArray","interfaceRef","nodeInterfaceRef","resolve","getColumnSerializer","idParser","getColumnParser","typeName","nodeRef","DrizzleNodeRef","modelLoader","PothosError","extendedOptions","interfaces","loadWithoutCache","context","info","record","loadSelectionForField","brandWithType","addInterfaces","configStore","onTypeConfig","nodeConfig","objectField","relayOptions","idFieldName","t","globalID","idFieldOptions","nullable","args","parent","completeValue","type","associateParamWithRef","drizzleObjectField","fieldName","field","addFields","tableName","drizzleInterfaceField","drizzleObjectFields","drizzleInterfaceFields","drizzleGraphQLOrderBy","addGraphQLInput","drizzleGraphQL","inputType","drizzleGraphQLFilters","drizzleGraphQLInsert","drizzleGraphQLUpdate"],"mappings":";;;;8DAUO;qCAEmC;8BACN;6BACR;yBACG;2BACE;yBAEoB;sBACrB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEhC,MAAMA,qBAAqBC,aAAa,CAACC,SAAS;AAElDF,mBAAmBG,aAAa,GAAG,SAASA,cAAcC,KAAK,EAAE,EAAEC,MAAM,EAAEC,MAAM,EAAE,GAAGC,SAAS;QAcrE;QAbXA,kBAAAA;IAAb,MAAMC,OAAOD,CAAAA,OAAAA,CAAAA,mBAAAA,QAAQE,OAAO,cAAfF,8BAAAA,mBAAmBA,QAAQC,IAAI,cAA/BD,kBAAAA,OAAmCH;IAEhD,MAAMM,MAAMH,QAAQE,OAAO,GACvB,IAAIE,2BAAgB,CAACJ,QAAQE,OAAO,EAAED,QACrCI,IAAAA,qBAAe,EAACR,OAAO,IAAI,EAAE;IAElCM,IAAIF,IAAI,GAAGA,iBAAAA,kBAAAA,OAAQJ;IAEnB,IAAI,CAACS,UAAU,CAACH,KAAK;QACnB,GAAIH,OAAO;QACXO,YAAY;YACV,GAAGP,QAAQO,UAAU;YACrBC,oBAAoBX;YACpBY,kBAAkB,GAAE,wCAAA,IAAI,CAACT,OAAO,CAACU,OAAO,CAACC,MAAM,CAACC,CAAC,CAACC,MAAM,cAApC,4DAAA,qCAAsC,CAAChB,MAAM;YACjEiB,qBAAqBhB,mBAAAA,oBAAAA,SAAU,CAAC;YAChCiB,qBAAqBC,wBAAW,CAACC,QAAQ,CAACpB,OAAO,IAAI;QACvD;QACAI;QACAF,QAAQA,SAAS,IAAMA,OAAO,IAAImB,8CAAyB,CAACf,IAAIF,IAAI,EAAE,IAAI,EAAEJ,UAAUsB;IACxF;IAEA,OAAOhB;AACT;AAEAV,mBAAmB2B,gBAAgB,GAAG,SAASA,iBAC7CvB,KAAK,EACL,EAAEC,MAAM,EAAEC,MAAM,EAAE,GAAGC,SAAS;QAeN;QAbXA;IAAb,MAAMC,OAAOD,CAAAA,gBAAAA,QAAQC,IAAI,cAAZD,2BAAAA,gBAAgBH;IAE7B,MAAMM,MAAMH,QAAQE,OAAO,GACvB,IAAImB,iCAAmB,CAACrB,QAAQE,OAAO,EAAED,QACxCI,IAAAA,qBAAe,EAACR,OAAO,IAAI,EAAE;IAElCM,IAAIF,IAAI,GAAGA,iBAAAA,kBAAAA,OAAQJ;IAEnB,IAAI,CAACyB,aAAa,CAACnB,KAAK;QACtB,GAAIH,OAAO;QACXO,YAAY;YACV,GAAGP,QAAQO,UAAU;YACrBC,oBAAoBX;YACpBY,kBAAkB,GAAE,wCAAA,IAAI,CAACT,OAAO,CAACU,OAAO,CAACC,MAAM,CAACC,CAAC,CAACC,MAAM,cAApC,4DAAA,qCAAsC,CAAChB,MAAM;YACjEiB,qBAAqBhB,mBAAAA,oBAAAA,SAAU,CAAC;YAChCiB,qBAAqBC,wBAAW,CAACC,QAAQ,CAACpB,OAAO,IAAI;QACvD;QACAI;QACAF,QAAQA,SACJ,IAAMA,OAAO,IAAImB,8CAAyB,CAACf,IAAIF,IAAI,EAAE,IAAI,EAAEJ,OAAO,gBAClEsB;IACN;IAEA,OAAOhB;AACT;AAEAV,mBAAmB8B,WAAW,GAAG,SAASA,YAIxC1B,KAAiD,EACjD,EACE2B,IAAI,EAAEC,MAAM,EAAE,GAAGC,WAAW,EAC5BzB,IAAI,EACJC,OAAO,EACP,GAAGF,SACqF;QAKrE,wBAAA;IAHrB,MAAM2B,cAAc,IAAI,CAAC3B,OAAO,CAACU,OAAO,CAACC,MAAM,CAACC,CAAC,CAACC,MAAM,AAAC,CAAChB,MAAM;IAChE,MAAM+B,WAAW,OAAOH,WAAW,aAAaA,OAAOE,YAAYE,OAAO,IAAIJ;IAC9E,MAAMK,YAAYC,MAAMC,OAAO,CAACJ,YAAYA,WAAW;QAACA;KAAS;IACjE,MAAMK,gBAAe,yBAAA,CAAA,QAAA,IAAI,EAACC,gBAAgB,cAArB,6CAAA,4BAAA;IACrB,MAAMC,UAAUC,IAAAA,4BAAmB,EAACN;IACpC,MAAMO,WAAWC,IAAAA,wBAAe,EAACR;QAChB5B;IAAjB,MAAMqC,WAAWrC,CAAAA,OAAAA,oBAAAA,qBAAAA,UAAWD,kBAAXC,kBAAAA,OAAmBL;IACpC,MAAM2C,UAAU,IAAIC,uBAAc,CAACF,UAAU1C;IAC7C,MAAM6C,cAAc1B,wBAAW,CAACC,QAAQ,CAACpB,OAAO,IAAI,EAAEiC;IAEtD,IAAI,CAACG,cAAc;QACjB,MAAM,IAAIU,iBAAW,CAAC;IACxB;IAEA,MAAMC,kBAAkB;QACtB,GAAG5C,OAAO;QACVC;QACAC;QACA2C,YAAY;YAACZ;SAAa;QAC1Ba,kBAAkB,OAChBtB,IACAuB,SACAC;YAEA,MAAMC,SAAS,MAAMP,YAAYK,SAASG,qBAAqB,CAACF,MAAMX,SAASb,KAAKe;YAEpF,IAAIU,QAAQ;gBACVE,IAAAA,mBAAa,EAACF,QAAQV;YACxB;YAEA,OAAOU;QACT;IACF;IAEA,MAAM9C,MAAM,IAAI,CAACP,aAAa,CAACC,OAAO+C;IAEtC,IAAI5C,QAAQ6C,UAAU,EAAE;QACtB1C,IAAIiD,aAAa,CAACpD,QAAQ6C,UAAU;IACtC;IAEA,IAAI,CAACQ,WAAW,CAACC,YAAY,CAACnD,KAAK,CAACoD;YAGhC;YAAA;QAFF,IAAI,CAACC,WAAW,CACdrD,KACA,CAAA,0CAAA,6BAAA,AAAC,IAAI,CAACH,OAAO,CAAiDyD,YAAY,cAA1E,iDAAA,2BAA4EC,WAAW,cAAvF,oDAAA,yCACE,MACF,CAACC;gBAMM;mBALL,AACEA,EAGAC,QAAQ,CAAC;oBACN,6BAAA,AAAC,IAAI,CAAC5D,OAAO,CAAoDyD,YAAY,cAA7E,iDAAA,2BACCI,cAAc,AADlB;gBAEA,GAAGnC,SAAS;gBACZoC,UAAU;gBACVC,MAAM,CAAC;gBACP5B,SAAS,CAAC6B,SACRC,IAAAA,mBAAa,EAAC9B,QAAQ6B,SAAS,CAACxC,KAAQ,CAAA;4BACtC0C,MAAMX,WAAWtD,IAAI;4BACrBuB;wBACF,CAAA;YACJ;;IAEN;IAEA,IAAI,CAAC6B,WAAW,CAACc,qBAAqB,CAAC3B,SAASrC;IAEhD,OAAOqC;AACT;AAEA/C,mBAAmB2E,kBAAkB,GAAG,SAASA,mBAAmBF,IAAI,EAAEG,SAAS,EAAEC,KAAK;IACxF,MAAMnE,MAAM,OAAO+D,SAAS,WAAW7D,IAAAA,qBAAe,EAAC6D,MAAM,IAAI,IAAKA;IACtE,IAAI,CAACb,WAAW,CAACC,YAAY,CAACnD,KAAK,CAAC,EAAEF,IAAI,EAAE;QAC1C,IAAI,CAACoD,WAAW,CAACkB,SAAS,CAACpE,KAAK,IAAO,CAAA;gBACrC,CAACkE,UAAU,EAAEC,MAAM,IAAIpD,8CAAyB,CAACjB,MAAM,IAAI,EAAEE,IAAIqE,SAAS;YAC5E,CAAA;IACF;AACF;AAEA/E,mBAAmBgF,qBAAqB,GAAG,SAASA,sBAAsBP,IAAI,EAAEG,SAAS,EAAEC,KAAK;IAC9F,MAAMnE,MAAM,OAAO+D,SAAS,WAAW7D,IAAAA,qBAAe,EAAC6D,MAAM,IAAI,IAAKA;IACtE,IAAI,CAACb,WAAW,CAACC,YAAY,CAACnD,KAAK,CAAC,EAAEF,IAAI,EAAE;QAC1C,IAAI,CAACoD,WAAW,CAACkB,SAAS,CAACpE,KAAK,IAAO,CAAA;gBACrC,CAACkE,UAAU,EAAEC,MAAM,IAAIpD,8CAAyB,CAACjB,MAAM,IAAI,EAAEE,IAAIqE,SAAS,EAAE;YAC9E,CAAA;IACF;AACF;AAEA/E,mBAAmBiF,mBAAmB,GAAG,SAASA,oBAAoBR,IAAI,EAAEnE,MAAM;IAChF,MAAMI,MAAM,OAAO+D,SAAS,WAAW7D,IAAAA,qBAAe,EAAC6D,MAAM,IAAI,IAAKA;IACtE,IAAI,CAACb,WAAW,CAACC,YAAY,CAACnD,KAAK,CAAC,EAAEF,IAAI,EAAE;QAC1C,IAAI,CAACoD,WAAW,CAACkB,SAAS,CAACpE,KAAK,IAC9BJ,OAAO,IAAImB,8CAAyB,CAACjB,MAAM,IAAI,EAAEE,IAAIqE,SAAS;IAElE;AACF;AAEA/E,mBAAmBkF,sBAAsB,GAAG,SAASA,uBAAuBT,IAAI,EAAEnE,MAAM;IACtF,MAAMI,MAAM,OAAO+D,SAAS,WAAW7D,IAAAA,qBAAe,EAAC6D,MAAM,IAAI,IAAKA;IACtE,IAAI,CAACb,WAAW,CAACC,YAAY,CAACnD,KAAK,CAAC,EAAEF,IAAI,EAAE;QAC1C,IAAI,CAACoD,WAAW,CAACkB,SAAS,CAACpE,KAAK,IAC9BJ,OAAO,IAAImB,8CAAyB,CAACjB,MAAM,IAAI,EAAEE,IAAIqE,SAAS,EAAE;IAEpE;AACF;AAEA/E,mBAAmBmF,qBAAqB,GAAG,SAASA,sBAElD/E,KAAa,EACbqE,IAA4B,EAC5BlE,OAAiC;QAgBd;IAdnB,OAAO,AACL,IAAI,CAMJ6E,eAAe,CAACX,MAAM;QACtB,GAAGlE,OAAO;QACVO,YAAY;eACPP,oBAAAA,8BAAAA,QAASO,UAAU,AAAtB;YACAuE,gBAAgB;gBACdC,WAAW;gBACXlF;gBACA8B,WAAW,GAAE,wCAAA,IAAI,CAAC3B,OAAO,CAACU,OAAO,CAACC,MAAM,CAACC,CAAC,CAACC,MAAM,cAApC,4DAAA,qCAAsC,CAAChB,MAAM;YAC5D;QACF;IACF;AACF;AAEAJ,mBAAmBuF,qBAAqB,GAAG,SAASA,sBAElDnF,KAAa,EACbqE,IAA4B,EAC5BlE,OAAiC;QAgBd;IAdnB,OAAO,AACL,IAAI,CAMJ6E,eAAe,CAACX,MAAM;QACtB,GAAGlE,OAAO;QACVO,YAAY;eACPP,oBAAAA,8BAAAA,QAASO,UAAU,AAAtB;YACAuE,gBAAgB;gBACdC,WAAW;gBACXlF;gBACA8B,WAAW,GAAE,wCAAA,IAAI,CAAC3B,OAAO,CAACU,OAAO,CAACC,MAAM,CAACC,CAAC,CAACC,MAAM,cAApC,4DAAA,qCAAsC,CAAChB,MAAM;YAC5D;QACF;IACF;AACF;AAEAJ,mBAAmBwF,oBAAoB,GAAG,SAASA,qBAEjDpF,KAAa,EACbqE,IAA4B,EAC5BlE,OAAiC;QAgBd;IAdnB,OAAO,AACL,IAAI,CAMJ6E,eAAe,CAACX,MAAM;QACtB,GAAGlE,OAAO;QACVO,YAAY;eACPP,oBAAAA,8BAAAA,QAASO,UAAU,AAAtB;YACAuE,gBAAgB;gBACdC,WAAW;gBACXlF;gBACA8B,WAAW,GAAE,wCAAA,IAAI,CAAC3B,OAAO,CAACU,OAAO,CAACC,MAAM,CAACC,CAAC,CAACC,MAAM,cAApC,4DAAA,qCAAsC,CAAChB,MAAM;YAC5D;QACF;IACF;AACF;AAEAJ,mBAAmByF,oBAAoB,GAAG,SAASA,qBAEjDrF,KAAa,EACbqE,IAA4B,EAC5BlE,OAAiC;QAgBd;IAdnB,OAAO,AACL,IAAI,CAMJ6E,eAAe,CAACX,MAAM;QACtB,GAAGlE,OAAO;QACVO,YAAY;eACPP,oBAAAA,8BAAAA,QAASO,UAAU,AAAtB;YACAuE,gBAAgB;gBACdC,WAAW;gBACXlF;gBACA8B,WAAW,GAAE,wCAAA,IAAI,CAAC3B,OAAO,CAACU,OAAO,CAACC,MAAM,CAACC,CAAC,CAACC,MAAM,cAApC,4DAAA,qCAAsC,CAAChB,MAAM;YAC5D;QACF;IACF;AACF"}
+{"version":3,"sources":["../../src/schema-builder.ts"],"names":["schemaBuilderProto","SchemaBuilder","prototype","drizzleObject","table","select","fields","options","getSchemaConfig","name","variant","ref","DrizzleObjectRef","getRefFromModel","objectType","extensions","pothosDrizzleModel","pothosDrizzleTable","schema","pothosDrizzleSelect","pothosDrizzleLoader","ModelLoader","forModel","DrizzleObjectFieldBuilder","undefined","drizzleInterface","DrizzleInterfaceRef","interfaceType","drizzleNode","id","column","idOptions","tableConfig","idColumn","columns","idColumns","Array","isArray","interfaceRef","nodeInterfaceRef","resolve","getColumnSerializer","idParser","getColumnParser","typeName","nodeRef","DrizzleNodeRef","modelLoader","PothosError","extendedOptions","interfaces","loadWithoutCache","context","info","record","loadSelectionForField","brandWithType","addInterfaces","configStore","onTypeConfig","nodeConfig","objectField","relayOptions","idFieldName","t","globalID","idFieldOptions","nullable","args","parent","completeValue","type","associateParamWithRef","drizzleObjectField","fieldName","field","addFields","tableName","drizzleInterfaceField","drizzleObjectFields","drizzleInterfaceFields","drizzleGraphQLOrderBy","addGraphQLInput","drizzleGraphQL","inputType","drizzleGraphQLFilters","drizzleGraphQLInsert","drizzleGraphQLUpdate"],"mappings":";;;;8DAUO;qCAEmC;8BACN;6BACR;yBACG;2BACE;wBAED;yBACqB;sBACrB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEhC,MAAMA,qBAAqBC,aAAa,CAACC,SAAS;AAElDF,mBAAmBG,aAAa,GAAG,SAASA,cAAcC,KAAK,EAAE,EAAEC,MAAM,EAAEC,MAAM,EAAE,GAAGC,SAAS;QAcrEC;QAbXD,kBAAAA;IAAb,MAAME,OAAOF,CAAAA,OAAAA,CAAAA,mBAAAA,QAAQG,OAAO,cAAfH,8BAAAA,mBAAmBA,QAAQE,IAAI,cAA/BF,kBAAAA,OAAmCH;IAEhD,MAAMO,MAAMJ,QAAQG,OAAO,GACvB,IAAIE,2BAAgB,CAACL,QAAQG,OAAO,EAAED,QACrCI,IAAAA,qBAAe,EAACT,OAAO,IAAI,EAAE;IAElCO,IAAIF,IAAI,GAAGA,iBAAAA,kBAAAA,OAAQL;IAEnB,IAAI,CAACU,UAAU,CAACH,KAAK;QACnB,GAAIJ,OAAO;QACXQ,YAAY;YACV,GAAGR,QAAQQ,UAAU;YACrBC,oBAAoBZ;YACpBa,kBAAkB,GAAET,0BAAAA,IAAAA,uBAAe,EAAC,IAAI,EAAEU,MAAM,cAA5BV,8CAAAA,uBAA8B,CAACJ,MAAM;YACzDe,qBAAqBd,mBAAAA,oBAAAA,SAAU,CAAC;YAChCe,qBAAqBC,wBAAW,CAACC,QAAQ,CAAClB,OAAO,IAAI;QACvD;QACAK;QACAH,QAAQA,SAAS,IAAMA,OAAO,IAAIiB,8CAAyB,CAACZ,IAAIF,IAAI,EAAE,IAAI,EAAEL,UAAUoB;IACxF;IAEA,OAAOb;AACT;AAEAX,mBAAmByB,gBAAgB,GAAG,SAASA,iBAC7CrB,KAAK,EACL,EAAEC,MAAM,EAAEC,MAAM,EAAE,GAAGC,SAAS;QAeNC;QAbXD;IAAb,MAAME,OAAOF,CAAAA,gBAAAA,QAAQE,IAAI,cAAZF,2BAAAA,gBAAgBH;IAE7B,MAAMO,MAAMJ,QAAQG,OAAO,GACvB,IAAIgB,iCAAmB,CAACnB,QAAQG,OAAO,EAAED,QACxCI,IAAAA,qBAAe,EAACT,OAAO,IAAI,EAAE;IAElCO,IAAIF,IAAI,GAAGA,iBAAAA,kBAAAA,OAAQL;IAEnB,IAAI,CAACuB,aAAa,CAAChB,KAAK;QACtB,GAAIJ,OAAO;QACXQ,YAAY;YACV,GAAGR,QAAQQ,UAAU;YACrBC,oBAAoBZ;YACpBa,kBAAkB,GAAET,0BAAAA,IAAAA,uBAAe,EAAC,IAAI,EAAEU,MAAM,cAA5BV,8CAAAA,uBAA8B,CAACJ,MAAM;YACzDe,qBAAqBd,mBAAAA,oBAAAA,SAAU,CAAC;YAChCe,qBAAqBC,wBAAW,CAACC,QAAQ,CAAClB,OAAO,IAAI;QACvD;QACAK;QACAH,QAAQA,SACJ,IAAMA,OAAO,IAAIiB,8CAAyB,CAACZ,IAAIF,IAAI,EAAE,IAAI,EAAEL,OAAO,gBAClEoB;IACN;IAEA,OAAOb;AACT;AAEAX,mBAAmB4B,WAAW,GAAG,SAASA,YAIxCxB,KAAiD,EACjD,EACEyB,IAAI,EAAEC,MAAM,EAAE,GAAGC,WAAW,EAC5BtB,IAAI,EACJC,OAAO,EACP,GAAGH,SACqF;QAKrE,wBAAA;IAHrB,MAAMyB,cAAcxB,IAAAA,uBAAe,EAAC,IAAI,EAAEU,MAAM,AAAC,CAACd,MAAM;IACxD,MAAM6B,WAAW,OAAOH,WAAW,aAAaA,OAAOE,YAAYE,OAAO,IAAIJ;IAC9E,MAAMK,YAAYC,MAAMC,OAAO,CAACJ,YAAYA,WAAW;QAACA;KAAS;IACjE,MAAMK,gBAAe,yBAAA,CAAA,QAAA,IAAI,EAACC,gBAAgB,cAArB,6CAAA,4BAAA;IACrB,MAAMC,UAAUC,IAAAA,4BAAmB,EAACN;IACpC,MAAMO,WAAWC,IAAAA,wBAAe,EAACR;QAChBzB;IAAjB,MAAMkC,WAAWlC,CAAAA,OAAAA,oBAAAA,qBAAAA,UAAWD,kBAAXC,kBAAAA,OAAmBN;IACpC,MAAMyC,UAAU,IAAIC,uBAAc,CAACF,UAAUxC;IAC7C,MAAM2C,cAAc1B,wBAAW,CAACC,QAAQ,CAAClB,OAAO,IAAI,EAAE+B;IAEtD,IAAI,CAACG,cAAc;QACjB,MAAM,IAAIU,iBAAW,CAAC;IACxB;IAEA,MAAMC,kBAAkB;QACtB,GAAG1C,OAAO;QACVE;QACAC;QACAwC,YAAY;YAACZ;SAAa;QAC1Ba,kBAAkB,OAChBtB,IACAuB,SACAC;YAEA,MAAMC,SAAS,MAAMP,YAAYK,SAASG,qBAAqB,CAACF,MAAMX,SAASb,KAAKe;YAEpF,IAAIU,QAAQ;gBACVE,IAAAA,mBAAa,EAACF,QAAQV;YACxB;YAEA,OAAOU;QACT;IACF;IAEA,MAAM3C,MAAM,IAAI,CAACR,aAAa,CAACC,OAAO6C;IAEtC,IAAI1C,QAAQ2C,UAAU,EAAE;QACtBvC,IAAI8C,aAAa,CAAClD,QAAQ2C,UAAU;IACtC;IAEA,IAAI,CAACQ,WAAW,CAACC,YAAY,CAAChD,KAAK,CAACiD;YAGhC;YAAA;QAFF,IAAI,CAACC,WAAW,CACdlD,KACA,CAAA,0CAAA,6BAAA,AAAC,IAAI,CAACJ,OAAO,CAAiDuD,YAAY,cAA1E,iDAAA,2BAA4EC,WAAW,cAAvF,oDAAA,yCACE,MACF,CAACC;gBAMM;mBALL,AACEA,EAGAC,QAAQ,CAAC;oBACN,6BAAA,AAAC,IAAI,CAAC1D,OAAO,CAAoDuD,YAAY,cAA7E,iDAAA,2BACCI,cAAc,AADlB;gBAEA,GAAGnC,SAAS;gBACZoC,UAAU;gBACVC,MAAM,CAAC;gBACP5B,SAAS,CAAC6B,SACRC,IAAAA,mBAAa,EAAC9B,QAAQ6B,SAAS,CAACxC,KAAQ,CAAA;4BACtC0C,MAAMX,WAAWnD,IAAI;4BACrBoB;wBACF,CAAA;YACJ;;IAEN;IAEA,IAAI,CAAC6B,WAAW,CAACc,qBAAqB,CAAC3B,SAASlC;IAEhD,OAAOkC;AACT;AAEA7C,mBAAmByE,kBAAkB,GAAG,SAASA,mBAAmBF,IAAI,EAAEG,SAAS,EAAEC,KAAK;IACxF,MAAMhE,MAAM,OAAO4D,SAAS,WAAW1D,IAAAA,qBAAe,EAAC0D,MAAM,IAAI,IAAKA;IACtE,IAAI,CAACb,WAAW,CAACC,YAAY,CAAChD,KAAK,CAAC,EAAEF,IAAI,EAAE;QAC1C,IAAI,CAACiD,WAAW,CAACkB,SAAS,CAACjE,KAAK,IAAO,CAAA;gBACrC,CAAC+D,UAAU,EAAEC,MAAM,IAAIpD,8CAAyB,CAACd,MAAM,IAAI,EAAEE,IAAIkE,SAAS;YAC5E,CAAA;IACF;AACF;AAEA7E,mBAAmB8E,qBAAqB,GAAG,SAASA,sBAAsBP,IAAI,EAAEG,SAAS,EAAEC,KAAK;IAC9F,MAAMhE,MAAM,OAAO4D,SAAS,WAAW1D,IAAAA,qBAAe,EAAC0D,MAAM,IAAI,IAAKA;IACtE,IAAI,CAACb,WAAW,CAACC,YAAY,CAAChD,KAAK,CAAC,EAAEF,IAAI,EAAE;QAC1C,IAAI,CAACiD,WAAW,CAACkB,SAAS,CAACjE,KAAK,IAAO,CAAA;gBACrC,CAAC+D,UAAU,EAAEC,MAAM,IAAIpD,8CAAyB,CAACd,MAAM,IAAI,EAAEE,IAAIkE,SAAS,EAAE;YAC9E,CAAA;IACF;AACF;AAEA7E,mBAAmB+E,mBAAmB,GAAG,SAASA,oBAAoBR,IAAI,EAAEjE,MAAM;IAChF,MAAMK,MAAM,OAAO4D,SAAS,WAAW1D,IAAAA,qBAAe,EAAC0D,MAAM,IAAI,IAAKA;IACtE,IAAI,CAACb,WAAW,CAACC,YAAY,CAAChD,KAAK,CAAC,EAAEF,IAAI,EAAE;QAC1C,IAAI,CAACiD,WAAW,CAACkB,SAAS,CAACjE,KAAK,IAC9BL,OAAO,IAAIiB,8CAAyB,CAACd,MAAM,IAAI,EAAEE,IAAIkE,SAAS;IAElE;AACF;AAEA7E,mBAAmBgF,sBAAsB,GAAG,SAASA,uBAAuBT,IAAI,EAAEjE,MAAM;IACtF,MAAMK,MAAM,OAAO4D,SAAS,WAAW1D,IAAAA,qBAAe,EAAC0D,MAAM,IAAI,IAAKA;IACtE,IAAI,CAACb,WAAW,CAACC,YAAY,CAAChD,KAAK,CAAC,EAAEF,IAAI,EAAE;QAC1C,IAAI,CAACiD,WAAW,CAACkB,SAAS,CAACjE,KAAK,IAC9BL,OAAO,IAAIiB,8CAAyB,CAACd,MAAM,IAAI,EAAEE,IAAIkE,SAAS,EAAE;IAEpE;AACF;AAEA7E,mBAAmBiF,qBAAqB,GAAG,SAASA,sBAElD7E,KAAa,EACbmE,IAA4B,EAC5BhE,OAAiC;QAgBdC;IAdnB,OAAO,AACL,IAAI,CAMJ0E,eAAe,CAACX,MAAM;QACtB,GAAGhE,OAAO;QACVQ,YAAY;eACPR,oBAAAA,8BAAAA,QAASQ,UAAU,AAAtB;YACAoE,gBAAgB;gBACdC,WAAW;gBACXhF;gBACA4B,WAAW,GAAExB,0BAAAA,IAAAA,uBAAe,EAAC,IAAI,EAAEU,MAAM,cAA5BV,8CAAAA,uBAA8B,CAACJ,MAAM;YACpD;QACF;IACF;AACF;AAEAJ,mBAAmBqF,qBAAqB,GAAG,SAASA,sBAElDjF,KAAa,EACbmE,IAA4B,EAC5BhE,OAAiC;QAgBdC;IAdnB,OAAO,AACL,IAAI,CAMJ0E,eAAe,CAACX,MAAM;QACtB,GAAGhE,OAAO;QACVQ,YAAY;eACPR,oBAAAA,8BAAAA,QAASQ,UAAU,AAAtB;YACAoE,gBAAgB;gBACdC,WAAW;gBACXhF;gBACA4B,WAAW,GAAExB,0BAAAA,IAAAA,uBAAe,EAAC,IAAI,EAAEU,MAAM,cAA5BV,8CAAAA,uBAA8B,CAACJ,MAAM;YACpD;QACF;IACF;AACF;AAEAJ,mBAAmBsF,oBAAoB,GAAG,SAASA,qBAEjDlF,KAAa,EACbmE,IAA4B,EAC5BhE,OAAiC;QAgBdC;IAdnB,OAAO,AACL,IAAI,CAMJ0E,eAAe,CAACX,MAAM;QACtB,GAAGhE,OAAO;QACVQ,YAAY;eACPR,oBAAAA,8BAAAA,QAASQ,UAAU,AAAtB;YACAoE,gBAAgB;gBACdC,WAAW;gBACXhF;gBACA4B,WAAW,GAAExB,0BAAAA,IAAAA,uBAAe,EAAC,IAAI,EAAEU,MAAM,cAA5BV,8CAAAA,uBAA8B,CAACJ,MAAM;YACpD;QACF;IACF;AACF;AAEAJ,mBAAmBuF,oBAAoB,GAAG,SAASA,qBAEjDnF,KAAa,EACbmE,IAA4B,EAC5BhE,OAAiC;QAgBdC;IAdnB,OAAO,AACL,IAAI,CAMJ0E,eAAe,CAACX,MAAM;QACtB,GAAGhE,OAAO;QACVQ,YAAY;eACPR,oBAAAA,8BAAAA,QAASQ,UAAU,AAAtB;YACAoE,gBAAgB;gBACdC,WAAW;gBACXhF;gBACA4B,WAAW,GAAExB,0BAAAA,IAAAA,uBAAe,EAAC,IAAI,EAAEU,MAAM,cAA5BV,8CAAAA,uBAA8B,CAACJ,MAAM;YACpD;QACF;IACF;AACF"}

package/lib/src/types.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/types.ts"],"names":["drizzleTableName","Symbol","for"],"mappings":";;;;+BAiDaA;;;eAAAA;;;AAAN,MAAMA,mBAAmBC,OAAOC,GAAG,CAAC"}
+{"version":3,"sources":["../../src/types.ts"],"names":["drizzleTableName","Symbol","for"],"mappings":";;;;+BA8DaA;;;eAAAA;;;AAAN,MAAMA,mBAAmBC,OAAOC,GAAG,CAAC"}

package/lib/src/utils/config.js

@@ -2,14 +2,34 @@
 Object.defineProperty(exports, "__esModule", {
     value: true
 });
-Object.defineProperty(exports, "getSchemaConfig", {
-    enumerable: true,
-    get: function() {
+function _export(target, all) {
+    for(var name in all)Object.defineProperty(target, name, {
+        enumerable: true,
+        get: all[name]
+    });
+}
+_export(exports, {
+    getClient: function() {
+        return getClient;
+    },
+    getSchemaConfig: function() {
         return getSchemaConfig;
     }
 });
 const _core = require("@pothos/core");
-const cache = (0, _core.createContextCache)((config)=>{
+const _drizzleorm = require("drizzle-orm");
+const configCache = (0, _core.createContextCache)((builder)=>{
+    let config;
+    if (builder.options.drizzle.schema) {
+        const tablesConfig = (0, _drizzleorm.extractTablesRelationalConfig)(builder.options.drizzle.schema, _drizzleorm.createTableRelationsHelpers);
+        config = {
+            fullSchema: builder.options.drizzle.schema,
+            schema: tablesConfig.tables,
+            tableNamesMap: tablesConfig.tableNamesMap
+        };
+    } else {
+        config = builder.options.drizzle.client._;
+    }
     const dbToSchema = Object.values(config.schema).reduce((acc, table)=>{
         acc[table.dbName] = table;
         return acc;
@@ -19,8 +39,19 @@ const cache = (0, _core.createContextCache)((config)=>{
         ...config
     };
 });
+const clientCache = (0, _core.createContextCache)((builder)=>{
+    const clientConfig = builder.options.drizzle.client;
+    const getClient = typeof clientConfig === 'function' ? (0, _core.createContextCache)((ctx)=>clientConfig(ctx)) : (_ctx)=>clientConfig;
+    return (0, _core.createContextCache)((context)=>{
+        const client = getClient(context);
+        return client;
+    });
+});
 function getSchemaConfig(builder) {
-    return cache(builder.options.drizzle.client._);
+    return configCache(builder);
+}
+function getClient(builder, context) {
+    return clientCache(builder)(context);
 }
 
 //# sourceMappingURL=config.js.map

package/lib/src/utils/config.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/utils/config.ts"],"names":["getSchemaConfig","cache","createContextCache","config","dbToSchema","Object","values","schema","reduce","acc","table","dbName","builder","options","drizzle","client","_"],"mappings":";;;;+BA2BgBA;;;eAAAA;;;sBA3BqC;AAUrD,MAAMC,QAAQC,IAAAA,wBAAkB,EAC9B,CAACC;IACC,MAAMC,aAAaC,OAAOC,MAAM,CAACH,OAAOI,MAAM,EAAEC,MAAM,CACpD,CAACC,KAAKC;QACJD,GAAG,CAACC,MAAMC,MAAM,CAAC,GAAGD;QACpB,OAAOD;IACT,GACA,CAAC;IAGH,OAAO;QACLL;QACA,GAAGD,MAAM;IACX;AACF;AAGK,SAASH,gBACdY,OAA+C;IAE/C,OAAOX,MAAMW,QAAQC,OAAO,CAACC,OAAO,CAACC,MAAM,CAACC,CAAC;AAC/C"}
+{"version":3,"sources":["../../../src/utils/config.ts"],"names":["getClient","getSchemaConfig","configCache","createContextCache","builder","config","options","drizzle","schema","tablesConfig","extractTablesRelationalConfig","createTableRelationsHelpers","fullSchema","tables","tableNamesMap","client","_","dbToSchema","Object","values","reduce","acc","table","dbName","clientCache","clientConfig","ctx","_ctx","context"],"mappings":";;;;;;;;;;;IAkEgBA,SAAS;eAATA;;IANAC,eAAe;eAAfA;;;sBA5DqC;4BAO9C;AAMP,MAAMC,cAAcC,IAAAA,wBAAkB,EACpC,CAACC;IACC,IAAIC;IACJ,IAAID,QAAQE,OAAO,CAACC,OAAO,CAACC,MAAM,EAAE;QAClC,MAAMC,eAAeC,IAAAA,yCAA6B,EAChDN,QAAQE,OAAO,CAACC,OAAO,CAACC,MAAM,EAC9BG,uCAA2B;QAE7BN,SAAS;YACPO,YAAYR,QAAQE,OAAO,CAACC,OAAO,CAACC,MAAM;YAC1CA,QAAQC,aAAaI,MAAM;YAC3BC,eAAeL,aAAaK,aAAa;QAC3C;IACF,OAAO;QACLT,SAAS,AAACD,QAAQE,OAAO,CAACC,OAAO,CAACQ,MAAM,CACrCC,CAAC;IACN;IAEA,MAAMC,aAAaC,OAAOC,MAAM,CAACd,OAAOG,MAAM,EAAEY,MAAM,CACpD,CAACC,KAAKC;QACJD,GAAG,CAACC,MAAMC,MAAM,CAAC,GAAGD;QACpB,OAAOD;IACT,GACA,CAAC;IAGH,OAAO;QACLJ;QACA,GAAGZ,MAAM;IACX;AACF;AAGF,MAAMmB,cAAcrB,IAAAA,wBAAkB,EAAC,CAACC;IACtC,MAAMqB,eAAerB,QAAQE,OAAO,CAACC,OAAO,CAACQ,MAAM;IACnD,MAAMf,YACJ,OAAOyB,iBAAiB,aACpBtB,IAAAA,wBAAkB,EAAC,CAACuB,MAAQD,aAAaC,QACzC,CAACC,OAAiBF;IAExB,OAAOtB,IAAAA,wBAAkB,EAAC,CAACyB;QACzB,MAAMb,SAASf,UAAU4B;QAEzB,OAAOb;IACT;AACF;AAEO,SAASd,gBACdG,OAA+C;IAE/C,OAAOF,YAAYE;AACrB;AAEO,SAASJ,UACdI,OAA+C,EAC/CwB,OAAe;IAEf,OAAOJ,YAAYpB,SAAkBwB;AACvC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pothos/plugin-drizzle",
-  "version": "0.1.0",
+  "version": "0.2.1",
   "description": "A Pothos plugin for drizzle",
   "main": "./lib/index.js",
   "types": "./dts/index.d.ts",
@@ -46,10 +46,10 @@
     "graphql-scalars": "^1.23.0",
     "graphql-tag": "^2.12.6",
     "postgres": "^3.4.4",
-    "@pothos/plugin-add-graphql": "4.1.0",
     "@pothos/core": "4.1.0",
-    "@pothos/plugin-relay": "4.1.0",
     "@pothos/plugin-scope-auth": "4.1.0",
+    "@pothos/plugin-relay": "4.1.0",
+    "@pothos/plugin-add-graphql": "4.1.0",
     "@pothos/plugin-with-input": "4.1.0",
     "@pothos/test-utils": "2.1.0"
   },

package/src/drizzle-field-builder.ts

@@ -32,6 +32,7 @@ import type {
   TypesForRelation,
   VariantFieldOptions,
 } from './types';
+import { getSchemaConfig } from './utils/config';
 import {
   drizzleCursorConnectionQuery,
   getCursorFormatter,
@@ -195,10 +196,12 @@ export class DrizzleObjectFieldBuilder<
     connectionOptions = {},
     edgeOptions = {},
   ) {
-    const relationField =
-      this.builder.options.drizzle.client._.schema?.[this.table].relations[name as string];
-    const relatedModel =
-      this.builder.options.drizzle.client._.schema?.[relationField?.referencedTableName!]!;
+    const relationField = getSchemaConfig(this.builder).schema?.[this.table].relations[
+      name as string
+    ];
+    const relatedModel = getSchemaConfig(this.builder).schema?.[
+      relationField?.referencedTableName!
+    ]!;
 
     if (!relationField) {
       throw new PothosSchemaError(
@@ -381,8 +384,9 @@ export class DrizzleObjectFieldBuilder<
     >
   ): FieldRef<Types, TypesForRelation<Types, TableConfig['relations'][Field]>, 'Object'> {
     const [options = {} as never] = allArgs;
-    const relationField =
-      this.builder.options.drizzle.client._.schema?.[this.table].relations[name as string];
+    const relationField = getSchemaConfig(this.builder).schema?.[this.table].relations[
+      name as string
+    ];
 
     if (!relationField) {
       throw new PothosSchemaError(

package/src/field-builder.ts

@@ -125,10 +125,9 @@ fieldBuilderProto.drizzleConnection = function drizzleConnection<
         context: {},
         info: GraphQLResolveInfo,
       ) => {
-        const drizzleModel =
-          this.builder.options.drizzle.client._.schema?.[
-            typeof type === 'string' ? type : (ref as DrizzleRef<SchemaTypes>).tableName
-          ]!;
+        const drizzleModel = getSchemaConfig(this.builder).schema?.[
+          typeof type === 'string' ? type : (ref as DrizzleRef<SchemaTypes>).tableName
+        ]!;
 
         return resolveDrizzleCursorConnection(
           drizzleModel,

package/src/model-loader.ts

@@ -1,7 +1,8 @@
 import { type SchemaTypes, createContextCache } from '@pothos/core';
 import { type Column, type TableRelationalConfig, inArray, sql } from 'drizzle-orm';
 import type { GraphQLResolveInfo } from 'graphql';
-import { type PothosDrizzleSchemaConfig, getSchemaConfig } from './utils/config';
+import type { DrizzleClient } from './types';
+import { type PothosDrizzleSchemaConfig, getClient, getSchemaConfig } from './utils/config';
 import { cacheKey, setLoaderMappings } from './utils/loader-map';
 import { selectionStateFromInfo, stateFromInfo } from './utils/map-query';
 import {
@@ -35,6 +36,7 @@ export class ModelLoader {
   table: TableRelationalConfig;
   columns: Column[];
   selectSQL;
+  client: DrizzleClient;
 
   constructor(
     context: object,
@@ -52,6 +54,8 @@ export class ModelLoader {
       this.columns.length > 1
         ? sql`(${sql.join(this.columns, sql`, `)})`
         : this.columns[0].getSQL();
+
+    this.client = getClient(builder, context);
   }
 
   static forModel<Types extends SchemaTypes>(
@@ -166,7 +170,7 @@ export class ModelLoader {
     nextTick.promise
       .then(() => {
         const api = (
-          this.builder.options.drizzle.client.query as Record<
+          this.client.query as Record<
             string,
             { findMany: (...args: unknown[]) => Promise<Record<string, unknown>[]> }
           >

package/src/schema-builder.ts

@@ -16,6 +16,7 @@ import { ModelLoader } from './model-loader';
 import { DrizzleNodeRef } from './node-ref';
 import { DrizzleObjectRef } from './object-ref';
 import type { DrizzleGraphQLInputExtensions, DrizzleNodeOptions } from './types';
+import { getSchemaConfig } from './utils/config';
 import { getColumnParser, getColumnSerializer } from './utils/cursors';
 import { getRefFromModel } from './utils/refs';
 
@@ -35,7 +36,7 @@ schemaBuilderProto.drizzleObject = function drizzleObject(table, { select, field
     extensions: {
       ...options.extensions,
       pothosDrizzleModel: table,
-      pothosDrizzleTable: this.options.drizzle.client._.schema?.[table],
+      pothosDrizzleTable: getSchemaConfig(this).schema?.[table],
       pothosDrizzleSelect: select ?? {},
       pothosDrizzleLoader: ModelLoader.forModel(table, this),
     },
@@ -63,7 +64,7 @@ schemaBuilderProto.drizzleInterface = function drizzleInterface(
     extensions: {
       ...options.extensions,
       pothosDrizzleModel: table,
-      pothosDrizzleTable: this.options.drizzle.client._.schema?.[table],
+      pothosDrizzleTable: getSchemaConfig(this).schema?.[table],
       pothosDrizzleSelect: select ?? {},
       pothosDrizzleLoader: ModelLoader.forModel(table, this),
     },
@@ -88,7 +89,7 @@ schemaBuilderProto.drizzleNode = function drizzleNode(
     ...options
   }: DrizzleNodeOptions<SchemaTypes, keyof SchemaTypes['DrizzleRelationSchema'], {}, {}, []>,
 ) {
-  const tableConfig = this.options.drizzle.client._.schema![table];
+  const tableConfig = getSchemaConfig(this).schema![table];
   const idColumn = typeof column === 'function' ? column(tableConfig.columns) : column;
   const idColumns = Array.isArray(idColumn) ? idColumn : [idColumn];
   const interfaceRef = this.nodeInterfaceRef?.();
@@ -214,7 +215,7 @@ schemaBuilderProto.drizzleGraphQLOrderBy = function drizzleGraphQLOrderBy(
       drizzleGraphQL: {
         inputType: 'orderBy',
         table,
-        tableConfig: this.options.drizzle.client._.schema?.[table]!,
+        tableConfig: getSchemaConfig(this).schema?.[table]!,
       } satisfies DrizzleGraphQLInputExtensions,
     },
   });
@@ -240,7 +241,7 @@ schemaBuilderProto.drizzleGraphQLFilters = function drizzleGraphQLFilters(
       drizzleGraphQL: {
         inputType: 'filters',
         table,
-        tableConfig: this.options.drizzle.client._.schema?.[table]!,
+        tableConfig: getSchemaConfig(this).schema?.[table]!,
       } satisfies DrizzleGraphQLInputExtensions,
     },
   });
@@ -266,7 +267,7 @@ schemaBuilderProto.drizzleGraphQLInsert = function drizzleGraphQLInsert(
       drizzleGraphQL: {
         inputType: 'insert',
         table,
-        tableConfig: this.options.drizzle.client._.schema?.[table]!,
+        tableConfig: getSchemaConfig(this).schema?.[table]!,
       } satisfies DrizzleGraphQLInputExtensions,
     },
   });
@@ -292,7 +293,7 @@ schemaBuilderProto.drizzleGraphQLUpdate = function drizzleGraphQLUpdate(
       drizzleGraphQL: {
         inputType: 'update',
         table,
-        tableConfig: this.options.drizzle.client._.schema?.[table]!,
+        tableConfig: getSchemaConfig(this).schema?.[table]!,
       } satisfies DrizzleGraphQLInputExtensions,
     },
   });

package/src/types.ts

@@ -41,11 +41,24 @@ import type { DrizzleRef } from './interface-ref';
 import type { IndirectInclude } from './utils/map-query';
 import type { SelectionMap } from './utils/selections';
 
-export interface DrizzlePluginOptions<_Types extends SchemaTypes> {
-  client: { _: Partial<RelationalSchemaConfig<TablesRelationalConfig>>; query: {} };
+export type DrizzleClient = {
+  _: Partial<RelationalSchemaConfig<TablesRelationalConfig>>;
+  query: {};
+};
+
+export type DrizzlePluginOptions<Types extends SchemaTypes> = {
   maxConnectionSize?: number;
   defaultConnectionSize?: number;
-}
+} & (
+  | {
+      client: DrizzleClient;
+      schema?: Record<string, unknown>;
+    }
+  | {
+      client: (ctx: Types['Context']) => DrizzleClient;
+      schema: Record<string, unknown>;
+    }
+);
 
 export const drizzleTableName = Symbol.for('Pothos.drizzleTableName');
 

package/src/utils/config.ts

@@ -1,15 +1,34 @@
 import { type SchemaTypes, createContextCache } from '@pothos/core';
-import type {
-  RelationalSchemaConfig,
-  TableRelationalConfig,
-  TablesRelationalConfig,
+import {
+  type RelationalSchemaConfig,
+  type TableRelationalConfig,
+  type TablesRelationalConfig,
+  createTableRelationsHelpers,
+  extractTablesRelationalConfig,
 } from 'drizzle-orm';
+import type { DrizzleClient } from '../types';
 
 export interface PothosDrizzleSchemaConfig extends RelationalSchemaConfig<TablesRelationalConfig> {
   dbToSchema: Record<string, TableRelationalConfig>;
 }
-const cache = createContextCache(
-  (config: RelationalSchemaConfig<TablesRelationalConfig>): PothosDrizzleSchemaConfig => {
+const configCache = createContextCache(
+  (builder: PothosSchemaTypes.SchemaBuilder<SchemaTypes>): PothosDrizzleSchemaConfig => {
+    let config: RelationalSchemaConfig<TablesRelationalConfig>;
+    if (builder.options.drizzle.schema) {
+      const tablesConfig = extractTablesRelationalConfig(
+        builder.options.drizzle.schema,
+        createTableRelationsHelpers,
+      );
+      config = {
+        fullSchema: builder.options.drizzle.schema,
+        schema: tablesConfig.tables,
+        tableNamesMap: tablesConfig.tableNamesMap,
+      };
+    } else {
+      config = (builder.options.drizzle.client as DrizzleClient)
+        ._ as RelationalSchemaConfig<TablesRelationalConfig>;
+    }
+
     const dbToSchema = Object.values(config.schema).reduce<Record<string, TableRelationalConfig>>(
       (acc, table) => {
         acc[table.dbName] = table;
@@ -25,8 +44,29 @@ const cache = createContextCache(
   },
 );
 
+const clientCache = createContextCache((builder: PothosSchemaTypes.SchemaBuilder<SchemaTypes>) => {
+  const clientConfig = builder.options.drizzle.client;
+  const getClient =
+    typeof clientConfig === 'function'
+      ? createContextCache((ctx) => clientConfig(ctx))
+      : (_ctx: object) => clientConfig;
+
+  return createContextCache((context: object) => {
+    const client = getClient(context);
+
+    return client;
+  });
+});
+
 export function getSchemaConfig<Types extends SchemaTypes>(
   builder: PothosSchemaTypes.SchemaBuilder<Types>,
 ) {
-  return cache(builder.options.drizzle.client._ as RelationalSchemaConfig<TablesRelationalConfig>);
+  return configCache(builder as never);
+}
+
+export function getClient<Types extends SchemaTypes>(
+  builder: PothosSchemaTypes.SchemaBuilder<Types>,
+  context: object,
+) {
+  return clientCache(builder as never)(context);
 }