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


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()
            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 _cell_text(self, cell):
        text = cell.text.strip()
        if not text:
            try:
                text = (self.driver.execute_script("return arguments[0].innerText;", cell) or "").strip()
            except Exception:
                pass
        return text

    def _normalize_id(self, s):
        return "".join(c for c in str(s) if c.isalnum()).lower()

    def _extract_data_from_page(self):
        wait = WebDriverWait(self.driver, 15)
        extracted = {}
        try:
            wait.until(
                EC.presence_of_element_located(
                    (By.XPATH, "//h4[text()='Eligible' or text()='Ineligible']/following::table[1]/tbody/tr")
                )
            )

            for status_label, elig_flag in [("Eligible", "Y"), ("Ineligible", "N")]:
                rows = self.driver.find_elements(
                    By.XPATH,
                    f"//h4[text()='{status_label}']/following::table[1]/tbody/tr"
                )
                for row in rows:
                    cells = row.find_elements(By.TAG_NAME, "td")
                    if len(cells) < 3:
                        continue
                    member_number = self._cell_text(cells[2])
                    norm_cell = self._normalize_id(member_number)
                    norm_self = self._normalize_id(self.memberId)
                    if norm_self and norm_cell and (norm_self in norm_cell or norm_cell in norm_self):
                        full_name = self._cell_text(cells[4]) if len(cells) > 4 else ""
                        plan_name = (
                            self._cell_text(cells[6]) if len(cells) > 6
                            else (self._cell_text(cells[-1]) if len(cells) > 4 else "")
                        )
                        name_parts = full_name.split()
                        extracted = {
                            "eligibility": elig_flag,
                            "firstName": name_parts[0] if name_parts else "",
                            "lastName": " ".join(name_parts[1:]) if len(name_parts) > 1 else "",
                            "insurance": plan_name,
                        }
                        print(f"[extraction] MATCHED {status_label} → name='{full_name}' plan='{plan_name}'")
                        return extracted

            print(f"[extraction] No matching row for memberId='{self.memberId}'")
            return {"eligibility": None}
        except Exception as e:
            print("Extraction error:", e)
            return {"eligibility": None}

    # ── 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:
            # Primary: ng-bind="member.memberNumber" is the stable Angular binding on the link
            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 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,
                    "history_pdf_path": None,
                    "file_type": "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,
                    "history_pdf_path": None,
                    "file_type": "pdf",
                }

            # Print history PDF
            history_pdf_path = self.step5_history_pdf()

            result = {
                "status": "success",
                "pdf_path": eligibility_pdf_path,
                "history_pdf_path": history_pdf_path,
                "file_type": "pdf",
                "message": "Eligibility and service history PDFs captured successfully",
            }
            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()