spcd/spcd/__init__.py
2024-08-19 10:11:39 +02:00

127 lines
3 KiB
Python

"""Shell to Python Continuous Deployment."""
__version__ = "0.0.1"
import sys
from os import environ, pathsep
from pathlib import Path
import env
from rwx import fs
from rwx.log import stream as log
from rwx.ps import run
from spcd import cmd
from spcd.ci import project, projects
from spcd.util import browse, cat, split, step
COMMANDS_PREFIX = "spcd-"
def clone_project_branch() -> None:
if not projects.environment.get("GITLAB_CI"):
step("Clone project branch")
log.info(projects)
split()
log.info(project)
split()
log.info(f"""\
{project.url}
""")
run(
"git",
"clone",
"--branch",
project.branch,
"--",
project.url,
project.root,
)
def install_commands(path: str) -> None:
step("Install commands")
user = Path("/usr/local/bin")
for command in [
"browse-workspace",
"build-project",
"check-project",
"synchronize",
]:
log.info(command)
(user / f"{COMMANDS_PREFIX}{command}").symlink_to(path)
def install_python_packages() -> None:
step("Install Python packages")
log.info("pip")
run("pip", "install", "--upgrade", "pip")
split()
packages = [
"mypy",
"pelican",
"pytest",
"ruff",
"sphinx",
"sphinx-rtd-theme",
"twine",
]
for package in packages:
log.info(package)
run("pip", "install", *packages)
def list_environment_variables() -> None:
step("List environment variables")
for variable, value in sorted(projects.environment.items()):
if variable not in ["SPCD", "SPCD_SSH_KEY"]:
log.info(f"{variable} = {value}")
else:
log.info(f"{variable}")
def main(main: str) -> None:
paths = environ["PATH"].split(pathsep)
if env.SPCD_PYTHON_VENV_BINARIES not in paths:
environ["PATH"] = pathsep.join([env.SPCD_PYTHON_VENV_BINARIES, *paths])
path, *arguments = sys.argv
name = Path(path).name
if name == "__main__.py":
list_environment_variables()
clone_project_branch()
set_ssh()
install_commands(main)
install_python_packages()
else:
function = getattr(cmd, name.replace("-", "_"))
function(*arguments)
def set_ssh() -> None:
step("Set SSH")
# get variables
ssh_hosts = projects.environment.get("SPCD_SSH_HOSTS")
ssh_key = projects.environment.get("SPCD_SSH_KEY")
# set key type
ssh_type = "ed25519"
# set home directory
home = Path("~").expanduser()
# make home directory
ssh = home / ".ssh"
ssh.mkdir(exist_ok=True, parents=True)
ssh.chmod(0o700)
# write private key
key = ssh / f"id_{ssh_type}"
if ssh_key:
fs.write(key, ssh_key)
key.chmod(0o400)
# write known hosts
known = ssh / "known_hosts"
if ssh_hosts:
fs.write(known, ssh_hosts)
known.chmod(0o400)
# display
browse(ssh)
if known.exists():
cat(known)