#!/usr/bin/env python3
"""Validate the proposed minimal XIE 2.3 client sequence on one connection:
   INIT_SESSION(1) -> QUERY_OP_DEFAULTS(3) -> QUERY_EVENTS(6) -> TERM_SESSION(2)
"""
import os, sys, struct
sys.path.insert(0, os.path.join(os.path.dirname(os.path.abspath(__file__)), "evidence"))
import xie_raw as X

s = X.socket.create_connection((X.HOST, X.PORT), timeout=5)
s.settimeout(5.0)
try:
    X.setup(s)
    present, opcode, ev, er = X.query_extension(s, "Xie")
    print("extension present=%d opcode=%d first_event=%d first_error=%d" % (present, opcode, ev, er))

    def send(minor, payload=b"", words=None):
        if words is None:
            words = 1 + len(payload) // 4
        data = payload.ljust((words * 4) - 4, b"\0")
        s.sendall(struct.pack("<BBH", opcode, minor, words) + data)
        try:
            kind, head, extra = X.read_packet(s)
        except X.socket.timeout:
            return "TIMEOUT"
        if kind == 1:
            return "REPLY len=%d  %s" % (struct.unpack("<I", head[4:8])[0], head.hex())
        if kind == 0:
            return "ERROR code=%d bad=%#x minor_op=%d" % (
                head[1], struct.unpack("<I", head[4:8])[0], struct.unpack("<H", head[8:10])[0])
        return "EVENT type=%d" % kind

    print("init   (1,2w):", send(1, struct.pack("<HH", 2, 3), 2))
    print("qdefs  (3,2w):", send(3, b"\0\0\0\0", 2))
    print("qevents(6,1w):", send(6, b"", 1))
    print("term   (2,1w):", send(2, b"", 1))
finally:
    s.close()