spcd/cd/__init__.py

import os

from cd.project import Project
from cd.projects import Projects
import env
from rwx import fs
from rwx import ps

COMMANDS_PREFIX = "cd-"

projects = Projects(os.environ)
project = Project(projects)


def cd_browse_workspace():
    browse(project.root)


def cd_build_project():
    for extension in ["py", "sh"]:
        path = os.path.join(project.root, f"build.{extension}")
        if os.path.exists(path):
            ps.run(path)
            break
    else:
        pass


def cd_clone_branch():
    print(f"""\
{project.url}
↓
""", end=str(), flush=True)
    ps.run("git",
        "clone",
        "--branch", project.branch,
        "--",
        project.url,
        project.root,
    )


def cd_list_environment():
    for variable, value in sorted(projects.environment.items()):
        print(variable, "=", value)


def cd_synchronize():
    host = "rwx.work"
    source = "out"
    user = "cd"
    #
    root = os.sep.join([str(),
                        user, project.branch, projects.group, project.name])
    #
    target = f"{user}@{host}:{root}"
    ps.run("rsync",
        "--archive",
        "--delete-before",
        "--verbose",
        f"{source}/",
        f"{target}/",
        "--dry-run",
    )


def browse(root: str):
    paths = []
    for directory, _, files in os.walk(root):
        for file in files:
            absolute_path = os.path.join(directory, file)
            relative_path = os.path.relpath(absolute_path, start=root)
            paths.append(relative_path)
    open(root)
    for path in sorted(paths):
        print(path)
    shut(root)


def cat(file: str):
    open(file)
    print(fs.read_file_text(file).rstrip())
    shut(file)


def install_commands(path):
    step("Install commands")
    user = "/usr/local/bin"
    for command in [
        "browse-workspace",
        "build-project",
        "clone-branch",
        "list-environment",
        "synchronize",
    ]:
        print(command)
        os.symlink(path, os.path.join(user, f"{COMMANDS_PREFIX}{command}"))


def set_ssh(*arguments):
    step("Set SSH")
    #
    ssh_key, ssh_hosts = arguments
    #
    ssh_type = "ed25519"
    #
    home = os.path.expanduser("~")
    #
    ssh = os.path.join(home, ".ssh")
    os.makedirs(ssh, exist_ok=True)
    os.chmod(ssh, 0o700)
    #
    key = os.path.join(ssh, f"id_{ssh_type}")
    if ssh_key:
        fs.write(key, ssh_key)
        os.chmod(key, 0o400)
    #
    known = os.path.join(ssh, "known_hosts")
    if ssh_hosts:
        fs.write(known, ssh_hosts)
        os.chmod(known, 0o400)
        cat(known)
    #
    browse(ssh)


def open(*arguments):
    print(env.CD_OPEN, end=str())
    print(*arguments, flush=True)


def shut(*arguments):
    print(env.CD_SHUT, end=str())
    print(*arguments, flush=True)


def split():
    print(env.CD_SPLT, flush=True)


def step(*arguments):
    env.CD_STEP += 1
    print(env.CD_DOWN)
    print(env.CD_VERT, env.CD_STEP, *arguments)
    print(env.CD___UP, flush=True)