#!/usr/bin/env python3
import logging
from datetime import datetime, timezone
from foodie_utils import (
    AUTHORS, check_author_rate_limit, load_json_file,
    get_x_rate_limit_status, update_system_activity, is_any_script_running
)
import time
import sys
import os

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

def display_author_status(author):
    """Display detailed status for a single author."""
    username = author['username']
    can_post, remaining, reset = check_author_rate_limit(author)
    reset_time = datetime.fromtimestamp(reset, tz=timezone.utc).strftime('%Y-%m-%d %H:%M:%S')
    
    # Only check API if no scripts are running
    if not is_any_script_running():
        api_remaining, api_reset = get_x_rate_limit_status(author)
        if api_remaining is not None:
            api_reset_time = datetime.fromtimestamp(api_reset, tz=timezone.utc).strftime('%Y-%m-%d %H:%M:%S')
            # Use API values as primary display
            remaining = api_remaining
            reset_time = api_reset_time
            can_post = remaining > 0
    
    status = "✅" if can_post else "❌"
    print(f"\n{status} {username}:")
    print(f"  • Remaining tweets: {remaining}/17")
    print(f"  • Reset time: {reset_time}")
    print(f"  • Can post: {'Yes' if can_post else 'No'}")
    
    # Show API status for verification
    if not is_any_script_running():
        api_remaining, api_reset = get_x_rate_limit_status(author)
        if api_remaining is not None:
            api_reset_time = datetime.fromtimestamp(api_reset, tz=timezone.utc).strftime('%Y-%m-%d %H:%M:%S')
            print(f"  • API Status: {api_remaining} remaining, resets at {api_reset_time}")

def display_total_capacity():
    """Display total capacity across all authors."""
    total_capacity = len(AUTHORS) * 17
    total_used = 0
    available_authors = 0
    
    print("\n=== X Posting Capacity Status ===")
    print(f"Total daily capacity: {total_capacity} tweets ({len(AUTHORS)} authors × 17 tweets)")
    print("\nAuthor Status:")
    
    for author in AUTHORS:
        can_post, remaining, _ = check_author_rate_limit(author)
        # Only check API if no scripts are running
        if not is_any_script_running():
            api_remaining, _ = get_x_rate_limit_status(author)
            if api_remaining is not None:
                remaining = api_remaining
                can_post = remaining > 0
        
        used = 17 - remaining
        total_used += used
        if can_post:
            available_authors += 1
        display_author_status(author)
    
    print("\n=== Summary ===")
    print(f"Total tweets used today: {total_used}")
    print(f"Total tweets remaining: {total_capacity - total_used}")
    print(f"Authors available to post: {available_authors}/{len(AUTHORS)}")
    
    # Calculate percentage used
    percent_used = (total_used / total_capacity) * 100
    print(f"Capacity used: {percent_used:.1f}%")
    
    if percent_used > 80:
        print("\n⚠️  Warning: High capacity usage! Consider adding more authors.")
    elif percent_used > 60:
        print("\nℹ️  Note: Moderate capacity usage. Monitor usage patterns.")

def main():
    try:
        # Update system activity
        update_system_activity("check_x_capacity", "running", os.getpid())
        
        # Display capacity status
        display_total_capacity()
        
        # Update system activity
        update_system_activity("check_x_capacity", "stopped")
        
    except KeyboardInterrupt:
        print("\nScript interrupted by user")
        update_system_activity("check_x_capacity", "stopped")
        sys.exit(0)
    except Exception as e:
        logger.error(f"Error: {e}")
        update_system_activity("check_x_capacity", "stopped")
        sys.exit(1)

if __name__ == "__main__":
    main()