#!/usr/bin/env python3
"""
Telethon Session Initialization Script
Инициализирует сессию Telethon с интерактивной авторизацией
Сохраняет сессию для использования в Docker контейнере

Использование:
    python3 init_telethon_session.py
"""

import asyncio
import os
import sys
from pathlib import Path
from dotenv import load_dotenv
from telethon import TelegramClient
from telethon.sessions import StringSession
import logging

# Configure logging
logging.basicConfig(
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    level=logging.INFO
)
logger = logging.getLogger(__name__)

# Load environment variables
env_path = Path(__file__).parent / '.env'
load_dotenv(env_path)

# Get Telethon credentials
TELETHON_API_ID = os.getenv('TELETHON_API_ID')
TELETHON_API_HASH = os.getenv('TELETHON_API_HASH')
TELETHON_PHONE = os.getenv('TELETHON_PHONE')

# Session directory
SESSION_DIR = Path(__file__).parent / 'app' / 'sessions'
SESSION_FILE = SESSION_DIR / 'userbot_session.session'
SESSION_STRING_FILE = SESSION_DIR / 'telethon_string_session.txt'


async def initialize_session():
    """Initialize Telethon session with interactive authentication"""
    
    print("\n" + "="*80)
    print("🚀 Telethon Session Initialization")
    print("="*80)
    
    # Verify credentials
    if not all([TELETHON_API_ID, TELETHON_API_HASH, TELETHON_PHONE]):
        print("❌ Missing Telethon credentials in .env:")
        print(f"   TELETHON_API_ID: {bool(TELETHON_API_ID)}")
        print(f"   TELETHON_API_HASH: {bool(TELETHON_API_HASH)}")
        print(f"   TELETHON_PHONE: {bool(TELETHON_PHONE)}")
        print("\nPlease set these variables in .env file")
        return False
    
    # Create session directory
    SESSION_DIR.mkdir(parents=True, exist_ok=True)
    
    try:
        # Create Telethon client
        client = TelegramClient(
            str(SESSION_FILE),
            api_id=int(TELETHON_API_ID),
            api_hash=TELETHON_API_HASH
        )
        
        print(f"\n📱 Phone: {TELETHON_PHONE}")
        print(f"🔐 API ID: {TELETHON_API_ID}")
        
        # Connect to Telegram
        print("\n🔗 Connecting to Telegram...")
        await client.connect()
        print("✅ Connected!")
        
        # Check if already authorized
        if await client.is_user_authorized():
            print("ℹ️  Session already authorized!")
            me = await client.get_me()
            print(f"👤 User: {me.first_name} (@{me.username})")
        else:
            # Start authorization flow
            print("\n🔐 Starting authorization process...")
            print(f"📲 You will receive an SMS code to {TELETHON_PHONE}")
            
            # Send auth code request
            await client.send_code_request(TELETHON_PHONE)
            
            # Get code from user
            try:
                code = input("\n📝 Enter the SMS code: ").strip()
                
                # Try to sign in
                try:
                    await client.sign_in(TELETHON_PHONE, code)
                    print("✅ Successfully signed in!")
                    
                except Exception as e:
                    # Might need 2FA
                    print(f"⚠️  {str(e)}")
                    password = input("🔒 Enter 2FA password (if prompted): ").strip()
                    if password:
                        await client.sign_in(password=password)
                        print("✅ Successfully signed in with 2FA!")
                    else:
                        print("❌ Failed to sign in")
                        return False
            
            except KeyboardInterrupt:
                print("\n❌ Authorization cancelled by user")
                return False
        
        # Verify and save session
        me = await client.get_me()
        print(f"\n✅ Session initialized for: {me.first_name} (@{me.username})")
        print(f"   ID: {me.id}")
        print(f"   Username: @{me.username}")
        
        # Get and save string session
        string_session = StringSession.save(client.session)
        with open(SESSION_STRING_FILE, 'w') as f:
            f.write(string_session)
        print(f"\n💾 Session saved to: {SESSION_FILE}")
        print(f"💾 String session saved to: {SESSION_STRING_FILE}")
        
        # Disconnect
        await client.disconnect()
        print("\n✅ Session initialization completed successfully!")
        return True
    
    except Exception as e:
        print(f"\n❌ Error during session initialization: {e}")
        logger.error(f"Error: {e}", exc_info=True)
        return False


async def verify_session():
    """Verify existing session"""
    
    print("\n" + "="*80)
    print("🔍 Verifying existing session...")
    print("="*80)
    
    if not SESSION_FILE.exists():
        print(f"❌ Session file not found: {SESSION_FILE}")
        return False
    
    try:
        client = TelegramClient(
            str(SESSION_FILE),
            api_id=int(TELETHON_API_ID),
            api_hash=TELETHON_API_HASH
        )
        
        print("🔗 Connecting with existing session...")
        await client.connect()
        
        if await client.is_user_authorized():
            me = await client.get_me()
            print(f"✅ Session is valid!")
            print(f"   User: {me.first_name} (@{me.username})")
            print(f"   ID: {me.id}")
            await client.disconnect()
            return True
        else:
            print("❌ Session is not authorized")
            await client.disconnect()
            return False
    
    except Exception as e:
        print(f"❌ Error verifying session: {e}")
        return False


def print_instructions():
    """Print instructions for using the session in Docker"""
    
    print("\n" + "="*80)
    print("📋 Instructions for Docker")
    print("="*80)
    
    print("""
1. After session is created, the files will be at:
   - app/sessions/userbot_session.session
   - app/sessions/telethon_string_session.txt

2. Ensure these files are copied to Docker container:
   docker cp app/sessions/ tg_autoposter_userbot:/app/

3. Or mount as volume in docker-compose.yml:
   volumes:
     - ./app/sessions:/app/app/sessions

4. Restart the UserBot container:
   docker-compose restart userbot

5. Check logs:
   docker-compose logs userbot -f
""")


async def main():
    """Main function"""
    
    if len(sys.argv) > 1 and sys.argv[1] == '--verify':
        # Verify existing session
        result = await verify_session()
        sys.exit(0 if result else 1)
    
    # Initialize new session
    result = await initialize_session()
    
    if result:
        print_instructions()
        sys.exit(0)
    else:
        sys.exit(1)


if __name__ == '__main__':
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        print("\n\n❌ Interrupted by user")
        sys.exit(1)