"""
Path: examples/pull_live_events.py
Author: @kaburagisec
Created: September 22, 2025
Tested devices: EZVIZ H8C (https://www.ezviz.com/inter/product/h8c/43162)

This script connects to an ONVIF-compliant device, creates a PullPoint subscription,
and continuously pulls live events for 15 minutes, printing event details to the console.

Uses ONVIFParser (>= v0.2.2 patch) to extract Topic elements that zeep doesn't parse correctly.
"""

import datetime

from onvif import ONVIFClient, ONVIFParser

HOST = "192.168.1.3"
PORT = 80
USERNAME = "admin"
PASSWORD = "admin123"

# Create parser to extract Topic elements from SOAP responses
parser = ONVIFParser(
    extract_xpaths={
        "topic": ".//{http://docs.oasis-open.org/wsn/b-2}Topic",
    }
)

# Pass parser as plugin to client (no capture_xml needed!)
client = ONVIFClient(HOST, PORT, USERNAME, PASSWORD, plugins=[parser])

# 1. Create PullPoint Subscription from Events service
subscription_ref = client.events().CreatePullPointSubscription()
print("Subscription Response:\n", subscription_ref)

# 2. Create PullPoint service instance with the Subscription reference
pullpoint = client.pullpoint(subscription_ref)

# 3. Pull events for 15 minutes
end_time = datetime.datetime.now() + datetime.timedelta(minutes=15)
print("\n✅ Start pulling events for 15 minutes...")

while datetime.datetime.now() < end_time:
    try:
        # PT5S = 5 seconds timeout, MessageLimit=1000
        # Timeout can be adjusted as needed and in ISO 8601 format
        msgs = pullpoint.PullMessages(Timeout="PT5S", MessageLimit=1000)
        notifications = getattr(msgs, "NotificationMessage", None)

        if notifications:
            if not isinstance(notifications, list):
                notifications = [notifications]

            # (>= v0.2.2 patch)
            # Extract topics using ONVIFParser (zeep workaround for simpleContent limitation)
            topics = parser.get_extracted_texts("topic", len(notifications))

            for idx, n in enumerate(notifications):
                # Get Topic text from parser
                # Zeep can't parse Topic correctly because it has both Dialect attribute and text content
                topic_val = topics[idx]

                # Extract Message (XML element)
                # After the patch (> v0.0.4), _value_1 is directly a list of Elements (not a dict)
                msg_elem = None
                if hasattr(n.Message, "_value_1") and n.Message._value_1:
                    # _value_1 is now a list of Elements, take the first one
                    if (
                        isinstance(n.Message._value_1, list)
                        and len(n.Message._value_1) > 0
                    ):
                        msg_elem = n.Message._value_1[0]
                    else:
                        msg_elem = n.Message._value_1

                if msg_elem is not None:
                    # Extract timestamp from UtcTime attribute
                    utc_str = msg_elem.get("UtcTime")
                    if utc_str:
                        ts_utc = datetime.datetime.fromisoformat(
                            utc_str.replace("Z", "+00:00")
                        )
                        ts = ts_utc.astimezone()
                    else:
                        ts = datetime.datetime.now().astimezone()

                    print("\n=== Event received @", ts, "===")

                    # print("PullMessages Response:\n", msgs)

                    print("Operation ->", msg_elem.get("PropertyOperation"))

                    print("Topic ->", topic_val)

                    # Extract <tt:Source>
                    source_tokens = []
                    for el in msg_elem.findall(
                        ".//{http://www.onvif.org/ver10/schema}Source/{http://www.onvif.org/ver10/schema}SimpleItem"
                    ):
                        name = el.get("Name")
                        value = el.get("Value")
                        source_tokens.append(f"{name}: {value}")
                    if source_tokens:
                        print("Source ->", ", ".join(source_tokens))

                    # Extract <tt:Data>
                    data_tokens = []
                    for el in msg_elem.findall(
                        ".//{http://www.onvif.org/ver10/schema}Data/{http://www.onvif.org/ver10/schema}SimpleItem"
                    ):
                        name = el.get("Name")
                        value = el.get("Value")
                        data_tokens.append(f"{name}: {value}")
                    if data_tokens:
                        print("Data ->", ", ".join(data_tokens))

                else:
                    print("\n-> Event without Message")
        else:
            print("\n-> 🔁 No new events.")
    except Exception as e:  # pylint: disable=broad-exception-caught
        print("\n-> ❌ Error while pulling events:", e)

    # time.sleep(1)  # small delay between requests

print("\n✅ Finished: 15 minutes event pulling ended")

# 4. Create Subscription service instance with the Subscription reference
subscription = client.subscription(subscription_ref)

# 5. Unsubscribe to clean up
unsubscribe = subscription.Unsubscribe()