screen-locker/screen_locker/_ui_flows_relaxed.py

"""Relaxed-day (Tue/Wed/Thu) UI flow methods mixin.

textgate: glyphs and numbers only (``_glyphs``); the verifier's reason goes
to the log, not the screen.
"""

from __future__ import annotations

from concurrent.futures import ThreadPoolExecutor
import logging

from screen_locker import _glyphs as g
from screen_locker._weekly_check import (
    RELAXED_DAY_SKIP_TYPE,
    WEEKLY_WORKOUT_MINIMUM,
    count_weekly_workouts,
)

_logger = logging.getLogger(__name__)


class UIFlowsRelaxedMixin:
    """Mixin providing the relaxed-day UI flow logic."""

    # ------------------------------------------------------------------
    # Relaxed-day flow (Tue/Wed/Thu — optional, no penalty for skipping)
    # ------------------------------------------------------------------

    def _start_relaxed_day_flow(self) -> None:
        """Show optional workout prompt for relaxed days (Tue-Thu).

        The screen is not locked — the user can skip freely or voluntarily
        import a Stronglift workout that counts toward the weekly minimum.
        Was "Optional Day (Tue / Wed / Thu)" / "Weekly workouts: N / M. No
        penalty for skipping today." / "Skip — No Penalty" / "Log Stronglift
        Workout": now the optional-day diamond, the week as dots + numbers,
        skip (green) and the barbell (go).
        """
        count = count_weekly_workouts(self.log_file)
        self.clear_container()
        self._label(
            g.OPTIONAL,
            role="display",
            color=self._colors.palette.warning,
            pad="md",
        )
        self._text(
            g.dots(count, WEEKLY_WORKOUT_MINIMUM),
            role="title",
            color=self._colors.palette.muted,
            pad="sm",
        )
        frame = self._button_row()
        self._button(
            frame,
            g.SKIP,
            bg=self._colors.palette.success,
            command=self._skip_relaxed_day,
            width=4,
        ).pack(side="left", padx=self._colors.space("sm"))
        self._button(
            frame,
            f"{g.GO} {g.LIFT}",
            bg=self._colors.palette.accent,
            command=self._start_relaxed_phone_check,
            width=8,
        ).pack(side="left", padx=self._colors.space("sm"))
        # Skipping is free here, but a sick day is more than a skip: it
        # stands every gate down and lifts tonight's shutdown to 20:00.
        self._sick_button(frame)

    def _skip_relaxed_day(self) -> None:
        """Dismiss today's relaxed-day prompt for the rest of the calendar day.

        Without a persisted marker here, the recurring 30-minute
        ``workout-locker.timer`` re-derives ``relaxed_day`` from scratch every
        tick and rebuilds this same prompt, since the underlying date check
        has no memory of its own of an earlier same-day dismissal.
        """
        self.workout_data = {"type": RELAXED_DAY_SKIP_TYPE}
        self.save_workout_log()
        self.close()

    def _start_relaxed_phone_check(self) -> None:
        """Run Stronglift check in relaxed mode (no screen grab)."""
        self.clear_container()
        self._label(
            f"{g.PHONE} {g.LIFT}",
            role="display",
            color=self._colors.palette.warning,
            pad="lg",
        )
        self._text(g.WAIT, role="display", color=self._colors.palette.muted)
        executor = ThreadPoolExecutor(max_workers=1)
        self._phone_future = executor.submit(self._verify_phone_workout)
        executor.shutdown(wait=False)
        self._poll_relaxed_phone_check()

    def _poll_relaxed_phone_check(self) -> None:
        """Poll background phone check in relaxed-day mode."""
        if self._phone_future is not None and self._phone_future.done():
            status, message = self._phone_future.result()
            self._handle_relaxed_phone_result(status, message)
        else:
            self.root.after(500, self._poll_relaxed_phone_check)

    def _handle_relaxed_phone_result(self, status: str, message: str) -> None:
        """Route phone check result in relaxed-day mode.

        On success saves the workout (counts toward weekly total) then closes.
        On failure shows retry, I'm sick and close.
        """
        if status == "verified":
            self.workout_data["type"] = "phone_verified"
            self.workout_data["source"] = message
            unlock_delay = 1500 if self.demo_mode else 2000
            self.root.after(unlock_delay, self.unlock_screen)
        else:
            self._show_relaxed_retry(message, status)

    def _show_relaxed_retry(self, message: str, status: str) -> None:
        """Show retry and skip-close when workout not found in relaxed mode.

        Was "No Workout Found" / the reason / "TRY AGAIN" / "Close (Skip)".
        """
        _logger.warning("Relaxed-day check found no workout (%s): %s", status, message)
        self.clear_container()
        self._label(
            f"{g.LIFT} {g.FAIL}",
            role="display",
            color=self._colors.palette.danger,
            pad="md",
        )
        self._text(
            g.status_glyph(status), role="display", color=self._colors.palette.warning
        )
        frame = self._button_row()
        self._button(
            frame,
            g.RETRY,
            bg=self._colors.palette.accent,
            command=self._start_relaxed_phone_check,
            width=4,
        ).pack(side="left", padx=self._colors.space("sm"))
        self._sick_button(frame)
        self._button(
            frame,
            g.SKIP,
            bg=self._colors.palette.success,
            command=self.close,
            width=4,
        ).pack(side="left", padx=self._colors.space("sm"))