part: set up overridable options

Author: RebeccaStevens

knip.jsonc

@@ -2,7 +2,7 @@
   "$schema": "node_modules/knip/schema-jsonc.json",
   "entry": ["src/index.ts!", "cz-adapter/index.js", "tests/**/*.test.ts"],
   "project": ["src/**/*.ts!", "cz-adapter/**/*.{js,ts}", "tests/**/*.{js,ts}"],
-  "ignore": ["src/utils/conditional-imports/esm/**/*"],
+  "ignore": ["src/utils/conditional-imports/esm/**/*", "src/utils/schemas.ts"],
   "ignoreDependencies": [
     "@stylistic/eslint-plugin",
     "@rebeccastevens/eslint-config",

package.json

@@ -80,7 +80,8 @@
     "deepmerge-ts": "^7.1.0",
     "escape-string-regexp": "^4.0.0",
     "is-immutable-type": "^4.0.0",
-    "ts-api-utils": "^1.3.0"
+    "ts-api-utils": "^1.3.0",
+    "ts-declaration-location": "^1.0.3"
   },
   "devDependencies": {
     "@babel/eslint-parser": "7.24.8",

pnpm-lock.yaml

@@ -1 +1,2 @@
 export * from "./ignore";
+export * from "./overrides";

src/options/index.ts

@@ -0,0 +1,185 @@
+import assert from "node:assert/strict";
+
+import { type TSESTree } from "@typescript-eslint/utils";
+import { type RuleContext } from "@typescript-eslint/utils/ts-eslint";
+import { deepmerge } from "deepmerge-ts";
+import typeMatchesSpecifier, {
+  type TypeDeclarationSpecifier,
+} from "ts-declaration-location";
+import { type Program, type Type, type TypeNode } from "typescript";
+
+import { getTypeDataOfNode } from "#/utils/rule";
+import {
+  type RawTypeSpecifier,
+  type TypeSpecifier,
+  typeMatchesPattern,
+} from "#/utils/type-specifier";
+
+/**
+ * Options that can be overridden.
+ */
+export type OverridableOptions<CoreOptions> = CoreOptions & {
+  overrides?: Array<
+    {
+      specifiers: TypeSpecifier | TypeSpecifier[];
+    } & (
+      | {
+          options: CoreOptions;
+          inherit?: boolean;
+          disable?: false;
+        }
+      | {
+          disable: true;
+        }
+    )
+  >;
+};
+
+export type RawOverridableOptions<CoreOptions> = CoreOptions & {
+  overrides?: Array<{
+    specifiers?: RawTypeSpecifier | RawTypeSpecifier[];
+    options?: CoreOptions;
+    inherit?: boolean;
+    disable?: boolean;
+  }>;
+};
+
+export function upgradeRawOverridableOptions<CoreOptions>(
+  raw: Readonly<RawOverridableOptions<CoreOptions>>,
+): OverridableOptions<CoreOptions> {
+  return {
+    ...raw,
+    overrides:
+      raw.overrides?.map((override) => ({
+        ...override,
+        specifiers:
+          override.specifiers === undefined
+            ? []
+            : Array.isArray(override.specifiers)
+              ? override.specifiers.map(upgradeRawTypeSpecifier)
+              : [upgradeRawTypeSpecifier(override.specifiers)],
+      })) ?? [],
+  } as OverridableOptions<CoreOptions>;
+}
+
+function upgradeRawTypeSpecifier(raw: RawTypeSpecifier): TypeSpecifier {
+  const { ignoreName, ignorePattern, name, pattern, ...rest } = raw;
+
+  const names = name === undefined ? [] : Array.isArray(name) ? name : [name];
+
+  const patterns = (
+    pattern === undefined ? [] : Array.isArray(pattern) ? pattern : [pattern]
+  ).map((p) => new RegExp(p, "u"));
+
+  const ignoreNames =
+    ignoreName === undefined
+      ? []
+      : Array.isArray(ignoreName)
+        ? ignoreName
+        : [ignoreName];
+
+  const ignorePatterns = (
+    ignorePattern === undefined
+      ? []
+      : Array.isArray(ignorePattern)
+        ? ignorePattern
+        : [ignorePattern]
+  ).map((p) => new RegExp(p, "u"));
+
+  const include = [...names, ...patterns];
+  const exclude = [...ignoreNames, ...ignorePatterns];
+
+  return {
+    ...rest,
+    include,
+    exclude,
+  };
+}
+
+/**
+ * Get the core options to use, taking into account overrides.
+ */
+export function getCoreOptions<
+  CoreOptions extends object,
+  Options extends Readonly<OverridableOptions<CoreOptions>>,
+>(
+  node: TSESTree.Node,
+  context: Readonly<RuleContext<string, unknown[]>>,
+  options: Readonly<Options>,
+): CoreOptions | null {
+  const program = context.sourceCode.parserServices?.program ?? undefined;
+  if (program === undefined) {
+    return options;
+  }
+
+  const [type, typeNode] = getTypeDataOfNode(node, context);
+  return getCoreOptionsForType(type, typeNode, context, options);
+}
+
+export function getCoreOptionsForType<
+  CoreOptions extends object,
+  Options extends Readonly<OverridableOptions<CoreOptions>>,
+>(
+  type: Type,
+  typeNode: TypeNode | null,
+  context: Readonly<RuleContext<string, unknown[]>>,
+  options: Readonly<Options>,
+): CoreOptions | null {
+  const program = context.sourceCode.parserServices?.program ?? undefined;
+  if (program === undefined) {
+    return options;
+  }
+
+  const found = options.overrides?.find((override) =>
+    (Array.isArray(override.specifiers)
+      ? override.specifiers
+      : [override.specifiers]
+    ).some(
+      (specifier) =>
+        typeMatchesSpecifierDeep(program, specifier, type) &&
+        (specifier.include === undefined ||
+          specifier.include.length === 0 ||
+          typeMatchesPattern(
+            program,
+            type,
+            typeNode,
+            specifier.include,
+            specifier.exclude,
+          )),
+    ),
+  );
+
+  if (found !== undefined) {
+    if (found.disable === true) {
+      return null;
+    }
+    if (found.inherit !== false) {
+      return deepmerge(options, found.options) as CoreOptions;
+    }
+    return found.options;
+  }
+
+  return options;
+}
+
+function typeMatchesSpecifierDeep(
+  program: Program,
+  specifier: TypeDeclarationSpecifier,
+  type: Type,
+) {
+  const m_stack = [type];
+  // eslint-disable-next-line functional/no-loop-statements -- best to do this iteratively.
+  while (m_stack.length > 0) {
+    const t = m_stack.pop() ?? assert.fail();
+
+    if (typeMatchesSpecifier(program, specifier, t)) {
+      return true;
+    }
+
+    if (t.aliasTypeArguments !== undefined) {
+      m_stack.push(...t.aliasTypeArguments);
+    }
+  }
+
+  return false;
+}

src/options/overrides.ts

@@ -1,3 +1,5 @@
+import assert from "node:assert/strict";
+
 import { type TSESTree } from "@typescript-eslint/utils";
 import {
   type NamedCreateRuleMeta,
@@ -22,6 +24,8 @@ import { getImmutabilityOverrides } from "#/settings";
 import { __VERSION__ } from "#/utils/constants";
 import { type ESFunction } from "#/utils/node-types";
 
+import { typeMatchesPattern } from "./type-specifier";
+
 /**
  * Any custom rule meta properties.
  */
@@ -187,12 +191,46 @@ export function getTypeOfNode<Context extends RuleContext<string, BaseOptions>>(
   node: TSESTree.Node,
   context: Context,
 ): Type {
+  assert(ts !== undefined);
+
   const { esTreeNodeToTSNodeMap } = getParserServices(context);
 
   const tsNode = esTreeNodeToTSNodeMap.get(node);
   return getTypeOfTSNode(tsNode, context);
 }
 
+/**
+ * Get the type of the the given node.
+ */
+export function getTypeNodeOfNode<
+  Context extends RuleContext<string, BaseOptions>,
+>(node: TSESTree.Node, context: Context): TypeNode | null {
+  assert(ts !== undefined);
+
+  const { esTreeNodeToTSNodeMap } = getParserServices(context);
+
+  const tsNode = esTreeNodeToTSNodeMap.get(node) as TSNode & {
+    type?: TypeNode;
+  };
+  return tsNode.type ?? null;
+}
+
+/**
+ * Get the type of the the given node.
+ */
+export function getTypeDataOfNode<
+  Context extends RuleContext<string, BaseOptions>,
+>(node: TSESTree.Node, context: Context): [Type, TypeNode | null] {
+  assert(ts !== undefined);
+
+  const { esTreeNodeToTSNodeMap } = getParserServices(context);
+
+  const tsNode = esTreeNodeToTSNodeMap.get(node) as TSNode & {
+    type?: TypeNode;
+  };
+  return [getTypeOfTSNode(tsNode, context), tsNode.type ?? null];
+}
+
 /**
  * Get the type of the the given ts node.
  */
@@ -280,6 +318,7 @@ export function getTypeImmutabilityOfNode<
     // Don't use the global cache in testing environments as it may cause errors when switching between different config options.
     process.env["NODE_ENV"] !== "test",
     maxImmutability,
+    typeMatchesPattern,
   );
 }