import sqlite3

import models
import ui


def log_workout(conn: sqlite3.Connection) -> None:
    exercises = models.list_workout_exercises(conn)
    if not exercises:
        print("No exercises defined. Add some first via Manage Workout Exercises.")
        return

    # Session metadata
    ui.clear_screen()
    ui.print_header("Log Workout")
    date_time = ui.prompt_datetime("Date/Time")
    session_note = ui.prompt_str("Session note (optional): ", required=False)

    # Check if user wants to start from a template
    entries: list[dict] = []
    templates = models.list_workout_templates(conn)
    if templates:
        print("\nStart from a template?")
        for i, t in enumerate(templates, 1):
            print(f"  {i}. {t['name']}")
        print("  0. No template (blank session)")
        choice = input("\n> ").strip()
        try:
            idx = int(choice)
            if 1 <= idx <= len(templates):
                # Load template exercises as defaults
                template_id = templates[idx - 1]["id"]
                _, tmpl_entries = models.get_workout_template_detail(conn, template_id)
                print(f'\nUsing template "{templates[idx - 1]["name"]}"')
                print("Press Enter to accept defaults, or type new values.\n")
                for te in tmpl_entries:
                    print(f"  -- {te['exercise_name']} --")
                    sets = ui.prompt_int(
                        f"  Sets [{te['sets'] or ''}]: ",
                        min_val=1,
                        default=te["sets"],
                    )
                    assert sets is not None
                    reps, weight, rest_time = ui.prompt_sets_detail(
                        sets,
                        default_reps=te["reps"],
                        default_rest=te["rest_time"],
                        bw_relative=bool(te["bw_relative"]),
                    )
                    lsrpe = ui.prompt_int("  Last Set RPE: ", min_val=1, max_val=10)
                    note = ui.prompt_str(
                        f"  Note [{te['note'] or ''}]: ",
                        required=False,
                        default=te["note"],
                    )
                    entries.append(
                        {
                            "exercise_id": te["exercise_id"],
                            "exercise_name": te["exercise_name"],
                            "bw_relative": bool(te["bw_relative"]),
                            "sets": sets,
                            "reps": reps,
                            "weight": weight,
                            "rest_time": rest_time,
                            "lsrpe": lsrpe,
                            "note": note,
                        }
                    )
                    print(f'  "{te["exercise_name"]}" added to session.')
        except (ValueError, IndexError):
            pass  # Invalid input or 0 — proceed without template

    # Build exercise list — collect everything in memory before saving
    while True:
        print("\nAvailable exercises:")
        for e in exercises:
            print(f"  {e['id']}. {e['name']}")
        print()
        choice = input("Select exercise ID ('d' = done): ").strip()

        if choice.lower() == "d":
            if not entries:
                print("No exercises added. Aborting.")
                return
            break

        # Select existing exercise and prompt for set details
        try:
            eid = int(choice)
        except ValueError:
            print("Invalid input.")
            continue
        ex = models.get_workout_exercise(conn, eid)
        if not ex:
            print("Exercise not found.")
            continue

        print(f"\n  -- {ex['name']} --")
        sets = ui.prompt_int("  Sets: ", min_val=1)
        assert sets is not None
        reps, weight, rest_time = ui.prompt_sets_detail(
            sets, bw_relative=bool(ex["bw_relative"])
        )
        lsrpe = ui.prompt_int("  Last Set RPE: ", min_val=1, max_val=10)
        note = ui.prompt_str("  Note (optional): ", required=False)

        entries.append(
            {
                "exercise_id": eid,
                "exercise_name": ex["name"],
                "bw_relative": bool(ex["bw_relative"]),
                "sets": sets,
                "reps": reps,
                "weight": weight,
                "rest_time": rest_time,
                "lsrpe": lsrpe,
                "note": note,
            }
        )
        print(f'  "{ex["name"]}" added to session.')

    # Show summary and confirm before writing to DB
    ui.clear_screen()
    ui.print_header("Session Summary")
    print(f"Date: {date_time}")
    if session_note:
        print(f"Note: {session_note}")
    ui.print_table(
        ["#", "Exercise", "Sets", "Reps", "Weight", "Rest", "LSRPE", "Note"],
        [
            [
                str(i),
                e["exercise_name"],
                str(e["sets"]),
                str(e["reps"]).replace(",", "/"),
                ui.format_weight(e["weight"], e["bw_relative"]),
                ui.format_rest_time(e["rest_time"]),
                str(e["lsrpe"]),
                e["note"] or "",
            ]
            for i, e in enumerate(entries, 1)
        ],
    )

    if ui.confirm("\nSave this session?"):
        models.save_workout_session(conn, date_time, session_note, entries)
        print("Session saved!")
    else:
        print("Session discarded.")
    ui.pause()


def view_workout_sessions(conn: sqlite3.Connection) -> None:
    while True:
        ui.clear_screen()
        # List all sessions with exercise name preview
        sessions = models.list_workout_sessions(conn)
        if not sessions:
            print("\nNo sessions recorded yet.")
            return
        ui.print_header("Past Workout Sessions")
        ui.print_table(
            ["#", "Date", "Exercises", "Note"],
            [
                [str(i), s["date_time"], s["exercises"] or "", s["note"] or ""]
                for i, s in enumerate(sessions, 1)
            ],
        )

        # Select a session to view details
        choice = input("\nSelect # for details ('b' = back): ").strip()
        if choice.lower() == "b":
            break
        try:
            idx = int(choice) - 1
            if idx < 0 or idx >= len(sessions):
                print("Invalid selection.")
                continue
        except ValueError:
            print("Invalid input.")
            continue

        # Show full session detail with all exercise entries
        ui.clear_screen()
        session_id = sessions[idx]["id"]
        session, entries = models.get_workout_session_detail(conn, session_id)
        assert session is not None
        ui.print_header(f"Session: {session['date_time']}")
        if session["note"]:
            print(f"Note: {session['note']}\n")
        ui.print_table(
            ["#", "Exercise", "Sets", "Reps", "Weight", "Rest", "LSRPE", "Note"],
            [
                [
                    str(e["position"]),
                    e["exercise_name"],
                    str(e["sets"]),
                    str(e["reps"]).replace(",", "/"),
                    ui.format_weight(str(e["weight"]), bool(e["bw_relative"])),
                    ui.format_rest_time(e["rest_time"]),
                    str(e["lsrpe"]),
                    e["note"] or "",
                ]
                for e in entries
            ],
        )
        action = input("\nActions: (b)ack, (d)elete session\n> ").strip().lower()
        if action == "d":
            if ui.confirm("Delete this session?"):
                models.delete_workout_session(conn, session_id)
                print("Session deleted.")
                ui.pause()