v1.0.2

- Seconds countdown below 61s - shows 1m at 61s, then 60, 59... 0 counting down each second, then now when reset time passes

- Fast polling at reset - when either timer shows now, polls the API every 5s to pick up fresh rate limit data

- Auto-restores normal polling - stops the 5s fast poll as soon as fresh data comes back with a new reset time

Author: CodeZeno

src/native_interop.rs

@@ -16,6 +16,7 @@ pub const WINEVENT_OUTOFCONTEXT: u32 = 0x0000;
 // Timer IDs
 pub const TIMER_POLL: usize = 1;
 pub const TIMER_COUNTDOWN: usize = 2;
+pub const TIMER_RESET_POLL: usize = 3;
 
 // Custom messages
 pub const WM_APP: u32 = 0x8000;

src/poller.rs

@@ -235,8 +235,10 @@ fn format_countdown(resets_at: Option<SystemTime>) -> String {
         format!("{total_days}d")
     } else if total_mins > 61 {
         format!("{total_hours}h")
-    } else {
+    } else if total_secs > 60 {
         format!("{total_mins}m")
+    } else {
+        format!("{total_secs}")
     }
 }
 
@@ -250,6 +252,11 @@ pub fn time_until_display_change(resets_at: Option<SystemTime>) -> Option<Durati
     let total_hours = total_secs / 3600;
     let total_days = total_secs / 86400;
 
+    if total_secs <= 60 {
+        // Update every second during final countdown
+        return Some(Duration::from_secs(1));
+    }
+
     let next_boundary = if total_days >= 1 {
         Duration::from_secs(total_days * 86400)
     } else if total_mins > 61 {
@@ -269,3 +276,10 @@ pub fn time_until_display_change(resets_at: Option<SystemTime>) -> Option<Durati
         Some(Duration::from_secs(1))
     }
 }
+
+/// Returns true if either section has reached "now" (reset time has passed).
+pub fn is_past_reset(data: &UsageData) -> bool {
+    let now = SystemTime::now();
+    let past = |s: &UsageSection| matches!(s.resets_at, Some(t) if now.duration_since(t).is_ok());
+    past(&data.session) || past(&data.weekly)
+}

src/window.rs

@@ -11,7 +11,7 @@ use windows::Win32::UI::WindowsAndMessaging::*;
 use windows::core::PCWSTR;
 
 use crate::models::UsageData;
-use crate::native_interop::{self, Color, TIMER_COUNTDOWN, TIMER_POLL, WM_APP_USAGE_UPDATED};
+use crate::native_interop::{self, Color, TIMER_COUNTDOWN, TIMER_POLL, TIMER_RESET_POLL, WM_APP_USAGE_UPDATED};
 use crate::poller;
 use crate::theme;
 
@@ -559,6 +559,13 @@ fn do_poll(send_hwnd: SendHwnd) {
                 s.weekly_percent = data.weekly.percentage;
                 s.session_text = session_text;
                 s.weekly_text = weekly_text;
+                // Stop fast-poll if reset data is now fresh
+                if !poller::is_past_reset(&data) {
+                    unsafe {
+                        let _ = KillTimer(hwnd, TIMER_RESET_POLL);
+                    }
+                }
+
                 s.data = Some(data);
                 s.last_poll_ok = true;
 
@@ -615,6 +622,14 @@ fn schedule_countdown_timer() {
     };
 
     let hwnd = s.hwnd.to_hwnd();
+
+    // If a reset time has passed, poll every 5s to pick up fresh data
+    if poller::is_past_reset(data) {
+        unsafe {
+            SetTimer(hwnd, TIMER_RESET_POLL, 5_000, None);
+        }
+    }
+
     let session_delay = poller::time_until_display_change(data.session.resets_at);
     let weekly_delay = poller::time_until_display_change(data.weekly.resets_at);
 
@@ -804,7 +819,12 @@ unsafe extern "system" fn wnd_proc(
                     render_layered();
                     schedule_countdown_timer();
                 }
-                3 => {}
+                TIMER_RESET_POLL => {
+                    let sh = SendHwnd::from_hwnd(hwnd);
+                    std::thread::spawn(move || {
+                        do_poll(sh);
+                    });
+                }
                 _ => {}
             }
             LRESULT(0)