86 lines
2.3 KiB
Python
86 lines
2.3 KiB
Python
import asyncio
|
|
from pathlib import Path
|
|
|
|
import yaml
|
|
|
|
from app.integrations.docker.compose import (
|
|
ComposeDocument,
|
|
find_docker_compose,
|
|
run_docker_command,
|
|
)
|
|
from app.integrations.docker.ports import _get_port_index, _replace_port_in_mapping
|
|
|
|
|
|
def _render_compose(document: ComposeDocument) -> str:
|
|
return yaml.dump(document, default_flow_style=False, sort_keys=False)
|
|
|
|
|
|
def _write_compose(path: Path, content: str) -> None:
|
|
path.write_text(content, encoding="utf-8")
|
|
|
|
|
|
def _read_compose_text(path: Path) -> str:
|
|
return path.read_text(encoding="utf-8")
|
|
|
|
|
|
async def change_port(
|
|
service_name: str,
|
|
folder: Path,
|
|
port_from: int,
|
|
port_to: int,
|
|
host_to: str | None = None,
|
|
rewrite_host: bool = False,
|
|
) -> bool:
|
|
compose_data = find_docker_compose(folder)
|
|
if compose_data is None:
|
|
return False
|
|
|
|
docker_compose, compose_path = compose_data
|
|
compose_services = docker_compose.get("services") or {}
|
|
if not isinstance(compose_services, dict) or not compose_services:
|
|
return False
|
|
|
|
service_config = compose_services.get(service_name)
|
|
if not isinstance(service_config, dict):
|
|
return False
|
|
|
|
mappings = service_config.get("ports")
|
|
if not isinstance(mappings, list):
|
|
return False
|
|
|
|
idx = _get_port_index(mappings, port_from)
|
|
if idx is None:
|
|
return False
|
|
|
|
mappings[idx] = _replace_port_in_mapping(
|
|
mappings[idx],
|
|
port_from,
|
|
port_to,
|
|
host_to=host_to,
|
|
rewrite_host=rewrite_host,
|
|
)
|
|
|
|
original_text = await asyncio.to_thread(_read_compose_text, compose_path)
|
|
updated_text = _render_compose(docker_compose)
|
|
|
|
try:
|
|
await asyncio.to_thread(_write_compose, compose_path, updated_text)
|
|
result = await run_docker_command(
|
|
compose_path,
|
|
["up", "-d", service_name],
|
|
)
|
|
except Exception:
|
|
try:
|
|
await asyncio.to_thread(_write_compose, compose_path, original_text)
|
|
except OSError as restore_error:
|
|
raise RuntimeError(
|
|
f"Failed to restore original compose file: {compose_path}"
|
|
) from restore_error
|
|
return False
|
|
|
|
if result is None or result.get("status") != "success":
|
|
await asyncio.to_thread(_write_compose, compose_path, original_text)
|
|
return False
|
|
|
|
return True
|