@pothos/plugin-drizzle 0.18.0 → 0.19.0

package/lib/index.js

@@ -32,7 +32,7 @@ require("./global-types.js");
 require("./field-builder.js");
 require("./schema-builder.js");
 const _core = /*#__PURE__*/ _interop_require_wildcard(require("@pothos/core"));
-const _loadermap = require("./utils/loader-map.js");
+const _selectionmapper = require("@pothos/selection-mapper");
 const _connectionhelpers = require("./connection-helpers.js");
 const _drizzlefieldbuilder = require("./drizzle-field-builder.js");
 const _types = _export_star(require("./types.js"), exports);
@@ -94,88 +94,49 @@ function _interop_require_wildcard(obj, nodeInterop) {
 const pluginName = 'drizzle';
 const _default = pluginName;
 class PothosDrizzlePlugin extends _core.BasePlugin {
-    // private mappingCache = new Map<
-    //   string,
-    //   InputTypeFieldsMapping<Types, DrizzleGraphQLInputExtensions>
-    // >();
     onTypeConfig(typeConfig) {
-        var _typeConfig_extensions;
+        var _typeConfig_extensions, _typeConfig_extensions1;
         if (typeConfig.kind !== 'Object' && typeConfig.kind !== 'Interface') {
             return typeConfig;
         }
         let model = (_typeConfig_extensions = typeConfig.extensions) === null || _typeConfig_extensions === void 0 ? void 0 : _typeConfig_extensions.pothosDrizzleModel;
+        let table = (_typeConfig_extensions1 = typeConfig.extensions) === null || _typeConfig_extensions1 === void 0 ? void 0 : _typeConfig_extensions1.pothosDrizzleTable;
         for (const iface of typeConfig.interfaces){
-            var _this_buildCache_getTypeConfig_extensions;
-            const interfaceModel = (_this_buildCache_getTypeConfig_extensions = this.buildCache.getTypeConfig(iface, 'Interface').extensions) === null || _this_buildCache_getTypeConfig_extensions === void 0 ? void 0 : _this_buildCache_getTypeConfig_extensions.pothosDrizzleModel;
+            var _interfaceConfig_extensions;
+            const interfaceConfig = this.buildCache.getTypeConfig(iface, 'Interface');
+            const interfaceModel = (_interfaceConfig_extensions = interfaceConfig.extensions) === null || _interfaceConfig_extensions === void 0 ? void 0 : _interfaceConfig_extensions.pothosDrizzleModel;
             if (interfaceModel) {
+                var _interfaceConfig_extensions1;
                 if (model && model !== interfaceModel) {
-                    throw new _core.PothosSchemaError(`DrizzleObjects must be based on the same prisma model as any DrizzleInterfaces they extend. ${typeConfig.name} uses ${model} and ${iface.name} uses ${interfaceModel}`);
+                    throw new _core.PothosSchemaError(`DrizzleObjects must be based on the same drizzle table as any DrizzleInterfaces they extend. ${typeConfig.name} uses ${model} and ${iface.name} uses ${interfaceModel}`);
                 }
                 model = interfaceModel;
+                // A plain object type implementing a drizzle interface is walked with the interface's
+                // table, so fragments on it plan the relations it inherits.
+                table !== null && table !== void 0 ? table : table = (_interfaceConfig_extensions1 = interfaceConfig.extensions) === null || _interfaceConfig_extensions1 === void 0 ? void 0 : _interfaceConfig_extensions1.pothosDrizzleTable;
             }
         }
         return {
             ...typeConfig,
             extensions: {
                 ...typeConfig.extensions,
-                pothosDrizzleModel: model
+                pothosDrizzleModel: model,
+                pothosDrizzleTable: table
             }
         };
     }
     onOutputFieldConfig(fieldConfig) {
         var _fieldConfig_extensions;
-        // const argMappings = mapInputFields<Types, DrizzleGraphQLInputExtensions>(
-        //   fieldConfig.args,
-        //   this.buildCache,
-        //   (inputField) => {
-        //     if (inputField.type.kind === 'InputObject') {
-        //       const config = this.buildCache.getTypeConfig(inputField.type.ref);
-        //       return (config.extensions?.drizzleGraphQL as DrizzleGraphQLInputExtensions) ?? null;
-        //     }
-        //     return null;
-        //   },
-        //   this.mappingCache,
-        // );
-        // const argMapper = argMappings
-        //   ? createInputValueMapper(argMappings, (input, mappings) => {
-        //       if (!mappings.value) {
-        //         return input;
-        //       }
-        //       const { table, tableConfig, inputType } = mappings.value;
-        //       switch (inputType) {
-        //         case 'orderBy':
-        //           return extractOrderBy(tableConfig, input as never);
-        //         case 'filters':
-        //           return extractFilters(tableConfig, table, input as never);
-        //         case 'insert':
-        //           return remapFromGraphQLSingleInput(input as never, tableConfig);
-        //         case 'update':
-        //           return remapFromGraphQLSingleInput(input as never, tableConfig);
-        //         default:
-        //           throw new Error(`Unknown drizzle input type: ${inputType}`);
-        //       }
-        //     })
-        //   : null;
-        // const argMappers: typeof fieldConfig.argMappers = argMapper
-        //   ? [...(fieldConfig.argMappers ?? []), (args) => argMapper(args, undefined)]
-        //   : fieldConfig.argMappers;
         if (fieldConfig.kind === 'DrizzleObject' && fieldConfig.pothosOptions.select && !((_fieldConfig_extensions = fieldConfig.extensions) === null || _fieldConfig_extensions === void 0 ? void 0 : _fieldConfig_extensions.pothosDrizzleSelect)) {
             const { select } = fieldConfig.pothosOptions;
             return {
                 ...fieldConfig,
-                // argMappers,
                 extensions: {
                     ...fieldConfig.extensions,
                     pothosDrizzleSelect: typeof select === 'function' ? (args, ctx, nestedQuery, _resolveSelection, pathInfo)=>{
-                        // Attach PathInfo properties to nestedQuery for t.field select callbacks
-                        const nestedQueryWithPath = Object.assign(nestedQuery, {
-                            path: pathInfo.path,
-                            segments: pathInfo.segments
-                        });
-                        return {
-                            columns: {},
-                            ...select(args, ctx, nestedQueryWithPath)
-                        };
+                        // A field's select also reads its metadata from the nested selection function.
+                        const nestedQueryWithPath = Object.assign(nestedQuery, pathInfo);
+                        return (0, _core.completeValue)(select(args, ctx, nestedQueryWithPath), wrapSelect);
                     } : {
                         columns: {},
                         ...select
@@ -188,11 +149,15 @@ class PothosDrizzlePlugin extends _core.BasePlugin {
         };
     }
     wrapResolve(resolver, fieldConfig) {
-        var _fieldConfig_extensions, _parentConfig_extensions;
+        var _fieldConfig_extensions, _fieldConfig_extensions1, _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);
+        // A field can declare how to tell whether its data is already on the parent row. When the
+        // row was not loaded from the planned query (a resolver returned a row it fetched itself),
+        // the mapping alone is not proof, and the field falls back to the model loader.
+        const loadedCheck = (_fieldConfig_extensions1 = fieldConfig.extensions) === null || _fieldConfig_extensions1 === void 0 ? void 0 : _fieldConfig_extensions1.pothosDrizzleLoaded;
         const parentTypes = new Set([
             fieldConfig.parentType
         ]);
@@ -210,23 +175,34 @@ class PothosDrizzlePlugin extends _core.BasePlugin {
             throw new Error(`ModelLoader not found for type ${parentConfig.name}`);
         }
         return (parent, args, context, info)=>{
-            let mapping = (0, _loadermap.getLoaderMapping)(context, info.path, info.parentType.name);
+            // Asked of the row: a mapping recorded for this row wins over the plan's, which answers
+            // for the rows the planned query loaded.
+            let mapping = (0, _selectionmapper.getLoaderMapping)(context, info.path, info.parentType.name, parent);
             if (!mapping) {
                 for (const parentType of parentTypes){
-                    mapping = (0, _loadermap.getLoaderMapping)(context, info.path, parentType);
+                    mapping = (0, _selectionmapper.getLoaderMapping)(context, info.path, parentType, parent);
                     if (mapping) {
                         break;
                     }
                 }
             }
-            if (mapping) {
-                (0, _loadermap.setLoaderMappings)(context, info, mapping);
+            if ((!loadedCheck || loadedCheck(parent, info, context)) && mapping) {
+                // Recorded under the field's own parent type as well, so its resolver finds the pathInfo
+                // it was planned with, whichever same-model type the plan was made for. Against the row
+                // it resolves with: a sibling row of the same list may have been loaded by another plan.
+                (0, _selectionmapper.setFieldMapping)(context, info, mapping, parent);
                 return resolver(parent, args, context, info);
             }
             return modelLoader(context).loadSelection(info, parent).then((result)=>resolver(result, args, context, info));
         };
     }
 }
+/** A falsy selection means the field selects nothing from the parent row. */ function wrapSelect(selected) {
+    return selected ? {
+        columns: {},
+        ...selected
+    } : null;
+}
 _core.default.registerPlugin(pluginName, PothosDrizzlePlugin, {});
 
 //# sourceMappingURL=index.js.map

package/lib/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts"],"names":["DrizzleObjectFieldBuilder","PothosDrizzlePlugin","drizzleClientCache","drizzleConnectionHelpers","drizzleTableName","pluginName","BasePlugin","onTypeConfig","typeConfig","kind","model","extensions","pothosDrizzleModel","iface","interfaces","interfaceModel","buildCache","getTypeConfig","PothosSchemaError","name","onOutputFieldConfig","fieldConfig","pothosOptions","select","pothosDrizzleSelect","args","ctx","nestedQuery","_resolveSelection","pathInfo","nestedQueryWithPath","Object","assign","path","segments","columns","wrapResolve","resolver","parentConfig","parentType","parentTypes","Set","interfaceConfig","add","modelLoader","pothosDrizzleLoader","Error","parent","context","info","mapping","getLoaderMapping","setLoaderMappings","loadSelection","then","result","SchemaBuilder","registerPlugin"],"mappings":";;;;;;;;;;;QAeSA;eAAAA,8CAAyB;;QASrBC;eAAAA;;QAFb;eAAA;;QAJSC;eAAAA,0BAAkB;;QAJlBC;eAAAA,2CAAwB;;QAGxBC;eAAAA,uBAAgB;;;QAjBlB;QACA;QACA;8DAOA;2BAG6C;mCAEX;qCACC;oCAC5B;wBAEqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEnC,MAAMC,aAAa;MAEnB,WAAeA;AAER,MAAMJ,4BAAuDK,gBAAU;IAC5E,kCAAkC;IAClC,YAAY;IACZ,iEAAiE;IACjE,OAAO;IACEC,aAAaC,UAA4B,EAAoB;YAKxDA;QAJZ,IAAIA,WAAWC,IAAI,KAAK,YAAYD,WAAWC,IAAI,KAAK,aAAa;YACnE,OAAOD;QACT;QAEA,IAAIE,SAAQF,yBAAAA,WAAWG,UAAU,cAArBH,6CAAAA,uBAAuBI,kBAAkB;QAErD,KAAK,MAAMC,SAASL,WAAWM,UAAU,CAAE;gBAClB;YAAvB,MAAMC,kBAAiB,4CAAA,IAAI,CAACC,UAAU,CAACC,aAAa,CAACJ,OAAO,aAAaF,UAAU,cAA5D,gEAAA,0CACnBC,kBAAkB;YAEtB,IAAIG,gBAAgB;gBAClB,IAAIL,SAASA,UAAUK,gBAAgB;oBACrC,MAAM,IAAIG,uBAAiB,CACzB,CAAC,4FAA4F,EAAEV,WAAWW,IAAI,CAAC,MAAM,EAAET,MAAM,KAAK,EAAEG,MAAMM,IAAI,CAAC,MAAM,EAAEJ,gBAAgB;gBAE3K;gBAEAL,QAAQK;YACV;QACF;QAEA,OAAO;YACL,GAAGP,UAAU;YACbG,YAAY;gBACV,GAAGH,WAAWG,UAAU;gBACxBC,oBAAoBF;YACtB;QACF;IACF;IAESU,oBACPC,WAA2C,EACJ;YA+CpCA;QA9CH,4EAA4E;QAC5E,sBAAsB;QACtB,qBAAqB;QACrB,sBAAsB;QACtB,oDAAoD;QACpD,2EAA2E;QAE3E,6FAA6F;QAC7F,QAAQ;QAER,mBAAmB;QACnB,OAAO;QACP,uBAAuB;QACvB,KAAK;QAEL,gCAAgC;QAChC,iEAAiE;QACjE,+BAA+B;QAC/B,wBAAwB;QACxB,UAAU;QAEV,kEAAkE;QAElE,6BAA6B;QAC7B,0BAA0B;QAC1B,gEAAgE;QAChE,0BAA0B;QAC1B,uEAAuE;QACvE,yBAAyB;QACzB,6EAA6E;QAC7E,yBAAyB;QACzB,6EAA6E;QAE7E,mBAAmB;QACnB,yEAAyE;QACzE,UAAU;QACV,SAAS;QACT,YAAY;QAEZ,8DAA8D;QAC9D,gFAAgF;QAChF,8BAA8B;QAE9B,IACEA,YAAYZ,IAAI,KAAK,mBACrBY,YAAYC,aAAa,CAACC,MAAM,IAChC,GAACF,0BAAAA,YAAYV,UAAU,cAAtBU,8CAAAA,wBAAwBG,mBAAmB,GAC5C;YACA,MAAM,EAAED,MAAM,EAAE,GAAGF,YAAYC,aAAa;YAC5C,OAAO;gBACL,GAAGD,WAAW;gBACd,cAAc;gBACdV,YAAY;oBACV,GAAGU,YAAYV,UAAU;oBACzBa,qBACE,OAAOD,WAAW,aACd,CACEE,MACAC,KACAC,aACAC,mBACAC;wBAEA,yEAAyE;wBACzE,MAAMC,sBAAsBC,OAAOC,MAAM,CAACL,aAAa;4BACrDM,MAAMJ,SAASI,IAAI;4BACnBC,UAAUL,SAASK,QAAQ;wBAC7B;wBACA,OAAO;4BACLC,SAAS,CAAC;4BACV,GAAG,AAACZ,OACFE,MACAC,KACAI,oBACD;wBACH;oBACF,IACA;wBACEK,SAAS,CAAC;wBACV,GAAGZ,MAAM;oBACX;gBACR;YACF;QACF;QAEA,OAAO;YACL,GAAGF,WAAW;QAEhB;IACF;IAESe,YACPC,QAA0E,EAC1EhB,WAA2C,EACc;YACZA,yBAiBzBiB;QAjBpB,IAAIjB,YAAYZ,IAAI,KAAK,mBAAmB,GAACY,0BAAAA,YAAYV,UAAU,cAAtBU,8CAAAA,wBAAwBG,mBAAmB,GAAE;YACxF,OAAOa;QACT;QAEA,MAAMC,eAAe,IAAI,CAACtB,UAAU,CAACC,aAAa,CAACI,YAAYkB,UAAU;QAEzE,MAAMC,cAAc,IAAIC,IAAI;YAACpB,YAAYkB,UAAU;SAAC;QAEpD,IAAID,aAAa7B,IAAI,KAAK,eAAe6B,aAAa7B,IAAI,KAAK,UAAU;YACvE,KAAK,MAAMI,SAASyB,aAAaxB,UAAU,CAAE;oBAEvC4B;gBADJ,MAAMA,kBAAkB,IAAI,CAAC1B,UAAU,CAACC,aAAa,CAACJ,OAAO;gBAC7D,KAAI6B,8BAAAA,gBAAgB/B,UAAU,cAA1B+B,kDAAAA,4BAA4B9B,kBAAkB,EAAE;oBAClD4B,YAAYG,GAAG,CAACD,gBAAgBvB,IAAI;gBACtC;YACF;QACF;QAEA,MAAMyB,eAAcN,2BAAAA,aAAa3B,UAAU,cAAvB2B,+CAAAA,yBAAyBO,mBAAmB;QAIhE,IAAI,CAACD,aAAa;YAChB,MAAM,IAAIE,MAAM,CAAC,+BAA+B,EAAER,aAAanB,IAAI,EAAE;QACvE;QAEA,OAAO,CAAC4B,QAAQtB,MAAMuB,SAASC;YAC7B,IAAIC,UAAUC,IAAAA,2BAAgB,EAACH,SAASC,KAAKhB,IAAI,EAAEgB,KAAKV,UAAU,CAACpB,IAAI;YAEvE,IAAI,CAAC+B,SAAS;gBACZ,KAAK,MAAMX,cAAcC,YAAa;oBACpCU,UAAUC,IAAAA,2BAAgB,EAACH,SAASC,KAAKhB,IAAI,EAAEM;oBAC/C,IAAIW,SAAS;wBACX;oBACF;gBACF;YACF;YAEA,IAAIA,SAAS;gBACXE,IAAAA,4BAAiB,EAACJ,SAASC,MAAMC;gBAEjC,OAAOb,SAASU,QAAQtB,MAAMuB,SAASC;YACzC;YAEA,OAAOL,YAAYI,SAChBK,aAAa,CAACJ,MAAMF,QACpBO,IAAI,CAAC,CAACC,SAAWlB,SAASkB,QAAQ9B,MAAMuB,SAASC;QACtD;IACF;AACF;AAEAO,aAAa,CAACC,cAAc,CAACpD,YAAYJ,qBAAqB,CAAC"}
+{"version":3,"sources":["../src/index.ts"],"names":["DrizzleObjectFieldBuilder","PothosDrizzlePlugin","drizzleClientCache","drizzleConnectionHelpers","drizzleTableName","pluginName","BasePlugin","onTypeConfig","typeConfig","kind","model","extensions","pothosDrizzleModel","table","pothosDrizzleTable","iface","interfaces","interfaceConfig","buildCache","getTypeConfig","interfaceModel","PothosSchemaError","name","onOutputFieldConfig","fieldConfig","pothosOptions","select","pothosDrizzleSelect","args","ctx","nestedQuery","_resolveSelection","pathInfo","nestedQueryWithPath","Object","assign","completeValue","wrapSelect","columns","wrapResolve","resolver","parentConfig","parentType","loadedCheck","pothosDrizzleLoaded","parentTypes","Set","add","modelLoader","pothosDrizzleLoader","Error","parent","context","info","mapping","getLoaderMapping","path","setFieldMapping","loadSelection","then","result","selected","SchemaBuilder","registerPlugin"],"mappings":";;;;;;;;;;;QAkBSA;eAAAA,8CAAyB;;QASrBC;eAAAA;;QAFb;eAAA;;QAJSC;eAAAA,0BAAkB;;QAJlBC;eAAAA,2CAAwB;;QAGxBC;eAAAA,uBAAgB;;;QApBlB;QACA;QACA;8DAQA;iCAC2C;mCAMT;qCACC;oCAC5B;wBAEqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEnC,MAAMC,aAAa;MAEnB,WAAeA;AAER,MAAMJ,4BAAuDK,gBAAU;IACnEC,aAAaC,UAA4B,EAAoB;YAKxDA,wBACAA;QALZ,IAAIA,WAAWC,IAAI,KAAK,YAAYD,WAAWC,IAAI,KAAK,aAAa;YACnE,OAAOD;QACT;QAEA,IAAIE,SAAQF,yBAAAA,WAAWG,UAAU,cAArBH,6CAAAA,uBAAuBI,kBAAkB;QACrD,IAAIC,SAAQL,0BAAAA,WAAWG,UAAU,cAArBH,8CAAAA,wBAAuBM,kBAAkB;QAErD,KAAK,MAAMC,SAASP,WAAWQ,UAAU,CAAE;gBAElBC;YADvB,MAAMA,kBAAkB,IAAI,CAACC,UAAU,CAACC,aAAa,CAACJ,OAAO;YAC7D,MAAMK,kBAAiBH,8BAAAA,gBAAgBN,UAAU,cAA1BM,kDAAAA,4BAA4BL,kBAAkB;YAErE,IAAIQ,gBAAgB;oBAURH;gBATV,IAAIP,SAASA,UAAUU,gBAAgB;oBACrC,MAAM,IAAIC,uBAAiB,CACzB,CAAC,6FAA6F,EAAEb,WAAWc,IAAI,CAAC,MAAM,EAAEZ,MAAM,KAAK,EAAEK,MAAMO,IAAI,CAAC,MAAM,EAAEF,gBAAgB;gBAE5K;gBAEAV,QAAQU;gBACR,sFAAsF;gBACtF,4DAA4D;gBAC5DP,kBAAAA,mBAAAA,QAAAA,SAAUI,+BAAAA,gBAAgBN,UAAU,cAA1BM,mDAAAA,6BAA4BH,kBAAkB;YAG1D;QACF;QAEA,OAAO;YACL,GAAGN,UAAU;YACbG,YAAY;gBACV,GAAGH,WAAWG,UAAU;gBACxBC,oBAAoBF;gBACpBI,oBAAoBD;YACtB;QACF;IACF;IAESU,oBACPC,WAA2C,EACJ;YAIpCA;QAHH,IACEA,YAAYf,IAAI,KAAK,mBACrBe,YAAYC,aAAa,CAACC,MAAM,IAChC,GAACF,0BAAAA,YAAYb,UAAU,cAAtBa,8CAAAA,wBAAwBG,mBAAmB,GAC5C;YACA,MAAM,EAAED,MAAM,EAAE,GAAGF,YAAYC,aAAa;YAC5C,OAAO;gBACL,GAAGD,WAAW;gBACdb,YAAY;oBACV,GAAGa,YAAYb,UAAU;oBACzBgB,qBACE,OAAOD,WAAW,aACd,CACEE,MACAC,KACAC,aACAC,mBACAC;wBAEA,+EAA+E;wBAC/E,MAAMC,sBAAsBC,OAAOC,MAAM,CAACL,aAAaE;wBACvD,OAAOI,IAAAA,mBAAa,EAClB,AAACV,OACCE,MACAC,KACAI,sBAEFI;oBAEJ,IACA;wBACEC,SAAS,CAAC;wBACV,GAAGZ,MAAM;oBACX;gBACR;YACF;QACF;QAEA,OAAO;YACL,GAAGF,WAAW;QAChB;IACF;IAESe,YACPC,QAA0E,EAC1EhB,WAA2C,EACc;YACZA,yBAQzBA,0BAeAiB;QAvBpB,IAAIjB,YAAYf,IAAI,KAAK,mBAAmB,GAACe,0BAAAA,YAAYb,UAAU,cAAtBa,8CAAAA,wBAAwBG,mBAAmB,GAAE;YACxF,OAAOa;QACT;QAEA,MAAMC,eAAe,IAAI,CAACvB,UAAU,CAACC,aAAa,CAACK,YAAYkB,UAAU;QACzE,0FAA0F;QAC1F,2FAA2F;QAC3F,gFAAgF;QAChF,MAAMC,eAAcnB,2BAAAA,YAAYb,UAAU,cAAtBa,+CAAAA,yBAAwBoB,mBAAmB;QAI/D,MAAMC,cAAc,IAAIC,IAAI;YAACtB,YAAYkB,UAAU;SAAC;QAEpD,IAAID,aAAahC,IAAI,KAAK,eAAegC,aAAahC,IAAI,KAAK,UAAU;YACvE,KAAK,MAAMM,SAAS0B,aAAazB,UAAU,CAAE;oBAEvCC;gBADJ,MAAMA,kBAAkB,IAAI,CAACC,UAAU,CAACC,aAAa,CAACJ,OAAO;gBAC7D,KAAIE,8BAAAA,gBAAgBN,UAAU,cAA1BM,kDAAAA,4BAA4BL,kBAAkB,EAAE;oBAClDiC,YAAYE,GAAG,CAAC9B,gBAAgBK,IAAI;gBACtC;YACF;QACF;QAEA,MAAM0B,eAAcP,2BAAAA,aAAa9B,UAAU,cAAvB8B,+CAAAA,yBAAyBQ,mBAAmB;QAIhE,IAAI,CAACD,aAAa;YAChB,MAAM,IAAIE,MAAM,CAAC,+BAA+B,EAAET,aAAanB,IAAI,EAAE;QACvE;QAEA,OAAO,CAAC6B,QAAQvB,MAAMwB,SAASC;YAC7B,wFAAwF;YACxF,yCAAyC;YACzC,IAAIC,UAAUC,IAAAA,iCAAgB,EAACH,SAASC,KAAKG,IAAI,EAAEH,KAAKX,UAAU,CAACpB,IAAI,EAAE6B;YAEzE,IAAI,CAACG,SAAS;gBACZ,KAAK,MAAMZ,cAAcG,YAAa;oBACpCS,UAAUC,IAAAA,iCAAgB,EAACH,SAASC,KAAKG,IAAI,EAAEd,YAAYS;oBAC3D,IAAIG,SAAS;wBACX;oBACF;gBACF;YACF;YAEA,IAAI,AAAC,CAAA,CAACX,eAAeA,YAAYQ,QAAQE,MAAMD,QAAO,KAAME,SAAS;gBACnE,yFAAyF;gBACzF,wFAAwF;gBACxF,yFAAyF;gBACzFG,IAAAA,gCAAe,EAACL,SAASC,MAAMC,SAASH;gBAExC,OAAOX,SAASW,QAAQvB,MAAMwB,SAASC;YACzC;YAEA,OAAOL,YAAYI,SAChBM,aAAa,CAACL,MAAMF,QACpBQ,IAAI,CAAC,CAACC,SAAWpB,SAASoB,QAAQhC,MAAMwB,SAASC;QACtD;IACF;AACF;AAEA,2EAA2E,GAC3E,SAAShB,WAAWwB,QAAmB;IACrC,OAAOA,WAAW;QAAEvB,SAAS,CAAC;QAAG,GAAGuB,QAAQ;IAAC,IAAI;AACnD;AAEAC,aAAa,CAACC,cAAc,CAAC1D,YAAYJ,qBAAqB,CAAC"}

package/lib/model-loader.js

@@ -9,11 +9,11 @@ Object.defineProperty(exports, "ModelLoader", {
     }
 });
 const _core = require("@pothos/core");
+const _selectionmapper = require("@pothos/selection-mapper");
 const _drizzleorm = require("drizzle-orm");
+const _adapter = require("./utils/adapter.js");
 const _config = require("./utils/config.js");
-const _loadermap = require("./utils/loader-map.js");
 const _mapquery = require("./utils/map-query.js");
-const _selections = require("./utils/selections.js");
 function _define_property(obj, key, value) {
     if (key in obj) {
         Object.defineProperty(obj, key, {
@@ -42,67 +42,82 @@ class ModelLoader {
         return this.columns.length > 1 ? values : values[0];
     }
     getSelection(info) {
-        const key = (0, _loadermap.cacheKey)(info.parentType.name, info.path);
-        if (!this.queryCache.has(key)) {
-            const selection = (0, _mapquery.selectionStateFromInfo)(this.config, this.context, info);
-            this.queryCache.set(key, {
-                selection,
-                query: (0, _selections.selectionToQuery)(this.config, selection)
-            });
+        const key = (0, _selectionmapper.cacheKey)(info.parentType.name, info.path);
+        if (!this.rowCache.has(key)) {
+            this.rowCache.set(key, _selectionmapper.Plan.forParentRow((0, _adapter.drizzleAdapter)(this.config), this.context, info));
         }
-        return this.queryCache.get(key);
+        return this.rowCache.get(key);
     }
     getSelectionForField(info, typeName) {
-        const key = (0, _loadermap.cacheKey)(typeName, info.path);
-        if (!this.queryCache.has(key)) {
-            const selection = (0, _mapquery.stateFromInfo)({
+        const key = (0, _selectionmapper.cacheKey)(typeName, info.path);
+        if (!this.planCache.has(key)) {
+            this.planCache.set(key, // Walked without paths, so there is always a plan.
+            (0, _mapquery.planFromInfo)({
                 config: this.config,
                 context: this.context,
                 info,
                 typeName
-            });
-            this.queryCache.set(key, {
-                selection,
-                query: (0, _selections.selectionToQuery)(this.config, selection)
-            });
+            }));
         }
-        return this.queryCache.get(key);
+        return this.planCache.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) {
-                (0, _loadermap.setLoaderMappings)(this.context, info, mappings.mappings);
+    /**
+   * `model` reloaded with the selection of the field `info` resolves. A synchronous selection
+   * stages synchronously, so every row resolved in a tick joins the same batch; only a selection
+   * with an async select beneath the field waits for it.
+   */ loadSelection(info, model) {
+        const selection = this.getSelection(info);
+        return (0, _core.isThenable)(selection) ? selection.then((settled)=>this.loadWith(settled, info, model)) : this.loadWith(selection, info, model);
+    }
+    loadWith(played, info, model) {
+        return this.stageQuery(played, model).then((result)=>{
+            if (result) {
+                const mapping = played.mappings[`${info.parentType.name}@${info.path.key}`];
+                if (mapping) {
+                    // Recorded for the field itself too, so its resolver finds the pathInfo it was planned
+                    // with, along with the mappings of the fields beneath it. Against `result` alone: this
+                    // load fetched one row, so the field's own mapping is not the whole field's to claim.
+                    // A sibling row that falls back on a later tick would otherwise read it as proof of
+                    // having been loaded and resolve against a row that never carried the data.
+                    (0, _selectionmapper.setRowFieldMapping)(this.context, info, mapping, result);
+                }
             }
-        }
-        return result;
+            return result;
+        });
     }
-    async loadSelectionForField(info, model, returnType) {
-        const { selection, query } = this.getSelectionForField(info, returnType);
-        const result = await this.stageQuery(selection, query, model);
-        if (result) {
-            (0, _loadermap.setLoaderMappings)(this.context, info, selection.mappings);
-        }
-        return result;
+    /** A node loaded by id with the selection beneath the field `info` resolves, as `returnType`. */ loadSelectionForField(info, model, returnType) {
+        const selection = this.getSelectionForField(info, returnType);
+        return (0, _core.isThenable)(selection) ? selection.then((settled)=>this.loadFieldWith(settled.play(), info, model)) : this.loadFieldWith(selection.play(), info, model);
+    }
+    loadFieldWith(played, info, model) {
+        return this.stageQuery(played, model).then((result)=>{
+            if (result) {
+                // Relay nodes of different types share this field path. Each node plan contributes
+                // its type-qualified mappings, as a queryFromInfo plan would.
+                (0, _selectionmapper.setLoaderMappings)(this.context, info, played.mappings);
+            }
+            return result;
+        });
     }
-    stageQuery(selection, query, model) {
+    stageQuery(played, model) {
         for (const entry of this.staged){
-            if ((0, _selections.selectionCompatible)(entry.state, query)) {
-                (0, _selections.mergeSelection)(this.config, entry.state, query);
+            // Node to node: the batch takes the field's play whole, never through a query.
+            if (this.adapter.canMergeNode(entry.root, played.root)) {
+                this.adapter.mergeNode(entry.root, played.root);
                 if (!entry.models.has(model)) {
                     entry.models.set(model, createResolvablePromise());
                 }
                 return entry.models.get(model).promise;
             }
         }
-        return this.initLoad(selection, model);
+        return this.initLoad(played, model);
     }
-    initLoad(selection, model) {
+    initLoad(played, model) {
         const promise = createResolvablePromise();
+        const root = this.adapter.createNode(played.root.model);
+        this.adapter.mergeNode(root, played.root);
         const entry = {
-            state: selection,
+            root,
             models: new Map([
                 [
                     model,
@@ -117,7 +132,7 @@ class ModelLoader {
             this.staged.delete(entry);
             const api = client.query[this.modelName];
             const query = api.findMany({
-                ...(0, _selections.selectionToQuery)(this.config, selection),
+                ...this.adapter.toQuery(entry.root),
                 where: {
                     RAW: (table)=>(0, _drizzleorm.inArray)(this.selectSQL(table), [
                             ...entry.models.keys()
@@ -126,9 +141,17 @@ class ModelLoader {
             });
             query.then((results)=>{
                 for (const [model, promise] of entry.models.entries()){
-                    const result = results.find((row)=>this.primaryKey.every((key)=>row[this.config.columnToTsName(key)] === model[this.config.columnToTsName(key)]));
+                    // Matched on the columns the batch was keyed by, which are the columns a row of
+                    // `entry.models` carries: a node loaded by a column other than the primary key
+                    // has only that one.
+                    const result = results.find((row)=>this.columns.every((key)=>{
+                            const name = this.config.columnToTsName(key);
+                            const actual = row[name];
+                            const expected = model[name];
+                            return actual === expected || actual instanceof Date && expected instanceof Date && actual.getTime() === expected.getTime();
+                        }));
                     if (result) {
-                        promise.resolve(result !== null && result !== void 0 ? result : null);
+                        promise.resolve(result);
                     } else {
                         promise.reject(new Error(`Model ${this.modelName}(${this.sqlForModel(model)}) not found`));
                     }
@@ -150,9 +173,16 @@ class ModelLoader {
         _define_property(this, "context", void 0);
         _define_property(this, "builder", void 0);
         _define_property(this, "modelName", void 0);
-        _define_property(this, "queryCache", new Map());
+        // One plan per `Type@path`, a promise while a select beneath the field is async. A parent-row
+        // plan is played where it is made, since it is never played behind another selection; a field's
+        // plan is played per load, so no two loads share a node.
+        _define_property(this, "rowCache", new Map());
+        _define_property(this, "planCache", new Map());
+        // Each batch owns the node it accumulates into, so nothing a cached plan or play holds is
+        // changed by a row joining the batch.
         _define_property(this, "staged", new Set());
         _define_property(this, "config", void 0);
+        _define_property(this, "adapter", void 0);
         _define_property(this, "table", void 0);
         _define_property(this, "columns", void 0);
         _define_property(this, "primaryKey", void 0);
@@ -161,6 +191,7 @@ class ModelLoader {
         this.builder = builder;
         this.modelName = modelName;
         this.config = (0, _config.getSchemaConfig)(builder);
+        this.adapter = (0, _adapter.drizzleAdapter)(this.config);
         this.table = this.config.relations[modelName];
         this.primaryKey = this.config.getPrimaryKey(modelName);
         this.columns = columns !== null && columns !== void 0 ? columns : this.primaryKey;

package/lib/model-loader.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/model-loader.ts"],"names":["ModelLoader","forModel","modelName","builder","columns","createContextCache","model","sqlForModel","values","map","key","columnName","config","columnToTsName","Error","length","getSelection","info","cacheKey","parentType","name","path","queryCache","has","selection","selectionStateFromInfo","context","set","query","selectionToQuery","get","getSelectionForField","typeName","stateFromInfo","loadSelection","result","stageQuery","mappings","setLoaderMappings","loadSelectionForField","returnType","entry","staged","selectionCompatible","state","mergeSelection","models","createResolvablePromise","promise","initLoad","Map","add","nextTick","client","getClient","then","delete","api","findMany","where","RAW","table","inArray","selectSQL","keys","results","entries","find","row","primaryKey","every","resolve","reject","err","catch","setTimeout","Set","getSchemaConfig","relations","getPrimaryKey","column","Object","getColumns","tblColumn","sql","join","getSQL","resolveFn","rejectFn","Promise"],"mappings":";;;;+BA2BaA;;;eAAAA;;;sBA3BwC;4BAS9C;wBAEoE;2BAC/B;0BACU;4BAO/C;;;;;;;;;;;;;;AAOA,MAAMA;IA2CX,OAAOC,SACLC,SAAiB,EACjBC,OAA+C,EAC/CC,OAAkB,EAClB;QACA,OAAOC,IAAAA,wBAAkB,EACvB,CAACC,QAAU,IAAIN,YAAYM,OAAOH,SAAkBD,WAAWE;IAEnE;IAEAG,YAAYD,KAAa,EAAE;QACzB,MAAME,SAAS,IAAI,CAACJ,OAAO,CAACK,GAAG,CAAC,CAACC;YAC/B,MAAMC,aAAa,IAAI,CAACC,MAAM,CAACC,cAAc,CAACH;YAC9C,IAAIC,cAAcL,OAAO;gBACvB,OAAOA,KAAK,CAACK,WAAW;YAC1B;YAEA,MAAM,IAAIG,MAAM,CAAC,mBAAmB,EAAEH,WAAW,cAAc,EAAE,IAAI,CAACT,SAAS,EAAE;QACnF;QAEA,OAAO,IAAI,CAACE,OAAO,CAACW,MAAM,GAAG,IAAIP,SAASA,MAAM,CAAC,EAAE;IACrD;IAEAQ,aAAaC,IAAwB,EAAE;QACrC,MAAMP,MAAMQ,IAAAA,mBAAQ,EAACD,KAAKE,UAAU,CAACC,IAAI,EAAEH,KAAKI,IAAI;QACpD,IAAI,CAAC,IAAI,CAACC,UAAU,CAACC,GAAG,CAACb,MAAM;YAC7B,MAAMc,YAAYC,IAAAA,gCAAsB,EAAC,IAAI,CAACb,MAAM,EAAE,IAAI,CAACc,OAAO,EAAET;YACpE,IAAI,CAACK,UAAU,CAACK,GAAG,CAACjB,KAAK;gBACvBc;gBACAI,OAAOC,IAAAA,4BAAgB,EAAC,IAAI,CAACjB,MAAM,EAAEY;YACvC;QACF;QAEA,OAAO,IAAI,CAACF,UAAU,CAACQ,GAAG,CAACpB;IAC7B;IAEAqB,qBAAqBd,IAAwB,EAAEe,QAAgB,EAAE;QAC/D,MAAMtB,MAAMQ,IAAAA,mBAAQ,EAACc,UAAUf,KAAKI,IAAI;QACxC,IAAI,CAAC,IAAI,CAACC,UAAU,CAACC,GAAG,CAACb,MAAM;YAC7B,MAAMc,YAAYS,IAAAA,uBAAa,EAAC;gBAC9BrB,QAAQ,IAAI,CAACA,MAAM;gBACnBc,SAAS,IAAI,CAACA,OAAO;gBACrBT;gBACAe;YACF;YAEA,IAAI,CAACV,UAAU,CAACK,GAAG,CAACjB,KAAK;gBACvBc;gBACAI,OAAOC,IAAAA,4BAAgB,EAAC,IAAI,CAACjB,MAAM,EAAEY;YACvC;QACF;QAEA,OAAO,IAAI,CAACF,UAAU,CAACQ,GAAG,CAACpB;IAC7B;IAEA,MAAMwB,cAAcjB,IAAwB,EAAEX,KAAa,EAAE;QAC3D,MAAM,EAAEkB,SAAS,EAAEI,KAAK,EAAE,GAAG,IAAI,CAACZ,YAAY,CAACC;QAE/C,MAAMkB,SAAS,MAAM,IAAI,CAACC,UAAU,CAACZ,WAAWI,OAAOtB;QAEvD,IAAI6B,QAAQ;YACV,MAAME,WAAWb,UAAUa,QAAQ,CAACpB,KAAKI,IAAI,CAACX,GAAG,CAAC;YAElD,IAAI2B,UAAU;gBACZC,IAAAA,4BAAiB,EAAC,IAAI,CAACZ,OAAO,EAAET,MAAMoB,SAASA,QAAQ;YACzD;QACF;QAEA,OAAOF;IACT;IAEA,MAAMI,sBAAsBtB,IAAwB,EAAEX,KAAa,EAAEkC,UAAkB,EAAE;QACvF,MAAM,EAAEhB,SAAS,EAAEI,KAAK,EAAE,GAAG,IAAI,CAACG,oBAAoB,CAACd,MAAMuB;QAE7D,MAAML,SAAS,MAAM,IAAI,CAACC,UAAU,CAACZ,WAAWI,OAAOtB;QAEvD,IAAI6B,QAAQ;YACVG,IAAAA,4BAAiB,EAAC,IAAI,CAACZ,OAAO,EAAET,MAAMO,UAAUa,QAAQ;QAC1D;QAEA,OAAOF;IACT;IAEAC,WAAWZ,SAAyB,EAAEI,KAAmB,EAAEtB,KAAa,EAAE;QACxE,KAAK,MAAMmC,SAAS,IAAI,CAACC,MAAM,CAAE;YAC/B,IAAIC,IAAAA,+BAAmB,EAACF,MAAMG,KAAK,EAAEhB,QAAQ;gBAC3CiB,IAAAA,0BAAc,EAAC,IAAI,CAACjC,MAAM,EAAE6B,MAAMG,KAAK,EAAEhB;gBAEzC,IAAI,CAACa,MAAMK,MAAM,CAACvB,GAAG,CAACjB,QAAQ;oBAC5BmC,MAAMK,MAAM,CAACnB,GAAG,CAACrB,OAAOyC;gBAC1B;gBAEA,OAAON,MAAMK,MAAM,CAAChB,GAAG,CAACxB,OAAQ0C,OAAO;YACzC;QACF;QAEA,OAAO,IAAI,CAACC,QAAQ,CAACzB,WAAWlB;IAClC;IAEA2C,SAASzB,SAAyB,EAAElB,KAAa,EAAE;QACjD,MAAM0C,UAAUD;QAChB,MAAMN,QAAQ;YACZG,OAAOpB;YACPsB,QAAQ,IAAII,IAAI;gBAAC;oBAAC5C;oBAAO0C;iBAAQ;aAAC;QACpC;QACA,IAAI,CAACN,MAAM,CAACS,GAAG,CAACV;QAEhB,MAAMW,WAAWL;QACjB,MAAMM,SAASC,IAAAA,iBAAS,EAAC,IAAI,CAACnD,OAAO,EAAE,IAAI,CAACuB,OAAO;QAEnD0B,SAASJ,OAAO,CACbO,IAAI,CAAC;YACJ,IAAI,CAACb,MAAM,CAACc,MAAM,CAACf;YACnB,MAAMgB,MAAM,AACVJ,OAAOzB,KAAK,AAIb,CAAC,IAAI,CAAC1B,SAAS,CAAC;YAEjB,MAAM0B,QAAQ6B,IAAIC,QAAQ,CAAC;gBACzB,GAAG7B,IAAAA,4BAAgB,EAAC,IAAI,CAACjB,MAAM,EAAEY,UAAU;gBAC3CmC,OAAO;oBACLC,KAAK,CAACC,QACJC,IAAAA,mBAAO,EACL,IAAI,CAACC,SAAS,CAACF,QACf;+BAAIpB,MAAMK,MAAM,CAACkB,IAAI;yBAAG,CAACvD,GAAG,CAAC,CAACH,QAAU,IAAI,CAACC,WAAW,CAACD;gBAE/D;YACF;YAEAsB,MAAM2B,IAAI,CACR,CAACU;gBACC,KAAK,MAAM,CAAC3D,OAAO0C,QAAQ,IAAIP,MAAMK,MAAM,CAACoB,OAAO,GAAI;oBACrD,MAAM/B,SAAS8B,QAAQE,IAAI,CAAC,CAACC,MAC3B,IAAI,CAACC,UAAU,CAACC,KAAK,CACnB,CAAC5D,MACC0D,GAAG,CAAC,IAAI,CAACxD,MAAM,CAACC,cAAc,CAACH,KAAyB,KACxD,AAACJ,KAAiC,CAAC,IAAI,CAACM,MAAM,CAACC,cAAc,CAACH,KAAK;oBAIzE,IAAIyB,QAAQ;wBACVa,QAAQuB,OAAO,CAACpC,mBAAAA,oBAAAA,SAAU;oBAC5B,OAAO;wBACLa,QAAQwB,MAAM,CACZ,IAAI1D,MAAM,CAAC,MAAM,EAAE,IAAI,CAACZ,SAAS,CAAC,CAAC,EAAE,IAAI,CAACK,WAAW,CAACD,OAAO,WAAW,CAAC;oBAE7E;gBACF;YACF,GACA,CAACmE;gBACC,KAAK,MAAMzB,WAAWP,MAAMK,MAAM,CAACtC,MAAM,GAAI;oBAC3CwC,QAAQwB,MAAM,CAACC;gBACjB;YACF;QAEJ,GACCC,KAAK,CAAC,CAACD;YACN,KAAK,MAAMzB,WAAWP,MAAMK,MAAM,CAACtC,MAAM,GAAI;gBAC3CwC,QAAQwB,MAAM,CAACC;YACjB;QACF;QAEFE,WAAW,IAAMvB,SAASmB,OAAO,IAAI;QAErC,OAAOvB,QAAQA,OAAO;IACxB;IA9LA,YACEtB,OAAe,EACfvB,OAA+C,EAC/CD,SAAiB,EACjBE,OAAkB,CAClB;QAxBFsB,uBAAAA,WAAAA,KAAAA;QAEAvB,uBAAAA,WAAAA,KAAAA;QAEAD,uBAAAA,aAAAA,KAAAA;QAEAoB,uBAAAA,cAAa,IAAI4B;QAEjBR,uBAAAA,UAAS,IAAIkC;QAKbhE,uBAAAA,UAAAA,KAAAA;QACAiD,uBAAAA,SAAAA,KAAAA;QACAzD,uBAAAA,WAAAA,KAAAA;QACAiE,uBAAAA,cAAAA,KAAAA;QACAN,uBAAAA,aAAAA,KAAAA;QAQE,IAAI,CAACrC,OAAO,GAAGA;QACf,IAAI,CAACvB,OAAO,GAAGA;QACf,IAAI,CAACD,SAAS,GAAGA;QACjB,IAAI,CAACU,MAAM,GAAGiE,IAAAA,uBAAe,EAAC1E;QAC9B,IAAI,CAAC0D,KAAK,GAAG,IAAI,CAACjD,MAAM,CAACkE,SAAS,CAAC5E,UAAU;QAC7C,IAAI,CAACmE,UAAU,GAAG,IAAI,CAACzD,MAAM,CAACmE,aAAa,CAAC7E;QAC5C,IAAI,CAACE,OAAO,GAAGA,oBAAAA,qBAAAA,UAAW,IAAI,CAACiE,UAAU;QACzC,IAAI,CAACN,SAAS,GAAG,CAACF;YAChB,MAAMzD,UAAU,IAAI,CAACA,OAAO,CAACK,GAAG,CAAC,CAACuE,SAChCC,OAAOzE,MAAM,CAAC0E,IAAAA,sBAAU,EAACrB,QAAQM,IAAI,CACnC,CAACgB,YAAc,gBAAgBA,aAAaA,UAAU/D,IAAI,KAAK4D,OAAO5D,IAAI;YAG9E,OAAOhB,QAAQW,MAAM,GAAG,IAAIqE,IAAAA,eAAG,CAAA,CAAC,CAAC,EAAEA,eAAG,CAACC,IAAI,CAACjF,SAASgF,IAAAA,eAAG,CAAA,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,GAAGhF,OAAO,CAAC,EAAE,CAAEkF,MAAM;QACvF;IACF;AA0KF;AAEA,SAASvC;IACP,IAAIwC;IACJ,IAAIC;IACJ,MAAMxC,UAAU,IAAIyC,QAAW,CAAClB,SAASC;QACvCe,YAAYhB;QACZiB,WAAWhB;IACb;IAEA,OAAO;QAAExB;QAASuB,SAASgB;QAAWf,QAAQgB;IAAS;AACzD"}
+{"version":3,"sources":["../src/model-loader.ts"],"names":["ModelLoader","forModel","modelName","builder","columns","createContextCache","model","sqlForModel","values","map","key","columnName","config","columnToTsName","Error","length","getSelection","info","cacheKey","parentType","name","path","rowCache","has","set","Plan","forParentRow","drizzleAdapter","context","get","getSelectionForField","typeName","planCache","planFromInfo","loadSelection","selection","isThenable","then","settled","loadWith","played","stageQuery","result","mapping","mappings","setRowFieldMapping","loadSelectionForField","returnType","loadFieldWith","play","setLoaderMappings","entry","staged","adapter","canMergeNode","root","mergeNode","models","createResolvablePromise","promise","initLoad","createNode","Map","add","nextTick","client","getClient","delete","api","query","findMany","toQuery","where","RAW","table","inArray","selectSQL","keys","results","entries","find","row","every","actual","expected","Date","getTime","resolve","reject","err","catch","setTimeout","Set","primaryKey","getSchemaConfig","relations","getPrimaryKey","column","Object","getColumns","tblColumn","sql","join","getSQL","resolveFn","rejectFn","Promise"],"mappings":";;;;+BA4BaA;;;eAAAA;;;sBA5BuE;iCACd;4BAS/D;yBAQA;wBACoE;0BAC9C;;;;;;;;;;;;;;AAQtB,MAAMA;IAoDX,OAAOC,SACLC,SAAiB,EACjBC,OAA+C,EAC/CC,OAAkB,EAClB;QACA,OAAOC,IAAAA,wBAAkB,EACvB,CAACC,QAAU,IAAIN,YAAYM,OAAOH,SAAkBD,WAAWE;IAEnE;IAEAG,YAAYD,KAAa,EAAE;QACzB,MAAME,SAAS,IAAI,CAACJ,OAAO,CAACK,GAAG,CAAC,CAACC;YAC/B,MAAMC,aAAa,IAAI,CAACC,MAAM,CAACC,cAAc,CAACH;YAC9C,IAAIC,cAAcL,OAAO;gBACvB,OAAOA,KAAK,CAACK,WAAW;YAC1B;YAEA,MAAM,IAAIG,MAAM,CAAC,mBAAmB,EAAEH,WAAW,cAAc,EAAE,IAAI,CAACT,SAAS,EAAE;QACnF;QAEA,OAAO,IAAI,CAACE,OAAO,CAACW,MAAM,GAAG,IAAIP,SAASA,MAAM,CAAC,EAAE;IACrD;IAEAQ,aAAaC,IAAwB,EAAE;QACrC,MAAMP,MAAMQ,IAAAA,yBAAQ,EAACD,KAAKE,UAAU,CAACC,IAAI,EAAEH,KAAKI,IAAI;QACpD,IAAI,CAAC,IAAI,CAACC,QAAQ,CAACC,GAAG,CAACb,MAAM;YAC3B,IAAI,CAACY,QAAQ,CAACE,GAAG,CAACd,KAAKe,qBAAI,CAACC,YAAY,CAACC,IAAAA,uBAAc,EAAC,IAAI,CAACf,MAAM,GAAG,IAAI,CAACgB,OAAO,EAAEX;QACtF;QAEA,OAAO,IAAI,CAACK,QAAQ,CAACO,GAAG,CAACnB;IAC3B;IAEAoB,qBAAqBb,IAAwB,EAAEc,QAAgB,EAAE;QAC/D,MAAMrB,MAAMQ,IAAAA,yBAAQ,EAACa,UAAUd,KAAKI,IAAI;QACxC,IAAI,CAAC,IAAI,CAACW,SAAS,CAACT,GAAG,CAACb,MAAM;YAC5B,IAAI,CAACsB,SAAS,CAACR,GAAG,CAChBd,KACA,mDAAmD;YACnDuB,IAAAA,sBAAY,EAAC;gBAAErB,QAAQ,IAAI,CAACA,MAAM;gBAAEgB,SAAS,IAAI,CAACA,OAAO;gBAAEX;gBAAMc;YAAS;QAE9E;QAEA,OAAO,IAAI,CAACC,SAAS,CAACH,GAAG,CAACnB;IAC5B;IAEA;;;;GAIC,GACDwB,cAAcjB,IAAwB,EAAEX,KAAa,EAA2C;QAC9F,MAAM6B,YAAY,IAAI,CAACnB,YAAY,CAACC;QAEpC,OAAOmB,IAAAA,gBAAU,EAACD,aACdA,UAAUE,IAAI,CAAC,CAACC,UAAY,IAAI,CAACC,QAAQ,CAACD,SAASrB,MAAMX,UACzD,IAAI,CAACiC,QAAQ,CAACJ,WAAWlB,MAAMX;IACrC;IAEQiC,SAASC,MAAyB,EAAEvB,IAAwB,EAAEX,KAAa,EAAE;QACnF,OAAO,IAAI,CAACmC,UAAU,CAACD,QAAQlC,OAAO+B,IAAI,CAAC,CAACK;YAC1C,IAAIA,QAAQ;gBACV,MAAMC,UAAUH,OAAOI,QAAQ,CAAC,GAAG3B,KAAKE,UAAU,CAACC,IAAI,CAAC,CAAC,EAAEH,KAAKI,IAAI,CAACX,GAAG,EAAE,CAAC;gBAE3E,IAAIiC,SAAS;oBACX,uFAAuF;oBACvF,uFAAuF;oBACvF,sFAAsF;oBACtF,oFAAoF;oBACpF,4EAA4E;oBAC5EE,IAAAA,mCAAkB,EAAC,IAAI,CAACjB,OAAO,EAAEX,MAAM0B,SAASD;gBAClD;YACF;YAEA,OAAOA;QACT;IACF;IAEA,+FAA+F,GAC/FI,sBACE7B,IAAwB,EACxBX,KAAa,EACbyC,UAAkB,EACuB;QACzC,MAAMZ,YAAY,IAAI,CAACL,oBAAoB,CAACb,MAAM8B;QAElD,OAAOX,IAAAA,gBAAU,EAACD,aACdA,UAAUE,IAAI,CAAC,CAACC,UAAY,IAAI,CAACU,aAAa,CAACV,QAAQW,IAAI,IAAIhC,MAAMX,UACrE,IAAI,CAAC0C,aAAa,CAACb,UAAUc,IAAI,IAAIhC,MAAMX;IACjD;IAEQ0C,cAAcR,MAAyB,EAAEvB,IAAwB,EAAEX,KAAa,EAAE;QACxF,OAAO,IAAI,CAACmC,UAAU,CAACD,QAAQlC,OAAO+B,IAAI,CAAC,CAACK;YAC1C,IAAIA,QAAQ;gBACV,mFAAmF;gBACnF,8DAA8D;gBAC9DQ,IAAAA,kCAAiB,EAAC,IAAI,CAACtB,OAAO,EAAEX,MAAMuB,OAAOI,QAAQ;YACvD;YAEA,OAAOF;QACT;IACF;IAEAD,WAAWD,MAAyB,EAAElC,KAAa,EAAE;QACnD,KAAK,MAAM6C,SAAS,IAAI,CAACC,MAAM,CAAE;YAC/B,+EAA+E;YAC/E,IAAI,IAAI,CAACC,OAAO,CAACC,YAAY,CAACH,MAAMI,IAAI,EAAEf,OAAOe,IAAI,GAAG;gBACtD,IAAI,CAACF,OAAO,CAACG,SAAS,CAACL,MAAMI,IAAI,EAAEf,OAAOe,IAAI;gBAE9C,IAAI,CAACJ,MAAMM,MAAM,CAAClC,GAAG,CAACjB,QAAQ;oBAC5B6C,MAAMM,MAAM,CAACjC,GAAG,CAAClB,OAAOoD;gBAC1B;gBAEA,OAAOP,MAAMM,MAAM,CAAC5B,GAAG,CAACvB,OAAQqD,OAAO;YACzC;QACF;QAEA,OAAO,IAAI,CAACC,QAAQ,CAACpB,QAAQlC;IAC/B;IAEAsD,SAASpB,MAAyB,EAAElC,KAAa,EAAE;QACjD,MAAMqD,UAAUD;QAChB,MAAMH,OAAO,IAAI,CAACF,OAAO,CAACQ,UAAU,CAACrB,OAAOe,IAAI,CAACjD,KAAK;QAEtD,IAAI,CAAC+C,OAAO,CAACG,SAAS,CAACD,MAAMf,OAAOe,IAAI;QAExC,MAAMJ,QAAQ;YACZI;YACAE,QAAQ,IAAIK,IAAI;gBAAC;oBAACxD;oBAAOqD;iBAAQ;aAAC;QACpC;QACA,IAAI,CAACP,MAAM,CAACW,GAAG,CAACZ;QAEhB,MAAMa,WAAWN;QACjB,MAAMO,SAASC,IAAAA,iBAAS,EAAC,IAAI,CAAC/D,OAAO,EAAE,IAAI,CAACyB,OAAO;QAEnDoC,SAASL,OAAO,CACbtB,IAAI,CAAC;YACJ,IAAI,CAACe,MAAM,CAACe,MAAM,CAAChB;YACnB,MAAMiB,MAAMH,OAAOI,KAAK,CAAC,IAAI,CAACnE,SAAS,CAAC;YAExC,MAAMmE,QAAQD,IAAIE,QAAQ,CAAC;gBACzB,GAAG,IAAI,CAACjB,OAAO,CAACkB,OAAO,CAACpB,MAAMI,IAAI,CAAC;gBACnCiB,OAAO;oBACLC,KAAK,CAACC,QACJC,IAAAA,mBAAO,EACL,IAAI,CAACC,SAAS,CAACF,QACf;+BAAIvB,MAAMM,MAAM,CAACoB,IAAI;yBAAG,CAACpE,GAAG,CAAC,CAACH,QAAU,IAAI,CAACC,WAAW,CAACD;gBAE/D;YACF;YAEA+D,MAAMhC,IAAI,CACR,CAACyC;gBACC,KAAK,MAAM,CAACxE,OAAOqD,QAAQ,IAAIR,MAAMM,MAAM,CAACsB,OAAO,GAAI;oBACrD,gFAAgF;oBAChF,+EAA+E;oBAC/E,qBAAqB;oBACrB,MAAMrC,SAASoC,QAAQE,IAAI,CAAC,CAACC,MAC3B,IAAI,CAAC7E,OAAO,CAAC8E,KAAK,CAAC,CAACxE;4BAClB,MAAMU,OAAO,IAAI,CAACR,MAAM,CAACC,cAAc,CAACH;4BACxC,MAAMyE,SAASF,GAAG,CAAC7D,KAAyB;4BAC5C,MAAMgE,WAAW,AAAC9E,KAAiC,CAACc,KAAK;4BAEzD,OACE+D,WAAWC,YACVD,kBAAkBE,QACjBD,oBAAoBC,QACpBF,OAAOG,OAAO,OAAOF,SAASE,OAAO;wBAE3C;oBAGF,IAAI5C,QAAQ;wBACViB,QAAQ4B,OAAO,CAAC7C;oBAClB,OAAO;wBACLiB,QAAQ6B,MAAM,CACZ,IAAI1E,MAAM,CAAC,MAAM,EAAE,IAAI,CAACZ,SAAS,CAAC,CAAC,EAAE,IAAI,CAACK,WAAW,CAACD,OAAO,WAAW,CAAC;oBAE7E;gBACF;YACF,GACA,CAACmF;gBACC,KAAK,MAAM9B,WAAWR,MAAMM,MAAM,CAACjD,MAAM,GAAI;oBAC3CmD,QAAQ6B,MAAM,CAACC;gBACjB;YACF;QAEJ,GACCC,KAAK,CAAC,CAACD;YACN,KAAK,MAAM9B,WAAWR,MAAMM,MAAM,CAACjD,MAAM,GAAI;gBAC3CmD,QAAQ6B,MAAM,CAACC;YACjB;QACF;QAEFE,WAAW,IAAM3B,SAASuB,OAAO,IAAI;QAErC,OAAO5B,QAAQA,OAAO;IACxB;IA5NA,YACE/B,OAAe,EACfzB,OAA+C,EAC/CD,SAAiB,EACjBE,OAAkB,CAClB;QAhCFwB,uBAAAA,WAAAA,KAAAA;QAEAzB,uBAAAA,WAAAA,KAAAA;QAEAD,uBAAAA,aAAAA,KAAAA;QAEA,8FAA8F;QAC9F,gGAAgG;QAChG,yDAAyD;QACzDoB,uBAAAA,YAAW,IAAIwC;QAEf9B,uBAAAA,aAAY,IAAI8B;QAEhB,0FAA0F;QAC1F,sCAAsC;QACtCV,uBAAAA,UAAS,IAAIwC;QAKbhF,uBAAAA,UAAAA,KAAAA;QACAyC,uBAAAA,WAAAA,KAAAA;QACAqB,uBAAAA,SAAAA,KAAAA;QACAtE,uBAAAA,WAAAA,KAAAA;QACAyF,uBAAAA,cAAAA,KAAAA;QACAjB,uBAAAA,aAAAA,KAAAA;QAQE,IAAI,CAAChD,OAAO,GAAGA;QACf,IAAI,CAACzB,OAAO,GAAGA;QACf,IAAI,CAACD,SAAS,GAAGA;QACjB,IAAI,CAACU,MAAM,GAAGkF,IAAAA,uBAAe,EAAC3F;QAC9B,IAAI,CAACkD,OAAO,GAAG1B,IAAAA,uBAAc,EAAC,IAAI,CAACf,MAAM;QACzC,IAAI,CAAC8D,KAAK,GAAG,IAAI,CAAC9D,MAAM,CAACmF,SAAS,CAAC7F,UAAU;QAC7C,IAAI,CAAC2F,UAAU,GAAG,IAAI,CAACjF,MAAM,CAACoF,aAAa,CAAC9F;QAC5C,IAAI,CAACE,OAAO,GAAGA,oBAAAA,qBAAAA,UAAW,IAAI,CAACyF,UAAU;QACzC,IAAI,CAACjB,SAAS,GAAG,CAACF;YAChB,MAAMtE,UAAU,IAAI,CAACA,OAAO,CAACK,GAAG,CAAC,CAACwF,SAChCC,OAAO1F,MAAM,CAAC2F,IAAAA,sBAAU,EAACzB,QAAQM,IAAI,CACnC,CAACoB,YAAc,gBAAgBA,aAAaA,UAAUhF,IAAI,KAAK6E,OAAO7E,IAAI;YAG9E,OAAOhB,QAAQW,MAAM,GAAG,IAAIsF,IAAAA,eAAG,CAAA,CAAC,CAAC,EAAEA,eAAG,CAACC,IAAI,CAAClG,SAASiG,IAAAA,eAAG,CAAA,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,GAAGjG,OAAO,CAAC,EAAE,CAAEmG,MAAM;QACvF;IACF;AAuMF;AAEA,SAAS7C;IACP,IAAI8C;IACJ,IAAIC;IACJ,MAAM9C,UAAU,IAAI+C,QAAW,CAACnB,SAASC;QACvCgB,YAAYjB;QACZkB,WAAWjB;IACb;IAEA,OAAO;QAAE7B;QAAS4B,SAASiB;QAAWhB,QAAQiB;IAAS;AACzD"}