R/module-select-aes.R

In esquisse: Explore and Visualize Your Data Interactively

#' @importFrom shiny NS uiOutput
#' @importFrom htmltools tagList
select_aes_ui <- function(id) {
  ns <- NS(id)
  tagList(
    uiOutput(outputId = ns("ui_aesthetics"))
  )
}

#' @importFrom shiny moduleServer reactiveValues renderUI is.reactive observeEvent
select_aes_server <- function(id,
                              data_r = reactive(NULL),
                              default_aes = c("fill", "color", "size", "group", "facet"),
                              input_aes = reactive(NULL)) {
  moduleServer(
    id,
    function(input, output, session) {

      ns <- session$ns
      res_rv <- reactiveValues(aes = list())

      # Generate drag-and-drop input
      output$ui_aesthetics <- renderUI({
        if (is.reactive(default_aes)) {
          aesthetics <- default_aes()
        } else {
          if (length(input_aes()) < 1) {
            aesthetics <- default_aes
          } else {
            aesthetics <- input_aes()
          }
        }
        data <- isolate(data_r())
        if (!is.null(data)) {
          var_choices <- get_col_names(data)
          choiceValues <- var_choices
          choiceNames <- badgeType(
            col_name = var_choices,
            col_type = col_type(data[, var_choices, drop = TRUE])
          )
          selected <- dropNulls(isolate(input$dragvars$target))
        } else {
          choiceValues <- ""
          choiceNames <- ""
          selected <- NULL
        }
        if ("facet" %in% aesthetics) {
          tagList(
            tags$style(
              HTML(sprintf(
                "#%s-target-6661636574 {grid-area: 1 / %s / 3 / %s; height: auto !important;}",
                ns("dragvars"), length(aesthetics) + 2, length(aesthetics) + 2 + 1
              )),
              HTML(sprintf(
                "#%s-source-container { grid-area: 1 / 1 / 2 / %s; }",
                ns("dragvars"), length(aesthetics) + 2
              ))
            ),
            dragulaInput(
              inputId = ns("dragvars"),
              sourceLabel = "Variables",
              targetsLabels = c("X", "Y", aesthetics),
              targetsIds = c("xvar", "yvar", aesthetics),
              choiceValues = choiceValues,
              choiceNames = choiceNames,
              selected = selected,
              badge = FALSE,
              ncolGrid = length(aesthetics) + 2,
              nrowGrid = 2,
              ncolSource = NULL,
              width = "100%",
              height = "70px",
              targetsHeight = "50px",
              replace = TRUE
            )
          )
        } else {
          dragulaInput(
            inputId = ns("dragvars"),
            sourceLabel = "Variables",
            targetsLabels = c("X", "Y", aesthetics),
            targetsIds = c("xvar", "yvar", aesthetics),
            choiceValues = choiceValues,
            choiceNames = choiceNames,
            selected = selected,
            badge = FALSE,
            width = "100%",
            height = "70px",
            targetsHeight = "50px",
            replace = TRUE
          )
        }
      })

      # Update drag-and-drop input when data changes
      observeEvent(data_r(), {
        data <- data_r()
        if (is.null(data)) {
          updateDragulaInput(
            session = session,
            inputId = "dragvars",
            status = NULL,
            choices = character(0),
            badge = FALSE
          )
        } else {
          var_choices <- get_col_names(data)
          var_badges <- badgeType(
            col_name = var_choices,
            col_type = col_type(data[, var_choices, drop = TRUE])
          )
          selected <- dropNulls(isolate(input$dragvars$target))
          var_selected <- unlist(selected, use.names = FALSE)
          if (!all(var_selected %in% var_choices))
            var_selected <- NULL
          updateDragulaInput(
            session = session,
            inputId = "dragvars",
            status = NULL,
            choiceValues = var_choices,
            choiceNames = var_badges,
            # selected = shiny::isolate(input$dragvars$target),
            selectedNames = if (length(var_selected) > 0) lapply(
              X = selected,
              function(x) {
                var_badges[var_choices == x]
              }
            ),
            selectedValues = if (length(var_selected) > 0) selected,
            badge = FALSE
          )
        }
      }, ignoreNULL = FALSE)

      observeEvent(input$dragvars$target, {
        res_rv$aes <- lapply(
          X = input$dragvars$target,
          FUN = function(x) {
            if (!is.null(x))
              return(as.character(x))
            x
          }
        )
      })

      return(reactive(res_rv$aes))
    }
  )
}