store_cookie.py

#!/usr/bin/env python3
"""Force authentication and store LinkedIn cookie."""

import os
from linkedin_mcp_server.authentication import store_authentication
from linkedin_mcp_server.drivers.chrome import get_or_create_driver


def main():
    # Get cookie from environment
    cookie = os.getenv("LINKEDIN_COOKIE")
    if not cookie:
        print("❌ No LINKEDIN_COOKIE environment variable found")
        print("Run: export LINKEDIN_COOKIE='li_at=YOUR_COOKIE'")
        return

    try:
        # Store the cookie
        success = store_authentication(cookie)
        if success:
            print("✅ Cookie stored in keychain")

            # Test authentication by creating driver
            print("🔄 Testing authentication...")
            driver = get_or_create_driver(cookie)
            print("✅ Authentication successful!")

            # Clean up
            driver.quit()

        else:
            print("❌ Failed to store cookie")

    except Exception as e:
        print(f"❌ Authentication failed: {e}")


if __name__ == "__main__":
    main()