99 lines
2.6 KiB
Python
99 lines
2.6 KiB
Python
import asyncio
|
|
from pathlib import Path
|
|
|
|
import yaml
|
|
from integrations.docker.models import DockerContainer
|
|
from integrations.docker.ports import _get_service_port
|
|
|
|
COMPOSE_NAMES = (
|
|
"compose.yml",
|
|
"compose.yaml",
|
|
"docker-compose.yml",
|
|
"docker-compose.yaml",
|
|
)
|
|
|
|
|
|
def _get_visible_folders(directory: Path) -> list[Path]:
|
|
return [
|
|
p
|
|
for p in Path(directory).iterdir()
|
|
if p.is_dir() and not p.name.startswith(".")
|
|
]
|
|
|
|
|
|
def find_docker_compose(path: Path) -> tuple[dict, Path] | None:
|
|
if not path or not path.is_dir():
|
|
return None
|
|
|
|
for name in COMPOSE_NAMES:
|
|
file = path / name
|
|
if file.is_file():
|
|
content = yaml.safe_load(file.read_text()) or {}
|
|
return content, file
|
|
|
|
return None
|
|
|
|
|
|
def get_services(folder: Path) -> list[DockerContainer] | None:
|
|
services: list[DockerContainer] = []
|
|
compose_data = find_docker_compose(folder)
|
|
if compose_data is None:
|
|
return None
|
|
|
|
docker_compose, _ = compose_data
|
|
compose_services = docker_compose.get("services") or {}
|
|
if not compose_services:
|
|
return None
|
|
|
|
for service_name, config in compose_services.items():
|
|
mappings = config.get("ports") if isinstance(config, dict) else None
|
|
if not isinstance(mappings, list) or not mappings:
|
|
continue
|
|
|
|
ports = [
|
|
p
|
|
for p in (_get_service_port(mapping) for mapping in mappings)
|
|
if p is not None
|
|
]
|
|
if not ports:
|
|
continue
|
|
|
|
services.append({"name": service_name, "ports": ports, "folder": folder})
|
|
|
|
return services
|
|
|
|
|
|
def get_all_services_in_dir(directory: str | Path) -> list[DockerContainer]:
|
|
path = Path(directory)
|
|
if not path.is_dir():
|
|
return []
|
|
|
|
services: list[DockerContainer] = []
|
|
for sub_folder in _get_visible_folders(path):
|
|
sub_services = get_services(sub_folder)
|
|
if sub_services:
|
|
services.extend(sub_services)
|
|
|
|
return services
|
|
|
|
|
|
async def run_docker_command(file: Path, command: list[str]) -> dict | None:
|
|
if not command:
|
|
return None
|
|
|
|
full_command = ["docker", "compose", "-f", f"{file}", *command]
|
|
|
|
process = await asyncio.create_subprocess_exec(
|
|
*full_command,
|
|
cwd=str(file.parent),
|
|
stdout=asyncio.subprocess.PIPE,
|
|
stderr=asyncio.subprocess.PIPE,
|
|
)
|
|
|
|
stdout, stderr = await process.communicate()
|
|
|
|
if process.returncode == 0:
|
|
return {"status": "success", "output": stdout.decode()}
|
|
else:
|
|
return {"status": "error", "reason": stderr.decode()}
|