spcd/cd/__init__.py
2024-06-05 11:55:25 +02:00

121 lines
2.5 KiB
Python

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():
print(f'''\
{project.root}
''', end=str())
paths = []
for directory, directories, files in os.walk(project.root):
for file in files:
absolute_path = os.path.join(directory, file)
relative_path = os.path.relpath(absolute_path, start=project.root)
paths.append(relative_path)
for path in sorted(paths):
print(path)
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 install_commands(path):
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):
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)
def split():
print(env.SPLIT, flush=True)
def step(*arguments):
env.STEP += 1
print(env.DOWN)
print(env.VERTICAL, env.STEP, *arguments)
print(env.UP, flush=True)