feat: implement check registry (#1396)

Implement a fully typed check registry and specification system.

- Port all checks to new specification system
- Add register exports to each module
- Add registry
- Refactor tests
- Backport `itertools.batched`
  
Benchmarks:
```
On main
Benchmark 1: uv run proselint --demo
  Time (mean ± σ):      1.090 s ±  0.265 s    [User: 1.041 s, System: 0.033 s]
  Range (min … max):    1.001 s …  1.843 s    10 runs
On #1396
Benchmark 1: uv run proselint --demo
  Time (mean ± σ):     432.2 ms ±   2.0 ms    [User: 405.4 ms, System: 24.8 ms]
  Range (min … max):   429.5 ms … 435.8 ms    10 runs
```

---------

Co-authored-by: drainpixie <121581793+drainpixie@users.noreply.github.com>

Author: Nytelife26

.gitignore

@@ -18,3 +18,4 @@ corpora/*
 dist/
 *.pstore
 .pre-commit-config.yaml
+.hypothesis

alejandra.toml

@@ -0,0 +1 @@
+indentation = "Tabs"

flake.lock

@@ -1,85 +1,93 @@
 # TODO: use uv2nix
 {
-  inputs = {
-    nixpkgs.url = "github:NixOS/nixpkgs/nixos-25.05";
-    hooks = {
-      url = "github:cachix/git-hooks.nix";
-      inputs.nixpkgs.follows = "nixpkgs";
-    };
-    pyproject = {
-      url = "github:pyproject-nix/pyproject.nix";
-      inputs.nixpkgs.follows = "nixpkgs";
-    };
-  };
+	inputs = {
+		nixpkgs.url = "github:NixOS/nixpkgs/nixos-25.05";
+		hooks = {
+			url = "github:cachix/git-hooks.nix";
+			inputs.nixpkgs.follows = "nixpkgs";
+		};
+		pyproject = {
+			url = "github:pyproject-nix/pyproject.nix";
+			inputs.nixpkgs.follows = "nixpkgs";
+		};
+	};
 
-  outputs = {
-    self,
-    hooks,
-    nixpkgs,
-    pyproject,
-    ...
-  }: let
-    forAllSystems = function:
-      nixpkgs.lib.genAttrs [
-        "x86_64-linux"
-        "aarch64-linux"
-        "x86_64-darwin"
-        "aarch64-darwin"
-      ] (system: function system nixpkgs.legacyPackages.${system});
-    project = pyproject.lib.project.loadPyproject {projectRoot = ./.;};
-  in {
-    devShells = forAllSystems (system: pkgs: let
-      python = pkgs.python312;
-      arg = project.renderers.withPackages {inherit python;};
-      pyenv = python.withPackages arg;
-      check = self.checks.${system}.pre-commit-check;
-    in {
-      default = pkgs.mkShell {
-        inherit (check) shellHook;
+	outputs = {
+		self,
+		hooks,
+		nixpkgs,
+		pyproject,
+		...
+	}: let
+		forAllSystems = function:
+			nixpkgs.lib.genAttrs [
+				"x86_64-linux"
+				"aarch64-linux"
+				"x86_64-darwin"
+				"aarch64-darwin"
+			] (system: function system nixpkgs.legacyPackages.${system});
+		project = pyproject.lib.project.loadPyproject {projectRoot = ./.;};
+	in {
+		devShells =
+			forAllSystems (system: pkgs: let
+					python = pkgs.python312;
+					arg = project.renderers.withPackages {inherit python;};
+					pyenv = python.withPackages arg;
+					check = self.checks.${system}.pre-commit-check;
+				in {
+					default =
+						pkgs.mkShell {
+							inherit (check) shellHook;
 
-        packages =
-          check.enabledPackages
-          ++ [
-            pyenv
-            pkgs.uv
-            pkgs.pyright
-            pkgs.hyperfine
-          ];
-      };
-    });
+							packages =
+								check.enabledPackages
+								++ [
+									pyenv
+									pkgs.uv
+								];
+						};
+				});
 
-    packages = forAllSystems (system: pkgs: let
-      python = pkgs.python312;
-      attrs = project.renderers.buildPythonPackage {inherit python;};
-    in {
-      default = python.pkgs.buildPythonPackage attrs;
-    });
+		packages =
+			forAllSystems (system: pkgs: let
+					python = pkgs.python312;
+					attrs = project.renderers.buildPythonPackage {inherit python;};
+				in {
+					default = python.pkgs.buildPythonPackage attrs;
+				});
 
-    apps = forAllSystems (system: _: {
-      default = {
-        type = "app";
-        program = "${self.packages.${system}.default}/bin/proselint";
-      };
-    });
+		apps =
+			forAllSystems (system: _: {
+					default = {
+						type = "app";
+						program = "${self.packages.${system}.default}/bin/proselint";
+					};
+				});
 
-    checks = forAllSystems (system: _: {
-      pre-commit-check = hooks.lib.${system}.run {
-        src = ./.;
-        hooks = {
-          trim-trailing-whitespace.enable = true;
-          end-of-file-fixer.enable = true;
-          mixed-line-endings.enable = true;
-          markdownlint.enable = true;
-          ruff.enable = true;
-          pyright.enable = true;
-          convco.enable = true;
-          alejandra.enable = true;
-          statix = {
-            enable = true;
-            settings.ignore = ["/.direnv"];
-          };
-        };
-      };
-    });
-  };
+		checks =
+			forAllSystems (system: pkgs: {
+					pre-commit-check =
+						hooks.lib.${system}.run {
+							src = ./.;
+							hooks = {
+								trim-trailing-whitespace.enable = true;
+								end-of-file-fixer.enable = true;
+								mixed-line-endings.enable = true;
+								markdownlint.enable = true;
+								ruff.enable = true;
+								pyright = {
+									enable = true;
+									package = pkgs.basedpyright;
+									entry = "${pkgs.basedpyright}/bin/basedpyright";
+								};
+								convco.enable = true;
+								alejandra.enable = true;
+								statix = {
+									enable = true;
+									settings.ignore = ["/.direnv"];
+								};
+							};
+						};
+				});
+	};
 }

flake.nix

@@ -0,0 +1,30 @@
+"""Backport features for older versions of Python."""
+
+from __future__ import annotations
+
+from itertools import islice
+from sys import version_info
+from typing import TYPE_CHECKING, TypeVar
+
+if TYPE_CHECKING:
+    from collections.abc import Generator, Iterable
+
+T = TypeVar("T")
+
+if version_info >= (3, 13):
+    from itertools import batched
+else:
+    def batched(
+        iterable: Iterable[T], n: int, *, strict: bool = False
+    ) -> Generator[tuple[T, ...]]:
+        """Batch data from the `iterable` into `n`-tuples."""
+        if n < 0:
+            raise ValueError("n must be at least one")
+        iterator = iter(iterable)
+        while batch := tuple(islice(iterator, n)):
+            if strict and len(batch) != n:
+                raise ValueError("batched(): incomplete batch")
+            yield batch
+
+
+__all__ = ("batched",)

proselint/backports.py

@@ -1 +1,33 @@
 """All the checks are organized into modules and places here."""
+
+from proselint.registry import build_modules_register
+
+__register__ = build_modules_register(
+    (
+        ".annotations",
+        ".archaism",
+        ".cliches",
+        ".dates_times",
+        ".hedging",
+        ".industrial_language",
+        ".lexical_illusions",
+        ".malapropisms",
+        ".misc",
+        ".mixed_metaphors",
+        ".mondegreens",
+        ".needless_variants",
+        ".nonwords",
+        ".oxymorons",
+        ".psychology",
+        ".redundancy",
+        ".restricted",
+        ".skunked_terms",
+        ".social_awareness",
+        ".spelling",
+        ".terms",
+        ".typography",
+        ".uncomparables",
+        ".weasel_words",
+    ),
+    "proselint.checks",
+)

proselint/checks/__init__.py

@@ -1,4 +1,5 @@
-"""Annotation left in text.
+"""
+Annotation left in text.
 
 ---
 layout:     post
@@ -12,23 +13,23 @@
 Annotation left in text.
 
 """
-from proselint.tools import existence_check
-
-
-
-def check(text):
-    """Check the text."""
-    err = "annotations.misc"
-    msg = "Annotation left in text."
-
-    annotations = [
-        "FIXME",
-        "FIX ME",
-        "TODO",
-        "todo",
-        "ERASE THIS",
-        "FIX THIS",
-    ]
 
-    return existence_check(
-        text, annotations, err, msg, ignore_case=False, join=True)
+from proselint.registry.checks import Check, types
+
+check = Check(
+    check_type=types.Existence(
+        items=(
+            "FIXME",
+            "FIX ME",
+            "TODO",
+            "todo",
+            "ERASE THIS",
+            "FIX THIS",
+        )
+    ),
+    path="annotations.misc",
+    message="Annotation left in text.",
+    ignore_case=False,
+)
+
+__register__ = (check,)