from selenium import webdriver
from selenium.common import TimeoutException
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait, Select
from selenium.webdriver.support import expected_conditions as EC
from webdriver_manager.chrome import ChromeDriverManager
import time
import os
import base64

from mh_eligibility_results import extract_eligibility_row
from mh_member_details import extract_member_demographics


class AutomationMassHealthEligibilityHistoryCheck:
    def __init__(self, data):
        self.headless = False
        self.driver = None
        self.extracted_data = {}
        self.eligibility_tab = None  # handle to the eligibility results tab

        self.data = data.get("data")

        self.massdhp_username = self.data.get("massdhpUsername", "")
        self.massdhp_password = self.data.get("massdhpPassword", "")
        self.dateOfBirth = self.data.get("dateOfBirth", "")
        self.memberId = self.data.get("memberId", "")
        self.firstName = self.data.get("firstName", "")
        self.lastName = self.data.get("lastName", "")

        # Convert dateOfBirth from YYYY-MM-DD to MMDDYYYY
        if self.dateOfBirth and "-" in self.dateOfBirth:
            parts = self.dateOfBirth.split("-")
            if len(parts) == 3:
                year, month, day = parts
                self.dateOfBirth = f"{month.zfill(2)}{day.zfill(2)}{year}"

        self.download_dir = os.path.abspath("downloads")
        os.makedirs(self.download_dir, exist_ok=True)

    # ── driver ──────────────────────────────────────────────────────────────────

    def config_driver(self):
        options = webdriver.ChromeOptions()
        if self.headless:
            options.add_argument("--headless")

        prefs = {
            "download.default_directory": self.download_dir,
            "plugins.always_open_pdf_externally": False,
            "download.prompt_for_download": False,
            "download.directory_upgrade": True,
        }
        options.add_experimental_option("prefs", prefs)

        s = Service(ChromeDriverManager().install())
        self.driver = webdriver.Chrome(service=s, options=options)

    # ── login ───────────────────────────────────────────────────────────────────

    def login(self):
        wait = WebDriverWait(self.driver, 30)
        try:
            signin_button = wait.until(
                EC.element_to_be_clickable(
                    (By.CSS_SELECTOR, "a.btn.btn-block.btn-primary[href='https://connectsso.masshealth-dental.org/mhprovider/index.html']")
                )
            )
            signin_button.click()
            time.sleep(3)

            email_field = wait.until(EC.presence_of_element_located((By.ID, "User")))
            email_field.clear()
            email_field.send_keys(self.massdhp_username)

            password_field = wait.until(EC.presence_of_element_located((By.ID, "Password")))
            password_field.clear()
            password_field.send_keys(self.massdhp_password)

            login_button = wait.until(
                EC.element_to_be_clickable(
                    (By.CSS_SELECTOR, "input[type='submit'][name='submit'][value='Login']")
                )
            )
            login_button.click()

            # Wait for the SSO redirect. On some machines Chrome opens the dashboard
            # in a new tab and closes the SSO tab — poll ALL window handles.
            print("[login] Waiting for SSO redirect to provider.masshealth-dental.org ...")
            deadline = time.time() + 30
            found = False
            while time.time() < deadline:
                time.sleep(0.5)
                for handle in self.driver.window_handles:
                    try:
                        self.driver.switch_to.window(handle)
                        if self.driver.current_url.startswith("https://provider.masshealth-dental.org"):
                            print(f"[login] Redirect complete. URL: {self.driver.current_url}")
                            found = True
                            break
                    except Exception:
                        continue
                if found:
                    break

            if not found:
                print("[login] Redirect timeout — portal window not found in any tab.")
                return "ERROR: Login redirect timed out — check credentials or portal availability."

            return "Success"
        except Exception as e:
            print(f"[login] Error: {e}")
            return "ERROR:LOGIN FAILED"

    # ── step 1 — search member, extract eligibility data ────────────────────────

    def step1(self):
        wait = WebDriverWait(self.driver, 30)
        substep = "init"
        try:
            print(f"[step1] URL: {self.driver.current_url}")

            substep = "patient_management"
            patient_mgmt = wait.until(
                EC.presence_of_element_located(
                    (By.XPATH, "//strong[@translate='Patient Management']")
                )
            )
            self.driver.execute_script("arguments[0].scrollIntoView(true);", patient_mgmt)
            self.driver.execute_script("arguments[0].click();", patient_mgmt)
            time.sleep(2)

            substep = "member_eligibility_link"
            eligibility_link = wait.until(
                EC.presence_of_element_located(
                    (By.XPATH, "//a[@translate='Member Eligibility']")
                )
            )
            self.driver.execute_script("arguments[0].click();", eligibility_link)
            time.sleep(2)

            substep = "provider_dropdown"
            provider_dropdown = wait.until(
                EC.presence_of_element_located((By.NAME, "provider"))
            )
            select_provider = Select(provider_dropdown)
            options = [o.text for o in select_provider.options]
            print(f"[step1] provider options: {options}")

            substep = "select_provider"
            first_option = next(
                (o for o in select_provider.options if o.get_attribute("value").strip()),
                select_provider.options[0]
            )
            print(f"[step1] selecting provider: '{first_option.text}'")
            select_provider.select_by_value(first_option.get_attribute("value"))
            time.sleep(2)

            substep = "member_dob"
            member_dob = wait.until(
                EC.presence_of_all_elements_located((By.NAME, "dateInput"))
            )[1]
            member_dob.clear()
            member_dob.send_keys(self.dateOfBirth)

            substep = "member_number"
            member_number = wait.until(
                EC.presence_of_element_located((By.NAME, "memberNumber"))
            )
            member_number.clear()
            member_number.send_keys(self.memberId)

            substep = "search_button"
            search_button = wait.until(
                EC.element_to_be_clickable((By.XPATH, "//button[contains(@class,'btn-primary')]"))
            )
            search_button.click()

            substep = "wait_results"
            wait.until(
                EC.presence_of_element_located(
                    (By.XPATH, "//h4[text()='Eligible' or text()='Ineligible']")
                )
            )

            self.extracted_data = self._extract_data_from_page()
            print(f"[step1] data extracted: {self.extracted_data}")
            return "Success"
        except Exception as e:
            print(f"[step1] FAILED at substep='{substep}': {e}")
            print(f"[step1] URL at failure: {self.driver.current_url}")
            return f"ERROR:STEP1:{substep}"

    # ── helpers ─────────────────────────────────────────────────────────────────

    def _extract_data_from_page(self):
        # Waits until the matching row's name and plan are filled in
        return extract_eligibility_row(self.driver, self.memberId)

    # ── step 2 — print eligibility PDF, stay on results tab ─────────────────────

    def step2_eligibility_pdf(self):
        wait = WebDriverWait(self.driver, 30)
        self.eligibility_tab = self.driver.current_window_handle

        try:
            download_button = wait.until(
                EC.element_to_be_clickable(
                    (By.XPATH, "//button[contains(.,'Printer Friendly Format')]")
                )
            )
            download_button.click()

            try:
                WebDriverWait(self.driver, 10).until(lambda d: len(d.window_handles) > 1)
                new_tabs = [t for t in self.driver.window_handles if t != self.eligibility_tab]
                self.driver.switch_to.window(new_tabs[0])
                wait.until(lambda d: d.execute_script("return document.readyState") == "complete")
                wait.until(EC.presence_of_element_located((By.TAG_NAME, "body")))
                time.sleep(2)
                print("Printer-friendly tab opened:", self.driver.current_url)
            except TimeoutException:
                print("No new tab for eligibility; printing current page directly")

            safe_member = "".join(c for c in str(self.memberId) if c.isalnum() or c in "-_.")
            pdf_filename = f"eligibility_{safe_member}.pdf"
            pdf_data = self.driver.execute_cdp_cmd("Page.printToPDF", {"printBackground": True})
            pdf_bytes = base64.b64decode(pdf_data["data"])
            pdf_path = os.path.join(self.download_dir, pdf_filename)
            with open(pdf_path, "wb") as f:
                f.write(pdf_bytes)
            print("Eligibility PDF saved at:", pdf_path)

            # Close printer-friendly tab and return to eligibility results
            if self.driver.current_window_handle != self.eligibility_tab:
                self.driver.close()
                self.driver.switch_to.window(self.eligibility_tab)
                time.sleep(1)

            return pdf_path
        except Exception as e:
            print(f"[step2_eligibility_pdf] failed: {e}")
            # Ensure we are back on the eligibility results tab even on error
            if self.eligibility_tab and self.driver.current_window_handle != self.eligibility_tab:
                try:
                    self.driver.close()
                    self.driver.switch_to.window(self.eligibility_tab)
                except Exception:
                    pass
            raise

    # ── step 3 — click member ID link → member details page ─────────────────────

    def step3_click_member_id(self):
        wait = WebDriverWait(self.driver, 30)
        substep = "init"
        try:
            substep = "ng_bind_link"
            member_link = wait.until(
                EC.element_to_be_clickable(
                    (By.CSS_SELECTOR, "a[ng-bind='member.memberNumber']")
                )
            )
            self.driver.execute_script("arguments[0].click();", member_link)
            time.sleep(2)
            print(f"[step3] clicked member ID link, URL: {self.driver.current_url}")
            return "Success"
        except Exception as e:
            print(f"[step3] FAILED at substep='{substep}': {e}")
            return f"ERROR:STEP3:{substep}"

    # ── step 3b — print member details page as PDF via CDP ───────────────────────

    def step3b_member_details_pdf(self):
        wait = WebDriverWait(self.driver, 30)
        try:
            wait.until(lambda d: d.execute_script("return document.readyState") == "complete")
            time.sleep(2)

            # Gender, address and phone from the Member Detail block
            details = extract_member_demographics(self.driver)
            member_name = details.pop("memberName", "")
            # Results table never filled the name in — use the Member Detail heading
            if member_name and not self.extracted_data.get("firstName"):
                parts = member_name.split()
                details["firstName"] = parts[0]
                details["lastName"] = " ".join(parts[1:])
            self.extracted_data.update(details)

            safe_member = "".join(c for c in str(self.memberId) if c.isalnum() or c in "-_.")
            pdf_filename = f"eligibility_member_details_{safe_member}.pdf"
            pdf_data = self.driver.execute_cdp_cmd("Page.printToPDF", {"printBackground": True})
            pdf_bytes = base64.b64decode(pdf_data["data"])
            pdf_path = os.path.join(self.download_dir, pdf_filename)
            with open(pdf_path, "wb") as f:
                f.write(pdf_bytes)
            print("Member details PDF saved at:", pdf_path)
            return pdf_path
        except Exception as e:
            print(f"[step3b_member_details_pdf] failed: {e}")
            return None

    # ── step 4 — click "View Service History" on member details page ─────────────

    def step4_view_service_history(self):
        wait = WebDriverWait(self.driver, 30)
        substep = "init"
        try:
            # Primary: href contains /service-history and has btn-primary class
            substep = "service_history_link"
            history_link = wait.until(
                EC.element_to_be_clickable(
                    (By.CSS_SELECTOR, "a.btn.btn-primary[href*='service-history']")
                )
            )
            self.driver.execute_script("arguments[0].scrollIntoView(true);", history_link)
            self.driver.execute_script("arguments[0].click();", history_link)
            time.sleep(2)
            print(f"[step4] navigated to service history, URL: {self.driver.current_url}")
            return "Success"
        except Exception as e:
            print(f"[step4] FAILED at substep='{substep}': {e}")
            return f"ERROR:STEP4:{substep}"

    # ── step 5 — print service history PDF ──────────────────────────────────────

    def step5_history_pdf(self):
        wait = WebDriverWait(self.driver, 30)
        current_tab = self.driver.current_window_handle
        try:
            print_button = wait.until(
                EC.element_to_be_clickable(
                    (By.CSS_SELECTOR, "button.btn.btn-primary[ng-click='vm.printPage()']")
                )
            )
            print_button.click()

            try:
                WebDriverWait(self.driver, 10).until(lambda d: len(d.window_handles) > 1)
                new_tabs = [t for t in self.driver.window_handles if t != current_tab]
                self.driver.switch_to.window(new_tabs[0])
                wait.until(lambda d: d.execute_script("return document.readyState") == "complete")
                wait.until(EC.presence_of_element_located((By.TAG_NAME, "body")))
                time.sleep(2)
                print("History printer-friendly tab opened:", self.driver.current_url)
            except TimeoutException:
                print("No new tab for history; printing current page directly")

            safe_member = "".join(c for c in str(self.memberId) if c.isalnum() or c in "-_.")
            pdf_filename = f"eligibility_history_{safe_member}.pdf"
            pdf_data = self.driver.execute_cdp_cmd("Page.printToPDF", {"printBackground": True})
            pdf_bytes = base64.b64decode(pdf_data["data"])
            pdf_path = os.path.join(self.download_dir, pdf_filename)
            with open(pdf_path, "wb") as f:
                f.write(pdf_bytes)
            print("History PDF saved at:", pdf_path)
            return pdf_path
        except Exception as e:
            print(f"[step5_history_pdf] failed: {e}")
            return None

    # ── main workflow ────────────────────────────────────────────────────────────

    def main_workflow(self, url):
        try:
            self.config_driver()
            self.driver.maximize_window()
            self.driver.get(url)
            time.sleep(3)

            login_result = self.login()
            if login_result.startswith("ERROR"):
                return {"status": "error", "message": login_result}

            step1_result = self.step1()
            if step1_result.startswith("ERROR"):
                return {"status": "error", "message": step1_result}

            # Print eligibility PDF; stays on results tab afterwards
            eligibility_pdf_path = self.step2_eligibility_pdf()

            # Click the member ID link → member details
            step3_result = self.step3_click_member_id()
            if step3_result.startswith("ERROR"):
                return {
                    "status": "partial",
                    "message": step3_result,
                    "pdf_path": eligibility_pdf_path,
                    "file_type": "pdf",
                }

            # Print member details page
            member_details_pdf_path = self.step3b_member_details_pdf()

            # Click "View Service History"
            step4_result = self.step4_view_service_history()
            if step4_result.startswith("ERROR"):
                return {
                    "status": "partial",
                    "message": step4_result,
                    "pdf_path": eligibility_pdf_path,
                    "member_details_pdf_path": member_details_pdf_path,
                    "file_type": "pdf",
                }

            # Print history PDF
            history_pdf_path = self.step5_history_pdf()

            result = {
                "status": "success",
                "pdf_path": eligibility_pdf_path,
                "member_details_pdf_path": member_details_pdf_path,
                "history_pdf_path": history_pdf_path,
                "file_type": "pdf",
                "message": "Eligibility, member details, and service history PDFs captured",
            }
            if self.extracted_data:
                result.update(self.extracted_data)
            return result

        except Exception as e:
            return {"status": "error", "message": str(e)}
        finally:
            if self.driver:
                self.driver.quit()