mirror of
https://github.com/chatmail/relay.git
synced 2026-05-13 01:24:36 +00:00
98 lines
3.6 KiB
Python
98 lines
3.6 KiB
Python
import subprocess
|
|
import sys
|
|
import time
|
|
import os
|
|
import shutil
|
|
|
|
# This script automates LXC container creation and relay deployment testing.
|
|
# It cleans up old containers, creates a new one, and runs the deployment.
|
|
|
|
CONTAINER_PREFIX = "testrelay"
|
|
CONTAINER_NAME = f"{CONTAINER_PREFIX}-auto"
|
|
RELAY_SRC = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
|
|
|
|
def run_cmd(cmd, check=True, shell=True):
|
|
print(f"Executing: {cmd}")
|
|
try:
|
|
result = subprocess.run(cmd, shell=shell, check=check, capture_output=True, text=True)
|
|
return result.stdout.strip()
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"Error executing command: {e}")
|
|
print(f"Stdout: {e.stdout}")
|
|
print(f"Stderr: {e.stderr}")
|
|
if check:
|
|
sys.exit(1)
|
|
return None
|
|
|
|
def cleanup():
|
|
print("--- Cleaning up old containers ---")
|
|
containers = run_cmd("sudo lxc-ls -1").splitlines()
|
|
for c in containers:
|
|
if c.startswith(CONTAINER_PREFIX):
|
|
print(f"Destroying container: {c}")
|
|
run_cmd(f"sudo lxc-destroy -n {c} -f", check=False)
|
|
|
|
def create_container():
|
|
print(f"--- Creating container: {CONTAINER_NAME} ---")
|
|
run_cmd(f"sudo lxc-create -n {CONTAINER_NAME} -t download -- -d debian -r bookworm -a amd64")
|
|
|
|
def run_in_container(name, command):
|
|
return run_cmd(f'sudo lxc-attach -n {name} -- bash -c "{command}"')
|
|
|
|
def prepare_container():
|
|
print("--- Preparing container ---")
|
|
run_cmd(f"sudo lxc-start -n {CONTAINER_NAME}")
|
|
|
|
# Wait for IP
|
|
print("Waiting for container IP...")
|
|
for _ in range(30):
|
|
ip = run_cmd(f"sudo lxc-info -n {CONTAINER_NAME} -i").split()
|
|
if ip and len(ip) > 1:
|
|
print(f"Container IP: {ip[1]}")
|
|
break
|
|
time.sleep(1)
|
|
else:
|
|
print("Failed to get container IP.")
|
|
sys.exit(1)
|
|
|
|
# Disable systemd-resolved to free port 53
|
|
run_in_container(CONTAINER_NAME, "systemctl disable --now systemd-resolved || true")
|
|
run_in_container(CONTAINER_NAME, "echo 'nameserver 8.8.8.8' > /etc/resolv.conf")
|
|
|
|
# Sync code
|
|
print("Syncing code to container...")
|
|
target_path = f"/var/lib/lxc/{CONTAINER_NAME}/rootfs/relay-ir"
|
|
run_cmd(f"sudo mkdir -p {target_path}")
|
|
run_cmd(f"sudo rsync -av --exclude=.git --exclude='*.log' --exclude='test' {RELAY_SRC}/ {target_path}/")
|
|
|
|
# Install dependencies
|
|
print("Installing dependencies inside container...")
|
|
run_in_container(CONTAINER_NAME, "apt-get update && apt-get install -y curl sudo python3-dev gcc")
|
|
run_in_container(CONTAINER_NAME, "apt-get remove --purge -y exim4*")
|
|
run_in_container(CONTAINER_NAME, "curl -LsSf https://astral.sh/uv/install.sh | sh")
|
|
|
|
def perform_deployment():
|
|
print("--- Running deployment test ---")
|
|
# Using the PATH where uv is installed
|
|
path_env = "export PATH='/root/.local/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin'"
|
|
deploy_cmd = f"cd /relay-ir && {path_env} && ./scripts/initenv.sh && ./scripts/cmdeploy run --ssh-host @local --skip-dns-check"
|
|
|
|
try:
|
|
output = run_in_container(CONTAINER_NAME, deploy_cmd)
|
|
print("Deployment Output Summary:")
|
|
print(output[-1000:] if output else "No output") # Show last bit of output
|
|
print("\nSUCCESS: Deployment completed.")
|
|
except Exception:
|
|
print("\nFAILURE: Deployment failed.")
|
|
sys.exit(1)
|
|
|
|
if __name__ == "__main__":
|
|
if os.geteuid() != 0:
|
|
print("This script must be run as root/sudo.")
|
|
# We don't exit here because the inner run_cmd uses sudo, but it's better to be root.
|
|
|
|
cleanup()
|
|
create_container()
|
|
prepare_container()
|
|
perform_deployment()
|