101 lines
2.5 KiB
Python
101 lines
2.5 KiB
Python
import socket
|
|
from pathlib import Path
|
|
|
|
import yaml
|
|
|
|
from utils.docker.ports import _get_service_port
|
|
from utils.docker.types import DockerService
|
|
|
|
COMPOSE_NAMES = (
|
|
"compose.yml",
|
|
"compose.yaml",
|
|
"docker-compose.yml",
|
|
"docker-compose.yaml"
|
|
)
|
|
|
|
|
|
def _get_random_port() -> int:
|
|
try:
|
|
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
|
|
s.bind(('', 0))
|
|
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
|
port = s.getsockname()[1]
|
|
return port
|
|
except Exception as e:
|
|
raise RuntimeError(f"Could not find an available port: {e}")
|
|
|
|
|
|
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, str] | None:
|
|
if not path or not path.is_dir():
|
|
return None
|
|
|
|
for name in COMPOSE_NAMES:
|
|
file = path / name
|
|
if file.is_file():
|
|
return yaml.safe_load(file.read_text()), file.name
|
|
|
|
return None
|
|
|
|
|
|
def get_services(folder: Path) -> list[DockerService] | None:
|
|
"""
|
|
Returns a list of all public services by automatically finding the docker file of a fodler
|
|
|
|
:param folder:
|
|
:return:
|
|
6
|
|
Usage::
|
|
services = get_services("/root/CApp")
|
|
"""
|
|
try:
|
|
docker_compose, _ = find_docker_compose(folder)
|
|
except TypeError: # nessun docker_compose nella cartella
|
|
return None
|
|
|
|
if not docker_compose or not docker_compose.get('services'):
|
|
return None
|
|
|
|
services: list[DockerService] = []
|
|
|
|
for service, config in docker_compose["services"].items():
|
|
if not config.get("ports"):
|
|
continue
|
|
ports = [p for p in (_get_service_port(port) for port in config["ports"]) if p is not None]
|
|
if not ports:
|
|
continue
|
|
services.append({"name": service, "ports": ports})
|
|
|
|
return services or None
|
|
|
|
|
|
def get_all_services(directory: str) -> list[DockerService] | None:
|
|
"""
|
|
Returns a list of all public services in all sub folders of a directory
|
|
:param directory:
|
|
:return:
|
|
|
|
Usage::
|
|
services = get_all_services("/root/")
|
|
"""
|
|
path = Path(directory)
|
|
if not path.is_dir():
|
|
return None
|
|
|
|
sub_folders = get_visible_folders(path)
|
|
if not sub_folders:
|
|
return None
|
|
|
|
services = []
|
|
for sub_folder in sub_folders:
|
|
if s := get_services(sub_folder):
|
|
services.append(s)
|
|
|
|
return services or None
|