from selenium import webdriver
from bs4 import BeautifulSoup
import mysql.connector
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.common.keys import Keys
import json
import time
import sys
import pyperclip

# === Настройки ===
CHROMEDRIVER_PATH = r"C:\Users\Evgen\Downloads\chromedriver-win64\chromedriver.exe"
COOKIES_FILE = "cookies_dzen.json"

DB_CONFIG = {
    'host': '192.168.8.100',
    'user': 'root',
    'password': 'BXiQLt3G',
    'database': 'led_topics_db',
    'port': 3307,
}

EDITOR_URL = "https://dzen.ru/profile/editor/id/669dd289f109381c8b33de06/publications"

# === Подключение к БД ===
def get_article_from_db():
    try:
        conn = mysql.connector.connect(**DB_CONFIG)
        cursor = conn.cursor(dictionary=True)
        query = """
        SELECT topic, content 
        FROM led_topics
        WHERE is_archived = 0 
          AND content IS NOT NULL 
          AND content != ''
        ORDER BY RAND()
        LIMIT 1;
        """
        cursor.execute(query)
        result = cursor.fetchone()
        cursor.close()
        conn.close()
        return result
    except Exception as e:
        print(f"❌ Ошибка при подключении к БД: {e}")
        return None

# === Инициализация браузера ===
def init_browser():
    chrome_options = Options()
    chrome_options.add_argument("--disable-gpu")
    chrome_options.add_argument("--disable-software-rasterizer")
    chrome_options.add_argument("--disable-extensions")
    chrome_options.add_argument("--no-sandbox")
    chrome_options.add_argument("--disable-popup-blocking")
    chrome_options.add_argument("--disable-notifications")
    chrome_options.add_argument("--log-level=3")
    chrome_options.add_experimental_option('excludeSwitches', ['enable-logging'])
    chrome_options.add_experimental_option("useAutomationExtension", False)
    chrome_options.add_experimental_option("excludeSwitches", ["enable-automation"])
    chrome_options.set_capability('goog:loggingPrefs', {'browser': 'SEVERE'})
    service = Service(CHROMEDRIVER_PATH)
    driver = webdriver.Chrome(service=service, options=chrome_options)
    return driver

# === Авторизация через куки ===
def load_cookies(driver):
    try:
        with open(COOKIES_FILE, "r", encoding="utf-8") as f:
            cookies = json.load(f)
        for cookie in cookies:
            cookie.pop("sameSite", None)
            cookie.pop("storeId", None)
            cookie.pop("session", None)
            if not cookie["domain"].startswith("."):
                cookie["domain"] = "." + cookie["domain"]
            try:
                driver.add_cookie(cookie)
            except Exception as e:
                print(f"❌ Ошибка добавления куки {cookie.get('name')}: {e}")
        print("✅ Куки успешно загружены")
    except Exception as e:
        print(f"❌ Ошибка загрузки кук: {e}")

# === Открытие редактора ===
def open_editor(driver):
    print("🌐 Открываем редактор...")
    driver.get(EDITOR_URL)
    time.sleep(2)
    driver.delete_all_cookies()
    load_cookies(driver)
    driver.get(EDITOR_URL)
    time.sleep(5)
    # Закрытие всплывающего окна (если есть)
    try:
        close_btn = WebDriverWait(driver, 5).until(
            EC.element_to_be_clickable((By.CSS_SELECTOR, 'button[data-testid="close-button"]'))
        )
        close_btn.click()
    except:
        pass
    # Проверка авторизации
    try:
        print("⏳ Ожидание загрузки страницы редактора...")
        WebDriverWait(driver, 10).until(
            EC.presence_of_element_located((By.CSS_SELECTOR, "button[data-testid='add-publication-button']"))
        )
        print("✅ Авторизация успешна")
    except Exception as e:
        print(f"❌ Не удалось авторизоваться. Проверь куки: {e}")
        return False
    return True

# === Создание новой статьи ===
def create_new_article(driver):
    try:
        add_button = WebDriverWait(driver, 10).until(
            EC.element_to_be_clickable((By.CSS_SELECTOR, "button[data-testid='add-publication-button']"))
        )
        add_button.click()
        print("✅ Клик по кнопке 'Написать статью'")
    except Exception as e:
        print(f"❌ Ошибка при клике на 'Написать статью': {e}")
        return False
    try:
        write_button = WebDriverWait(driver, 10).until(
            EC.element_to_be_clickable((By.XPATH, '//span[text()="Написать статью"]'))
        )
        write_button.click()
        print("✅ Выбрана публикация 'Написать статью'")
    except Exception as e:
        print(f"❌ Ошибка при выборе 'Написать статью': {e}")
        return False
    return True

# === Закрытие подсказок ===
def close_tips(driver):
    try:
        close_button = WebDriverWait(driver, 10).until(
            EC.element_to_be_clickable((By.XPATH, '//div[@role="button" and @aria-label="Закрыть"]'))
        )
        close_button.click()
        print("✅ Подсказка закрыта")
    except:
        print("ℹ️ Подсказка не найдена")

# === Вставка заголовка через send_keys ===
def insert_title(driver, title):
    try:
        title_field = WebDriverWait(driver, 10).until(
            EC.presence_of_element_located((By.CSS_SELECTOR, "div.public-DraftEditor-content[contenteditable='true']"))
        )
        title_field.click()
        time.sleep(0.5)
        actions = ActionChains(driver)
        actions.send_keys(title)
        actions.perform()
        print("✅ Заголовок успешно вставлен")
    except Exception as e:
        print(f"❌ Не удалось вставить заголовок: {e}")

# === Вставка форматированного контента через горячие клавиши с поддержкой ссылок и других тегов ===
def send_formatted_content(driver, html):
    from selenium.webdriver.common.action_chains import ActionChains
    from selenium.webdriver.common.keys import Keys
    from bs4 import BeautifulSoup
    editor_xpath = "//div[@aria-describedby='placeholder-ZenDraftEditor' and contains(@class, 'public-DraftEditor-content') and @contenteditable='true']"
    editor = WebDriverWait(driver, 10).until(
        EC.element_to_be_clickable((By.XPATH, editor_xpath))
    )
    editor.click()
    time.sleep(0.5)
    soup = BeautifulSoup(html, 'html.parser')
    actions = ActionChains(driver)
    for tag in soup.contents:
        if tag.name in ['p', 'h1', 'h2', 'h3', 'h4', 'h5', 'h6']:
            if tag.find('a'):
                a_tag = tag.find('a')
                link_text = a_tag.get_text(strip=True)
                link_url = a_tag['href']
                full_text = tag.get_text()
                before, after = full_text.split(link_text, 1)
                if before:
                    actions.send_keys(before)
                actions.send_keys(link_text)
                actions.perform()
                # выделить текст ссылки
                for _ in range(len(link_text)):
                    actions = ActionChains(driver)
                    actions.key_down(Keys.SHIFT).send_keys(Keys.LEFT).key_up(Keys.SHIFT)
                    actions.perform()
                # Клик по кнопке "ссылка" в тулбаре
                try:
                    print("[DEBUG] Ждем появления кнопки-ссылки (linkTools-9h)...")
                    link_button = WebDriverWait(driver, 5).until(
                        EC.element_to_be_clickable((By.XPATH, "//div[contains(@class, 'editor-toolbar__linkTools-9h')]"))
                    )
                    print(f"[DEBUG] Найдена кнопка-ссылка: {link_button.get_attribute('outerHTML')}")
                    time.sleep(1)
                    link_button.click()
                    print("✅ Клик по кнопке-ссылке (linkTools-9h) в тулбаре")
                    time.sleep(1)
                    print("[DEBUG] Ждем появления input для ссылки...")
                    link_input = WebDriverWait(driver, 5).until(
                        EC.element_to_be_clickable((By.XPATH, "//input[@type='text' and (contains(@placeholder, 'Введите ссылку') or contains(@placeholder, 'http') or contains(@placeholder, 'ссылк'))]"))
                    )
                    print(f"✅ Найден input для ссылки с placeholder: {link_input.get_attribute('placeholder')}, HTML: {link_input.get_attribute('outerHTML')}")
                    time.sleep(1)
                    link_input.clear()
                    link_input.send_keys(link_url)
                    time.sleep(1)
                    link_input.send_keys(Keys.ENTER)
                    print(f"✅ Вставлена ссылка через тулбар: {link_text} -> {link_url}")
                    editor_text = editor.text
                    print(f"[DEBUG] Текущий текст в редакторе после вставки ссылки: {editor_text}")
                except Exception as e:
                    print(f"❌ Не удалось кликнуть по кнопке-ссылке (linkTools-9h) или ввести ссылку: {e}")
                    with open('debug_toolbar.html', 'w', encoding='utf-8') as f:
                        f.write(driver.page_source)
                    print('[DEBUG] Сохранён HTML для отладки: debug_toolbar.html')
                if after:
                    actions = ActionChains(driver)
                    actions.send_keys(after)
                    actions.send_keys(Keys.ENTER)
                    actions.perform()
                else:
                    actions = ActionChains(driver)
                    actions.send_keys(Keys.ENTER)
                    actions.perform()
                continue
            text = tag.get_text(strip=True)
            if not text:
                continue
            if tag.find(['b', 'strong']):
                actions.key_down(Keys.CONTROL).send_keys('b').key_up(Keys.CONTROL)
            if tag.find(['i', 'em']):
                actions.key_down(Keys.CONTROL).send_keys('i').key_up(Keys.CONTROL)
            actions.send_keys(text)
            if tag.find(['b', 'strong']):
                actions.key_down(Keys.CONTROL).send_keys('b').key_up(Keys.CONTROL)
            if tag.find(['i', 'em']):
                actions.key_down(Keys.CONTROL).send_keys('i').key_up(Keys.CONTROL)
            actions.send_keys(Keys.ENTER)
        elif tag.name in ['ul', 'ol']:
            for li in tag.find_all('li'):
                li_text = li.get_text(strip=True)
                if li_text:
                    actions.send_keys('• ' + li_text)
                    actions.send_keys(Keys.ENTER)
            actions.perform()
        elif tag.name == 'img':
            # Можно реализовать вставку изображения через UI редактора, если нужно
            pass
    actions.perform()
    print("✅ Форматированный контент вставлен через горячие клавиши (с поддержкой ссылок через тулбар)")

# === Основная функция ===
def main():
    driver = init_browser()
    article = get_article_from_db()
    if not article:
        print("❌ Не удалось загрузить статью из базы данных.")
        driver.quit()
        return
    soup = BeautifulSoup(article['content'], "html.parser")
    first_paragraph = soup.find("p")
    title = first_paragraph.get_text(strip=True) if first_paragraph else "Без заголовка"
    if first_paragraph:
        first_paragraph.decompose()
    # Собрать только внутренние теги (без <html>, <body>)
    if soup.body:
        cleaned_article = ''.join(str(tag) for tag in soup.body.contents)
    else:
        cleaned_article = ''.join(str(tag) for tag in soup.contents)
    if not open_editor(driver):
        driver.quit()
        return
    if not create_new_article(driver):
        driver.quit()
        return
    close_tips(driver)
    insert_title(driver, title)
    send_formatted_content(driver, cleaned_article)
    input("✅ Нажмите Enter, чтобы закрыть браузер...")
    driver.quit()

if __name__ == "__main__":
    main() сохраи сформатированный текст статьи в файл. прерви выполнение скрипта