import sqlite3

from db import get_connection, init_db
from screens import (
    log_weight,
    log_workout,
    manage_workout_exercises,
    manage_workout_templates,
    view_weight_logs,
    view_workout_sessions,
)


def main_menu(conn: sqlite3.Connection) -> None:
    import ui

    while True:
        ui.clear_screen()
        print("=== EgoMetrics ===\n")
        print("0. Quit\n")

        print("=== Workout ===\n")
        print("1. Log Session")
        print("2. View Sessions Logs")
        print("3. Manage Exercises")
        print("4. Manage Session Templates\n")

        print("=== Weight ===\n")
        print("5. Log Weight")
        print("6. View Weight Logs")

        choice = input("\n> ").strip()
        if choice == "0":
            break
        if choice == "1":
            log_workout(conn)
        elif choice == "2":
            view_workout_sessions(conn)
        elif choice == "3":
            manage_workout_exercises(conn)
        elif choice == "4":
            manage_workout_templates(conn)
        elif choice == "5":
            log_weight(conn)
        elif choice == "6":
            view_weight_logs(conn)


def main() -> None:
    init_db()
    conn = get_connection()
    try:
        main_menu(conn)
    except KeyboardInterrupt:
        print("\nGoodbye!")
    finally:
        conn.close()


if __name__ == "__main__":
    main()