from invoke import task
"""



as root on ditto, run:
% inv init-workspace
then
% inv setup-pi ga-iot 5811792
"""


cmd = '''
HOME=/root
export HOME
eval `keychain --quiet --eval id_ecdsa`
'''

@task
def init_workspace(ctx):
    ctx.run(cmd + 'pdm run -p .. setup_pi.py --init', pty=True)

@task
def setup_pi(ctx, hostname, serial):
    ctx.run("qemu-img --version") # fail fast
    ctx.run(cmd + f'pdm run -p .. setup_pi.py {hostname} {serial}', pty=True)


@task
def iscsi(ctx, hostname):
    ctx.run(cmd + f'pdm run -p .. setup_pi.py --iscsi {hostname}', pty=True)