auth.py

"""
Browser authentication and cookie detection for MyFitnessPal.

Handles automatic cookie detection from Firefox, Edge, and Chrome.
If no cookies found, opens browser for manual login.
Can also clear cookies to force fresh login.
"""

import time
import webbrowser
from pathlib import Path
import os
import sqlite3

import browser_cookie3

try:
    import config
except ImportError:
    print("ERROR: config.py not found!")
    print("Please copy config.example.py to config.py and fill in your settings.")
    raise


def find_firefox_cookie_file():
    """
    Find Firefox cookie file on Linux (standard, Snap, or Flatpak).
    
    Returns:
        str: Path to cookies.sqlite file, or None if not found
    """
    import platform
    
    home = Path.home()
    paths = []
    
    if platform.system() == 'Linux':
        # Linux Firefox locations
        paths = [
            home / '.mozilla' / 'firefox',                              # Standard Firefox
            home / 'snap' / 'firefox' / 'common' / '.mozilla' / 'firefox',  # Snap Firefox
            home / '.var' / 'app' / 'org.mozilla.firefox' / '.mozilla' / 'firefox',  # Flatpak Firefox
        ]
    else:
        # Windows Firefox locations (fallback)
        paths = [
            Path(os.environ.get('APPDATA', '')) / 'Mozilla' / 'Firefox',
            Path(os.environ.get('LOCALAPPDATA', '')) / 'Mozilla' / 'Firefox',
        ]
        
        # Microsoft Store Firefox location
        local_app_data = os.environ.get('LOCALAPPDATA', '')
        if local_app_data:
            packages_dir = Path(local_app_data) / 'Packages'
            if packages_dir.exists():
                mozilla_packages = [p for p in packages_dir.iterdir() 
                                  if 'mozilla' in p.name.lower() and 'firefox' in p.name.lower()]
                for pkg in mozilla_packages:
                    store_path = pkg / 'LocalCache' / 'Roaming' / 'Mozilla' / 'Firefox'
                    if store_path.exists():
                        paths.insert(0, store_path)
    
    # Look for cookies.sqlite in profiles
    for firefox_path in paths:
        if not firefox_path.exists():
            continue
        # Check for Profiles subdirectory (standard structure)
        profiles_dir = firefox_path / 'Profiles'
        if profiles_dir.exists():
            for profile in profiles_dir.iterdir():
                cookie_file = profile / 'cookies.sqlite'
                if cookie_file.exists():
                    print(f"  Found cookie file: {cookie_file}")
                    return str(cookie_file)
        # Also check directly in firefox_path (some installations)
        for item in firefox_path.iterdir():
            if item.is_dir():
                cookie_file = item / 'cookies.sqlite'
                if cookie_file.exists():
                    print(f"  Found cookie file: {cookie_file}")
                    return str(cookie_file)
    
    print("  No Firefox cookie file found in standard locations.")
    return None


def get_working_cookies():
    """
    Get cookies from Firefox only.
    
    Returns:
        tuple: (cookies, "Firefox") or (None, None) if no cookies found
    """
    # Try Firefox with Microsoft Store support
    cookie_file = find_firefox_cookie_file()
    if cookie_file:
        try:
            cookies = browser_cookie3.firefox(
                cookie_file=cookie_file,
                domain_name='www.myfitnesspal.com'
            )
            cookie_list = list(cookies)
            if cookie_list:
                return cookies, "Firefox"
        except:
            pass
    
    # Try standard Firefox
    try:
        cookies = browser_cookie3.firefox(domain_name='www.myfitnesspal.com')
        cookie_list = list(cookies)
        if cookie_list:
            return cookies, "Firefox"
    except:
        pass
    
    return None, None


def wait_for_login():
    """
    Poll for cookies after browser login.
    
    Returns:
        myfitnesspal.Client: Authenticated client
        
    Raises:
        TimeoutError: If login doesn't complete within timeout
    """
    import myfitnesspal
    
    start_time = time.time()
    poll_interval = 5
    cookies_found = False
    
    while True:
        elapsed = time.time() - start_time
        
        if elapsed > config.MAX_LOGIN_WAIT_SECONDS:
            raise TimeoutError(
                f"Timeout waiting for login. Please log in to MyFitnessPal "
                f"in your browser and try again."
            )
        
        cookies, browser_name = get_working_cookies()
        
        # Only try to authenticate when we FIRST detect cookies
        # Don't keep retrying - that triggers rate limiting
        if cookies and not cookies_found:
            cookies_found = True
            print(f"\n\nCookies detected in {browser_name}!")
            print("Attempting authentication...")
            
            try:
                client = myfitnesspal.Client(cookiejar=cookies)
                print("Authentication successful!")
                return client
            except Exception as e:
                print(f"\nAuthentication failed: {str(e)}")
                print("\nThis usually means MyFitnessPal is blocking API access.")
                print("Possible solutions:")
                print("  1. Wait 30-60 minutes and try again")
                print("  2. Clear MyFitnessPal cookies in your browser and log in fresh")
                print("  3. Try from a different network/IP")
                raise
        
        if not cookies:
            remaining = int(config.MAX_LOGIN_WAIT_SECONDS - elapsed)
            print(f"Waiting for login... ({remaining}s remaining)", end='\r')
        
        time.sleep(poll_interval)


def ensure_logged_in():
    """
    Ensure user is logged in to MyFitnessPal via Firefox cookies.
    
    Tries Firefox cookies first. If not found or invalid, opens Firefox
    for user to log in, then waits for valid cookies.
    
    Returns:
        myfitnesspal.Client: Authenticated client
        
    Raises:
        TimeoutError: If user doesn't log in within MAX_LOGIN_WAIT_SECONDS
    """
    import myfitnesspal
    
    print("Checking for Firefox cookies...")
    cookies, browser_name = get_working_cookies()
    
    if cookies:
        print(f"Found cookies in {browser_name}!")
        try:
            client = myfitnesspal.Client(cookiejar=cookies)
            print("Authentication successful!")
            return client
        except Exception as e:
            print(f"\nWarning: Firefox cookies found but authentication failed: {e}")
            print("Opening Firefox for fresh login...")
            # Fall through to open browser
    else:
        print("\nNo Firefox cookies found.")
        print("Opening Firefox for login...")
    
    # Open Firefox and wait for login
    webbrowser.get('firefox').open(config.MFP_LOGIN_URL)
    print(f"Please log in to MyFitnessPal in Firefox.")
    print(f"Waiting up to {config.MAX_LOGIN_WAIT_SECONDS} seconds...\n")
    
    return wait_for_login()