import os
import time
import requests

COOKIE_FILE = r"C:\Users\Administrator\Desktop\TikTokDownload-main\douyin_cookie.txt"

def get_douyin_cookie_auto(force_refresh=False, headless=True):
    """
    自动获取抖音访客Cookie
    支持多种方式：Edge、Chrome、Firefox、纯HTTP请求
    """
    
    # 检查是否已有Cookie且不需要强制刷新
    if not force_refresh and os.path.exists(COOKIE_FILE):
        try:
            with open(COOKIE_FILE, 'r', encoding='utf-8') as f:
                cookie = f.read().strip()
                if cookie and len(cookie) > 100:
                    print("[自动Cookie] 使用已保存的Cookie")
                    return cookie
        except Exception as e:
            print(f"[自动Cookie] 读取Cookie失败: {e}")
    
    print("[自动Cookie] 开始自动获取...")
    
    # 方法1: 尝试使用Selenium（支持Edge/Chrome/Firefox）
    cookie = try_selenium_cookie(headless)
    if cookie:
        return cookie
    
    # 方法2: 纯HTTP请求方式（不依赖浏览器）
    print("[自动Cookie] Selenium不可用，尝试HTTP请求方式...")
    cookie = try_http_cookie()
    if cookie:
        return cookie
    
    print("[自动Cookie] 错误: 所有自动获取方式都失败")
    print("[自动Cookie] 提示: 请手动获取Cookie")
    return None

def try_selenium_cookie(headless=True):
    """尝试使用Selenium获取Cookie（支持Edge/Chrome/Firefox）"""
    
    try:
        from selenium import webdriver
        
        driver = None
        
        # 尝试Edge
        try:
            from selenium.webdriver.edge.options import Options as EdgeOptions
            print("[自动Cookie] 尝试启动Edge浏览器...")
            
            options = EdgeOptions()
            if headless:
                options.add_argument("--headless")
                options.add_argument("--disable-gpu")
            options.add_argument("--no-sandbox")
            options.add_argument("--disable-dev-shm-usage")
            options.add_argument("--disable-blink-features=AutomationControlled")
            
            driver = webdriver.Edge(options=options)
            print("[自动Cookie] Edge浏览器启动成功")
            
        except Exception as e:
            print(f"[自动Cookie] Edge失败: {str(e)[:50]}")
            
            # 尝试Chrome
            try:
                from selenium.webdriver.chrome.options import Options as ChromeOptions
                print("[自动Cookie] 尝试启动Chrome浏览器...")
                
                options = ChromeOptions()
                if headless:
                    options.add_argument("--headless")
                    options.add_argument("--disable-gpu")
                options.add_argument("--no-sandbox")
                options.add_argument("--disable-dev-shm-usage")
                options.add_argument("--disable-blink-features=AutomationControlled")
                
                driver = webdriver.Chrome(options=options)
                print("[自动Cookie] Chrome浏览器启动成功")
                
            except Exception as e2:
                print(f"[自动Cookie] Chrome失败: {str(e2)[:50]}")
                
                # 尝试Firefox
                try:
                    from selenium.webdriver.firefox.options import Options as FirefoxOptions
                    print("[自动Cookie] 尝试启动Firefox浏览器...")
                    
                    options = FirefoxOptions()
                    if headless:
                        options.add_argument("--headless")
                    
                    driver = webdriver.Firefox(options=options)
                    print("[自动Cookie] Firefox浏览器启动成功")
                    
                except Exception as e3:
                    print(f"[自动Cookie] Firefox失败: {str(e3)[:50]}")
        
        if driver is None:
            return None
        
        # 访问抖音
        print("[自动Cookie] 访问抖音网站...")
        driver.get("https://www.douyin.com")
        
        print("[自动Cookie] 等待Cookie生成（5秒）...")
        time.sleep(5)
        
        # 提取Cookie
        cookies = driver.get_cookies()
        
        if not cookies:
            print("[自动Cookie] 警告: 未获取到Cookie")
            driver.quit()
            return None
        
        # 转换为字符串
        cookie_str = "; ".join([f"{c['name']}={c['value']}" for c in cookies])
        
        print(f"[自动Cookie] 成功获取 {len(cookies)} 个Cookie项")
        print(f"[自动Cookie] Cookie长度: {len(cookie_str)}")
        
        # 保存
        try:
            with open(COOKIE_FILE, 'w', encoding='utf-8') as f:
                f.write(cookie_str)
            print(f"[自动Cookie] Cookie已保存")
        except Exception as e:
            print(f"[自动Cookie] 保存失败: {e}")
        
        driver.quit()
        return cookie_str
        
    except ImportError:
        print("[自动Cookie] Selenium未安装，跳过浏览器方式")
        return None
    except Exception as e:
        print(f"[自动Cookie] Selenium方式失败: {e}")
        return None

def try_http_cookie():
    """使用纯HTTP请求获取Cookie（不依赖浏览器）"""
    
    try:
        print("[自动Cookie] 使用HTTP请求方式...")
        
        session = requests.Session()
        
        # 模拟正常浏览器访问
        headers = {
            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
            'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
            'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8',
            'Accept-Encoding': 'gzip, deflate, br',
            'Connection': 'keep-alive',
            'Upgrade-Insecure-Requests': '1',
            'Sec-Fetch-Dest': 'document',
            'Sec-Fetch-Mode': 'navigate',
            'Sec-Fetch-Site': 'none',
        }
        
        session.headers.update(headers)
        
        # 访问抖音首页
        print("[自动Cookie] 访问抖音首页...")
        resp = session.get("https://www.douyin.com", timeout=15)
        
        print(f"[自动Cookie] 响应状态: {resp.status_code}")
        
        # 提取Cookie
        cookies = session.cookies.get_dict()
        
        if cookies:
            # 转换为字符串
            cookie_str = "; ".join([f"{name}={value}" for name, value in cookies.items()])
            
            print(f"[自动Cookie] 成功获取 {len(cookies)} 个Cookie项")
            print(f"[自动Cookie] Cookie项: {', '.join(cookies.keys())}")
            print(f"[自动Cookie] Cookie长度: {len(cookie_str)}")
            
            # 保存
            try:
                with open(COOKIE_FILE, 'w', encoding='utf-8') as f:
                    f.write(cookie_str)
                print(f"[自动Cookie] Cookie已保存")
            except Exception as e:
                print(f"[自动Cookie] 保存失败: {e}")
            
            return cookie_str
        else:
            print("[自动Cookie] HTTP方式未获取到Cookie")
            return None
            
    except requests.exceptions.Timeout:
        print("[自动Cookie] HTTP请求超时")
        return None
    except requests.exceptions.ConnectionError:
        print("[自动Cookie] 网络连接失败")
        return None
    except Exception as e:
        print(f"[自动Cookie] HTTP方式失败: {e}")
        return None

if __name__ == "__main__":
    print("="*60)
    print("测试：自动获取Cookie（多浏览器+HTTP方式）")
    print("="*60)
    
    cookie = get_douyin_cookie_auto(force_refresh=True, headless=True)
    
    if cookie:
        print(f"\n成功！Cookie长度: {len(cookie)}")
        print(f"Cookie前200字符:\n{cookie[:200]}...")
    else:
        print("\n失败：无法自动获取Cookie")
        print("\n请手动获取Cookie：")
        print("1. 打开浏览器访问 www.douyin.com")
        print("2. 按F12打开开发者工具")
        print("3. 刷新页面，点击Network标签")
        print("4. 点击任意请求，复制Cookie值")