#!/usr/bin/env python3

# Native modules
import json
import logging
from datetime import datetime, timedelta
import os
import sys
import time
import difflib

# Third-party modules
import requests
from dateutil.parser import parse
import pytz
from dotenv import load_dotenv
from urllib.parse import urlencode
from typing import List, Dict, Optional

load_dotenv()

# Configuration
USERNAME = os.environ.get("CROSSFIT_USERNAME")
PASSWORD = os.environ.get("CROSSFIT_PASSWORD")

if not all([USERNAME, PASSWORD]):
    raise ValueError("Missing environment variables: CROSSFIT_USERNAME and/or CROSSFIT_PASSWORD")

APPLICATION_ID = "81560887"
CATEGORY_ID = "677"  # Activity category ID for CrossFit
TIMEZONE = "Europe/Paris"  # Adjust to your timezone
TARGET_RESERVATION_TIME = "20:01"  # When bookings open (8 PM)
DEVICE_TYPE = "3"

# Retry configuration
RETRY_MAX = 3
RETRY_BACKOFF = 1
APP_VERSION = "5.09.21"

# Define your preferred sessions
# Format: List of tuples (day_of_week, start_time, session_name_contains)
# day_of_week: 0=Monday, 6=Sunday
PREFERRED_SESSIONS = [
    (4, "17:00", "WEIGHTLIFTING"),  # Friday 17:00 WEIGHTLIFTING
    (5, "12:30", "HYROX"),          # Saturday 12:30 HYROX
    (2, "18:30", "CONDITIONING"),   # Wednesday 18:30 CONDITIONING
    # (5, "15:15", "BIG WOD")       # Saturday 15:15 BIG WOD
]

# Configure logging once at script startup
logging.basicConfig(
    level=logging.DEBUG,  # Change to DEBUG for more detailed logs
    format='%(asctime)s - %(levelname)s - %(message)s',
    handlers=[
        logging.FileHandler("log/crossfit_booking.log"),
        logging.StreamHandler()
    ]
)
logging.getLogger("requests").setLevel(logging.WARNING)
logging.info("Logging enhanced with request library noise reduction")


class CrossFitBooker:
    def __init__(self):
        self.auth_token = None
        self.user_id = None
        self.session = requests.Session()
        self.base_headers = {
            "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.15; rv:140.0) Gecko/20100101 Firefox/140.0",
            "Content-Type": "application/x-www-form-urlencoded",
            "Nubapp-Origin": "user_apps",
        }
        self.session.headers.update(self.base_headers)

        # Define mandatory parameters for API calls
        self.mandatory_params = {
            "app_version": APP_VERSION,
            "device_type": DEVICE_TYPE,
            "id_application": APPLICATION_ID,
            "id_category_activity": CATEGORY_ID
        }

    def get_auth_headers(self) -> Dict:
        """Return headers with authorization if available"""
        headers = self.base_headers.copy()
        if self.auth_token:
            headers["Authorization"] = f"Bearer {self.auth_token}"
        return headers

    def login(self) -> bool:
        """Authenticate and get the bearer token"""
        try:
            # First login endpoint
            login_params = {
                "app_version": APP_VERSION,
                "device_type": DEVICE_TYPE,
                "username": USERNAME,
                "password": PASSWORD
            }

            response = self.session.post(
                "https://sport.nubapp.com/api/v4/users/checkUser.php",
                headers={"Content-Type": "application/x-www-form-urlencoded"},
                data=urlencode(login_params))

            if not response.ok:
                logging.error(f"First login step failed: {response.status_code} - {response.text} - Response: {response.text[:100]}")
                return False

            try:
                login_data = response.json()
                self.user_id = str(login_data["data"]["user"]["id_user"])
            except KeyError as ke:
                logging.error(f"Key error during login: {str(ke)} - Response: {response.text}")
                return False
            except ValueError as ve:
                logging.error(f"Value error during login: {str(ve)} - Response: {response.text}")
                return False

            # Second login endpoint
            response = self.session.post(
                "https://sport.nubapp.com/api/v4/login",
                headers={"Content-Type": "application/x-www-form-urlencoded;charset=UTF-8"},
                data=urlencode({
                    "device_type": DEVICE_TYPE,
                    "username": USERNAME,
                    "password": PASSWORD
                }))

            if response.ok:
                try:
                    login_data = response.json()
                    self.auth_token = login_data.get("token")
                except KeyError as ke:
                    logging.error(f"Key error during login: {str(ke)} - Response: {response.text}")
                    return False
                except ValueError as ve:
                    logging.error(f"Value error during login: {str(ve)} - Response: {response.text}")
                    return False

                if self.auth_token and self.user_id:
                    logging.info("Successfully logged in")
                    return True
            else:
                logging.error(f"Login failed: {response.status_code} - {response.text} - Response: {response.text[:100]}")
                return False

        except requests.exceptions.JSONDecodeError:
            logging.error("Failed to decode JSON response during login")
            return False
        except requests.exceptions.RequestException as e:
            logging.error(f"Request error during login: {str(e)}")
            return False
        except Exception as e:
            logging.error(f"Unexpected error during login: {str(e)}")
            return False

    def get_available_sessions(self, start_date: datetime, end_date: datetime) -> Optional[Dict]:
        """Fetch available sessions from the API with comprehensive error handling"""
        if not self.auth_token or not self.user_id:
            logging.error("Authentication required - missing token or user ID")
            return None

        url = "https://sport.nubapp.com/api/v4/activities/getActivitiesCalendar.php"

        # Prepare request with mandatory parameters
        request_data = self.mandatory_params.copy()
        request_data.update({
            "id_user": self.user_id,
            "start_timestamp": start_date.strftime("%d-%m-%Y"),
            "end_timestamp": end_date.strftime("%d-%m-%Y")
        })

        # Add retry logic with exponential backoff
        for retry in range(RETRY_MAX):
            try:
                try:
                    response = self.session.post(
                        url,
                        headers=self.get_auth_headers(),
                        data=urlencode(request_data),
                        timeout=10
                    )
                except requests.exceptions.Timeout:
                    logging.error(f"Request timed out after 10 seconds for URL: {url}")
                    return None
                except requests.exceptions.RequestException as e:
                    logging.error(f"Request failed for URL: {url} - Error: {str(e)}")
                    return None
                break  # Success, exit retry loop
            except requests.exceptions.JSONDecodeError:
                logging.error("Failed to decode JSON response")
                return None
            except requests.exceptions.RequestException as e:
                if retry == RETRY_MAX - 1:
                    logging.error(f"Final retry failed: {str(e)}")
                    raise  # Propagate error
                wait_time = RETRY_BACKOFF * (2 ** retry)
                logging.warning(f"Request failed (attempt {retry+1}/{RETRY_MAX}): {str(e)}. Retrying in {wait_time}s...")
                time.sleep(wait_time)
        else:
            # All retries exhausted
            logging.error(f"Failed after {RETRY_MAX} attempts")
            return None
        
        # Handle response
        if response.status_code == 200:
            try:
                json_response = response.json()
                return json_response
            except ValueError:
                logging.error("Failed to decode JSON response")
                return None
        elif response.status_code == 400:
            logging.error("400 Bad Request - likely missing or invalid parameters")
            logging.error(f"Request Data: {request_data}")
            logging.error(f"Response: {response.text[:100]}")
            return None
        elif response.status_code == 401:
            logging.error("401 Unauthorized - token may be expired or invalid")
            logging.error(f"Response: {response.text[:100]}")
            return None
        elif 500 <= response.status_code < 600:
            logging.error(f"Server error {response.status_code} - Response: {response.text[:100]}")
            raise requests.exceptions.ConnectionError(f"Server error {response.status_code}")
        else:
            logging.error(f"Unexpected status code: {response.status_code}")
            return None

    def book_session(self, session_id: str) -> bool:
        """Book a specific session with debug logging."""
        return self._make_request(
            url="https://sport.nubapp.com/api/v4/activities/bookActivityCalendar.php",
            data=self._prepare_booking_data(session_id),
            success_msg=f"Successfully booked session {session_id}"
        )

    def _prepare_booking_data(self, session_id: str) -> Dict:
        """Prepare request data for booking a session"""
        return {
            **self.mandatory_params,
            "id_activity_calendar": session_id,
            "id_user": self.user_id,
            "action_by": self.user_id,
            "n_guests": "0",
            "booked_on": "3"
        }

    def _make_request(self, url: str, data: Dict, success_msg: str) -> bool:
        """Handle API requests with retry logic and response processing"""
        for retry in range(RETRY_MAX):
            try:
                response = self.session.post(
                    url,
                    headers=self.get_auth_headers(),
                    data=urlencode(data),
                    timeout=10
                )
                
                if response.status_code == 200:
                    json_response = response.json()
                    if json_response.get("success", False):
                        logging.info(success_msg)
                        return True
                    logging.error(f"API returned success:false: {json_response}")
                    return False
                
                logging.error(f"HTTP {response.status_code}: {response.text[:100]}")
                return False

            except requests.exceptions.JSONDecodeError:
                logging.error("Failed to decode JSON response")
                return False
            except requests.exceptions.RequestException as e:
                if retry == RETRY_MAX - 1:
                    logging.error(f"Final retry failed: {str(e)}")
                    raise  # Propagate error
                wait_time = RETRY_BACKOFF * (2 ** retry)
                logging.warning(f"Request failed (attempt {retry+1}/{RETRY_MAX}): {str(e)}. Retrying in {wait_time}s...")
                time.sleep(wait_time)
        
        logging.error(f"Failed to complete request after {RETRY_MAX} attempts")
        return False

    def is_session_bookable(self, session: Dict, current_time: datetime) -> bool:
        """Check if a session is bookable based on user_info, ignoring error codes."""
        user_info = session.get("user_info", {})

        # First check if can_join is true (primary condition)
        if user_info.get("can_join", False):
            logging.debug("Session is bookable: can_join is True")
            return True
            
        # If can_join is False, check if there's a booking window
        booking_date_str = user_info.get("unableToBookUntilDate", "")
        booking_time_str = user_info.get("unableToBookUntilTime", "")
        
        if booking_date_str and booking_time_str:
            try:
                booking_datetime = datetime.strptime(
                    f"{booking_date_str} {booking_time_str}", 
                    "%d-%m-%Y %H:%M"
                )
                booking_datetime = pytz.timezone(TIMEZONE).localize(booking_datetime)
                
                if current_time >= booking_datetime:
                    logging.debug(f"Session is bookable: current_time {current_time} >= booking_datetime {booking_datetime}")
                    return True  # Booking window is open
                else:
                    return False  # Still waiting for booking to open
            except ValueError:
                pass  # Ignore invalid date formats
        
        # Default case: not bookable
        return False

    def matches_preferred_session(self, session: Dict, current_time: datetime) -> bool:
        """Check if session matches one of your preferred sessions with fuzzy matching."""
        try:
            session_time = parse(session["start_timestamp"])
            if not session_time.tzinfo:
                session_time = pytz.timezone(TIMEZONE).localize(session_time)

            day_of_week = session_time.weekday()
            session_time_str = session_time.strftime("%H:%M")
            session_name = session.get("name_activity", "").upper()

            for preferred_day, preferred_time, preferred_name in PREFERRED_SESSIONS:
                # Exact match first
                if (day_of_week == preferred_day and
                    session_time_str == preferred_time and
                    preferred_name in session_name):
                    return True

                # Fuzzy match fallback (80% similarity)
                ratio = difflib.SequenceMatcher(
                    None,
                    session_name.lower(),
                    preferred_name.lower()
                ).ratio()
                
                if (day_of_week == preferred_day and
                    abs(session_time.hour - int(preferred_time.split(':')[0])) <= 1 and
                    ratio >= 0.8):
                    logging.debug(f"Fuzzy match: {session_name} → {preferred_name} ({ratio:.2%})")
                    return True

            return False

        except Exception as e:
            logging.error(f"Failed to check session: {str(e)} - Session: {session}")
            return False

    def run_booking_cycle(self, current_time: datetime):
        """Run one cycle of checking and booking sessions"""
        # Calculate date range to check (next 3 days)
        start_date = current_time.date()
        end_date = start_date + timedelta(days=3)

        # Get available sessions
        sessions_data = self.get_available_sessions(start_date, end_date)
        if not sessions_data or not sessions_data.get("success", False):
            logging.error("No sessions available or error fetching sessions - Sessions Data: {sessions_data}")
            return

        activities = sessions_data.get("data", {}).get("activities_calendar", [])

        # Find sessions to book (both preferred and any available)
        sessions_to_book = []
        for session in activities:
            if not self.is_session_bookable(session, current_time):
                continue

            if self.matches_preferred_session(session, current_time):
                sessions_to_book.append(("Preferred", session))
            elif current_time.strftime("%H:%M") == TARGET_RESERVATION_TIME:
                # At booking time, consider all available sessions
                sessions_to_book.append(("Available", session))

        if not sessions_to_book:
            logging.info("No matching sessions found to book")
            return

        # Book sessions (preferred first)
        sessions_to_book.sort(key=lambda x: 0 if x[0] == "Preferred" else 1)
        for session_type, session in sessions_to_book:
            session_time = datetime.strptime(session["start_datetime"], "%Y-%m-%d %H:%M:%S")
            logging.info(f"Attempting to book {session_type} session at {session_time} ({session['name_activity']})")
            if self.book_session(session["id_activity_calendar"]):
                logging.info(f"Successfully booked {session_type} session at {session_time}")
            else:
                logging.error(f"Failed to book {session_type} session at {session_time} - Session: {session}")

    def run(self):
        """Main execution loop"""
        # Set up timezone
        tz = pytz.timezone(TIMEZONE)

        # Initial login
        if not self.login():
            logging.error("Authentication failed - exiting program")
            return

        while True:
            try:
                current_time = datetime.now(tz)
                logging.info(f"Current time: {current_time}")
                
                # Run booking cycle at the target time or if it's a test
                if current_time.strftime("%H:%M") == TARGET_RESERVATION_TIME:
                    self.run_booking_cycle(current_time)
                    # Wait a minute to avoid checking again immediately
                    time.sleep(60)
                else:
                    # Check again in 30 seconds
                    time.sleep(30)
            except Exception as e:
                logging.error(f"Unexpected error in booking cycle: {str(e)} - Traceback: {traceback.format_exc()}")
                time.sleep(60)  # Wait before retrying after error


if __name__ == "__main__":
    booker = CrossFitBooker()
    if not booker.login():
        logging.error("Failed to login - Traceback: {traceback.format_exc()}")
        exit(1)
    
    # Start continuous booking loop
    booker.run()