import socket, argparse
import select, sys, os, tty, atexit, termios, signal, shutil
import fcntl, termios, struct

hasVarlink = False
try:
    from varlink import (Client, VarlinkError)
    hasVarlink = True
except ImportError:
    pass


sock = socket.socket(socket.AF_UNIX, socket.SOCK_SEQPACKET)
address = "unix:/run/io.projectatomic.podman"

# stdin 1
# stdout 2
# stderr 3


parser = argparse.ArgumentParser()
parser.add_argument('container', help='container name or id')
args = parser.parse_args()
socket_path = ""
control_socket_path = ""

if hasVarlink:
    with Client(address=address) as client:
        podman = client.open('io.projectatomic.podman')
        try:
            response = podman.GetAttachSockets(args.container)
        except VarlinkError as e:
            sys.exit(e.error())

        socket_path = response["sockets"]["io_socket"]
        control_socket_path = response["sockets"]["control_socket"]
else:
    sockets_dir = os.path.join("/var/run/libpod/socket/", args.container)
    socket_path = os.path.join(sockets_dir, "attach")
    control_socket_path = os.path.join(sockets_dir, "ctl")


def reset_fd(my_fd, old_settings):
    termios.tcsetattr(my_fd, termios.TCSADRAIN, old_settings)
    os.close(my_fd)


def error_print(*args, **kwargs):
    print(*args, file=sys.stderr, **kwargs)

def terminal_size():
    h, w, hp, wp = struct.unpack('HHHH',
        fcntl.ioctl(1, termios.TIOCGWINSZ,
        struct.pack('HHHH', 0, 0, 0, 0)))
    return w, h

def resize_handler(signum, frame):
    control_socket = open(control_socket_path, 'w')
    # "%d %d %d\n", 1, size.Height, size.Width)
    w, h = terminal_size()
    #term_size = shutil.get_terminal_size()
    print(w,h)
    control_socket.write("{} {} {}\n".format(1, h, w))
    control_socket.close()
    #control_socket.flush()


signal.signal(signal.SIGWINCH, resize_handler)

sock.connect(socket_path)

stdin = os.open("/dev/stdin", os.O_RDWR)
old_settings = termios.tcgetattr(stdin)
atexit.register(reset_fd, stdin, old_settings)
tty.setraw(stdin)

inputs = [sock, stdin]
outputs = []

while inputs:
    try:
        readable, writable, exceptional = select.select(inputs, outputs, inputs)
        for r in readable:
            if isinstance(r, socket.socket):
                data = r.recv(1024)
                os.write(2, data[1:])

            if isinstance(r, int):
                data = os.read(stdin, 1024)
                if 29 in data:  # Bail on ctrl-[
                    sys.exit(0)
                sock.send(data)
    except BrokenPipeError:
        error_print("socket quit on container end")
        sys.exit(1)