Set focus to new and used columns after mutate()

tidyverse · Aug 28, 2022 · 4204ca9 · 4204ca9
1 parent f2fb03c
commit 4204ca9
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -80,3 +80,5 @@ Config/Needs/website:
     shiny,
     pkgdown,
     tidyverse/tidytemplate
+Remotes: 
+    r-lib/pillar@f-focus-attr
diff --git a/R/compute-collect.r b/R/compute-collect.r
@@ -57,7 +57,11 @@ collapse <- function(x, ...) {
 }
 
 #' @export
-collect.data.frame <- function(x, ...) x
+collect.data.frame <- function(x, ...) {
+  attr(x, "pillar_focus") <- NULL
+  x
+}
+
 #' @export
 compute.data.frame <- function(x, ...) x
 #' @export

diff --git a/R/mutate.R b/R/mutate.R
@@ -221,7 +221,9 @@ mutate.data.frame <- function(.data,
     cols_retain <- setdiff(cols_out, c(cols_used, cols_unused))
   }
 
-  dplyr_col_select(out, cols_retain)
+  out <- dplyr_col_select(out, cols_retain)
+  attr(out, "pillar_focus") <- intersect(cols_retain, c(cols_expr, cols_used))
+  out
 }
 
 #' @rdname mutate