EOS/scripts/update_version.py

#!.venv/bin/python
"""General version replacement script.

Usage:
    python scripts/update_version.py <version> <file1> [file2 ...]
"""

#!/usr/bin/env python3
import re
import sys
from datetime import timezone
from pathlib import Path
from typing import List

PROJECT_ROOT = Path(__file__).parent.parent
PACKAGE_DIR = PROJECT_ROOT / "src" / "akkudoktoreos"

# Add the src directory to sys.path so import akkudoktoreos works in all cases
SRC_DIR = PROJECT_ROOT / "src"
sys.path.insert(0, str(SRC_DIR))

DEFAULT_VERSION_FILES = [
    PROJECT_ROOT / ".env",          # Docker compose default environment
    PROJECT_ROOT / "config.yaml",   # Home Assistant config
]


# --- Patterns to match version strings ---
VERSION_PATTERNS = [
    # Python: __version__ = "1.2.3"
    re.compile(
        r'(?<![A-Za-z0-9_])(__version__\s*=\s*")'
        r'(?P<ver>\d+\.\d+\.\d+(?:[\.\+\-][0-9A-Za-z]+)?)'
        r'(")'
    ),

    # Python: version = "1.2.3"
    re.compile(
        r'(?<![A-Za-z0-9_])(version\s*=\s*")'
        r'(?P<ver>\d+\.\d+\.\d+(?:[\.\+\-][0-9A-Za-z]+)?)'
        r'(")'
    ),

    # JSON: "version": "1.2.3"
    re.compile(
        r'(?<![A-Za-z0-9_])("version"\s*:\s*")'
        r'(?P<ver>\d+\.\d+\.\d+(?:[\.\+\-][0-9A-Za-z]+)?)'
        r'(")'
    ),

    # Makefile-style: VERSION ?= 1.2.3
    re.compile(
        r'(?<![A-Za-z0-9_])(VERSION\s*\?=\s*)'
        r'(?P<ver>\d+\.\d+\.\d+(?:[\.\+\-][0-9A-Za-z]+)?)'
    ),

    # Environment-style: VERSION = 1.2.3
    re.compile(
        r'(?<![A-Za-z0-9_])(VERSION\s*\=\s*)'
        r'(?P<ver>\d+\.\d+\.\d+(?:[\.\+\-][0-9A-Za-z]+)?)'
    ),

    # YAML: version: "1.2.3"
    re.compile(
        r'(?m)^(version\s*:\s*["\']?)'
        r'(?P<ver>\d+\.\d+\.\d+(?:[\.\+\-][0-9A-Za-z]+)?)'
        r'(["\']?)\s*$'
    ),
]


def update_version_in_file(file_path: Path, new_version: str) -> bool:
    """Replace version strings in a file based on VERSION_PATTERNS.

    Returns True if the file was updated.
    """
    content = file_path.read_text()
    new_content = content
    file_would_be_updated = False

    for pattern in VERSION_PATTERNS:
        def repl(match):
            nonlocal file_would_be_updated
            ver = match.group("ver")
            if ver != new_version:
                file_would_be_updated = True

                # Three-group patterns (__version__, JSON, YAML)
                if len(match.groups()) == 3:
                    return f"{match.group(1)}{new_version}{match.group(3)}"

                # Two-group patterns (Makefile)
                return f"{match.group(1)}{new_version}"

            return match.group(0)

        new_content = pattern.sub(repl, new_content)

    if file_would_be_updated:
        with file_path.open("w", encoding="utf-8", newline="\n") as file:
            file.write(new_content)

    return file_would_be_updated


def update_version_date_file() -> str:
    """Write current version date to _version_date.py, only if changed."""
    from akkudoktoreos.core.version import VERSION_DATE_FILE, _version_date_hash
    version_date, _ = _version_date_hash()
    version_date_utc = version_date.astimezone(timezone.utc)
    version_date_str = version_date_utc.isoformat()
    new_content = f'VERSION_DATE = "{version_date_str}"\n'

    if VERSION_DATE_FILE.exists() and VERSION_DATE_FILE.read_text(encoding="utf-8") == new_content:
        print(f"No change to {VERSION_DATE_FILE}")
        return str(VERSION_DATE_FILE)

    with VERSION_DATE_FILE.open("w", encoding="utf-8", newline="\n") as file:
        file.write(new_content)
    print(f"Updated {VERSION_DATE_FILE} with UTC date {version_date_str}")
    return str(VERSION_DATE_FILE)


def main(version: str, files: List[str]):
    if not version:
        raise ValueError("No version provided")
    if not files:
        raise ValueError("No files provided")

    updated_files = []
    for f in files:
        path = Path(f)
        if not path.exists():
            print(f"Warning: {path} does not exist, skipping")
            continue
        if update_version_in_file(path, version):
            updated_files.append(str(path))

    if updated_files:
        print(f"Updated files: {', '.join(updated_files)}")

        # Only update VERSION_DATE_FILE if a real package file was touched
        # Exclude VERSION_DATE_FILE itself to avoid a self-referencing loop
        from akkudoktoreos.core.version import VERSION_DATE_FILE
        package_files_updated = any(
            str(PACKAGE_DIR) in f and Path(f).resolve() != VERSION_DATE_FILE.resolve()
            for f in updated_files
        )
        if package_files_updated:
            updated_files.append(update_version_date_file())
    else:
        print("No files updated.")


if __name__ == "__main__":
    if len(sys.argv) < 2:
        print("Usage: python update_version.py <version> [<file1> [file2 ...]]")
        sys.exit(1)

    version_arg = sys.argv[1]
    if len(sys.argv) == 2:
        files_arg = [str(f) for f in DEFAULT_VERSION_FILES]
    else:
        files_arg = sys.argv[2:]
    main(version_arg, files_arg)