import socket
import sys
import time


def main():
    host = sys.argv[1] if len(sys.argv) > 1 else "127.0.0.1"
    port = int(sys.argv[2]) if len(sys.argv) > 2 else 8080
    idle_timeout = int(sys.argv[3]) if len(sys.argv) > 3 else 3

    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.settimeout(10)
    s.connect((host, port))

    # CONNECT to echo server
    req = b"CONNECT tcp-echo:5678 HTTP/1.1\r\nHost: tcp-echo:5678\r\n\r\n"
    s.sendall(req)

    # Read 200 OK response
    resp = b""
    while b"\r\n\r\n" not in resp:
        chunk = s.recv(4096)
        if not chunk:
            break
        resp += chunk

    if b"200" not in resp:
        print(f"FAIL: expected 200, got {resp.decode(errors='replace')}")
        sys.exit(1)

    # Send request through the CONNECT tunnel
    s.sendall(b"ping-1\n")
    data = s.recv(4096)
    if b"ping-1" not in data:
        print(f"FAIL: expected ping-1 echo, got {data!r}")
        sys.exit(1)
    print(f"PASS: first request through tunnel succeeded")

    # Wait longer than idleTimeout
    wait = idle_timeout + 2
    print(f"Waiting {wait}s for idle timeout...")
    time.sleep(wait)

    # Try to send more data — idle timeout should have closed the pipe
    try:
        s.sendall(b"ping-2\n")
        data = s.recv(4096)
        if not data:
            print("PASS: connection closed after idle timeout (empty recv)")
            sys.exit(0)
        # Got data means the pipe is still alive
        print(f"FAIL: connection still alive after idle timeout, got {data!r}")
        sys.exit(1)
    except (socket.timeout, ConnectionResetError, BrokenPipeError, OSError) as e:
        print(f"PASS: connection closed after idle timeout: {e}")
        sys.exit(0)


if __name__ == "__main__":
    main()