110 lines
3.3 KiB
Python
Executable File
110 lines
3.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import sys
|
|
import subprocess
|
|
from pathlib import Path
|
|
from typing import Tuple
|
|
|
|
try:
|
|
import yaml
|
|
except Exception:
|
|
sys.stderr.write("PyYAML is required to run this check.\n")
|
|
sys.exit(2)
|
|
|
|
EXPECTED_IMAGE = "cooldockerizer93/spotizerr"
|
|
|
|
|
|
def load_compose(path: Path):
|
|
with path.open("r", encoding="utf-8") as f:
|
|
return yaml.safe_load(f) or {}
|
|
|
|
|
|
def save_compose(path: Path, data) -> None:
|
|
with path.open("w", encoding="utf-8") as f:
|
|
yaml.safe_dump(data, f, sort_keys=False)
|
|
|
|
|
|
def ensure_image_unversioned(data) -> Tuple[bool, str, str]:
|
|
"""
|
|
Returns (changed, old_image, new_image)
|
|
"""
|
|
services = (data or {}).get("services", {})
|
|
svc = services.get("spotizerr", {})
|
|
image = svc.get("image")
|
|
if image == EXPECTED_IMAGE:
|
|
return False, image, image
|
|
|
|
# Normalize to expected image if it has a tag/digest or is different
|
|
svc["image"] = EXPECTED_IMAGE
|
|
services["spotizerr"] = svc
|
|
data["services"] = services
|
|
return True, image, EXPECTED_IMAGE
|
|
|
|
|
|
def git(*args: str) -> subprocess.CompletedProcess:
|
|
return subprocess.run(["git", *args], check=False, text=True, capture_output=True)
|
|
|
|
|
|
def autocommit(file_path: str) -> None:
|
|
# Configure git identity if missing
|
|
git("config", "user.name").stdout
|
|
if git("config", "user.name").stdout.strip() == "":
|
|
git("config", "user.name", "github-actions[bot]")
|
|
if git("config", "user.email").stdout.strip() == "":
|
|
git("config", "user.email", "github-actions[bot]@users.noreply.github.com")
|
|
# Stage and commit
|
|
git("add", file_path)
|
|
status = git("status", "--porcelain").stdout.strip()
|
|
if status:
|
|
msg = "chore: normalize docker-compose image to cooldockerizer93/spotizerr"
|
|
commit_res = git("commit", "-m", msg)
|
|
if commit_res.returncode != 0:
|
|
sys.stderr.write(f"Git commit failed: {commit_res.stderr}\n")
|
|
sys.exit(1)
|
|
push_res = git("push")
|
|
if push_res.returncode != 0:
|
|
sys.stderr.write(f"Git push failed: {push_res.stderr}\n")
|
|
sys.exit(1)
|
|
print("Pushed normalization commit")
|
|
else:
|
|
print("No changes to commit")
|
|
|
|
|
|
def main(argv: list[str]) -> int:
|
|
# Usage: ensure_compose_image.py [docker-compose.yaml] [--autocommit]
|
|
compose_path = (
|
|
Path(argv[1])
|
|
if len(argv) > 1 and not argv[1].startswith("-")
|
|
else Path("docker-compose.yaml")
|
|
)
|
|
do_autocommit = any(arg == "--autocommit" for arg in argv[1:])
|
|
|
|
if not compose_path.exists():
|
|
sys.stderr.write(f"File not found: {compose_path}\n")
|
|
return 1
|
|
|
|
try:
|
|
data = load_compose(compose_path)
|
|
except Exception as e:
|
|
sys.stderr.write(f"Failed to parse YAML from {compose_path}: {e}\n")
|
|
return 1
|
|
|
|
changed, old_image, new_image = ensure_image_unversioned(data)
|
|
|
|
if changed:
|
|
save_compose(compose_path, data)
|
|
sys.stderr.write(
|
|
f"Normalized services.spotizerr.image from '{old_image}' to '{new_image}'\n"
|
|
)
|
|
# For pre-commit: exit non-zero to force user to re-stage
|
|
if do_autocommit:
|
|
autocommit(str(compose_path))
|
|
return 0
|
|
return 1
|
|
|
|
print(f"OK: docker-compose image is '{EXPECTED_IMAGE}'")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main(sys.argv))
|