feat: add py.typed during ap init

Author: softwareentrepreneer

src/afterpython/cli/commands/init.py

@@ -46,6 +46,24 @@ def init_website():
     subprocess.run(["ap", "update", "website"])
 
 
+def init_py_typed():
+    from afterpython.tools.pyproject import find_package_directory
+
+    try:
+        package_dir = find_package_directory()
+    except FileNotFoundError:
+        click.echo(
+            "Could not find package directory (__init__.py not found), skipping py-typed initialization"
+        )
+        return
+    py_typed_path = package_dir / "py.typed"
+    if py_typed_path.exists():
+        click.echo(f"py.typed file already exists at {py_typed_path}")
+        return
+    py_typed_path.touch()
+    click.echo(f"Created {py_typed_path}")
+
+
 @click.command()
 @click.option(
     "--yes",
@@ -85,6 +103,9 @@ def init(ctx, yes):
 
     init_website()
 
+    # TODO: add type checking related stuff here
+    init_py_typed()
+
     create_workflow("deploy")
     create_workflow("ci")
 

src/afterpython/tools/pyproject.py

@@ -6,6 +6,7 @@
     from tomlkit.toml_document import TOMLDocument
 
 import asyncio
+from pathlib import Path
 
 from pyproject_metadata import StandardMetadata
 
@@ -27,6 +28,41 @@ def read_metadata() -> StandardMetadata:
     return StandardMetadata.from_pyproject(read_pyproject())
 
 
+def find_package_directory() -> Path:
+    """Find the user's package directory.
+
+    Supports both common Python project layouts:
+    - src layout: user_project/src/package_name/__init__.py
+    - flat layout: user_project/package_name/__init__.py
+
+    Returns:
+        Path to the package directory containing __init__.py
+
+    Raises:
+        FileNotFoundError: If package directory cannot be found in either layout
+    """
+    metadata = read_metadata()
+    package_name = metadata.name
+    project_root = ap.paths.user_path
+
+    # Try src layout first (recommended layout)
+    src_layout_dir = project_root / "src" / package_name
+    if (src_layout_dir / "__init__.py").exists():
+        return src_layout_dir
+
+    # Try flat layout
+    flat_layout_dir = project_root / package_name
+    if (flat_layout_dir / "__init__.py").exists():
+        return flat_layout_dir
+
+    # Package not found in either location
+    raise FileNotFoundError(
+        f"Could not find package '{package_name}' in either src layout "
+        f"({src_layout_dir}) or flat layout ({flat_layout_dir}). "
+        f"Expected to find __init__.py in one of these locations."
+    )
+
+
 def init_pyproject():
     """Initialize pyproject.toml with sensible defaults
     - add [build-system] section with uv build backend (same as `uv init --package`)