tt_group.R

tt_group <- function(x, ..., by = NULL, hide_group = TRUE) {
  x <- dplyr::group_by(x, {{ by }})
  vars <- dplyr::group_vars(x)

  # Error if more than a single group is supplied
  stopifnot(rlang::has_length(vars, 1))

  x <- dplyr::arrange(
    dplyr::ungroup(x),
    .data[[vars]]
  )

  if (hide_group) {
    df <- dplyr::select(
      x,
      !tidyselect::any_of(vars)
    )
  }

  tinytable::group_tt(
    tinytable::tt(df, ...),
    i = x[[vars]]
  )
}