R/rules-line-break.R

In mlr-org/styler.mlr: Third-Party Style Guide for 'mlr'

#' Force the assignment operator `=` when possible
#'
#' Because you can't determine from a nest with `<-` what the context is,
#' we need to do it from the parent.
#' @keywords internal
force_assignment_eq = function(pd) {
  if (styler:::is_function_call(pd)) {
    return(pd)
  }
  before_assignemnt = styler:::previous_non_comment(
    pd,
    which(pd$token == "expr")[1]
  )

  # don't replace if before is (
  if (length(before_assignemnt) > 0 && pd$token[before_assignemnt] == "'('") {
    return(pd)
  }
  pd$child[pd$token == "expr"] = purrr::map(
    pd$child[pd$token == "expr"], force_assignment_eq_impl
  )
  pd
}

force_assignment_eq_impl = function(pd_child) {
  to_replace = pd_child$token == "LEFT_ASSIGN" & pd_child$text == "<-"
  if (any(to_replace)) {
    pd_child$token[to_replace] = "EQ_ASSIGN"
    pd_child$text[to_replace] = "="
  }
  return(pd_child)
}

# set_line_break_after_fun_dec_header = function(pd) {
#   if (any(pd$token == "FUNCTION") &&
#     pd$child[[nrow(pd)]]$token[1] == "'{'"
#   ) {
#     pos = styler:::next_non_comment(pd$child[[nrow(pd)]], 1)
#     pd$child[[nrow(pd)]]$lag_newlines[pos] = min(
#       2L,
#       pd$child[[nrow(pd)]]$lag_newlines[pos]
#     )
#   }
#   pd
# }


# if ) follows on }, add line break
add_line_break_before_round_closing_after_curly = function(pd) {
  round_after_curly = pd$token == "')'" & (pd$token_before == "'}'")
  pd$lag_newlines[round_after_curly] = min(pd$lag_newlines[round_after_curly], 1L)
  pd
}

style_line_break_around_curly = function(strict, pd) {
  if (styler:::is_curly_expr(pd) && nrow(pd) > 2) {
    closing_before = pd$token == "'}'"
    opening_before = (pd$token == "'{'") & (pd$token_after != "COMMENT")
    to_break = styler:::lag(opening_before, default = FALSE) | closing_before
    pd$lag_newlines[to_break] = pmin(2L, pmax(1L, pd$lag_newlines[to_break]))
  }
  pd
}

#' Like `styler:::set_line_break_before_closing_call()` but only adding a line
#' break if multi-line function call **and** the token before the closing brace
#' is a curly brace (otherwise indention does not work nicely).
#' @keywords internal
set_line_break_before_closing_call = function(pd, except_token_before) {
  if (!styler:::is_function_call(pd) && !styler:::is_subset_expr(pd)) {
    return(pd)
  }

  npd = nrow(pd)
  is_multi_line = any(pd$lag_newlines[rlang::seq2(3L, npd - 1L)] > 0)
  if (!is_multi_line) {
    exception = which(pd$token_before %in% except_token_before)
    pd$lag_newlines[setdiff(npd, exception)] = 0L
    return(pd)
  } else if (pd$token_before[npd] == "'}'") {
    pd$lag_newlines[npd] = 1L
  }

  pd
}