view_events.py

from gcsa.google_calendar import GoogleCalendar
from dotenv import load_dotenv
import os
from datetime import datetime, timedelta
from authenticate_calendar import authenticate_calendar
from colorama import Fore, Style, init

# Initialize colorama
init(autoreset=True)


def view_events():
    gc = authenticate_calendar()

    # Prompt the user to select a time range
    print(
        f"\n{Style.BRIGHT}{Fore.CYAN}☀️🌙 Select a time range option:{Style.RESET_ALL}\n"
    )
    print(f"{Fore.YELLOW}  1) Next 2 weeks{Style.RESET_ALL}")
    print(f"{Fore.YELLOW}  2) Last 2 weeks{Style.RESET_ALL}")
    print(f"{Fore.YELLOW}  3) Next month{Style.RESET_ALL}")
    print(f"{Fore.YELLOW}  4) Last month{Style.RESET_ALL}")
    print(f"{Fore.YELLOW}  5) Summarize most recent{Style.RESET_ALL}")
    print(f"{Fore.YELLOW}  6) Go back to home{Style.RESET_ALL}\n")
    choice = input(
        f"{Style.BRIGHT}Enter the option number (1/2/3/4/5/6): {Style.RESET_ALL}"
    )

    if choice == "1":
        time_min = datetime.now() - timedelta(days=0)
        time_max = datetime.now() + timedelta(days=14)
    elif choice == "2":
        time_min = datetime.now() - timedelta(days=14)
        time_max = datetime.now() + timedelta(days=0)
    elif choice == "3":
        time_min = datetime.now()
        time_max = datetime.now() + timedelta(days=30)
    elif choice == "4":
        time_min = datetime.now() - timedelta(days=30)
        time_max = datetime.now()
    elif choice == "5":
        time_min = datetime.now() - timedelta(days=3)
        time_max = datetime.now() + timedelta(days=7)
    elif choice == "6":
        print(f"\n{Fore.GREEN}Returning to main menu...{Style.RESET_ALL}")
        return
    else:
        print(
            f"{Fore.RED}\nInvalid choice. Please select 1, 2, 3, 4, 5, or 6.{Style.RESET_ALL}"
        )
        return

    events = list(
        gc.get_events(time_min, time_max, order_by="startTime", single_events=True)
    )

    if not events:
        print(f"{Fore.RED}No events found in the selected time range.{Style.RESET_ALL}")
        return

    print(
        f"\n{Style.BRIGHT}{Fore.CYAN}Events from {time_min.date()} to {time_max.date()}:{Style.RESET_ALL}"
    )
    print(f"{Fore.YELLOW}{'-' * 80}{Style.RESET_ALL}")
    print(f"{Fore.YELLOW}{'-' * 80}{Style.RESET_ALL}")

    for event in events:
        start_time = (
            event.start.strftime("%Y-%m-%d %H:%M")
            if isinstance(event.start, datetime)
            else event.start
        )
        end_time = (
            event.end.strftime("%Y-%m-%d %H:%M")
            if isinstance(event.end, datetime)
            else event.end
        )

        print(f"{Fore.YELLOW}Summary: {Style.RESET_ALL}{event.summary}")
        print(f"{Fore.CYAN}Start: {Style.RESET_ALL}{start_time}")
        print(f"{Fore.CYAN}End: {Style.RESET_ALL}{end_time}")

        if event.description:
            print(f"{Fore.CYAN}Description: {Style.RESET_ALL}{event.description}")
        print(f"{Fore.YELLOW}{'-' * 80}{Style.RESET_ALL}")


if __name__ == "__main__":
    view_events()