From cd5880bd79535f16849cc43e6c0c1a44260f544b Mon Sep 17 00:00:00 2001
From: Rich FitzJohn <r.fitzjohn@imperial.ac.uk>
Date: Fri, 13 Dec 2024 16:20:14 +0000
Subject: [PATCH 1/3] Add a spinner to progress bar

---
 R/progress.R                   | 16 ++++++++++++----
 tests/testthat/test-progress.R | 15 +++++++++++++++
 2 files changed, 27 insertions(+), 4 deletions(-)

diff --git a/R/progress.R b/R/progress.R
index 3ebc567f..898cc528 100644
--- a/R/progress.R
+++ b/R/progress.R
@@ -52,13 +52,13 @@ progress_bar_simple <- function(n_steps, every_s = 1, min_updates = 20) {
 progress_bar_fancy <- function(n_chains, n_steps, show_overall,
                                single_chain = FALSE) {
   ## We're expecting to take a while, so we show immediately, if enabled:
-  oo <- options(cli.progress_show_after = 0)
-  on.exit(options(oo))
+  oo <- options(cli.progress_show_after = 0,
+                cli.spinner = monty_spinner())
 
   e <- new.env()
   e$n <- rep(0, n_chains)
   overall <- progress_overall(n_chains, n_steps, show_overall, single_chain)
-  fmt <- paste("Sampling {overall(e$n)} {cli::pb_bar} |",
+  fmt <- paste("{cli::pb_spin} Sampling {overall(e$n)} {cli::pb_bar} |",
                "{cli::pb_percent} ETA: {cli::pb_eta}")
   fmt_done <- paste(
     "{cli::col_green(cli::symbol$tick)} Sampled {cli::pb_total} steps",
@@ -82,11 +82,12 @@ progress_bar_fancy <- function(n_chains, n_steps, show_overall,
     changed <- any(e$n[chain_id] != at, na.rm = TRUE)
     if (changed) {
       e$n[chain_id] <- at
-      cli::cli_progress_update(id = id, set = sum(e$n))
     }
+    cli::cli_progress_update(id = id, set = sum(e$n))
   }
 
   fail <- function() {
+    options(oo)
     cli::cli_progress_done(id, result = "failed")
   }
 
@@ -155,3 +156,10 @@ with_progress_fail_on_error <- function(progress, code) {
     error = function(e) progress$fail(),
     interrupt = function(e) progress$fail())
 }
+
+
+monty_spinner <- function(date = Sys.Date()) {
+  getOption(
+    "cli.spinner",
+    if (format(date, "%m") == "12") "christmas" else "dots12")
+}
diff --git a/tests/testthat/test-progress.R b/tests/testthat/test-progress.R
index b22ca716..b2901b84 100644
--- a/tests/testthat/test-progress.R
+++ b/tests/testthat/test-progress.R
@@ -150,3 +150,18 @@ test_that("can fail progress bar nicely", {
   expect_s3_class(res$result, "simpleError")
   expect_equal(conditionMessage(res$result), "some error")
 })
+
+
+test_that("can get default spinner", {
+  withr::with_options(list(cli.spinner = NULL), {
+    expect_equal(monty_spinner(), monty_spinner(Sys.Date()))
+    expect_equal(monty_spinner(as.Date("2025-03-01")), "dots12")
+    expect_equal(monty_spinner(as.Date("2025-12-05")), "christmas")
+  })
+
+  withr::with_options(list(cli.spinner = "dots"), {
+    expect_equal(monty_spinner(), "dots")
+    expect_equal(monty_spinner(as.Date("2025-03-01")), "dots")
+    expect_equal(monty_spinner(as.Date("2025-12-05")), "dots")
+  })
+})

From 6df764d2483145d4cf2684b70ac134f8e011f45b Mon Sep 17 00:00:00 2001
From: Rich FitzJohn <r.fitzjohn@imperial.ac.uk>
Date: Fri, 13 Dec 2024 16:20:28 +0000
Subject: [PATCH 2/3] Bump version

---
 DESCRIPTION | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/DESCRIPTION b/DESCRIPTION
index def22610..06316c5f 100644
--- a/DESCRIPTION
+++ b/DESCRIPTION
@@ -1,6 +1,6 @@
 Package: monty
 Title: Monte Carlo Models
-Version: 0.3.19
+Version: 0.3.20
 Authors@R: c(person("Rich", "FitzJohn", role = c("aut", "cre"),
                     email = "rich.fitzjohn@gmail.com"),
              person("Wes", "Hinsley", role = "aut"),

From 1df64dd52b339d83fc8f179c0d801144f9513117 Mon Sep 17 00:00:00 2001
From: Rich FitzJohn <r.fitzjohn@imperial.ac.uk>
Date: Fri, 13 Dec 2024 16:38:11 +0000
Subject: [PATCH 3/3] Avoid writing to closed progress bar

---
 R/progress.R | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/R/progress.R b/R/progress.R
index 898cc528..d5b017da 100644
--- a/R/progress.R
+++ b/R/progress.R
@@ -78,12 +78,12 @@ progress_bar_fancy <- function(n_chains, n_steps, show_overall,
   update <- function(chain_id, at) {
     ## Avoid writing into a closed progress bar, it will cause an
     ## error.  We do this by checking to see if progress has changed
-    ## from last time we tried updating.
-    changed <- any(e$n[chain_id] != at, na.rm = TRUE)
-    if (changed) {
+    ## from last time we tried updating, or if we're simply
+    ## incomplete.
+    if (any(at < n_steps | at > e$n)) {
       e$n[chain_id] <- at
+      cli::cli_progress_update(id = id, set = sum(e$n))
     }
-    cli::cli_progress_update(id = id, set = sum(e$n))
   }
 
   fail <- function() {