feat(mise): detect lazy loader from project dependencies

Select the generated __init__.py template from pyproject dependencies so
init generation works with both lazy-loader variants. Resolve the project
root from pyproject.toml as well, which makes generation work reliably
from nested paths while cleaning up local editor artifacts.

Author: liblaf

.config/copier/.copier-answers.release.yaml

@@ -1,5 +1,4 @@
 # This file is @generated by <https://github.com/liblaf/copier-release>.
 # DO NOT EDIT!
-# prettier-ignore
-_commit: v0.2.4
+_commit: v0.2.5
 _src_path: gh:liblaf/copier-release

.config/copier/.copier-answers.shared.yaml

@@ -1,6 +1,6 @@
 # This file is @generated by <https://github.com/liblaf/copier-shared>.
 # DO NOT EDIT!
-_commit: v0.2.6
+_commit: v0.2.7
 _src_path: gh:liblaf/copier-share
 author_email: 30631553+liblaf@users.noreply.github.com
 author_name: liblaf

.github/workflows/release-draft.yaml

@@ -23,7 +23,7 @@ jobs:
     steps:
       - id: auth
         name: Auth
-        uses: liblaf/actions/auth@cadf1f6ac2bed9a0ebc033e74a2c0994813e7f04 # v1
+        uses: liblaf/actions/auth@1f83732587ff97f1babbe9218fdc5436dda1a16f # v1
         with:
           app-id: ${{ vars.APP_ID }}
           private-key: ${{ secrets.PRIVATE_KEY }}

.github/workflows/release-pr.yaml

@@ -37,7 +37,7 @@ jobs:
           tool: git-cliff
       - id: version
         name: Next version
-        uses: liblaf/actions/next-version@cadf1f6ac2bed9a0ebc033e74a2c0994813e7f04 # v1
+        uses: liblaf/actions/next-version@1f83732587ff97f1babbe9218fdc5436dda1a16f # v1
         env:
           GIT_CLIFF_CONFIG_URL: ${{ env.GIT_CLIFF_CONFIG_URL }}
           GITHUB_TOKEN: ${{ github.token }}
@@ -58,7 +58,7 @@ jobs:
     steps:
       - id: auth
         name: Auth
-        uses: liblaf/actions/auth@cadf1f6ac2bed9a0ebc033e74a2c0994813e7f04 # v1
+        uses: liblaf/actions/auth@1f83732587ff97f1babbe9218fdc5436dda1a16f # v1
         with:
           app-id: ${{ vars.APP_ID }}
           private-key: ${{ secrets.PRIVATE_KEY }}

.github/workflows/release-publish.yaml

@@ -20,12 +20,12 @@ jobs:
     steps:
       - id: auth
         name: Auth
-        uses: liblaf/actions/auth@cadf1f6ac2bed9a0ebc033e74a2c0994813e7f04 # v1
+        uses: liblaf/actions/auth@1f83732587ff97f1babbe9218fdc5436dda1a16f # v1
         with:
           app-id: ${{ vars.APP_ID }}
           private-key: ${{ secrets.PRIVATE_KEY }}
       - name: Publish drafts
-        uses: liblaf/actions/publish-drafts@cadf1f6ac2bed9a0ebc033e74a2c0994813e7f04 # v1
+        uses: liblaf/actions/publish-drafts@1f83732587ff97f1babbe9218fdc5436dda1a16f # v1
         with:
           older-than: 6h
           token: ${{ steps.auth.outputs.token }}

.github/workflows/shared-mega-linter.yaml

@@ -44,7 +44,7 @@ jobs:
         name: MegaLinter
         uses: liblaf/megalinter-custom-flavor-all@4c245a0bdb312d24c66108433d4d701fb0875746 # main
         env:
-          GITHUB_TOKEN: ${{ github.token }}
+          GITHUB_TOKEN: ${{ steps.auth.outputs.token }}
           MEGALINTER_CONFIG: https://raw.githubusercontent.com/liblaf/megalinter-custom-flavor-all/refs/heads/main/.mega-linter.yml
       - if: success() || failure()
         name: Upload reports

.gitignore

@@ -0,0 +1 @@
+__pycache__/

.vscode/settings.json

@@ -1,78 +1,145 @@
 #!/usr/bin/env python
 import argparse
 import ast
+import enum
 import os
 import subprocess
 from pathlib import Path
+from typing import Any, Never
 
-TEMPLATE: str = """\
+import tomllib
+from packaging.requirements import Requirement
+
+
+class LazyLoader(enum.StrEnum):
+    LAZY_LOADER = enum.auto()
+    LIBLAF_LAZY_LOADER = enum.auto()
+
+
+TEMPLATES: dict[LazyLoader, str] = {
+    LazyLoader.LAZY_LOADER: """\
+from lazy_loader import attach_stub
+
+__getattr__, __dir__, __all__ = attach_stub(__name__, __file__)
+
+del attach_stub
+""",
+    LazyLoader.LIBLAF_LAZY_LOADER: """\
 from liblaf.lazy_loader import attach_stub
 
-__getattr__, __dir__, __all__ = attach_stub(__name__, __package__, __file__)
+__getattr__, __dir__, __all__ = attach_stub(__name__, __file__, __package__)
 
 del attach_stub
-"""
+""",
+}
 
 
 class Args(argparse.Namespace):
     path: Path
 
 
-def parse_args() -> Args:
-    parser: argparse.ArgumentParser = argparse.ArgumentParser()
-    parser.add_argument("path", nargs="?", type=Path)
-    args: argparse.Namespace = parser.parse_args(namespace=Args())
-    if args.path is None:
-        if (path := os.getenv("MISE_PROJECT_ROOT")) is not None:
-            args.path = Path(path)
-        else:
-            args.path = Path.cwd()
-    return args
+def detect_lazy_loader(project_root: Path) -> LazyLoader:
+    pyproject: Path = project_root / "pyproject.toml"
+    with pyproject.open("rb") as fp:
+        data: dict[str, Any] = tomllib.load(fp)
+    project: Any = data.get("project")
+    if not isinstance(project, dict):
+        error(f"`project` table not found in: {pyproject}")
+    dependencies: Any = project.get("dependencies")
+    if not isinstance(dependencies, list):
+        error(f"`project.dependencies` not found in: {pyproject}")
+    dependency_names: set[str] = set()
+    for dep in dependencies:
+        req: Requirement = Requirement(dep)
+        dependency_names.add(req.name.lower())
+    if "liblaf-lazy-loader" in dependency_names:
+        return LazyLoader.LIBLAF_LAZY_LOADER
+    if "lazy-loader" in dependency_names:
+        return LazyLoader.LAZY_LOADER
+    error(
+        "no supported lazy loader found in project dependencies: "
+        "expected `liblaf-lazy-loader` or `lazy-loader`"
+    )
+
+
+def error(message: str) -> Never:
+    raise SystemExit(message)
+
+
+def find_project_root(path: Path) -> Path:
+    for candidate in [path, *path.parents]:
+        if (candidate / "pyproject.toml").exists():
+            return candidate
+    error(f"pyproject.toml not found for: {path}")
 
 
-def git_ls_files(path: Path) -> list[Path]:
+def get_docstring(file: Path) -> str | None:
+    if not file.exists():
+        return None
+    source: str = file.read_text()
+    module: ast.Module = ast.parse(source)
+    if not (module.body and isinstance(module.body[0], ast.Expr)):
+        return None
+    node: ast.expr = module.body[0].value
+    if isinstance(node, ast.Constant) and isinstance(node.value, str):
+        return ast.get_source_segment(source, node)
+    return None
+
+
+def git_ls_files(*, project_root: Path, path: Path) -> list[Path]:
+    try:
+        relative_path: Path = path.relative_to(project_root)
+    except ValueError:
+        error(f"path is not under project root: {path}")
+
     result: subprocess.CompletedProcess[str] = subprocess.run(
         [
             "git",
+            "-C",
+            project_root,
             "ls-files",
             "--cached",
             "--others",
             "--exclude-standard",
             "--",
-            f"{path}/**/__init__.pyi",
+            relative_path / "__init__.pyi",
+            relative_path / "**/__init__.pyi",
         ],
         stdout=subprocess.PIPE,
         check=True,
         text=True,
     )
-    return [Path(line) for line in result.stdout.splitlines() if line]
+    return [project_root / line for line in result.stdout.splitlines() if line]
 
 
-def get_docstring(file: Path) -> str | None:
-    if not file.exists():
-        return None
-    source: str = file.read_text()
-    module: ast.Module = ast.parse(source)
-    if not (module.body and isinstance(module.body[0], ast.Expr)):
-        return None
-    node: ast.expr = module.body[0].value
-    if isinstance(node, ast.Constant) and isinstance(node.value, str):
-        return ast.get_source_segment(source, node)
-    return None
+def parse_args() -> Args:
+    parser: argparse.ArgumentParser = argparse.ArgumentParser()
+    parser.add_argument("path", nargs="?", type=Path)
+    args: argparse.Namespace = parser.parse_args(namespace=Args())
+    if args.path is None:
+        if (path := os.getenv("MISE_PROJECT_ROOT")) is not None:
+            args.path = Path(path)
+        else:
+            args.path = Path.cwd()
+    args.path = args.path.resolve()
+    return args
 
 
-def render_init(file: Path) -> str:
+def render_init(file: Path, *, loader: LazyLoader) -> str:
     docstring: str | None = get_docstring(file)
+    template: str = TEMPLATES[loader]
     if docstring is None:
-        return TEMPLATE
-    return f"{docstring}\n\n{TEMPLATE}"
+        return template
+    return f"{docstring}\n\n{template}"
 
 
 def main() -> None:
     args: Args = parse_args()
-    for pyi in git_ls_files(args.path):
+    project_root: Path = find_project_root(args.path)
+    loader: LazyLoader = detect_lazy_loader(project_root)
+    for pyi in git_ls_files(project_root=project_root, path=args.path):
         py: Path = pyi.with_suffix(".py")
-        content: str = render_init(py)
+        content: str = render_init(py, loader=loader)
         if py.exists():
             if py.read_text() == content:
                 print("skipped:", py)