tlaup 0.8.2__py3-none-any.whl

tlaup/__init__.py

@@ -0,0 +1,6 @@
+import truststore
+
+# Inject truststore as early as possible so that all standard library
+# and third-party tools relying on `ssl` (such as `requests` and `urllib3`)
+# utilize the native OS trust store.
+truststore.inject_into_ssl()

tlaup/cache/github.py

@@ -0,0 +1,28 @@
+import json
+from dataclasses import asdict
+from pathlib import Path
+
+from tlaup.ui import warn
+from tlaup.versioning.schema import RemoteVersion
+
+
+def load_github_cache(cache_file: Path) -> list[RemoteVersion] | None:
+    """Load the remote versions cache from disk."""
+    if cache_file.exists():
+        try:
+            with cache_file.open("r", encoding="utf-8") as f:
+                data = json.load(f)
+            return [RemoteVersion(**item) for item in data]
+        except (json.JSONDecodeError, OSError, KeyError, TypeError) as e:
+            warn(f"Failed to read cache: {e}")
+    return None
+
+
+def save_github_cache(cache_file: Path, versions: list[RemoteVersion]) -> None:
+    """Save the remote versions cache to disk."""
+    try:
+        cache_file.parent.mkdir(parents=True, exist_ok=True)
+        with cache_file.open("w", encoding="utf-8") as f:
+            json.dump([asdict(v) for v in versions], f)
+    except OSError as e:
+        warn(f"Failed to save cache: {e}")

tlaup/cli.py

@@ -0,0 +1,62 @@
+"""TLA+ CLI tool - entry point."""
+
+import importlib.metadata
+
+import typer
+
+from tlaup.cmd.check_java import check_java
+from tlaup.cmd.config import app as config_app
+from tlaup.cmd.fetch_cache import app as fetch_cache_app
+from tlaup.cmd.modules import app as modules_app
+from tlaup.cmd.tlc import tlc as run_tlc_cmd
+from tlaup.cmd.tools import app as tools_app
+from tlaup.config.loader import load_config
+from tlaup.versioning import get_app_version
+
+app = typer.Typer(
+    name="tlaup",
+    help="TLA+ tools: download TLA+ toolset distribution, compile custom modules, run model checker.",
+    no_args_is_help=True,
+    add_completion=False,
+)
+
+
+def version_callback(value: bool) -> None:
+    if value:
+        meta = importlib.metadata.metadata("tlaup")
+        app_ver = get_app_version(base_version=meta["Version"])
+        typer.echo(f"{meta['Name']} v{app_ver}")
+        typer.echo(meta["Summary"])
+        raise typer.Exit()
+
+
+@app.callback()
+def root(
+    _: bool = typer.Option(
+        None,
+        "--version",
+        "-v",
+        help="Show the application's version and exit.",
+        callback=version_callback,
+        is_eager=True,
+    ),
+) -> None:
+    """TLA+ CLI tool."""
+    # Load config early to trigger first-run copy
+    load_config()
+
+
+# --- Subcommands ---
+
+app.add_typer(modules_app, name="modules")
+app.add_typer(tools_app, name="tools")
+app.add_typer(fetch_cache_app, name="fetch-cache")
+app.add_typer(config_app, name="config")
+
+app.command(name="tlc")(run_tlc_cmd)
+app.command(name="check-java")(check_java)
+
+
+def main() -> None:
+    """Entry point for [project.scripts]."""
+    app()

tlaup/cmd/check_java.py

@@ -0,0 +1,19 @@
+import typer
+
+from tlaup.config.loader import load_config
+from tlaup.java import get_java_version, validate_java_version
+
+
+def check_java() -> None:
+    """Check if Java is installed and meets the minimum version requirement."""
+    config = load_config()
+    version = get_java_version()
+    if version:
+        typer.echo(f"Detected Java version: {version}")
+
+    try:
+        validate_java_version(config.java.min_version)
+        typer.echo(f"Java version is compatible (>= {config.java.min_version}).")
+    except RuntimeError as e:
+        typer.echo(f"Error: {e}", err=True)
+        raise typer.Exit(1) from None

tlaup/cmd/config/__init__.py

@@ -0,0 +1,6 @@
+import typer
+
+app = typer.Typer(name="config", help="Manage application configuration.", no_args_is_help=True)
+
+from . import edit  # noqa: F401, E402
+from . import list as list_cmd  # noqa: F401, E402

tlaup/cmd/config/edit.py

@@ -0,0 +1,34 @@
+import os
+import shutil
+import subprocess
+
+import typer
+
+from tlaup.cmd.config import app
+from tlaup.config.loader import config_path, load_config
+
+
+@app.command(name="edit")
+def edit_config(
+    editor: str | None = typer.Argument(None, help="The editor to use (defaults to $EDITOR or 'vim')."),
+) -> None:
+    """Open the configuration file in an editor."""
+    cp = config_path()
+
+    # Ensure config exists by loading it
+    load_config()
+
+    if not editor:
+        editor = os.environ.get("EDITOR", "vim")
+
+    editor_path = shutil.which(editor)
+    if not editor_path:
+        typer.echo(f"Error: Editor '{editor}' not found in PATH.", err=True)
+        raise typer.Exit(1)
+
+    typer.echo(f"Opening {cp} with {editor} ...")
+    try:
+        subprocess.run([editor_path, str(cp)], check=True)
+    except subprocess.CalledProcessError as e:
+        typer.echo(f"Editor exited with error: {e}", err=True)
+        raise typer.Exit(1) from None

tlaup/cmd/config/list.py

@@ -0,0 +1,16 @@
+import typer
+
+from tlaup.cmd.config import app
+from tlaup.config.loader import config_path
+
+
+@app.command(name="list")
+def list_config() -> None:
+    """Display the current configuration file content."""
+    cp = config_path()
+    if not cp.exists():
+        typer.echo("Configuration file not found.", err=True)
+        raise typer.Exit(1)
+
+    with cp.open(encoding="utf-8") as f:
+        typer.echo(f.read())

tlaup/cmd/fetch_cache/__init__.py

@@ -0,0 +1,5 @@
+import typer
+
+app = typer.Typer(name="fetch-cache", help="Manage API fetch cache.", no_args_is_help=True)
+
+from . import clear  # noqa: F401, E402

tlaup/cmd/fetch_cache/clear.py

@@ -0,0 +1,11 @@
+import typer
+
+from tlaup.cmd.fetch_cache import app
+from tlaup.versioning import clear_cache
+
+
+@app.command(name="clear")
+def cmd_clear_cache() -> None:
+    """Clear the local cache of remote GitHub versions."""
+    clear_cache()
+    typer.echo("GitHub versions cache cleared.")

tlaup/cmd/modules/__init__.py

@@ -0,0 +1,16 @@
+import typer
+
+app = typer.Typer(name="modules", help="Manage TLA+ Java modules.", no_args_is_help=True)
+
+
+# Import functions for new commands
+from .add import add_module  # noqa: E402
+from .list import list_modules  # noqa: E402
+from .path import show_modules_path  # noqa: E402
+from .remove import remove_module  # noqa: E402
+
+# Register new commands with the typer app
+app.command(name="add")(add_module)
+app.command(name="list")(list_modules)
+app.command(name="remove")(remove_module)
+app.command(name="path")(show_modules_path)

tlaup/cmd/modules/add.py

@@ -0,0 +1,140 @@
+import datetime
+import json
+import os
+import re
+import shutil
+import subprocess
+from pathlib import Path
+
+import typer
+
+from tlaup.config.loader import load_config
+from tlaup.java.classpath import ClasspathResolver, ResolveMode
+from tlaup.tlc.compiler import get_tlc_jar_path
+from tlaup.tlc.run_cache import clear_tlc_run_cache
+from tlaup.versioning.paths import get_modules_dir
+
+
+def find_override_classes(java_files: list[Path]) -> list[str]:
+    """Find all class names implementing ITLCOverrides in Java files."""
+    override_classes = []
+    package_pattern = re.compile(r"package\s+([a-zA-Z0-9_.]+)\s*;")
+    class_pattern = re.compile(
+        r"\bclass\s+([a-zA-Z0-9_]+)\b[^{]*\bimplements\b[^{]*\b(tlc2\.overrides\.)?ITLCOverrides\b"
+    )
+
+    for jf in java_files:
+        try:
+            content = jf.read_text(encoding="utf-8", errors="ignore")
+        except OSError:
+            continue
+
+        # Strip comments to prevent false matches
+        content_no_comments = re.sub(r"//.*", "", content)
+        content_no_comments = re.sub(r"/\*.*?\*/", "", content_no_comments, flags=re.DOTALL)
+
+        package_match = package_pattern.search(content_no_comments)
+        package_name = package_match.group(1) if package_match else ""
+
+        for match in class_pattern.finditer(content_no_comments):
+            class_name = match.group(1)
+            full_class_name = f"{package_name}.{class_name}" if package_name else class_name
+            override_classes.append(full_class_name)
+
+    return override_classes
+
+
+def add_module(  # noqa: PLR0912, PLR0915
+    path: str = typer.Argument(..., help="Path to the module directory to add."),
+) -> None:
+    """Compile custom Java overrides and add/update the module in the cache."""
+    source_dir = Path(path).resolve()
+    if not source_dir.is_dir():
+        typer.echo(f"Error: Path does not exist or is not a directory: {source_dir}", err=True)
+        raise typer.Exit(1)
+
+    config = load_config()
+    module_name = source_dir.name
+    target_dir = get_modules_dir() / module_name
+
+    # Overwrite if exists
+    if target_dir.exists():
+        shutil.rmtree(target_dir)
+    target_dir.mkdir(parents=True, exist_ok=True)
+
+    # 1. Copy directories (lib, modules, tlc2, and test/tests)
+    def copy_subfolder(subname: str) -> None:
+        src = source_dir / subname
+        if src.is_dir():
+            shutil.copytree(src, target_dir / subname)
+
+    typer.echo("Copying subfolders (lib, modules, test, tlc2)...")
+    copy_subfolder("lib")
+    copy_subfolder("modules")
+    copy_subfolder("tlc2")
+
+    if (source_dir / "test").is_dir():
+        copy_subfolder("test")
+    elif (source_dir / "tests").is_dir():
+        copy_subfolder("tests")
+
+    # 2. Compile Java overrides in cached version if tlc2/overrides exists
+    java_files = []
+    for candidate in [target_dir / "modules" / "tlc2" / "overrides", target_dir / "tlc2" / "overrides"]:
+        if candidate.is_dir():
+            java_files.extend(list(candidate.rglob("*.java")))
+
+    if java_files:
+        # Build classpath including tool_jar and local libs from the cached folder
+        cached_lib = target_dir / "lib"
+        extra_libs = []
+        if cached_lib.is_dir():
+            extra_libs.extend([str(j.absolute()) for j in cached_lib.glob("*.jar")])
+
+        tool_jar = get_tlc_jar_path()
+        if not tool_jar.exists():
+            typer.echo("Error: tla2tools.jar not found. Run 'tlaup tools install' first.", err=True)
+            shutil.rmtree(target_dir, ignore_errors=True)
+            raise typer.Exit(1)
+
+        resolver = ClasspathResolver(config, project_root=None, tool_jar=tool_jar, extra_libs=extra_libs)
+        classpath = os.pathsep.join(resolver.resolve(ResolveMode.COMPILE))
+
+        classes_dir = target_dir / "classes"
+        classes_dir.mkdir(parents=True, exist_ok=True)
+
+        cmd = ["javac", "-cp", classpath, "-d", str(classes_dir), *[str(f) for f in java_files]]
+        typer.echo("Compiling Java files in cached tlc2/overrides...")
+        try:
+            subprocess.run(cmd, check=True, capture_output=True, text=True)
+        except FileNotFoundError as err:
+            typer.echo("Error: 'javac' not found. Ensure JDK is installed.", err=True)
+            shutil.rmtree(target_dir, ignore_errors=True)
+            raise typer.Exit(1) from err
+        except subprocess.CalledProcessError as err:
+            typer.echo("Compilation failed!", err=True)
+            typer.echo(err.stdout, err=True)
+            typer.echo(err.stderr, err=True)
+            shutil.rmtree(target_dir, ignore_errors=True)
+            raise typer.Exit(1) from err
+
+        # Write service file
+        meta_inf = classes_dir / "META-INF" / "services"
+        meta_inf.mkdir(parents=True, exist_ok=True)
+        service_file = meta_inf / "tlc2.overrides.ITLCOverrides"
+
+        override_classes = find_override_classes(java_files)
+        if not override_classes:
+            override_classes = [config.tlc.overrides_class]
+
+        with service_file.open("w") as f:
+            for cls in override_classes:
+                f.write(f"{cls}\n")
+
+    # 3. Create metadata.json
+    metadata = {"name": module_name, "built_at": datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
+    with (target_dir / "metadata.json").open("w") as f:
+        json.dump(metadata, f, indent=2)
+
+    clear_tlc_run_cache()
+    typer.echo(f"Module '{module_name}' successfully added to cache.")

tlaup/cmd/modules/list.py

@@ -0,0 +1,45 @@
+import json
+
+import typer
+from rich.console import Console
+from rich.table import Table
+
+from tlaup.versioning.paths import get_modules_dir
+
+
+def list_modules() -> None:
+    """List all custom TLA+ modules added to the cache."""
+    modules_dir = get_modules_dir()
+
+    if not modules_dir.is_dir():
+        typer.echo("No modules currently added.")
+        return
+
+    modules_data = []
+    for item in sorted(modules_dir.iterdir()):
+        if item.is_dir():
+            metadata_file = item / "metadata.json"
+            if metadata_file.exists():
+                try:
+                    meta = json.loads(metadata_file.read_text())
+                    name = meta.get("name", item.name)
+                    built_at = meta.get("built_at", "Unknown")
+                    modules_data.append((name, built_at))
+                except Exception:
+                    modules_data.append((item.name, "Corrupt Metadata"))
+            else:
+                modules_data.append((item.name, "Unknown"))
+
+    if not modules_data:
+        typer.echo("No modules currently added.")
+        return
+
+    table = Table()
+    table.add_column("Module Name", style="cyan")
+    table.add_column("Last Built/Modified", style="magenta")
+
+    for name, built_at in modules_data:
+        table.add_row(name, built_at)
+
+    console = Console()
+    console.print(table)

tlaup/cmd/modules/path.py

@@ -0,0 +1,68 @@
+import os
+from pathlib import Path
+
+import typer
+
+from tlaup.config.loader import load_config
+from tlaup.java.classpath import ClasspathResolver, ResolveMode, SubdirType
+from tlaup.tlc.compiler import get_tlc_jar_path
+from tlaup.versioning.paths import get_modules_dir
+
+
+def show_modules_path() -> None:  # noqa: PLR0912
+    """Display the active paths that will be included when running TLC."""
+    config_obj = load_config()
+    tool_jar = get_tlc_jar_path()
+
+    resolver = ClasspathResolver(config_obj, project_root=None, tool_jar=tool_jar)
+
+    # 1. Source Paths (-DTLA-Library)
+    tla_library = resolver.get_tla_library_property()
+    typer.echo("Source Paths (-DTLA-Library):")
+    if tla_library:
+        for p in tla_library.split(os.pathsep):
+            typer.echo(f"  - {p}")
+    else:
+        typer.echo("  None")
+
+    # 2. Classes Paths
+    classes_paths = []
+    project_classes = resolver.get_project_path(SubdirType.CLASSES)
+    if project_classes:
+        classes_paths.append(str(project_classes.absolute()))
+
+    managed_dir = get_modules_dir()
+    if managed_dir.is_dir():
+        for item in sorted(managed_dir.iterdir()):
+            if item.is_dir():
+                mod_classes = item / "classes"
+                if mod_classes.is_dir():
+                    classes_paths.append(str(mod_classes.absolute()))
+
+    typer.echo("\nClasses Paths:")
+    if classes_paths:
+        for p in classes_paths:
+            typer.echo(f"  - {p}")
+    else:
+        typer.echo("  None")
+
+    # 3. Library Paths (JARs)
+    libs = resolver.resolve(ResolveMode.RUNTIME)
+    tool_jar_abs = str(tool_jar.absolute()) if tool_jar.exists() else ""
+
+    # Filter out classes directories and tool_jar to show only external library JARs
+    only_libs = []
+    for p in libs:
+        if p == tool_jar_abs:
+            continue
+        p_path = Path(p)
+        if p_path.name == "classes" and p_path.is_dir():
+            continue
+        only_libs.append(p)
+
+    typer.echo("\nLibrary Paths (JARs):")
+    if only_libs:
+        for p in only_libs:
+            typer.echo(f"  - {p}")
+    else:
+        typer.echo("  None")

tlaup/cmd/modules/remove.py

@@ -0,0 +1,22 @@
+import shutil
+
+import typer
+
+from tlaup.tlc.run_cache import clear_tlc_run_cache
+from tlaup.versioning.paths import get_modules_dir
+
+
+def remove_module(module_name: str = typer.Argument(..., help="Name of the module to remove from cache.")) -> None:
+    """Remove a custom module from the cache."""
+    target_dir = get_modules_dir() / module_name
+    if not target_dir.is_dir():
+        typer.echo(f"Error: Module '{module_name}' not found in cache.", err=True)
+        raise typer.Exit(1)
+
+    try:
+        shutil.rmtree(target_dir)
+        clear_tlc_run_cache()
+        typer.echo(f"Module '{module_name}' successfully deleted from cache.")
+    except OSError as e:
+        typer.echo(f"Error: Failed to delete module: {e}", err=True)
+        raise typer.Exit(1) from e