@baeta/extension-complexity 0.0.1

package/CHANGELOG.md

@@ -0,0 +1,11 @@
+# @baeta/extension-complexity
+
+## 0.0.1
+
+### Patch Changes
+
+- [#176](https://github.com/andreisergiu98/baeta/pull/176) [`d77cd8a`](https://github.com/andreisergiu98/baeta/commit/d77cd8a1810fdf72cfbbb08d05c207bbc893c822) Thanks [@andreisergiu98](https://github.com/andreisergiu98)! - feat: add complexity extension
+
+- Updated dependencies [[`59bbb9c`](https://github.com/andreisergiu98/baeta/commit/59bbb9c4baaf716f27dc251fe7aeb0231e6c5321), [`d77cd8a`](https://github.com/andreisergiu98/baeta/commit/d77cd8a1810fdf72cfbbb08d05c207bbc893c822), [`cf9f094`](https://github.com/andreisergiu98/baeta/commit/cf9f09468f84d99b069eb0f55e1fc207e2a41dd8)]:
+  - @baeta/core@0.1.2
+  - @baeta/errors@0.1.2

package/dist/index.cjs

@@ -0,0 +1,330 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class;// lib/complexity-extension.ts
+
+
+var _sdk = require('@baeta/core/sdk');
+
+// lib/complexity-calculator.ts
+var _errors = require('@baeta/errors');
+
+
+
+
+
+
+var _graphql = require('graphql');
+
+// utils/graphlq.ts
+
+function isListOrNullableList(type) {
+  if (type instanceof _graphql.GraphQLList) {
+    return true;
+  }
+  if (type instanceof _graphql.GraphQLNonNull) {
+    return isListOrNullableList(type.ofType);
+  }
+  return false;
+}
+
+// utils/string.ts
+function capitalize(str) {
+  return str.charAt(0).toUpperCase() + str.slice(1);
+}
+
+// lib/field-settings.ts
+
+
+
+function getFieldComplexitySettings(ctx, info, type, field, selection, fieldName, fieldSettingsMap) {
+  const args = _graphql.getArgumentValues.call(void 0, field, selection, info.variableValues);
+  const customComplexity = _optionalChain([fieldSettingsMap, 'access', _ => _[type.name], 'optionalAccess', _2 => _2[fieldName]]);
+  if (customComplexity) {
+    return customComplexity({ args, ctx });
+  }
+  const wildCardComplexity = _optionalChain([fieldSettingsMap, 'access', _3 => _3[type.name], 'optionalAccess', _4 => _4["*"]]);
+  if (wildCardComplexity) {
+    return wildCardComplexity({ args, ctx });
+  }
+}
+function registerFieldSettingsSetter(type, field, fn, map) {
+  map[type] ??= {};
+  map[type][field] = fn;
+}
+
+// lib/complexity-calculator.ts
+function calculateComplexity(ctx, info, fieldSettingsMap, defaults) {
+  const operationName = capitalize(info.operation.operation);
+  const operationType = info.schema.getType(operationName);
+  if (!operationType || !_graphql.isOutputType.call(void 0, operationType)) {
+    throw new (0, _errors.ValidationError)(`Unsupported operation ${operationName}`);
+  }
+  return complexityFromSelectionSet(
+    ctx,
+    info,
+    operationType,
+    info.operation.selectionSet,
+    fieldSettingsMap,
+    defaults
+  );
+}
+function complexityFromSelectionSet(ctx, info, type, selectionSet, fieldSettingsMap, defaults) {
+  const result = {
+    depth: 0,
+    breadth: 0,
+    complexity: 0
+  };
+  for (const selection of selectionSet.selections) {
+    const selectionResult = complexityFromSelection(
+      ctx,
+      info,
+      type,
+      selection,
+      fieldSettingsMap,
+      defaults
+    );
+    result.complexity += selectionResult.complexity;
+    result.breadth += selectionResult.breadth;
+    result.depth = Math.max(result.depth, selectionResult.depth);
+  }
+  return result;
+}
+function complexityFromSelection(ctx, info, type, selection, fieldSettingsMap, defaults) {
+  if (selection.kind === _graphql.Kind.FIELD) {
+    return complexityFromField(ctx, info, type, selection, fieldSettingsMap, defaults);
+  }
+  if (selection.kind === _graphql.Kind.FRAGMENT_SPREAD) {
+    const fragment = info.fragments[selection.name.value];
+    if (!fragment) {
+      throw new (0, _errors.ValidationError)(`Fragment ${selection.name.value} not found`);
+    }
+    return complexityFromFragment(ctx, info, type, fragment, fieldSettingsMap, defaults);
+  }
+  return complexityFromFragment(ctx, info, type, selection, fieldSettingsMap, defaults);
+}
+function complexityFromFragment(ctx, info, type, fragment, fieldSettingsMap, defaults) {
+  const fragmentType = fragment.typeCondition ? info.schema.getType(fragment.typeCondition.name.value) : type;
+  if (!_graphql.isOutputType.call(void 0, fragmentType)) {
+    throw new (0, _errors.ValidationError)(`Unsupported fragment type ${fragmentType}`);
+  }
+  if (!fragmentType) {
+    throw new (0, _errors.ValidationError)(`Fragment type ${fragmentType} not found`);
+  }
+  return complexityFromSelectionSet(
+    ctx,
+    info,
+    fragmentType,
+    fragment.selectionSet,
+    fieldSettingsMap,
+    defaults
+  );
+}
+function complexityFromField(ctx, info, type, selection, fieldSettingsMap, defaults) {
+  const fieldName = selection.name.value;
+  const field = _graphql.isObjectType.call(void 0, type) || _graphql.isInterfaceType.call(void 0, type) ? type.getFields()[fieldName] : void 0;
+  if (!field && !fieldName.startsWith("__")) {
+    throw new (0, _errors.ValidationError)(`Field ${fieldName} not found on type ${type.name}`);
+  }
+  if (!field) {
+    return {
+      depth: 1,
+      breadth: 1,
+      complexity: 1
+    };
+  }
+  const fieldComplexitySettings = getFieldComplexitySettings(
+    ctx,
+    info,
+    type,
+    field,
+    selection,
+    fieldName,
+    fieldSettingsMap
+  );
+  if (fieldComplexitySettings === false) {
+    return {
+      depth: 0,
+      breadth: 0,
+      complexity: 0
+    };
+  }
+  let depth = 1;
+  let breadth = 1;
+  let complexity = 0;
+  const listMultiplier = _nullishCoalesce(_optionalChain([fieldComplexitySettings, 'optionalAccess', _5 => _5.multiplier]), () => ( defaults.multiplier));
+  const multiplier = field && isListOrNullableList(field.type) ? listMultiplier : 1;
+  complexity += (_nullishCoalesce(_optionalChain([fieldComplexitySettings, 'optionalAccess', _6 => _6.complexity]), () => ( defaults.complexity))) * multiplier;
+  if (!field || !selection.selectionSet) {
+    return {
+      depth,
+      breadth,
+      complexity
+    };
+  }
+  const subSelection = complexityFromSelectionSet(
+    ctx,
+    info,
+    _graphql.getNamedType.call(void 0, field.type),
+    selection.selectionSet,
+    fieldSettingsMap,
+    defaults
+  );
+  depth += subSelection.depth;
+  breadth += subSelection.breadth;
+  complexity += subSelection.complexity * Math.max(multiplier, 1);
+  return {
+    depth,
+    breadth,
+    complexity
+  };
+}
+
+// lib/complexity-errors.ts
+
+var ComplexityErrorKind = /* @__PURE__ */ ((ComplexityErrorKind2) => {
+  ComplexityErrorKind2["Depth"] = "DepthExceeded";
+  ComplexityErrorKind2["Breadth"] = "BreadthExceeded";
+  ComplexityErrorKind2["Complexity"] = "ComplexityExceeded";
+  return ComplexityErrorKind2;
+})(ComplexityErrorKind || {});
+function getDefaultComplexityError(kind, limit, result) {
+  if (kind === "DepthExceeded" /* Depth */) {
+    return new (0, _errors.ValidationError)(`Depth limit of ${limit} exceeded, got: ${result}`);
+  }
+  if (kind === "BreadthExceeded" /* Breadth */) {
+    return new (0, _errors.ValidationError)(`Breadth limit of ${limit} exceeded, got: ${result}`);
+  }
+  if (kind === "ComplexityExceeded" /* Complexity */) {
+    return new (0, _errors.ValidationError)(`Complexity limit of ${limit} exceeded, got: ${result}`);
+  }
+  throw new Error("Unknown complexity error kind");
+}
+
+// lib/complexity-options.ts
+var defaultLimits = {
+  depth: 10,
+  breadth: 50,
+  complexity: 1e3
+};
+function normalizeOptions(options) {
+  return {
+    limit: _nullishCoalesce(options.limit, () => ( defaultLimits)),
+    defaultComplexity: _nullishCoalesce(options.defaultComplexity, () => ( 1)),
+    defaultListMultiplier: _nullishCoalesce(options.defaultListMultiplier, () => ( 10)),
+    complexityError: _nullishCoalesce(options.complexityError, () => ( getDefaultComplexityError))
+  };
+}
+
+// lib/store.ts
+var _core = require('@baeta/core');
+var complexityStoreKey = Symbol("complexity-extension");
+var [getComplexityStore, setComplexityStoreLoader] = _core.createContextStore.call(void 0, complexityStoreKey);
+
+// lib/store-loader.ts
+function loadComplexityStore(ctx, getLimits, defaultLimits2) {
+  setComplexityStoreLoader(ctx, async () => {
+    const limits = typeof getLimits === "function" ? await getLimits(ctx) : getLimits;
+    let cache = void 0;
+    const cacheComplexity = (fn) => {
+      if (cache) {
+        return cache;
+      }
+      cache = fn();
+      return cache;
+    };
+    return {
+      limits: {
+        depth: _nullishCoalesce(_optionalChain([limits, 'optionalAccess', _7 => _7.depth]), () => ( defaultLimits2.depth)),
+        breadth: _nullishCoalesce(_optionalChain([limits, 'optionalAccess', _8 => _8.breadth]), () => ( defaultLimits2.breadth)),
+        complexity: _nullishCoalesce(_optionalChain([limits, 'optionalAccess', _9 => _9.complexity]), () => ( defaultLimits2.complexity))
+      },
+      cacheComplexity
+    };
+  });
+}
+
+// lib/complexity-extension.ts
+var ComplexityExtension = (_class = class extends _sdk.Extension {
+  
+  constructor(options = {}) {
+    super();_class.prototype.__init.call(this);_class.prototype.__init2.call(this);_class.prototype.__init3.call(this);_class.prototype.__init4.call(this);_class.prototype.__init5.call(this);;
+    this.options = normalizeOptions(options);
+  }
+  __init() {this.fieldSettingsMap = {}}
+  __init2() {this.getTypeExtensions = (_module, type) => {
+    return {
+      $complexity: (fn) => {
+        registerFieldSettingsSetter(type, "*", fn, this.fieldSettingsMap);
+      }
+    };
+  }}
+  __init3() {this.getResolverExtensions = (_module, type, field) => {
+    return {
+      $complexity: (fn) => {
+        registerFieldSettingsSetter(type, field, fn, this.fieldSettingsMap);
+      }
+    };
+  }}
+  __init4() {this.getSubscriptionExtensions = (_module, field) => {
+    return {
+      $complexity: (fn) => {
+        registerFieldSettingsSetter("Subscription", field, fn, this.fieldSettingsMap);
+      }
+    };
+  }}
+  createComplexityMiddleware() {
+    return (next) => async (root, args, ctx, info) => {
+      loadComplexityStore(ctx, this.options.limit, defaultLimits);
+      const store = await getComplexityStore(ctx);
+      const limits = store.limits;
+      const results = store.cacheComplexity(() => {
+        return calculateComplexity(ctx, info, this.fieldSettingsMap, {
+          complexity: this.options.defaultComplexity,
+          multiplier: this.options.defaultListMultiplier
+        });
+      });
+      if (results.depth > limits.depth) {
+        throw this.options.complexityError("DepthExceeded" /* Depth */, limits.depth, results.depth);
+      }
+      if (results.breadth > limits.breadth) {
+        throw this.options.complexityError(
+          "BreadthExceeded" /* Breadth */,
+          limits.breadth,
+          results.breadth
+        );
+      }
+      if (results.complexity > limits.complexity) {
+        throw this.options.complexityError(
+          "ComplexityExceeded" /* Complexity */,
+          limits.complexity,
+          results.complexity
+        );
+      }
+      return next(root, args, ctx, info);
+    };
+  }
+  __init5() {this.build = (_module, mapper) => {
+    for (const type of mapper.getTypes()) {
+      if (type !== "Query" && type !== "Mutation" && type !== "Subscription") {
+        continue;
+      }
+      if (type !== "Subscription") {
+        for (const field of mapper.getTypeFields(type)) {
+          mapper.addMiddleware(type, field, this.createComplexityMiddleware());
+        }
+        continue;
+      }
+      for (const field of mapper.getTypeFields(type)) {
+        mapper.addMiddleware(type, `${field}.subscribe`, this.createComplexityMiddleware());
+      }
+    }
+  }}
+}, _class);
+
+// index.ts
+function complexityExtension(options) {
+  return () => new ComplexityExtension(options);
+}
+
+
+
+exports.ComplexityErrorKind = ComplexityErrorKind; exports.complexityExtension = complexityExtension;
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/home/runner/work/baeta/baeta/packages/extension-complexity/dist/index.cjs","../lib/complexity-extension.ts","../lib/complexity-calculator.ts","../utils/graphlq.ts","../utils/string.ts","../lib/field-settings.ts","../lib/complexity-errors.ts","../lib/complexity-options.ts","../lib/store.ts","../lib/store-loader.ts","../index.ts"],"names":["ComplexityErrorKind"],"mappings":"AAAA;ACAA;AACC;AAAA,sCAIM;ADDP;AACA;AELA,uCAAgC;AAChC;AAMC;AAGA;AACA;AACA;AACA;AAAA,kCACM;AFAP;AACA;AGfA;AAEO,SAAS,oBAAA,CAAqB,IAAA,EAAkC;AACtE,EAAA,GAAA,CAAI,KAAA,WAAgB,oBAAA,EAAa;AAChC,IAAA,OAAO,IAAA;AAAA,EACR;AAEA,EAAA,GAAA,CAAI,KAAA,WAAgB,uBAAA,EAAgB;AACnC,IAAA,OAAO,oBAAA,CAAqB,IAAA,CAAK,MAAM,CAAA;AAAA,EACxC;AAEA,EAAA,OAAO,KAAA;AACR;AHcA;AACA;AI3BO,SAAS,UAAA,CAAW,GAAA,EAAa;AACvC,EAAA,OAAO,GAAA,CAAI,MAAA,CAAO,CAAC,CAAA,CAAE,WAAA,CAAY,EAAA,EAAI,GAAA,CAAI,KAAA,CAAM,CAAC,CAAA;AACjD;AJ6BA;AACA;AKhCA;AAKC;AAAA;AAuBM,SAAS,0BAAA,CACf,GAAA,EACA,IAAA,EACA,IAAA,EACA,KAAA,EACA,SAAA,EACA,SAAA,EACA,gBAAA,EACC;AACD,EAAA,MAAM,KAAA,EAAO,wCAAA,KAAkB,EAAO,SAAA,EAAW,IAAA,CAAK,cAAc,CAAA;AACpE,EAAA,MAAM,iBAAA,kBAAmB,gBAAA,mBAAiB,IAAA,CAAK,IAAI,CAAA,4BAAA,CAAI,SAAS,GAAA;AAEhE,EAAA,GAAA,CAAI,gBAAA,EAAkB;AACrB,IAAA,OAAO,gBAAA,CAAiB,EAAE,IAAA,EAAM,IAAI,CAAC,CAAA;AAAA,EACtC;AAEA,EAAA,MAAM,mBAAA,kBAAqB,gBAAA,qBAAiB,IAAA,CAAK,IAAI,CAAA,4BAAA,CAAI,GAAG,GAAA;AAE5D,EAAA,GAAA,CAAI,kBAAA,EAAoB;AACvB,IAAA,OAAO,kBAAA,CAAmB,EAAE,IAAA,EAAM,IAAI,CAAC,CAAA;AAAA,EACxC;AACD;AAEO,SAAS,2BAAA,CACf,IAAA,EACA,KAAA,EACA,EAAA,EACA,GAAA,EACC;AACD,EAAA,GAAA,CAAI,IAAI,EAAA,IAAM,CAAC,CAAA;AACf,EAAA,GAAA,CAAI,IAAI,CAAA,CAAE,KAAK,EAAA,EAAI,EAAA;AACpB;ALRA;AACA;AE5BO,SAAS,mBAAA,CACf,GAAA,EACA,IAAA,EACA,gBAAA,EACA,QAAA,EACC;AACD,EAAA,MAAM,cAAA,EAAgB,UAAA,CAAW,IAAA,CAAK,SAAA,CAAU,SAAS,CAAA;AACzD,EAAA,MAAM,cAAA,EAAgB,IAAA,CAAK,MAAA,CAAO,OAAA,CAAQ,aAAa,CAAA;AAEvD,EAAA,GAAA,CAAI,CAAC,cAAA,GAAiB,CAAC,mCAAA,aAA0B,CAAA,EAAG;AACnD,IAAA,MAAM,IAAI,4BAAA,CAAgB,CAAA,sBAAA,EAAyB,aAAa,CAAA,CAAA;AACjE,EAAA;AAEO,EAAA;AACN,IAAA;AACA,IAAA;AACA,IAAA;AACe,IAAA;AACf,IAAA;AACA,IAAA;AACD,EAAA;AACD;AAOC;AAGe,EAAA;AACP,IAAA;AACE,IAAA;AACG,IAAA;AACb,EAAA;AAEiD,EAAA;AACxB,IAAA;AACvB,MAAA;AACA,MAAA;AACA,MAAA;AACA,MAAA;AACA,MAAA;AACA,MAAA;AACD,IAAA;AACqC,IAAA;AACH,IAAA;AACyB,IAAA;AAC5D,EAAA;AAEO,EAAA;AACR;AAOC;AAGmC,EAAA;AACqB,IAAA;AACxD,EAAA;AAE6C,EAAA;AACQ,IAAA;AAErC,IAAA;AAC4C,MAAA;AAC3D,IAAA;AAEyD,IAAA;AAC1D,EAAA;AAE0D,EAAA;AAC3D;AAOC;AAIuB,EAAA;AAGU,EAAA;AACuB,IAAA;AACxD,EAAA;AAEmB,EAAA;AACqC,IAAA;AACxD,EAAA;AAEO,EAAA;AACN,IAAA;AACA,IAAA;AACA,IAAA;AACS,IAAA;AACT,IAAA;AACA,IAAA;AACD,EAAA;AACD;AAOC;AAGiC,EAAA;AAGmB,EAAA;AAET,EAAA;AACE,IAAA;AAC7C,EAAA;AAEY,EAAA;AACJ,IAAA;AACC,MAAA;AACE,MAAA;AACG,MAAA;AACb,IAAA;AACD,EAAA;AAEgC,EAAA;AAC/B,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACD,EAAA;AAEuC,EAAA;AAC/B,IAAA;AACC,MAAA;AACE,MAAA;AACG,MAAA;AACb,IAAA;AACD,EAAA;AAEY,EAAA;AACE,EAAA;AACG,EAAA;AAE6C,EAAA;AACC,EAAA;AAEA,EAAA;AAExB,EAAA;AAC/B,IAAA;AACN,MAAA;AACA,MAAA;AACA,MAAA;AACD,IAAA;AACD,EAAA;AAEqB,EAAA;AACpB,IAAA;AACA,IAAA;AACuB,IAAA;AACb,IAAA;AACV,IAAA;AACA,IAAA;AACD,EAAA;AAEsB,EAAA;AACE,EAAA;AACsC,EAAA;AAEvD,EAAA;AACN,IAAA;AACA,IAAA;AACA,IAAA;AACD,EAAA;AACD;AFjCqE;AACA;AMnLrC;AAGzB;AACE,EAAA;AACE,EAAA;AACG,EAAA;AAHFA,EAAAA;AAAA;AAgBI;AACyB,EAAA;AACW,IAAA;AACnD,EAAA;AAE0C,EAAA;AACW,IAAA;AACrD,EAAA;AAE6C,EAAA;AACW,IAAA;AACxD,EAAA;AAE+C,EAAA;AAChD;ANsKqE;AACA;AO9LxC;AACrB,EAAA;AACE,EAAA;AACG,EAAA;AACb;AAM8B;AACtB,EAAA;AACkB,IAAA;AACwB,IAAA;AACQ,IAAA;AACZ,IAAA;AAC7C,EAAA;AACD;AP2LqE;AACA;AQvNlC;AAQ4B;AAG9D;ARgNoE;AACA;ASrNnE;AACyC,EAAA;AACwB,IAAA;AAEd,IAAA;AAEc,IAAA;AACrD,MAAA;AACH,QAAA;AACR,MAAA;AAEW,MAAA;AACJ,MAAA;AACR,IAAA;AAEO,IAAA;AACE,MAAA;AAC+B,QAAA;AACI,QAAA;AACM,QAAA;AACjD,MAAA;AACA,MAAA;AACD,IAAA;AACA,EAAA;AACF;ATmNqE;AACA;AClOb;AACtC,EAAA;AAE0C,EAAA;AACpD,IAAA;AACiC,IAAA;AACxC,EAAA;AAE8C,iBAAA;AAKM,kBAAA;AAC5C,IAAA;AACe,MAAA;AAC4B,QAAA;AACjD,MAAA;AACD,IAAA;AACD,EAAA;AAMsE,kBAAA;AAC9D,IAAA;AACe,MAAA;AAC8B,QAAA;AACnD,MAAA;AACD,IAAA;AACD,EAAA;AAKkE,kBAAA;AAC1D,IAAA;AACe,MAAA;AACwC,QAAA;AAC7D,MAAA;AACD,IAAA;AACD,EAAA;AAOE,EAAA;AACiD,IAAA;AAC2B,MAAA;AAElC,MAAA;AAErB,MAAA;AAEuB,MAAA;AACoC,QAAA;AACrD,UAAA;AACA,UAAA;AACzB,QAAA;AACD,MAAA;AAEiC,MAAA;AACd,QAAA;AACpB,MAAA;AAEsC,MAAA;AAClB,QAAA;AAAA,UAAA;AAEX,UAAA;AACC,UAAA;AACT,QAAA;AACD,MAAA;AAE4C,MAAA;AACxB,QAAA;AAAA,UAAA;AAEX,UAAA;AACC,UAAA;AACT,QAAA;AACD,MAAA;AAEiC,MAAA;AAClC,IAAA;AACD,EAAA;AAE4D,kBAAA;AACrB,IAAA;AACmB,MAAA;AACvD,QAAA;AACD,MAAA;AAE6B,MAAA;AACoB,QAAA;AACR,UAAA;AACxC,QAAA;AACA,QAAA;AACD,MAAA;AAEgD,MAAA;AACO,QAAA;AACvD,MAAA;AACD,IAAA;AACD,EAAA;AACD;ADqMqE;AACA;AUxTe;AACvC,EAAA;AAC7C;AV0TqE;AACA;AACA;AACA","file":"/home/runner/work/baeta/baeta/packages/extension-complexity/dist/index.cjs","sourcesContent":[null,"import {\n\tExtension,\n\ttype ModuleBuilder,\n\ttype NativeMiddleware,\n\ttype ResolverMapper,\n} from '@baeta/core/sdk';\nimport { calculateComplexity } from './complexity-calculator.ts';\nimport { ComplexityErrorKind } from './complexity-errors.ts';\nimport {\n\ttype ComplexityExtensionOptions,\n\tdefaultLimits,\n\tnormalizeOptions,\n} from './complexity-options.ts';\nimport { type FieldSettingsMap, registerFieldSettingsSetter } from './field-settings.ts';\nimport { loadComplexityStore } from './store-loader.ts';\nimport { getComplexityStore } from './store.ts';\n\nexport class ComplexityExtension<Ctx> extends Extension {\n\tprivate readonly options: Required<ComplexityExtensionOptions<Ctx>>;\n\n\tconstructor(options: ComplexityExtensionOptions<Ctx> = {}) {\n\t\tsuper();\n\t\tthis.options = normalizeOptions(options);\n\t}\n\n\tprivate fieldSettingsMap: FieldSettingsMap = {};\n\n\tgetTypeExtensions = <Root, Context>(\n\t\t_module: ModuleBuilder,\n\t\ttype: string,\n\t): BaetaExtensions.TypeExtensions<Root, Context> => {\n\t\treturn {\n\t\t\t$complexity: (fn) => {\n\t\t\t\tregisterFieldSettingsSetter(type, '*', fn, this.fieldSettingsMap);\n\t\t\t},\n\t\t};\n\t};\n\n\tgetResolverExtensions = <Result, Root, Context, Args>(\n\t\t_module: ModuleBuilder,\n\t\ttype: string,\n\t\tfield: string,\n\t): BaetaExtensions.ResolverExtensions<Result, Root, Context, Args> => {\n\t\treturn {\n\t\t\t$complexity: (fn) => {\n\t\t\t\tregisterFieldSettingsSetter(type, field, fn, this.fieldSettingsMap);\n\t\t\t},\n\t\t};\n\t};\n\n\tgetSubscriptionExtensions = <Root, Context, Args>(\n\t\t_module: ModuleBuilder,\n\t\tfield: string,\n\t): BaetaExtensions.SubscriptionExtensions<Root, Context, Args> => {\n\t\treturn {\n\t\t\t$complexity: (fn) => {\n\t\t\t\tregisterFieldSettingsSetter('Subscription', field, fn, this.fieldSettingsMap);\n\t\t\t},\n\t\t};\n\t};\n\n\tcreateComplexityMiddleware<Result, Root, Context, Args>(): NativeMiddleware<\n\t\tResult,\n\t\tRoot,\n\t\tContext,\n\t\tArgs\n\t> {\n\t\treturn (next) => async (root, args, ctx, info) => {\n\t\t\tloadComplexityStore(ctx as unknown as Ctx, this.options.limit, defaultLimits);\n\n\t\t\tconst store = await getComplexityStore(ctx);\n\n\t\t\tconst limits = store.limits;\n\n\t\t\tconst results = store.cacheComplexity(() => {\n\t\t\t\treturn calculateComplexity(ctx as unknown as Ctx, info, this.fieldSettingsMap, {\n\t\t\t\t\tcomplexity: this.options.defaultComplexity,\n\t\t\t\t\tmultiplier: this.options.defaultListMultiplier,\n\t\t\t\t});\n\t\t\t});\n\n\t\t\tif (results.depth > limits.depth) {\n\t\t\t\tthrow this.options.complexityError(ComplexityErrorKind.Depth, limits.depth, results.depth);\n\t\t\t}\n\n\t\t\tif (results.breadth > limits.breadth) {\n\t\t\t\tthrow this.options.complexityError(\n\t\t\t\t\tComplexityErrorKind.Breadth,\n\t\t\t\t\tlimits.breadth,\n\t\t\t\t\tresults.breadth,\n\t\t\t\t);\n\t\t\t}\n\n\t\t\tif (results.complexity > limits.complexity) {\n\t\t\t\tthrow this.options.complexityError(\n\t\t\t\t\tComplexityErrorKind.Complexity,\n\t\t\t\t\tlimits.complexity,\n\t\t\t\t\tresults.complexity,\n\t\t\t\t);\n\t\t\t}\n\n\t\t\treturn next(root, args, ctx, info);\n\t\t};\n\t}\n\n\tbuild = (_module: ModuleBuilder, mapper: ResolverMapper) => {\n\t\tfor (const type of mapper.getTypes()) {\n\t\t\tif (type !== 'Query' && type !== 'Mutation' && type !== 'Subscription') {\n\t\t\t\tcontinue;\n\t\t\t}\n\n\t\t\tif (type !== 'Subscription') {\n\t\t\t\tfor (const field of mapper.getTypeFields(type)) {\n\t\t\t\t\tmapper.addMiddleware(type, field, this.createComplexityMiddleware());\n\t\t\t\t}\n\t\t\t\tcontinue;\n\t\t\t}\n\n\t\t\tfor (const field of mapper.getTypeFields(type)) {\n\t\t\t\tmapper.addMiddleware(type, `${field}.subscribe`, this.createComplexityMiddleware());\n\t\t\t}\n\t\t}\n\t};\n}\n","import { ValidationError } from '@baeta/errors';\nimport {\n\ttype FieldNode,\n\ttype FragmentDefinitionNode,\n\ttype GraphQLNamedType,\n\ttype GraphQLResolveInfo,\n\ttype InlineFragmentNode,\n\tKind,\n\ttype SelectionNode,\n\ttype SelectionSetNode,\n\tgetNamedType,\n\tisInterfaceType,\n\tisObjectType,\n\tisOutputType,\n} from 'graphql';\nimport { isListOrNullableList } from '../utils/graphlq.ts';\nimport { capitalize } from '../utils/string.ts';\nimport { type FieldSettingsMap, getFieldComplexitySettings } from './field-settings.ts';\n\ninterface ComplexityDefaults {\n\tcomplexity: number;\n\tmultiplier: number;\n}\n\nexport function calculateComplexity<Context>(\n\tctx: Context,\n\tinfo: GraphQLResolveInfo,\n\tfieldSettingsMap: FieldSettingsMap,\n\tdefaults: ComplexityDefaults,\n) {\n\tconst operationName = capitalize(info.operation.operation);\n\tconst operationType = info.schema.getType(operationName);\n\n\tif (!operationType || !isOutputType(operationType)) {\n\t\tthrow new ValidationError(`Unsupported operation ${operationName}`);\n\t}\n\n\treturn complexityFromSelectionSet(\n\t\tctx,\n\t\tinfo,\n\t\toperationType,\n\t\tinfo.operation.selectionSet,\n\t\tfieldSettingsMap,\n\t\tdefaults,\n\t);\n}\n\nfunction complexityFromSelectionSet<Context>(\n\tctx: Context,\n\tinfo: GraphQLResolveInfo,\n\ttype: GraphQLNamedType,\n\tselectionSet: SelectionSetNode,\n\tfieldSettingsMap: FieldSettingsMap,\n\tdefaults: ComplexityDefaults,\n) {\n\tconst result = {\n\t\tdepth: 0,\n\t\tbreadth: 0,\n\t\tcomplexity: 0,\n\t};\n\n\tfor (const selection of selectionSet.selections) {\n\t\tconst selectionResult = complexityFromSelection(\n\t\t\tctx,\n\t\t\tinfo,\n\t\t\ttype,\n\t\t\tselection,\n\t\t\tfieldSettingsMap,\n\t\t\tdefaults,\n\t\t);\n\t\tresult.complexity += selectionResult.complexity;\n\t\tresult.breadth += selectionResult.breadth;\n\t\tresult.depth = Math.max(result.depth, selectionResult.depth);\n\t}\n\n\treturn result;\n}\n\nfunction complexityFromSelection<Context>(\n\tctx: Context,\n\tinfo: GraphQLResolveInfo,\n\ttype: GraphQLNamedType,\n\tselection: SelectionNode,\n\tfieldSettingsMap: FieldSettingsMap,\n\tdefaults: ComplexityDefaults,\n) {\n\tif (selection.kind === Kind.FIELD) {\n\t\treturn complexityFromField(ctx, info, type, selection, fieldSettingsMap, defaults);\n\t}\n\n\tif (selection.kind === Kind.FRAGMENT_SPREAD) {\n\t\tconst fragment = info.fragments[selection.name.value];\n\n\t\tif (!fragment) {\n\t\t\tthrow new ValidationError(`Fragment ${selection.name.value} not found`);\n\t\t}\n\n\t\treturn complexityFromFragment(ctx, info, type, fragment, fieldSettingsMap, defaults);\n\t}\n\n\treturn complexityFromFragment(ctx, info, type, selection, fieldSettingsMap, defaults);\n}\n\nfunction complexityFromFragment<Context>(\n\tctx: Context,\n\tinfo: GraphQLResolveInfo,\n\ttype: GraphQLNamedType,\n\tfragment: FragmentDefinitionNode | InlineFragmentNode,\n\tfieldSettingsMap: FieldSettingsMap,\n\tdefaults: ComplexityDefaults,\n) {\n\tconst fragmentType = fragment.typeCondition\n\t\t? info.schema.getType(fragment.typeCondition.name.value)\n\t\t: type;\n\n\tif (!isOutputType(fragmentType)) {\n\t\tthrow new ValidationError(`Unsupported fragment type ${fragmentType}`);\n\t}\n\n\tif (!fragmentType) {\n\t\tthrow new ValidationError(`Fragment type ${fragmentType} not found`);\n\t}\n\n\treturn complexityFromSelectionSet(\n\t\tctx,\n\t\tinfo,\n\t\tfragmentType,\n\t\tfragment.selectionSet,\n\t\tfieldSettingsMap,\n\t\tdefaults,\n\t);\n}\n\nfunction complexityFromField<Context>(\n\tctx: Context,\n\tinfo: GraphQLResolveInfo,\n\ttype: GraphQLNamedType,\n\tselection: FieldNode,\n\tfieldSettingsMap: FieldSettingsMap,\n\tdefaults: ComplexityDefaults,\n) {\n\tconst fieldName = selection.name.value;\n\n\tconst field =\n\t\tisObjectType(type) || isInterfaceType(type) ? type.getFields()[fieldName] : undefined;\n\n\tif (!field && !fieldName.startsWith('__')) {\n\t\tthrow new ValidationError(`Field ${fieldName} not found on type ${type.name}`);\n\t}\n\n\tif (!field) {\n\t\treturn {\n\t\t\tdepth: 1,\n\t\t\tbreadth: 1,\n\t\t\tcomplexity: 1,\n\t\t};\n\t}\n\n\tconst fieldComplexitySettings = getFieldComplexitySettings(\n\t\tctx,\n\t\tinfo,\n\t\ttype,\n\t\tfield,\n\t\tselection,\n\t\tfieldName,\n\t\tfieldSettingsMap,\n\t);\n\n\tif (fieldComplexitySettings === false) {\n\t\treturn {\n\t\t\tdepth: 0,\n\t\t\tbreadth: 0,\n\t\t\tcomplexity: 0,\n\t\t};\n\t}\n\n\tlet depth = 1;\n\tlet breadth = 1;\n\tlet complexity = 0;\n\n\tconst listMultiplier = fieldComplexitySettings?.multiplier ?? defaults.multiplier;\n\tconst multiplier = field && isListOrNullableList(field.type) ? listMultiplier : 1;\n\n\tcomplexity += (fieldComplexitySettings?.complexity ?? defaults.complexity) * multiplier;\n\n\tif (!field || !selection.selectionSet) {\n\t\treturn {\n\t\t\tdepth,\n\t\t\tbreadth,\n\t\t\tcomplexity,\n\t\t};\n\t}\n\n\tconst subSelection = complexityFromSelectionSet(\n\t\tctx,\n\t\tinfo,\n\t\tgetNamedType(field.type),\n\t\tselection.selectionSet,\n\t\tfieldSettingsMap,\n\t\tdefaults,\n\t);\n\n\tdepth += subSelection.depth;\n\tbreadth += subSelection.breadth;\n\tcomplexity += subSelection.complexity * Math.max(multiplier, 1);\n\n\treturn {\n\t\tdepth,\n\t\tbreadth,\n\t\tcomplexity,\n\t};\n}\n","import { GraphQLList, GraphQLNonNull, type GraphQLOutputType } from 'graphql';\n\nexport function isListOrNullableList(type: GraphQLOutputType): boolean {\n\tif (type instanceof GraphQLList) {\n\t\treturn true;\n\t}\n\n\tif (type instanceof GraphQLNonNull) {\n\t\treturn isListOrNullableList(type.ofType);\n\t}\n\n\treturn false;\n}\n","export function capitalize(str: string) {\n\treturn str.charAt(0).toUpperCase() + str.slice(1);\n}\n","import {\n\ttype FieldNode,\n\ttype GraphQLField,\n\ttype GraphQLNamedType,\n\ttype GraphQLResolveInfo,\n\tgetArgumentValues,\n} from 'graphql';\n\nexport type FieldSettingsMap = Record<\n\tstring,\n\t// biome-ignore lint/suspicious/noExplicitAny: Match any type from the original code\n\tRecord<string, GetFieldSettings<any, any> | undefined> | undefined\n>;\n\nexport type FieldSettings = {\n\tcomplexity?: number;\n\tmultiplier?: number;\n};\n\nexport type GetFieldSettingsArgs<Context, Args> = {\n\targs: Args;\n\tctx: Context;\n};\n\nexport type GetFieldSettings<Context, Args> = (\n\tparams: GetFieldSettingsArgs<Context, Args>,\n) => FieldSettings | false;\n\nexport function getFieldComplexitySettings<Context>(\n\tctx: Context,\n\tinfo: GraphQLResolveInfo,\n\ttype: GraphQLNamedType,\n\tfield: GraphQLField<unknown, unknown>,\n\tselection: FieldNode,\n\tfieldName: string,\n\tfieldSettingsMap: FieldSettingsMap,\n) {\n\tconst args = getArgumentValues(field, selection, info.variableValues);\n\tconst customComplexity = fieldSettingsMap[type.name]?.[fieldName];\n\n\tif (customComplexity) {\n\t\treturn customComplexity({ args, ctx });\n\t}\n\n\tconst wildCardComplexity = fieldSettingsMap[type.name]?.['*'];\n\n\tif (wildCardComplexity) {\n\t\treturn wildCardComplexity({ args, ctx });\n\t}\n}\n\nexport function registerFieldSettingsSetter<Context, Args>(\n\ttype: string,\n\tfield: string,\n\tfn: GetFieldSettings<Context, Args>,\n\tmap: FieldSettingsMap,\n) {\n\tmap[type] ??= {};\n\tmap[type][field] = fn;\n}\n","import { ValidationError } from '@baeta/errors';\nimport type { GraphQLError } from 'graphql';\n\nexport enum ComplexityErrorKind {\n\tDepth = 'DepthExceeded',\n\tBreadth = 'BreadthExceeded',\n\tComplexity = 'ComplexityExceeded',\n}\n\nexport type GetComplexityError = (\n\tkind: ComplexityErrorKind,\n\tlimits: number,\n\tresults: number,\n) => GraphQLError;\n\nexport function getDefaultComplexityError(\n\tkind: ComplexityErrorKind,\n\tlimit: number,\n\tresult: number,\n): GraphQLError {\n\tif (kind === ComplexityErrorKind.Depth) {\n\t\treturn new ValidationError(`Depth limit of ${limit} exceeded, got: ${result}`);\n\t}\n\n\tif (kind === ComplexityErrorKind.Breadth) {\n\t\treturn new ValidationError(`Breadth limit of ${limit} exceeded, got: ${result}`);\n\t}\n\n\tif (kind === ComplexityErrorKind.Complexity) {\n\t\treturn new ValidationError(`Complexity limit of ${limit} exceeded, got: ${result}`);\n\t}\n\n\tthrow new Error('Unknown complexity error kind');\n}\n","import { type GetComplexityError, getDefaultComplexityError } from './complexity-errors.ts';\nimport type { GetComplexityLimit } from './complexity-limits.ts';\n\nexport interface ComplexityExtensionOptions<Context> {\n\tlimit?: GetComplexityLimit<Context>;\n\tdefaultComplexity?: number;\n\tdefaultListMultiplier?: number;\n\tcomplexityError?: GetComplexityError;\n}\n\nexport const defaultLimits = {\n\tdepth: 10,\n\tbreadth: 50,\n\tcomplexity: 1000,\n};\n\nexport type NormalizedOptions<Context> = Required<ComplexityExtensionOptions<Context>>;\n\nexport function normalizeOptions<Context>(\n\toptions: ComplexityExtensionOptions<Context>,\n): NormalizedOptions<Context> {\n\treturn {\n\t\tlimit: options.limit ?? defaultLimits,\n\t\tdefaultComplexity: options.defaultComplexity ?? 1,\n\t\tdefaultListMultiplier: options.defaultListMultiplier ?? 10,\n\t\tcomplexityError: options.complexityError ?? getDefaultComplexityError,\n\t};\n}\n","import { createContextStore } from '@baeta/core';\nimport type { ComplexityLimit } from './complexity-limits.ts';\n\nexport interface ComplexityStore {\n\tlimits: Required<ComplexityLimit>;\n\tcacheComplexity: (fn: () => Required<ComplexityLimit>) => Required<ComplexityLimit>;\n}\n\nexport const complexityStoreKey = Symbol('complexity-extension');\n\nexport const [getComplexityStore, setComplexityStoreLoader] =\n\tcreateContextStore<ComplexityStore>(complexityStoreKey);\n","import type { ComplexityLimit } from './complexity-limits.ts';\nimport { setComplexityStoreLoader } from './store.ts';\n\nexport function loadComplexityStore<T>(\n\tctx: T,\n\tgetLimits: ComplexityLimit | ((ctx: T) => ComplexityLimit | Promise<ComplexityLimit>) | undefined,\n\tdefaultLimits: Required<ComplexityLimit>,\n) {\n\tsetComplexityStoreLoader(ctx, async () => {\n\t\tconst limits = typeof getLimits === 'function' ? await getLimits(ctx) : getLimits;\n\n\t\tlet cache: Required<ComplexityLimit> | undefined = undefined;\n\n\t\tconst cacheComplexity = (fn: () => Required<ComplexityLimit>) => {\n\t\t\tif (cache) {\n\t\t\t\treturn cache;\n\t\t\t}\n\n\t\t\tcache = fn();\n\t\t\treturn cache;\n\t\t};\n\n\t\treturn {\n\t\t\tlimits: {\n\t\t\t\tdepth: limits?.depth ?? defaultLimits.depth,\n\t\t\t\tbreadth: limits?.breadth ?? defaultLimits.breadth,\n\t\t\t\tcomplexity: limits?.complexity ?? defaultLimits.complexity,\n\t\t\t},\n\t\t\tcacheComplexity,\n\t\t};\n\t});\n}\n","import './lib/global-types.ts';\n\nimport { ComplexityExtension } from './lib/complexity-extension.ts';\nimport type { ComplexityExtensionOptions } from './lib/complexity-options.ts';\n\nexport { ComplexityErrorKind, type GetComplexityError } from './lib/complexity-errors.ts';\nexport type { ComplexityLimit, GetComplexityLimit } from './lib/complexity-limits.ts';\nexport type { ComplexityExtensionOptions } from './lib/complexity-options.ts';\n\nexport function complexityExtension<Ctx>(options?: ComplexityExtensionOptions<Ctx>) {\n\treturn () => new ComplexityExtension(options);\n}\n"]}

package/dist/index.d.cts

@@ -0,0 +1,62 @@
+import { Extension, ModuleBuilder, NativeMiddleware, ResolverMapper } from '@baeta/core/sdk';
+import { GraphQLError } from 'graphql';
+
+type FieldSettings = {
+    complexity?: number;
+    multiplier?: number;
+};
+type GetFieldSettingsArgs<Context, Args> = {
+    args: Args;
+    ctx: Context;
+};
+type GetFieldSettings<Context, Args> = (params: GetFieldSettingsArgs<Context, Args>) => FieldSettings | false;
+
+declare global {
+    export namespace BaetaExtensions {
+        interface ResolverExtensions<Result, Root, Context, Args> {
+            $complexity: (fn: GetFieldSettings<Context, Args>) => void;
+        }
+        interface TypeExtensions<Root, Context> {
+            $complexity: (fn: GetFieldSettings<Context, unknown>) => void;
+        }
+        interface SubscriptionExtensions<Root, Context, Args> {
+            $complexity: (fn: GetFieldSettings<Context, Args>) => void;
+        }
+    }
+}
+
+declare enum ComplexityErrorKind {
+    Depth = "DepthExceeded",
+    Breadth = "BreadthExceeded",
+    Complexity = "ComplexityExceeded"
+}
+type GetComplexityError = (kind: ComplexityErrorKind, limits: number, results: number) => GraphQLError;
+
+interface ComplexityLimit {
+    depth?: number;
+    breadth?: number;
+    complexity?: number;
+}
+type GetComplexityLimit<Context> = ComplexityLimit | ((ctx: Context) => ComplexityLimit | Promise<ComplexityLimit>);
+
+interface ComplexityExtensionOptions<Context> {
+    limit?: GetComplexityLimit<Context>;
+    defaultComplexity?: number;
+    defaultListMultiplier?: number;
+    complexityError?: GetComplexityError;
+}
+
+declare class ComplexityExtension<Ctx> extends Extension {
+    private readonly options;
+    constructor(options?: ComplexityExtensionOptions<Ctx>);
+    private fieldSettingsMap;
+    getTypeExtensions: <Root, Context>(_module: ModuleBuilder, type: string) => BaetaExtensions.TypeExtensions<Root, Context>;
+    getResolverExtensions: <Result, Root, Context, Args>(_module: ModuleBuilder, type: string, field: string) => BaetaExtensions.ResolverExtensions<Result, Root, Context, Args>;
+    getSubscriptionExtensions: <Root, Context, Args>(_module: ModuleBuilder, field: string) => BaetaExtensions.SubscriptionExtensions<Root, Context, Args>;
+    createComplexityMiddleware<Result, Root, Context, Args>(): NativeMiddleware<Result, Root, Context, Args>;
+    build: (_module: ModuleBuilder, mapper: ResolverMapper) => void;
+}
+
+declare function complexityExtension<Ctx>(options?: ComplexityExtensionOptions<Ctx>): () => ComplexityExtension<Ctx>;
+
+export { ComplexityErrorKind, type ComplexityExtensionOptions, type ComplexityLimit, type GetComplexityError, type GetComplexityLimit, complexityExtension };

package/dist/index.d.ts

@@ -0,0 +1,62 @@
+import { Extension, ModuleBuilder, NativeMiddleware, ResolverMapper } from '@baeta/core/sdk';
+import { GraphQLError } from 'graphql';
+
+type FieldSettings = {
+    complexity?: number;
+    multiplier?: number;
+};
+type GetFieldSettingsArgs<Context, Args> = {
+    args: Args;
+    ctx: Context;
+};
+type GetFieldSettings<Context, Args> = (params: GetFieldSettingsArgs<Context, Args>) => FieldSettings | false;
+
+declare global {
+    export namespace BaetaExtensions {
+        interface ResolverExtensions<Result, Root, Context, Args> {
+            $complexity: (fn: GetFieldSettings<Context, Args>) => void;
+        }
+        interface TypeExtensions<Root, Context> {
+            $complexity: (fn: GetFieldSettings<Context, unknown>) => void;
+        }
+        interface SubscriptionExtensions<Root, Context, Args> {
+            $complexity: (fn: GetFieldSettings<Context, Args>) => void;
+        }
+    }
+}
+
+declare enum ComplexityErrorKind {
+    Depth = "DepthExceeded",
+    Breadth = "BreadthExceeded",
+    Complexity = "ComplexityExceeded"
+}
+type GetComplexityError = (kind: ComplexityErrorKind, limits: number, results: number) => GraphQLError;
+
+interface ComplexityLimit {
+    depth?: number;
+    breadth?: number;
+    complexity?: number;
+}
+type GetComplexityLimit<Context> = ComplexityLimit | ((ctx: Context) => ComplexityLimit | Promise<ComplexityLimit>);
+
+interface ComplexityExtensionOptions<Context> {
+    limit?: GetComplexityLimit<Context>;
+    defaultComplexity?: number;
+    defaultListMultiplier?: number;
+    complexityError?: GetComplexityError;
+}
+
+declare class ComplexityExtension<Ctx> extends Extension {
+    private readonly options;
+    constructor(options?: ComplexityExtensionOptions<Ctx>);
+    private fieldSettingsMap;
+    getTypeExtensions: <Root, Context>(_module: ModuleBuilder, type: string) => BaetaExtensions.TypeExtensions<Root, Context>;
+    getResolverExtensions: <Result, Root, Context, Args>(_module: ModuleBuilder, type: string, field: string) => BaetaExtensions.ResolverExtensions<Result, Root, Context, Args>;
+    getSubscriptionExtensions: <Root, Context, Args>(_module: ModuleBuilder, field: string) => BaetaExtensions.SubscriptionExtensions<Root, Context, Args>;
+    createComplexityMiddleware<Result, Root, Context, Args>(): NativeMiddleware<Result, Root, Context, Args>;
+    build: (_module: ModuleBuilder, mapper: ResolverMapper) => void;
+}
+
+declare function complexityExtension<Ctx>(options?: ComplexityExtensionOptions<Ctx>): () => ComplexityExtension<Ctx>;
+
+export { ComplexityErrorKind, type ComplexityExtensionOptions, type ComplexityLimit, type GetComplexityError, type GetComplexityLimit, complexityExtension };

package/dist/index.js

@@ -0,0 +1,330 @@
+// lib/complexity-extension.ts
+import {
+  Extension
+} from "@baeta/core/sdk";
+
+// lib/complexity-calculator.ts
+import { ValidationError } from "@baeta/errors";
+import {
+  Kind,
+  getNamedType,
+  isInterfaceType,
+  isObjectType,
+  isOutputType
+} from "graphql";
+
+// utils/graphlq.ts
+import { GraphQLList, GraphQLNonNull } from "graphql";
+function isListOrNullableList(type) {
+  if (type instanceof GraphQLList) {
+    return true;
+  }
+  if (type instanceof GraphQLNonNull) {
+    return isListOrNullableList(type.ofType);
+  }
+  return false;
+}
+
+// utils/string.ts
+function capitalize(str) {
+  return str.charAt(0).toUpperCase() + str.slice(1);
+}
+
+// lib/field-settings.ts
+import {
+  getArgumentValues
+} from "graphql";
+function getFieldComplexitySettings(ctx, info, type, field, selection, fieldName, fieldSettingsMap) {
+  const args = getArgumentValues(field, selection, info.variableValues);
+  const customComplexity = fieldSettingsMap[type.name]?.[fieldName];
+  if (customComplexity) {
+    return customComplexity({ args, ctx });
+  }
+  const wildCardComplexity = fieldSettingsMap[type.name]?.["*"];
+  if (wildCardComplexity) {
+    return wildCardComplexity({ args, ctx });
+  }
+}
+function registerFieldSettingsSetter(type, field, fn, map) {
+  map[type] ??= {};
+  map[type][field] = fn;
+}
+
+// lib/complexity-calculator.ts
+function calculateComplexity(ctx, info, fieldSettingsMap, defaults) {
+  const operationName = capitalize(info.operation.operation);
+  const operationType = info.schema.getType(operationName);
+  if (!operationType || !isOutputType(operationType)) {
+    throw new ValidationError(`Unsupported operation ${operationName}`);
+  }
+  return complexityFromSelectionSet(
+    ctx,
+    info,
+    operationType,
+    info.operation.selectionSet,
+    fieldSettingsMap,
+    defaults
+  );
+}
+function complexityFromSelectionSet(ctx, info, type, selectionSet, fieldSettingsMap, defaults) {
+  const result = {
+    depth: 0,
+    breadth: 0,
+    complexity: 0
+  };
+  for (const selection of selectionSet.selections) {
+    const selectionResult = complexityFromSelection(
+      ctx,
+      info,
+      type,
+      selection,
+      fieldSettingsMap,
+      defaults
+    );
+    result.complexity += selectionResult.complexity;
+    result.breadth += selectionResult.breadth;
+    result.depth = Math.max(result.depth, selectionResult.depth);
+  }
+  return result;
+}
+function complexityFromSelection(ctx, info, type, selection, fieldSettingsMap, defaults) {
+  if (selection.kind === Kind.FIELD) {
+    return complexityFromField(ctx, info, type, selection, fieldSettingsMap, defaults);
+  }
+  if (selection.kind === Kind.FRAGMENT_SPREAD) {
+    const fragment = info.fragments[selection.name.value];
+    if (!fragment) {
+      throw new ValidationError(`Fragment ${selection.name.value} not found`);
+    }
+    return complexityFromFragment(ctx, info, type, fragment, fieldSettingsMap, defaults);
+  }
+  return complexityFromFragment(ctx, info, type, selection, fieldSettingsMap, defaults);
+}
+function complexityFromFragment(ctx, info, type, fragment, fieldSettingsMap, defaults) {
+  const fragmentType = fragment.typeCondition ? info.schema.getType(fragment.typeCondition.name.value) : type;
+  if (!isOutputType(fragmentType)) {
+    throw new ValidationError(`Unsupported fragment type ${fragmentType}`);
+  }
+  if (!fragmentType) {
+    throw new ValidationError(`Fragment type ${fragmentType} not found`);
+  }
+  return complexityFromSelectionSet(
+    ctx,
+    info,
+    fragmentType,
+    fragment.selectionSet,
+    fieldSettingsMap,
+    defaults
+  );
+}
+function complexityFromField(ctx, info, type, selection, fieldSettingsMap, defaults) {
+  const fieldName = selection.name.value;
+  const field = isObjectType(type) || isInterfaceType(type) ? type.getFields()[fieldName] : void 0;
+  if (!field && !fieldName.startsWith("__")) {
+    throw new ValidationError(`Field ${fieldName} not found on type ${type.name}`);
+  }
+  if (!field) {
+    return {
+      depth: 1,
+      breadth: 1,
+      complexity: 1
+    };
+  }
+  const fieldComplexitySettings = getFieldComplexitySettings(
+    ctx,
+    info,
+    type,
+    field,
+    selection,
+    fieldName,
+    fieldSettingsMap
+  );
+  if (fieldComplexitySettings === false) {
+    return {
+      depth: 0,
+      breadth: 0,
+      complexity: 0
+    };
+  }
+  let depth = 1;
+  let breadth = 1;
+  let complexity = 0;
+  const listMultiplier = fieldComplexitySettings?.multiplier ?? defaults.multiplier;
+  const multiplier = field && isListOrNullableList(field.type) ? listMultiplier : 1;
+  complexity += (fieldComplexitySettings?.complexity ?? defaults.complexity) * multiplier;
+  if (!field || !selection.selectionSet) {
+    return {
+      depth,
+      breadth,
+      complexity
+    };
+  }
+  const subSelection = complexityFromSelectionSet(
+    ctx,
+    info,
+    getNamedType(field.type),
+    selection.selectionSet,
+    fieldSettingsMap,
+    defaults
+  );
+  depth += subSelection.depth;
+  breadth += subSelection.breadth;
+  complexity += subSelection.complexity * Math.max(multiplier, 1);
+  return {
+    depth,
+    breadth,
+    complexity
+  };
+}
+
+// lib/complexity-errors.ts
+import { ValidationError as ValidationError2 } from "@baeta/errors";
+var ComplexityErrorKind = /* @__PURE__ */ ((ComplexityErrorKind2) => {
+  ComplexityErrorKind2["Depth"] = "DepthExceeded";
+  ComplexityErrorKind2["Breadth"] = "BreadthExceeded";
+  ComplexityErrorKind2["Complexity"] = "ComplexityExceeded";
+  return ComplexityErrorKind2;
+})(ComplexityErrorKind || {});
+function getDefaultComplexityError(kind, limit, result) {
+  if (kind === "DepthExceeded" /* Depth */) {
+    return new ValidationError2(`Depth limit of ${limit} exceeded, got: ${result}`);
+  }
+  if (kind === "BreadthExceeded" /* Breadth */) {
+    return new ValidationError2(`Breadth limit of ${limit} exceeded, got: ${result}`);
+  }
+  if (kind === "ComplexityExceeded" /* Complexity */) {
+    return new ValidationError2(`Complexity limit of ${limit} exceeded, got: ${result}`);
+  }
+  throw new Error("Unknown complexity error kind");
+}
+
+// lib/complexity-options.ts
+var defaultLimits = {
+  depth: 10,
+  breadth: 50,
+  complexity: 1e3
+};
+function normalizeOptions(options) {
+  return {
+    limit: options.limit ?? defaultLimits,
+    defaultComplexity: options.defaultComplexity ?? 1,
+    defaultListMultiplier: options.defaultListMultiplier ?? 10,
+    complexityError: options.complexityError ?? getDefaultComplexityError
+  };
+}
+
+// lib/store.ts
+import { createContextStore } from "@baeta/core";
+var complexityStoreKey = Symbol("complexity-extension");
+var [getComplexityStore, setComplexityStoreLoader] = createContextStore(complexityStoreKey);
+
+// lib/store-loader.ts
+function loadComplexityStore(ctx, getLimits, defaultLimits2) {
+  setComplexityStoreLoader(ctx, async () => {
+    const limits = typeof getLimits === "function" ? await getLimits(ctx) : getLimits;
+    let cache = void 0;
+    const cacheComplexity = (fn) => {
+      if (cache) {
+        return cache;
+      }
+      cache = fn();
+      return cache;
+    };
+    return {
+      limits: {
+        depth: limits?.depth ?? defaultLimits2.depth,
+        breadth: limits?.breadth ?? defaultLimits2.breadth,
+        complexity: limits?.complexity ?? defaultLimits2.complexity
+      },
+      cacheComplexity
+    };
+  });
+}
+
+// lib/complexity-extension.ts
+var ComplexityExtension = class extends Extension {
+  options;
+  constructor(options = {}) {
+    super();
+    this.options = normalizeOptions(options);
+  }
+  fieldSettingsMap = {};
+  getTypeExtensions = (_module, type) => {
+    return {
+      $complexity: (fn) => {
+        registerFieldSettingsSetter(type, "*", fn, this.fieldSettingsMap);
+      }
+    };
+  };
+  getResolverExtensions = (_module, type, field) => {
+    return {
+      $complexity: (fn) => {
+        registerFieldSettingsSetter(type, field, fn, this.fieldSettingsMap);
+      }
+    };
+  };
+  getSubscriptionExtensions = (_module, field) => {
+    return {
+      $complexity: (fn) => {
+        registerFieldSettingsSetter("Subscription", field, fn, this.fieldSettingsMap);
+      }
+    };
+  };
+  createComplexityMiddleware() {
+    return (next) => async (root, args, ctx, info) => {
+      loadComplexityStore(ctx, this.options.limit, defaultLimits);
+      const store = await getComplexityStore(ctx);
+      const limits = store.limits;
+      const results = store.cacheComplexity(() => {
+        return calculateComplexity(ctx, info, this.fieldSettingsMap, {
+          complexity: this.options.defaultComplexity,
+          multiplier: this.options.defaultListMultiplier
+        });
+      });
+      if (results.depth > limits.depth) {
+        throw this.options.complexityError("DepthExceeded" /* Depth */, limits.depth, results.depth);
+      }
+      if (results.breadth > limits.breadth) {
+        throw this.options.complexityError(
+          "BreadthExceeded" /* Breadth */,
+          limits.breadth,
+          results.breadth
+        );
+      }
+      if (results.complexity > limits.complexity) {
+        throw this.options.complexityError(
+          "ComplexityExceeded" /* Complexity */,
+          limits.complexity,
+          results.complexity
+        );
+      }
+      return next(root, args, ctx, info);
+    };
+  }
+  build = (_module, mapper) => {
+    for (const type of mapper.getTypes()) {
+      if (type !== "Query" && type !== "Mutation" && type !== "Subscription") {
+        continue;
+      }
+      if (type !== "Subscription") {
+        for (const field of mapper.getTypeFields(type)) {
+          mapper.addMiddleware(type, field, this.createComplexityMiddleware());
+        }
+        continue;
+      }
+      for (const field of mapper.getTypeFields(type)) {
+        mapper.addMiddleware(type, `${field}.subscribe`, this.createComplexityMiddleware());
+      }
+    }
+  };
+};
+
+// index.ts
+function complexityExtension(options) {
+  return () => new ComplexityExtension(options);
+}
+export {
+  ComplexityErrorKind,
+  complexityExtension
+};
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../lib/complexity-extension.ts","../lib/complexity-calculator.ts","../utils/graphlq.ts","../utils/string.ts","../lib/field-settings.ts","../lib/complexity-errors.ts","../lib/complexity-options.ts","../lib/store.ts","../lib/store-loader.ts","../index.ts"],"sourcesContent":["import {\n\tExtension,\n\ttype ModuleBuilder,\n\ttype NativeMiddleware,\n\ttype ResolverMapper,\n} from '@baeta/core/sdk';\nimport { calculateComplexity } from './complexity-calculator.ts';\nimport { ComplexityErrorKind } from './complexity-errors.ts';\nimport {\n\ttype ComplexityExtensionOptions,\n\tdefaultLimits,\n\tnormalizeOptions,\n} from './complexity-options.ts';\nimport { type FieldSettingsMap, registerFieldSettingsSetter } from './field-settings.ts';\nimport { loadComplexityStore } from './store-loader.ts';\nimport { getComplexityStore } from './store.ts';\n\nexport class ComplexityExtension<Ctx> extends Extension {\n\tprivate readonly options: Required<ComplexityExtensionOptions<Ctx>>;\n\n\tconstructor(options: ComplexityExtensionOptions<Ctx> = {}) {\n\t\tsuper();\n\t\tthis.options = normalizeOptions(options);\n\t}\n\n\tprivate fieldSettingsMap: FieldSettingsMap = {};\n\n\tgetTypeExtensions = <Root, Context>(\n\t\t_module: ModuleBuilder,\n\t\ttype: string,\n\t): BaetaExtensions.TypeExtensions<Root, Context> => {\n\t\treturn {\n\t\t\t$complexity: (fn) => {\n\t\t\t\tregisterFieldSettingsSetter(type, '*', fn, this.fieldSettingsMap);\n\t\t\t},\n\t\t};\n\t};\n\n\tgetResolverExtensions = <Result, Root, Context, Args>(\n\t\t_module: ModuleBuilder,\n\t\ttype: string,\n\t\tfield: string,\n\t): BaetaExtensions.ResolverExtensions<Result, Root, Context, Args> => {\n\t\treturn {\n\t\t\t$complexity: (fn) => {\n\t\t\t\tregisterFieldSettingsSetter(type, field, fn, this.fieldSettingsMap);\n\t\t\t},\n\t\t};\n\t};\n\n\tgetSubscriptionExtensions = <Root, Context, Args>(\n\t\t_module: ModuleBuilder,\n\t\tfield: string,\n\t): BaetaExtensions.SubscriptionExtensions<Root, Context, Args> => {\n\t\treturn {\n\t\t\t$complexity: (fn) => {\n\t\t\t\tregisterFieldSettingsSetter('Subscription', field, fn, this.fieldSettingsMap);\n\t\t\t},\n\t\t};\n\t};\n\n\tcreateComplexityMiddleware<Result, Root, Context, Args>(): NativeMiddleware<\n\t\tResult,\n\t\tRoot,\n\t\tContext,\n\t\tArgs\n\t> {\n\t\treturn (next) => async (root, args, ctx, info) => {\n\t\t\tloadComplexityStore(ctx as unknown as Ctx, this.options.limit, defaultLimits);\n\n\t\t\tconst store = await getComplexityStore(ctx);\n\n\t\t\tconst limits = store.limits;\n\n\t\t\tconst results = store.cacheComplexity(() => {\n\t\t\t\treturn calculateComplexity(ctx as unknown as Ctx, info, this.fieldSettingsMap, {\n\t\t\t\t\tcomplexity: this.options.defaultComplexity,\n\t\t\t\t\tmultiplier: this.options.defaultListMultiplier,\n\t\t\t\t});\n\t\t\t});\n\n\t\t\tif (results.depth > limits.depth) {\n\t\t\t\tthrow this.options.complexityError(ComplexityErrorKind.Depth, limits.depth, results.depth);\n\t\t\t}\n\n\t\t\tif (results.breadth > limits.breadth) {\n\t\t\t\tthrow this.options.complexityError(\n\t\t\t\t\tComplexityErrorKind.Breadth,\n\t\t\t\t\tlimits.breadth,\n\t\t\t\t\tresults.breadth,\n\t\t\t\t);\n\t\t\t}\n\n\t\t\tif (results.complexity > limits.complexity) {\n\t\t\t\tthrow this.options.complexityError(\n\t\t\t\t\tComplexityErrorKind.Complexity,\n\t\t\t\t\tlimits.complexity,\n\t\t\t\t\tresults.complexity,\n\t\t\t\t);\n\t\t\t}\n\n\t\t\treturn next(root, args, ctx, info);\n\t\t};\n\t}\n\n\tbuild = (_module: ModuleBuilder, mapper: ResolverMapper) => {\n\t\tfor (const type of mapper.getTypes()) {\n\t\t\tif (type !== 'Query' && type !== 'Mutation' && type !== 'Subscription') {\n\t\t\t\tcontinue;\n\t\t\t}\n\n\t\t\tif (type !== 'Subscription') {\n\t\t\t\tfor (const field of mapper.getTypeFields(type)) {\n\t\t\t\t\tmapper.addMiddleware(type, field, this.createComplexityMiddleware());\n\t\t\t\t}\n\t\t\t\tcontinue;\n\t\t\t}\n\n\t\t\tfor (const field of mapper.getTypeFields(type)) {\n\t\t\t\tmapper.addMiddleware(type, `${field}.subscribe`, this.createComplexityMiddleware());\n\t\t\t}\n\t\t}\n\t};\n}\n","import { ValidationError } from '@baeta/errors';\nimport {\n\ttype FieldNode,\n\ttype FragmentDefinitionNode,\n\ttype GraphQLNamedType,\n\ttype GraphQLResolveInfo,\n\ttype InlineFragmentNode,\n\tKind,\n\ttype SelectionNode,\n\ttype SelectionSetNode,\n\tgetNamedType,\n\tisInterfaceType,\n\tisObjectType,\n\tisOutputType,\n} from 'graphql';\nimport { isListOrNullableList } from '../utils/graphlq.ts';\nimport { capitalize } from '../utils/string.ts';\nimport { type FieldSettingsMap, getFieldComplexitySettings } from './field-settings.ts';\n\ninterface ComplexityDefaults {\n\tcomplexity: number;\n\tmultiplier: number;\n}\n\nexport function calculateComplexity<Context>(\n\tctx: Context,\n\tinfo: GraphQLResolveInfo,\n\tfieldSettingsMap: FieldSettingsMap,\n\tdefaults: ComplexityDefaults,\n) {\n\tconst operationName = capitalize(info.operation.operation);\n\tconst operationType = info.schema.getType(operationName);\n\n\tif (!operationType || !isOutputType(operationType)) {\n\t\tthrow new ValidationError(`Unsupported operation ${operationName}`);\n\t}\n\n\treturn complexityFromSelectionSet(\n\t\tctx,\n\t\tinfo,\n\t\toperationType,\n\t\tinfo.operation.selectionSet,\n\t\tfieldSettingsMap,\n\t\tdefaults,\n\t);\n}\n\nfunction complexityFromSelectionSet<Context>(\n\tctx: Context,\n\tinfo: GraphQLResolveInfo,\n\ttype: GraphQLNamedType,\n\tselectionSet: SelectionSetNode,\n\tfieldSettingsMap: FieldSettingsMap,\n\tdefaults: ComplexityDefaults,\n) {\n\tconst result = {\n\t\tdepth: 0,\n\t\tbreadth: 0,\n\t\tcomplexity: 0,\n\t};\n\n\tfor (const selection of selectionSet.selections) {\n\t\tconst selectionResult = complexityFromSelection(\n\t\t\tctx,\n\t\t\tinfo,\n\t\t\ttype,\n\t\t\tselection,\n\t\t\tfieldSettingsMap,\n\t\t\tdefaults,\n\t\t);\n\t\tresult.complexity += selectionResult.complexity;\n\t\tresult.breadth += selectionResult.breadth;\n\t\tresult.depth = Math.max(result.depth, selectionResult.depth);\n\t}\n\n\treturn result;\n}\n\nfunction complexityFromSelection<Context>(\n\tctx: Context,\n\tinfo: GraphQLResolveInfo,\n\ttype: GraphQLNamedType,\n\tselection: SelectionNode,\n\tfieldSettingsMap: FieldSettingsMap,\n\tdefaults: ComplexityDefaults,\n) {\n\tif (selection.kind === Kind.FIELD) {\n\t\treturn complexityFromField(ctx, info, type, selection, fieldSettingsMap, defaults);\n\t}\n\n\tif (selection.kind === Kind.FRAGMENT_SPREAD) {\n\t\tconst fragment = info.fragments[selection.name.value];\n\n\t\tif (!fragment) {\n\t\t\tthrow new ValidationError(`Fragment ${selection.name.value} not found`);\n\t\t}\n\n\t\treturn complexityFromFragment(ctx, info, type, fragment, fieldSettingsMap, defaults);\n\t}\n\n\treturn complexityFromFragment(ctx, info, type, selection, fieldSettingsMap, defaults);\n}\n\nfunction complexityFromFragment<Context>(\n\tctx: Context,\n\tinfo: GraphQLResolveInfo,\n\ttype: GraphQLNamedType,\n\tfragment: FragmentDefinitionNode | InlineFragmentNode,\n\tfieldSettingsMap: FieldSettingsMap,\n\tdefaults: ComplexityDefaults,\n) {\n\tconst fragmentType = fragment.typeCondition\n\t\t? info.schema.getType(fragment.typeCondition.name.value)\n\t\t: type;\n\n\tif (!isOutputType(fragmentType)) {\n\t\tthrow new ValidationError(`Unsupported fragment type ${fragmentType}`);\n\t}\n\n\tif (!fragmentType) {\n\t\tthrow new ValidationError(`Fragment type ${fragmentType} not found`);\n\t}\n\n\treturn complexityFromSelectionSet(\n\t\tctx,\n\t\tinfo,\n\t\tfragmentType,\n\t\tfragment.selectionSet,\n\t\tfieldSettingsMap,\n\t\tdefaults,\n\t);\n}\n\nfunction complexityFromField<Context>(\n\tctx: Context,\n\tinfo: GraphQLResolveInfo,\n\ttype: GraphQLNamedType,\n\tselection: FieldNode,\n\tfieldSettingsMap: FieldSettingsMap,\n\tdefaults: ComplexityDefaults,\n) {\n\tconst fieldName = selection.name.value;\n\n\tconst field =\n\t\tisObjectType(type) || isInterfaceType(type) ? type.getFields()[fieldName] : undefined;\n\n\tif (!field && !fieldName.startsWith('__')) {\n\t\tthrow new ValidationError(`Field ${fieldName} not found on type ${type.name}`);\n\t}\n\n\tif (!field) {\n\t\treturn {\n\t\t\tdepth: 1,\n\t\t\tbreadth: 1,\n\t\t\tcomplexity: 1,\n\t\t};\n\t}\n\n\tconst fieldComplexitySettings = getFieldComplexitySettings(\n\t\tctx,\n\t\tinfo,\n\t\ttype,\n\t\tfield,\n\t\tselection,\n\t\tfieldName,\n\t\tfieldSettingsMap,\n\t);\n\n\tif (fieldComplexitySettings === false) {\n\t\treturn {\n\t\t\tdepth: 0,\n\t\t\tbreadth: 0,\n\t\t\tcomplexity: 0,\n\t\t};\n\t}\n\n\tlet depth = 1;\n\tlet breadth = 1;\n\tlet complexity = 0;\n\n\tconst listMultiplier = fieldComplexitySettings?.multiplier ?? defaults.multiplier;\n\tconst multiplier = field && isListOrNullableList(field.type) ? listMultiplier : 1;\n\n\tcomplexity += (fieldComplexitySettings?.complexity ?? defaults.complexity) * multiplier;\n\n\tif (!field || !selection.selectionSet) {\n\t\treturn {\n\t\t\tdepth,\n\t\t\tbreadth,\n\t\t\tcomplexity,\n\t\t};\n\t}\n\n\tconst subSelection = complexityFromSelectionSet(\n\t\tctx,\n\t\tinfo,\n\t\tgetNamedType(field.type),\n\t\tselection.selectionSet,\n\t\tfieldSettingsMap,\n\t\tdefaults,\n\t);\n\n\tdepth += subSelection.depth;\n\tbreadth += subSelection.breadth;\n\tcomplexity += subSelection.complexity * Math.max(multiplier, 1);\n\n\treturn {\n\t\tdepth,\n\t\tbreadth,\n\t\tcomplexity,\n\t};\n}\n","import { GraphQLList, GraphQLNonNull, type GraphQLOutputType } from 'graphql';\n\nexport function isListOrNullableList(type: GraphQLOutputType): boolean {\n\tif (type instanceof GraphQLList) {\n\t\treturn true;\n\t}\n\n\tif (type instanceof GraphQLNonNull) {\n\t\treturn isListOrNullableList(type.ofType);\n\t}\n\n\treturn false;\n}\n","export function capitalize(str: string) {\n\treturn str.charAt(0).toUpperCase() + str.slice(1);\n}\n","import {\n\ttype FieldNode,\n\ttype GraphQLField,\n\ttype GraphQLNamedType,\n\ttype GraphQLResolveInfo,\n\tgetArgumentValues,\n} from 'graphql';\n\nexport type FieldSettingsMap = Record<\n\tstring,\n\t// biome-ignore lint/suspicious/noExplicitAny: Match any type from the original code\n\tRecord<string, GetFieldSettings<any, any> | undefined> | undefined\n>;\n\nexport type FieldSettings = {\n\tcomplexity?: number;\n\tmultiplier?: number;\n};\n\nexport type GetFieldSettingsArgs<Context, Args> = {\n\targs: Args;\n\tctx: Context;\n};\n\nexport type GetFieldSettings<Context, Args> = (\n\tparams: GetFieldSettingsArgs<Context, Args>,\n) => FieldSettings | false;\n\nexport function getFieldComplexitySettings<Context>(\n\tctx: Context,\n\tinfo: GraphQLResolveInfo,\n\ttype: GraphQLNamedType,\n\tfield: GraphQLField<unknown, unknown>,\n\tselection: FieldNode,\n\tfieldName: string,\n\tfieldSettingsMap: FieldSettingsMap,\n) {\n\tconst args = getArgumentValues(field, selection, info.variableValues);\n\tconst customComplexity = fieldSettingsMap[type.name]?.[fieldName];\n\n\tif (customComplexity) {\n\t\treturn customComplexity({ args, ctx });\n\t}\n\n\tconst wildCardComplexity = fieldSettingsMap[type.name]?.['*'];\n\n\tif (wildCardComplexity) {\n\t\treturn wildCardComplexity({ args, ctx });\n\t}\n}\n\nexport function registerFieldSettingsSetter<Context, Args>(\n\ttype: string,\n\tfield: string,\n\tfn: GetFieldSettings<Context, Args>,\n\tmap: FieldSettingsMap,\n) {\n\tmap[type] ??= {};\n\tmap[type][field] = fn;\n}\n","import { ValidationError } from '@baeta/errors';\nimport type { GraphQLError } from 'graphql';\n\nexport enum ComplexityErrorKind {\n\tDepth = 'DepthExceeded',\n\tBreadth = 'BreadthExceeded',\n\tComplexity = 'ComplexityExceeded',\n}\n\nexport type GetComplexityError = (\n\tkind: ComplexityErrorKind,\n\tlimits: number,\n\tresults: number,\n) => GraphQLError;\n\nexport function getDefaultComplexityError(\n\tkind: ComplexityErrorKind,\n\tlimit: number,\n\tresult: number,\n): GraphQLError {\n\tif (kind === ComplexityErrorKind.Depth) {\n\t\treturn new ValidationError(`Depth limit of ${limit} exceeded, got: ${result}`);\n\t}\n\n\tif (kind === ComplexityErrorKind.Breadth) {\n\t\treturn new ValidationError(`Breadth limit of ${limit} exceeded, got: ${result}`);\n\t}\n\n\tif (kind === ComplexityErrorKind.Complexity) {\n\t\treturn new ValidationError(`Complexity limit of ${limit} exceeded, got: ${result}`);\n\t}\n\n\tthrow new Error('Unknown complexity error kind');\n}\n","import { type GetComplexityError, getDefaultComplexityError } from './complexity-errors.ts';\nimport type { GetComplexityLimit } from './complexity-limits.ts';\n\nexport interface ComplexityExtensionOptions<Context> {\n\tlimit?: GetComplexityLimit<Context>;\n\tdefaultComplexity?: number;\n\tdefaultListMultiplier?: number;\n\tcomplexityError?: GetComplexityError;\n}\n\nexport const defaultLimits = {\n\tdepth: 10,\n\tbreadth: 50,\n\tcomplexity: 1000,\n};\n\nexport type NormalizedOptions<Context> = Required<ComplexityExtensionOptions<Context>>;\n\nexport function normalizeOptions<Context>(\n\toptions: ComplexityExtensionOptions<Context>,\n): NormalizedOptions<Context> {\n\treturn {\n\t\tlimit: options.limit ?? defaultLimits,\n\t\tdefaultComplexity: options.defaultComplexity ?? 1,\n\t\tdefaultListMultiplier: options.defaultListMultiplier ?? 10,\n\t\tcomplexityError: options.complexityError ?? getDefaultComplexityError,\n\t};\n}\n","import { createContextStore } from '@baeta/core';\nimport type { ComplexityLimit } from './complexity-limits.ts';\n\nexport interface ComplexityStore {\n\tlimits: Required<ComplexityLimit>;\n\tcacheComplexity: (fn: () => Required<ComplexityLimit>) => Required<ComplexityLimit>;\n}\n\nexport const complexityStoreKey = Symbol('complexity-extension');\n\nexport const [getComplexityStore, setComplexityStoreLoader] =\n\tcreateContextStore<ComplexityStore>(complexityStoreKey);\n","import type { ComplexityLimit } from './complexity-limits.ts';\nimport { setComplexityStoreLoader } from './store.ts';\n\nexport function loadComplexityStore<T>(\n\tctx: T,\n\tgetLimits: ComplexityLimit | ((ctx: T) => ComplexityLimit | Promise<ComplexityLimit>) | undefined,\n\tdefaultLimits: Required<ComplexityLimit>,\n) {\n\tsetComplexityStoreLoader(ctx, async () => {\n\t\tconst limits = typeof getLimits === 'function' ? await getLimits(ctx) : getLimits;\n\n\t\tlet cache: Required<ComplexityLimit> | undefined = undefined;\n\n\t\tconst cacheComplexity = (fn: () => Required<ComplexityLimit>) => {\n\t\t\tif (cache) {\n\t\t\t\treturn cache;\n\t\t\t}\n\n\t\t\tcache = fn();\n\t\t\treturn cache;\n\t\t};\n\n\t\treturn {\n\t\t\tlimits: {\n\t\t\t\tdepth: limits?.depth ?? defaultLimits.depth,\n\t\t\t\tbreadth: limits?.breadth ?? defaultLimits.breadth,\n\t\t\t\tcomplexity: limits?.complexity ?? defaultLimits.complexity,\n\t\t\t},\n\t\t\tcacheComplexity,\n\t\t};\n\t});\n}\n","import './lib/global-types.ts';\n\nimport { ComplexityExtension } from './lib/complexity-extension.ts';\nimport type { ComplexityExtensionOptions } from './lib/complexity-options.ts';\n\nexport { ComplexityErrorKind, type GetComplexityError } from './lib/complexity-errors.ts';\nexport type { ComplexityLimit, GetComplexityLimit } from './lib/complexity-limits.ts';\nexport type { ComplexityExtensionOptions } from './lib/complexity-options.ts';\n\nexport function complexityExtension<Ctx>(options?: ComplexityExtensionOptions<Ctx>) {\n\treturn () => new ComplexityExtension(options);\n}\n"],"mappings":";AAAA;AAAA,EACC;AAAA,OAIM;;;ACLP,SAAS,uBAAuB;AAChC;AAAA,EAMC;AAAA,EAGA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACM;;;ACdP,SAAS,aAAa,sBAA8C;AAE7D,SAAS,qBAAqB,MAAkC;AACtE,MAAI,gBAAgB,aAAa;AAChC,WAAO;AAAA,EACR;AAEA,MAAI,gBAAgB,gBAAgB;AACnC,WAAO,qBAAqB,KAAK,MAAM;AAAA,EACxC;AAEA,SAAO;AACR;;;ACZO,SAAS,WAAW,KAAa;AACvC,SAAO,IAAI,OAAO,CAAC,EAAE,YAAY,IAAI,IAAI,MAAM,CAAC;AACjD;;;ACFA;AAAA,EAKC;AAAA,OACM;AAsBA,SAAS,2BACf,KACA,MACA,MACA,OACA,WACA,WACA,kBACC;AACD,QAAM,OAAO,kBAAkB,OAAO,WAAW,KAAK,cAAc;AACpE,QAAM,mBAAmB,iBAAiB,KAAK,IAAI,IAAI,SAAS;AAEhE,MAAI,kBAAkB;AACrB,WAAO,iBAAiB,EAAE,MAAM,IAAI,CAAC;AAAA,EACtC;AAEA,QAAM,qBAAqB,iBAAiB,KAAK,IAAI,IAAI,GAAG;AAE5D,MAAI,oBAAoB;AACvB,WAAO,mBAAmB,EAAE,MAAM,IAAI,CAAC;AAAA,EACxC;AACD;AAEO,SAAS,4BACf,MACA,OACA,IACA,KACC;AACD,MAAI,IAAI,MAAM,CAAC;AACf,MAAI,IAAI,EAAE,KAAK,IAAI;AACpB;;;AHnCO,SAAS,oBACf,KACA,MACA,kBACA,UACC;AACD,QAAM,gBAAgB,WAAW,KAAK,UAAU,SAAS;AACzD,QAAM,gBAAgB,KAAK,OAAO,QAAQ,aAAa;AAEvD,MAAI,CAAC,iBAAiB,CAAC,aAAa,aAAa,GAAG;AACnD,UAAM,IAAI,gBAAgB,yBAAyB,aAAa,EAAE;AAAA,EACnE;AAEA,SAAO;AAAA,IACN;AAAA,IACA;AAAA,IACA;AAAA,IACA,KAAK,UAAU;AAAA,IACf;AAAA,IACA;AAAA,EACD;AACD;AAEA,SAAS,2BACR,KACA,MACA,MACA,cACA,kBACA,UACC;AACD,QAAM,SAAS;AAAA,IACd,OAAO;AAAA,IACP,SAAS;AAAA,IACT,YAAY;AAAA,EACb;AAEA,aAAW,aAAa,aAAa,YAAY;AAChD,UAAM,kBAAkB;AAAA,MACvB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACD;AACA,WAAO,cAAc,gBAAgB;AACrC,WAAO,WAAW,gBAAgB;AAClC,WAAO,QAAQ,KAAK,IAAI,OAAO,OAAO,gBAAgB,KAAK;AAAA,EAC5D;AAEA,SAAO;AACR;AAEA,SAAS,wBACR,KACA,MACA,MACA,WACA,kBACA,UACC;AACD,MAAI,UAAU,SAAS,KAAK,OAAO;AAClC,WAAO,oBAAoB,KAAK,MAAM,MAAM,WAAW,kBAAkB,QAAQ;AAAA,EAClF;AAEA,MAAI,UAAU,SAAS,KAAK,iBAAiB;AAC5C,UAAM,WAAW,KAAK,UAAU,UAAU,KAAK,KAAK;AAEpD,QAAI,CAAC,UAAU;AACd,YAAM,IAAI,gBAAgB,YAAY,UAAU,KAAK,KAAK,YAAY;AAAA,IACvE;AAEA,WAAO,uBAAuB,KAAK,MAAM,MAAM,UAAU,kBAAkB,QAAQ;AAAA,EACpF;AAEA,SAAO,uBAAuB,KAAK,MAAM,MAAM,WAAW,kBAAkB,QAAQ;AACrF;AAEA,SAAS,uBACR,KACA,MACA,MACA,UACA,kBACA,UACC;AACD,QAAM,eAAe,SAAS,gBAC3B,KAAK,OAAO,QAAQ,SAAS,cAAc,KAAK,KAAK,IACrD;AAEH,MAAI,CAAC,aAAa,YAAY,GAAG;AAChC,UAAM,IAAI,gBAAgB,6BAA6B,YAAY,EAAE;AAAA,EACtE;AAEA,MAAI,CAAC,cAAc;AAClB,UAAM,IAAI,gBAAgB,iBAAiB,YAAY,YAAY;AAAA,EACpE;AAEA,SAAO;AAAA,IACN;AAAA,IACA;AAAA,IACA;AAAA,IACA,SAAS;AAAA,IACT;AAAA,IACA;AAAA,EACD;AACD;AAEA,SAAS,oBACR,KACA,MACA,MACA,WACA,kBACA,UACC;AACD,QAAM,YAAY,UAAU,KAAK;AAEjC,QAAM,QACL,aAAa,IAAI,KAAK,gBAAgB,IAAI,IAAI,KAAK,UAAU,EAAE,SAAS,IAAI;AAE7E,MAAI,CAAC,SAAS,CAAC,UAAU,WAAW,IAAI,GAAG;AAC1C,UAAM,IAAI,gBAAgB,SAAS,SAAS,sBAAsB,KAAK,IAAI,EAAE;AAAA,EAC9E;AAEA,MAAI,CAAC,OAAO;AACX,WAAO;AAAA,MACN,OAAO;AAAA,MACP,SAAS;AAAA,MACT,YAAY;AAAA,IACb;AAAA,EACD;AAEA,QAAM,0BAA0B;AAAA,IAC/B;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACD;AAEA,MAAI,4BAA4B,OAAO;AACtC,WAAO;AAAA,MACN,OAAO;AAAA,MACP,SAAS;AAAA,MACT,YAAY;AAAA,IACb;AAAA,EACD;AAEA,MAAI,QAAQ;AACZ,MAAI,UAAU;AACd,MAAI,aAAa;AAEjB,QAAM,iBAAiB,yBAAyB,cAAc,SAAS;AACvE,QAAM,aAAa,SAAS,qBAAqB,MAAM,IAAI,IAAI,iBAAiB;AAEhF,iBAAe,yBAAyB,cAAc,SAAS,cAAc;AAE7E,MAAI,CAAC,SAAS,CAAC,UAAU,cAAc;AACtC,WAAO;AAAA,MACN;AAAA,MACA;AAAA,MACA;AAAA,IACD;AAAA,EACD;AAEA,QAAM,eAAe;AAAA,IACpB;AAAA,IACA;AAAA,IACA,aAAa,MAAM,IAAI;AAAA,IACvB,UAAU;AAAA,IACV;AAAA,IACA;AAAA,EACD;AAEA,WAAS,aAAa;AACtB,aAAW,aAAa;AACxB,gBAAc,aAAa,aAAa,KAAK,IAAI,YAAY,CAAC;AAE9D,SAAO;AAAA,IACN;AAAA,IACA;AAAA,IACA;AAAA,EACD;AACD;;;AInNA,SAAS,mBAAAA,wBAAuB;AAGzB,IAAK,sBAAL,kBAAKC,yBAAL;AACN,EAAAA,qBAAA,WAAQ;AACR,EAAAA,qBAAA,aAAU;AACV,EAAAA,qBAAA,gBAAa;AAHF,SAAAA;AAAA,GAAA;AAYL,SAAS,0BACf,MACA,OACA,QACe;AACf,MAAI,SAAS,6BAA2B;AACvC,WAAO,IAAID,iBAAgB,kBAAkB,KAAK,mBAAmB,MAAM,EAAE;AAAA,EAC9E;AAEA,MAAI,SAAS,iCAA6B;AACzC,WAAO,IAAIA,iBAAgB,oBAAoB,KAAK,mBAAmB,MAAM,EAAE;AAAA,EAChF;AAEA,MAAI,SAAS,uCAAgC;AAC5C,WAAO,IAAIA,iBAAgB,uBAAuB,KAAK,mBAAmB,MAAM,EAAE;AAAA,EACnF;AAEA,QAAM,IAAI,MAAM,+BAA+B;AAChD;;;ACvBO,IAAM,gBAAgB;AAAA,EAC5B,OAAO;AAAA,EACP,SAAS;AAAA,EACT,YAAY;AACb;AAIO,SAAS,iBACf,SAC6B;AAC7B,SAAO;AAAA,IACN,OAAO,QAAQ,SAAS;AAAA,IACxB,mBAAmB,QAAQ,qBAAqB;AAAA,IAChD,uBAAuB,QAAQ,yBAAyB;AAAA,IACxD,iBAAiB,QAAQ,mBAAmB;AAAA,EAC7C;AACD;;;AC3BA,SAAS,0BAA0B;AAQ5B,IAAM,qBAAqB,OAAO,sBAAsB;AAExD,IAAM,CAAC,oBAAoB,wBAAwB,IACzD,mBAAoC,kBAAkB;;;ACRhD,SAAS,oBACf,KACA,WACAE,gBACC;AACD,2BAAyB,KAAK,YAAY;AACzC,UAAM,SAAS,OAAO,cAAc,aAAa,MAAM,UAAU,GAAG,IAAI;AAExE,QAAI,QAA+C;AAEnD,UAAM,kBAAkB,CAAC,OAAwC;AAChE,UAAI,OAAO;AACV,eAAO;AAAA,MACR;AAEA,cAAQ,GAAG;AACX,aAAO;AAAA,IACR;AAEA,WAAO;AAAA,MACN,QAAQ;AAAA,QACP,OAAO,QAAQ,SAASA,eAAc;AAAA,QACtC,SAAS,QAAQ,WAAWA,eAAc;AAAA,QAC1C,YAAY,QAAQ,cAAcA,eAAc;AAAA,MACjD;AAAA,MACA;AAAA,IACD;AAAA,EACD,CAAC;AACF;;;ARdO,IAAM,sBAAN,cAAuC,UAAU;AAAA,EACtC;AAAA,EAEjB,YAAY,UAA2C,CAAC,GAAG;AAC1D,UAAM;AACN,SAAK,UAAU,iBAAiB,OAAO;AAAA,EACxC;AAAA,EAEQ,mBAAqC,CAAC;AAAA,EAE9C,oBAAoB,CACnB,SACA,SACmD;AACnD,WAAO;AAAA,MACN,aAAa,CAAC,OAAO;AACpB,oCAA4B,MAAM,KAAK,IAAI,KAAK,gBAAgB;AAAA,MACjE;AAAA,IACD;AAAA,EACD;AAAA,EAEA,wBAAwB,CACvB,SACA,MACA,UACqE;AACrE,WAAO;AAAA,MACN,aAAa,CAAC,OAAO;AACpB,oCAA4B,MAAM,OAAO,IAAI,KAAK,gBAAgB;AAAA,MACnE;AAAA,IACD;AAAA,EACD;AAAA,EAEA,4BAA4B,CAC3B,SACA,UACiE;AACjE,WAAO;AAAA,MACN,aAAa,CAAC,OAAO;AACpB,oCAA4B,gBAAgB,OAAO,IAAI,KAAK,gBAAgB;AAAA,MAC7E;AAAA,IACD;AAAA,EACD;AAAA,EAEA,6BAKE;AACD,WAAO,CAAC,SAAS,OAAO,MAAM,MAAM,KAAK,SAAS;AACjD,0BAAoB,KAAuB,KAAK,QAAQ,OAAO,aAAa;AAE5E,YAAM,QAAQ,MAAM,mBAAmB,GAAG;AAE1C,YAAM,SAAS,MAAM;AAErB,YAAM,UAAU,MAAM,gBAAgB,MAAM;AAC3C,eAAO,oBAAoB,KAAuB,MAAM,KAAK,kBAAkB;AAAA,UAC9E,YAAY,KAAK,QAAQ;AAAA,UACzB,YAAY,KAAK,QAAQ;AAAA,QAC1B,CAAC;AAAA,MACF,CAAC;AAED,UAAI,QAAQ,QAAQ,OAAO,OAAO;AACjC,cAAM,KAAK,QAAQ,6CAA2C,OAAO,OAAO,QAAQ,KAAK;AAAA,MAC1F;AAEA,UAAI,QAAQ,UAAU,OAAO,SAAS;AACrC,cAAM,KAAK,QAAQ;AAAA;AAAA,UAElB,OAAO;AAAA,UACP,QAAQ;AAAA,QACT;AAAA,MACD;AAEA,UAAI,QAAQ,aAAa,OAAO,YAAY;AAC3C,cAAM,KAAK,QAAQ;AAAA;AAAA,UAElB,OAAO;AAAA,UACP,QAAQ;AAAA,QACT;AAAA,MACD;AAEA,aAAO,KAAK,MAAM,MAAM,KAAK,IAAI;AAAA,IAClC;AAAA,EACD;AAAA,EAEA,QAAQ,CAAC,SAAwB,WAA2B;AAC3D,eAAW,QAAQ,OAAO,SAAS,GAAG;AACrC,UAAI,SAAS,WAAW,SAAS,cAAc,SAAS,gBAAgB;AACvE;AAAA,MACD;AAEA,UAAI,SAAS,gBAAgB;AAC5B,mBAAW,SAAS,OAAO,cAAc,IAAI,GAAG;AAC/C,iBAAO,cAAc,MAAM,OAAO,KAAK,2BAA2B,CAAC;AAAA,QACpE;AACA;AAAA,MACD;AAEA,iBAAW,SAAS,OAAO,cAAc,IAAI,GAAG;AAC/C,eAAO,cAAc,MAAM,GAAG,KAAK,cAAc,KAAK,2BAA2B,CAAC;AAAA,MACnF;AAAA,IACD;AAAA,EACD;AACD;;;ASlHO,SAAS,oBAAyB,SAA2C;AACnF,SAAO,MAAM,IAAI,oBAAoB,OAAO;AAC7C;","names":["ValidationError","ComplexityErrorKind","defaultLimits"]}

package/package.json

@@ -0,0 +1,82 @@
+{
+  "name": "@baeta/extension-complexity",
+  "version": "0.0.1",
+  "keywords": [
+    "baeta",
+    "graphql",
+    "schema",
+    "types",
+    "typescript",
+    "framework",
+    "builder"
+  ],
+  "homepage": "https://github.com/andreisergiu98/baeta#readme",
+  "bugs": {
+    "url": "https://github.com/andreisergiu98/baeta/issues"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/andreisergiu98/baeta.git",
+    "directory": "packages/extension-complexity"
+  },
+  "license": "MIT",
+  "author": {
+    "name": "Andrei Pampu",
+    "url": "https://github.com/andreisergiu98"
+  },
+  "type": "module",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "require": "./dist/index.cjs"
+    }
+  },
+  "types": "dist/index.d.ts",
+  "files": [
+    "dist",
+    "package.json"
+  ],
+  "scripts": {
+    "build": "tsup",
+    "prepack": "prep",
+    "postpack": "prep --clean",
+    "types": "tsc --noEmit"
+  },
+  "devDependencies": {
+    "@baeta/builder": "^0.0.0",
+    "@baeta/core": "^0.1.2",
+    "@baeta/tsconfig": "^0.0.0",
+    "graphql": "^16.9.0",
+    "typescript": "^5.6.3"
+  },
+  "peerDependencies": {
+    "@baeta/core": "^0.1.2",
+    "graphql": "^16.6.0"
+  },
+  "engines": {
+    "node": ">=22.0.0"
+  },
+  "publishConfig": {
+    "access": "public",
+    "exports": {
+      ".": {
+        "types": "./dist/index.d.ts",
+        "import": "./dist/index.js",
+        "require": "./dist/index.cjs"
+      }
+    }
+  },
+  "ava": {
+    "extensions": {
+      "ts": "module"
+    },
+    "nodeArguments": [
+      "--no-warnings",
+      "--experimental-transform-types"
+    ]
+  },
+  "dependencies": {
+    "@baeta/errors": "^0.1.2"
+  }
+}