57 lines
4.2 KiB
Python
Executable File
57 lines
4.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""Detect mixed source trees and preserve the known obsolete 0.0.1 auth module."""
|
|
from __future__ import annotations
|
|
import argparse
|
|
import hashlib
|
|
from pathlib import Path
|
|
import sys
|
|
|
|
LEGACY_AUTH = "4df9d473c9214365d91c83072f97aafab6ede4edcf4e1369b5c3679ded5cff85"
|
|
|
|
|
|
def check(root: Path, repair_known: bool = False) -> list[str]:
|
|
messages: list[str] = []
|
|
legacy = root / "src/auth.rs"
|
|
if legacy.exists() and (root / "src/auth/mod.rs").is_file():
|
|
if legacy.is_symlink() or hashlib.sha256(legacy.read_bytes()).hexdigest() != LEGACY_AUTH:
|
|
raise ValueError("Both src/auth.rs and src/auth/mod.rs exist. The single-file module is not the known 0.0.1 file. Preserve your edits and move it outside src before rebuilding. See docs/MIGRATION.md.")
|
|
if not repair_known:
|
|
raise ValueError("Known obsolete src/auth.rs from 0.0.1 found. Run ./dev.sh repair or start ./dev.sh to preserve it in .upgrade-backup/legacy-auth/.")
|
|
directory = root / ".upgrade-backup/legacy-auth"
|
|
directory.mkdir(parents=True, exist_ok=True)
|
|
target = directory / "auth.rs.0.0.1"
|
|
if target.exists():
|
|
if hashlib.sha256(target.read_bytes()).hexdigest() != LEGACY_AUTH:
|
|
raise ValueError("Legacy backup path contains a different file; refusing to overwrite it.")
|
|
legacy.unlink() # Byte-identical content was already preserved.
|
|
else:
|
|
legacy.rename(target)
|
|
messages.append("Preserved obsolete src/auth.rs as .upgrade-backup/legacy-auth/auth.rs.0.0.1")
|
|
for module in (root / "src").rglob("mod.rs"):
|
|
flat = module.parent.with_suffix(".rs")
|
|
if flat.is_file():
|
|
raise ValueError(f"Conflicting Rust modules: {flat.relative_to(root)} and {module.relative_to(root)}")
|
|
for path in ["src/auth/mod.rs", "src/auth/store.rs", "src/upstream.rs", "src/tls.rs", "src/proxy/websocket.rs", "src/cache/mod.rs", "src/cache/model.rs", "src/cache/lifecycle.rs", "src/cache/cold.rs", "src/cache/dashboard.rs", "src/cache/tickets.rs", "src/cache/lookup.rs", "src/cache/policy.rs", "src/cache/fill.rs", "src/cache/tests.rs", "src/settings.rs", "src/compression.rs", "src/diagnostics.rs", "src/system_stats/mod.rs", "src/system_stats/linux.rs", "src/vhost.rs", "src/balance.rs", "src/security.rs", "src/admin/web/vhosts.js", "src/admin/web/streams.js", "src/admin/web/host404.js", "src/admin/web/management.js", "src/pages.rs", "scripts/mock_server.py", "scripts/mock_auth.py", "src/forwarding.rs", "src/tinyauth.rs", "src/http3.rs", "src/stream.rs", "src/admin/web/integrations.js", "src/prometheus.rs", "src/admin/web/settings.js", "src/admin/web/docker.js", "src/docker.rs", "docker-agent/main.go", "docker-agent/main_test.go", "docker-agent/rustproxy-docker-agent.service", "docker-agent/install.sh", "docs/DOCKER_DISCOVERY.md", "scripts/acme_provider.py", "src/basic_auth.rs", "scripts/deploy.py", "scripts/migrate_npm.py", "docs/NPM_MIGRATION.md", "docs/STREAMS.md", "install.sh", "install-debian13.sh", "update.sh", "build.rs", "scripts/swagger_assets.py", "src/admin/documentation.rs", "src/admin/openapi.json", "langs/en_US.json", "langs/pl_pl.json", "Dockerfile", ".dockerignore", "docs/CONTAINER.md", "docs/PROMETHEUS.md", "docs/NOT_FOUND_HOSTS.md", "debian/control", "debian/rules", "debian/rustproxy.service", "debian/rustproxy.logrotate", "grafana/rustproxy-overview.json"]:
|
|
if not (root / path).is_file():
|
|
raise ValueError(f"Incomplete RustProxy 0.24.10 source tree: missing {path}")
|
|
text = (root / "src/lib.rs").read_text(encoding="utf-8")
|
|
if '#[path = "auth/mod.rs"]' not in text:
|
|
raise ValueError("src/lib.rs is not the 0.24.10 entry point; extract the complete package, not individual files.")
|
|
return messages
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument("--root", type=Path, default=Path(__file__).resolve().parents[1])
|
|
parser.add_argument("--repair-known", action="store_true")
|
|
args = parser.parse_args()
|
|
try:
|
|
for line in check(args.root.resolve(), args.repair_known):
|
|
print(line, file=sys.stderr)
|
|
except (OSError, ValueError) as error:
|
|
parser.exit(2, f"Source preflight failed: {error}\n")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|