#!/usr/bin/python3

import os
import subprocess
import sys
from osbuild.util import jsoncomm


def setup_stdio():
    with jsoncomm.Socket.new_client("/run/osbuild/api/osbuild") as client:
        req = {'method': 'setup-stdio'}
        client.send(req)
        msg, fds, _ = client.recv()
        for io in ['stdin', 'stdout', 'stderr']:
            target = getattr(sys, io)
            source = fds[msg[io]]
            os.dup2(source, target.fileno())
        fds.close()


if __name__ == "__main__":
    setup_stdio()

    r = subprocess.run(sys.argv[1:], check=False)
    sys.exit(r.returncode)