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 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)
    #
    browse(home)


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)