import random
import string
import time

from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

from helper import signup

def has_punctuation(s):
    return any(c in string.punctuation for c in s)

def login(driver, home, uname, password):
    driver.get(home)
    WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.LINK_TEXT, '登录'))).click()
    driver.find_element(By.ID, 'username').send_keys(uname)
    driver.find_element(By.ID, 'password').send_keys(password)
    driver.find_element(By.XPATH, '//button[text()="登录"]').click()
    WebDriverWait(driver, 10).until(EC.title_is(f"EnglishPal Study Room for {uname}"))


def select_valid_word(driver):
    elem = driver.find_element(By.ID, 'text-content')
    essay_content = elem.text
    valid_word = random.choice([word for word in essay_content.split() if len(word) >= 6 and not has_punctuation(
        word) and 'font>' not in word and 'br>' not in word and 'p>' not in word])
    driver.find_element(By.ID, 'selected-words').send_keys(valid_word)
    return valid_word


def select_more_words(driver):
    elem = driver.find_element(By.ID, 'text-content')
    essay_content = elem.text
    valid_words = [word for word in essay_content.split() if len(word) >= 6 and not has_punctuation(
        word) and 'font>' not in word and 'br>' not in word and 'p>' not in word]
    words_to_send = ', '.join(valid_words[:7])
    driver.find_element(By.ID, 'selected-words').send_keys(words_to_send)


def test_unfamiliar_button(driver, URL):
    try:
        username, password = signup(URL, driver)
        time.sleep(5)

        word = select_valid_word(driver)
        
       # Add selected word to the list of unfamiliar words
        time.sleep(5)
        add_to_unfamiliar_button =  driver.find_element(By.XPATH, '//button[text()="把生词加入我的生词库"]')
        add_to_unfamiliar_button.click()
        time.sleep(3)

        add_words_button =  driver.find_element(By.XPATH, '//button[text()="加入我的生词簿"]') # Replace with actual ID or selector
        add_words_button.click()
        # Wait for the page to update again
        time.sleep(2)

      # Check if the added word appears in the word container
        word_container = driver.find_element(By.CLASS_NAME, 'word-container')
        words = word_container.find_elements(By.TAG_NAME, 'p')
        added_word_exists = any(word.text == word for word in words)
        time.sleep(2)

        # Mark word as unfamiliar
        time.sleep(5)
        for _ in range(16):
            time.sleep(1)
            unfamiliar_btn = driver.find_element(By.XPATH, '//a[contains(@class, "btn-warning") and text()="不熟悉"]')
            unfamiliar_btn.click()
        # Wait for the page to updatec
        time.sleep(5)

        
        #Get unfimiliar word count 
        initial_count = driver.find_element(By.ID, 'freq_'+ word).text 
        
        #add new words
        word = select_more_words(driver)
        add_to_unfamiliar_button =  driver.find_element(By.XPATH, '//button[text()="把生词加入我的生词库"]')
        add_to_unfamiliar_button.click()
        time.sleep(5)

        #add the words to the list of words
        add_words_again =  driver.find_element(By.XPATH, '//button[text()="加入我的生词簿"]') # Replace with actual ID or selector
        add_words_again.click()
        time.sleep(3)
      
      
        #Count unfamiliar words
        current_unfamiliar_count = driver.find_element(By.XPATH, '//a[contains(@class, "btn-warning") and text()="不熟悉"]')
        current_unfamiliar_count = int(unfamiliar_btn.text)
        time.sleep(4)

         # Verify that the number of unfamiliar words has increased
        assert current_unfamiliar_count > initial_count, "Unfamiliar word count did not increase as expected"
        time.sleep(3)


        # Log out and close the browser
        driver.find_element(By.XPATH, '//a[contains(@class, "btn-secondary") and text()="退出"]')
        driver.execute_script("window.open('');window.close();")
        WebDriverWait(driver, 2)
        driver.execute_script("window.open('');")
        driver.switch_to.window(driver.window_handles[-1])

    finally:
        driver.quit()