#!/usr/bin/env python3
"""Probe getdatevisits to see XML structure for one date."""
import json, re, sys
from datetime import datetime
from pathlib import Path
from urllib.request import Request, urlopen
from urllib.error import HTTPError

CREDS_PATH = Path("/Users/exult/claude-workspace/config/credentials/advancedmd.json")
with open(CREDS_PATH) as f:
    creds = json.load(f)
api_user = creds["api_users"][0]
OFFICE_KEY = creds["office_key"]
USERNAME = api_user["user"]
PASSWORD = api_user["password"]
APPNAME = api_user.get("appname", "ABS-AVMD")
PARTNER_LOGIN_URL = "https://partnerlogin.advancedmd.com/practicemanager/xmlrpc/processrequest.aspx"

def now_str():
    return datetime.now().strftime("%m/%d/%Y %I:%M:%S %p")

def esc(s):
    return (s.replace("&", "&amp;").replace("<", "&lt;")
             .replace(">", "&gt;").replace('"', "&quot;").replace("'", "&apos;"))

def post_xml(url, body, token=None):
    headers = {"Content-Type": "text/xml"}
    if token:
        headers["Cookie"] = f"token={token}"
    req = Request(url, data=body.encode("utf-8"), headers=headers, method="POST")
    try:
        with urlopen(req, timeout=60) as resp:
            return resp.read().decode("utf-8"), resp.status
    except HTTPError as e:
        return e.read().decode("utf-8"), e.code

body = (f'<ppmdmsg action="login" class="login" msgtime="{esc(now_str())}" '
        f'username="{esc(USERNAME)}" psw="{esc(PASSWORD)}" '
        f'officecode="{esc(OFFICE_KEY)}" appname="{esc(APPNAME)}"/>')
text1, _ = post_xml(PARTNER_LOGIN_URL, body)
ws = re.search(r'webserver="([^"]+)"', text1)
webserver = ws.group(1) if ws else None
if not webserver:
    m = re.search(r'https://[^\s.]+\.advancedmd\.com/processrequest/[^\s"<]+', text1)
    webserver = m.group(0) if m else None
print("STEP1 webserver=", webserver, file=sys.stderr)
print("STEP1 RESP[:500]=", text1[:500], file=sys.stderr)
api_url = f"{webserver}/xmlrpc/processrequest.aspx"
text2, _ = post_xml(api_url, body)
tm = re.search(r'<usercontext[^>]*?>([^<]+)</usercontext>', text2, re.IGNORECASE | re.DOTALL)
token = tm.group(1).strip()
print("LOGIN OK token len=", len(token), file=sys.stderr)

date_arg = sys.argv[1] if len(sys.argv) > 1 else "2026-06-15"
body = (f'<ppmdmsg action="getdatevisits" class="api" msgtime="{esc(now_str())}" '
        f'visitdate="{date_arg}"><visit columnheading="ColumnHeading"/>'
        f'<patient name="Name"/></ppmdmsg>')
text, status = post_xml(api_url, body, token)
print("STATUS=", status, file=sys.stderr)
print(text[:12000])