#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import pymysql
import requests
from pathlib import Path
from datetime import datetime
from dateutil import parser
import sys

# Force UTF-8 output even under Windows Task Scheduler
import sys
try:
    sys.stdout.reconfigure(encoding='utf-8')
    sys.stderr.reconfigure(encoding='utf-8')
except AttributeError:
    # Python < 3.7 fallback (not needed for you, but safe)
    import io
    sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')
    sys.stderr = io.TextIOWrapper(sys.stderr.buffer, encoding='utf-8')

# ================================
# 🛡 SAFE PRINT FOR CP1250 / Emoji
# ================================
def safe_print(text: str):
    enc = sys.stdout.encoding or ""
    if not enc.lower().startswith("utf"):
        # strip emoji + characters outside BMP
        text = ''.join(ch for ch in text if ord(ch) < 65536)
    try:
        print(text)
    except UnicodeEncodeError:
        # final fallback to ASCII only
        text = ''.join(ch for ch in text if ord(ch) < 128)
        print(text)


# ================================
# 🔧 CONFIGURATION
# ================================
TOKEN_PATH = Path("token.txt")
CLINIC_SLUG = "mudr-buzalkova"
LIMIT = 300

DB_CONFIG = {
    "host": "192.168.1.76",
    "port": 3307,
    "user": "root",
    "password": "Vlado9674+",
    "database": "medevio",
    "charset": "utf8mb4",
    "cursorclass": pymysql.cursors.DictCursor,
}

# ⭐ GraphQL query
GRAPHQL_QUERY = r"""
query ClinicRequestList2(
  $clinicSlug: String!,
  $queueId: String,
  $queueAssignment: QueueAssignmentFilter!,
  $state: PatientRequestState,
  $pageInfo: PageInfo!,
  $locale: Locale!
) {
  requestsResponse: listPatientRequestsForClinic2(
    clinicSlug: $clinicSlug,
    queueId: $queueId,
    queueAssignment: $queueAssignment,
    state: $state,
    pageInfo: $pageInfo
  ) {
    count
    patientRequests {
      id
      displayTitle(locale: $locale)
      createdAt
      updatedAt
      doneAt
      removedAt
      extendedPatient {
        name
        surname
        identificationNumber
      }
      lastMessage {
        createdAt
      }
    }
  }
}
"""

# ================================
# TOKEN
# ================================
def read_token(path: Path) -> str:
    tok = path.read_text(encoding="utf-8").strip()
    if tok.startswith("Bearer "):
        return tok.split(" ", 1)[1]
    return tok


# ================================
# DATETIME PARSER
# ================================
def to_mysql_dt(iso_str):
    if not iso_str:
        return None
    try:
        dt = parser.isoparse(iso_str)
        dt = dt.astimezone()
        return dt.strftime("%Y-%m-%d %H:%M:%S")
    except:
        return None


# ================================
# UPSERT
# ================================
def upsert(conn, r):
    p = r.get("extendedPatient") or {}

    api_updated = to_mysql_dt(r.get("updatedAt"))
    last_msg = r.get("lastMessage") or {}
    msg_at = to_mysql_dt(last_msg.get("createdAt"))

    def max_dt(a, b):
        if a and b:
            return max(a, b)
        return a or b

    final_updated = max_dt(api_updated, msg_at)

    sql = """
        INSERT INTO pozadavky (
            id, displayTitle, createdAt, updatedAt, doneAt, removedAt,
            pacient_jmeno, pacient_prijmeni, pacient_rodnecislo
        ) VALUES (%s,%s,%s,%s,%s,%s,%s,%s,%s)
        ON DUPLICATE KEY UPDATE
            displayTitle=VALUES(displayTitle),
            updatedAt=VALUES(updatedAt),
            doneAt=VALUES(doneAt),
            removedAt=VALUES(removedAt),
            pacient_jmeno=VALUES(pacient_jmeno),
            pacient_prijmeni=VALUES(pacient_prijmeni),
            pacient_rodnecislo=VALUES(pacient_rodnecislo)
    """

    vals = (
        r.get("id"),
        r.get("displayTitle"),
        to_mysql_dt(r.get("createdAt")),
        final_updated,
        to_mysql_dt(r.get("doneAt")),
        to_mysql_dt(r.get("removedAt")),
        p.get("name"),
        p.get("surname"),
        p.get("identificationNumber"),
    )

    with conn.cursor() as cur:
        cur.execute(sql, vals)

    conn.commit()


# ================================
# FETCH LAST 300 DONE REQUESTS
# ================================
def fetch_done(headers):
    vars = {
        "clinicSlug": CLINIC_SLUG,
        "queueId": None,
        "queueAssignment": "ANY",
        "pageInfo": {"first": LIMIT, "offset": 0},
        "locale": "cs",
        "state": "DONE",
    }

    payload = {
        "operationName": "ClinicRequestList2",
        "query": GRAPHQL_QUERY,
        "variables": vars,
    }

    r = requests.post("https://api.medevio.cz/graphql", json=payload, headers=headers)
    r.raise_for_status()

    data = r.json()["data"]["requestsResponse"]
    return data.get("patientRequests", [])


# ================================
# MAIN
# ================================
def main():
    token = read_token(TOKEN_PATH)
    headers = {
        "Authorization": f"Bearer {token}",
        "Content-Type": "application/json",
        "Accept": "application/json",
    }

    conn = pymysql.connect(**DB_CONFIG)

    safe_print(f"\n=== Downloading last {LIMIT} DONE requests @ {datetime.now():%Y-%m-%d %H:%M:%S} ===")

    requests_list = fetch_done(headers)
    safe_print(f"📌 Requests returned: {len(requests_list)}")

    for r in requests_list:
        upsert(conn, r)

    conn.close()
    safe_print("\n\u2705 DONE - latest closed requests synced.\n")


# ================================
if __name__ == "__main__":
    main()