mirror of
https://github.com/Cian-H/I-Form_Server_Node_Deployer.git
synced 2025-12-23 06:32:08 +00:00
98 lines
2.7 KiB
Python
98 lines
2.7 KiB
Python
from fnmatch import fnmatch
|
|
from typing import Annotated
|
|
|
|
import typer
|
|
|
|
from config import (
|
|
CLEANUP_IMAGES,
|
|
CLIENT,
|
|
CWD_MOUNTDIR,
|
|
DOCKERFILE_DIR,
|
|
ROOT_DIR,
|
|
)
|
|
from create_img import create_img
|
|
from debug import debug_mode
|
|
import docker
|
|
|
|
|
|
def filter_validation_response(response: str) -> str:
|
|
return "\n".join(
|
|
filter(
|
|
# Filter out the warning about unused key human_readable, this always exists in
|
|
# configurations produced by fuel-ignition
|
|
lambda x: not fnmatch(x.strip(), "warning at*Unused key human_read"),
|
|
response.split("\n"),
|
|
)
|
|
).strip()
|
|
|
|
|
|
def validation_result() -> str:
|
|
dockerfile = DOCKERFILE_DIR / "validate.dockerfile"
|
|
image = CLIENT.images.build(
|
|
path=".",
|
|
dockerfile=str(dockerfile),
|
|
tag="validate",
|
|
buildargs={"CWD_MOUNTDIR": str(CWD_MOUNTDIR)},
|
|
rm=CLEANUP_IMAGES,
|
|
pull=True,
|
|
quiet=True,
|
|
)
|
|
response = CLIENT.containers.run(
|
|
image,
|
|
mounts=[
|
|
docker.types.Mount(
|
|
target=str(CWD_MOUNTDIR),
|
|
source=str(ROOT_DIR),
|
|
type="bind",
|
|
)
|
|
],
|
|
remove=True,
|
|
)
|
|
if CLEANUP_IMAGES:
|
|
image.remove(force=True)
|
|
return response
|
|
|
|
|
|
def validate() -> (bool, str):
|
|
response = validation_result().decode()
|
|
response = filter_validation_response(response)
|
|
return (not bool(response), response)
|
|
|
|
|
|
def main(
|
|
hostname: Annotated[str, typer.Option(help="Hostname for the new node", prompt=True)],
|
|
password: Annotated[
|
|
str,
|
|
typer.Option(
|
|
help="Password for the root user on the new node",
|
|
prompt=True,
|
|
confirmation_prompt=True,
|
|
hide_input=True,
|
|
),
|
|
],
|
|
switch_ip_address: Annotated[
|
|
str, typer.Option(help="IP address of the switch to connect to", prompt=True)
|
|
],
|
|
switch_port: Annotated[int, typer.Option(help="Port on the switch to connect to", prompt=True)],
|
|
swarm_token: Annotated[
|
|
str, typer.Option(help="Swarm token for connecting to the swarm", prompt=True)
|
|
],
|
|
debug: Annotated[bool, typer.Option(help="Enable debug mode")] = False,
|
|
) -> None:
|
|
debug_mode(debug)
|
|
# f = create_img
|
|
# if debug:
|
|
# f = ss(f) # noqa: F821, # type: ignore #? ss is installed in debug_mode
|
|
# f(hostname, password, switch_ip_address, switch_port, swarm_token)
|
|
create_img(hostname, password, switch_ip_address, switch_port, swarm_token)
|
|
valid, response = validate()
|
|
if not valid:
|
|
print(response)
|
|
raise typer.Exit(1)
|
|
else:
|
|
print("Valid ignition image created!")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
typer.run(main)
|