perf: Do not recreate and fill the environment in .savvy_wrap_PlRExpr() (#1444)

Co-authored-by: eitsupi <[email protected]>

Author: etiennebacher

.lintr.R

@@ -9,6 +9,7 @@ linters <- linters_with_defaults(
 exclusions <- as.list(
   c(
     "R/000-wrappers.R",
+    "R/001-wrappers.R",
     fs::dir_ls("R", glob = "**/import-standalone-*.R") |>
       rlang::set_names(NULL)
   )

NAMESPACE

@@ -3,6 +3,7 @@
 S3method("!",polars_expr)
 S3method("!=",polars_expr)
 S3method("!=",polars_series)
+S3method("$",PlRExpr)
 S3method("$",polars_data_frame)
 S3method("$",polars_dtype)
 S3method("$",polars_dtype_enum)

R/001-wrappers.R

@@ -0,0 +1,27 @@
+# Override the default `.savvy_wrap_*` functions for avoiding assignment many members to the env.
+# See <https://github.com/pola-rs/r-polars/issues/1439> for details.
+# TODO: generate this file automatically
+
+PlRExprMethods <- new.env(parent = emptyenv())
+
+`.savvy_wrap_PlRExpr` <- function(ptr) {
+  e <- new.env(parent = emptyenv())
+  e$.ptr <- ptr
+
+  class(e) <- c("PlRExpr", "savvy_polars__sealed")
+  e
+}
+
+#' @export
+`$.PlRExpr` <- function(x, name) {
+  method_names <- names(PlRExprMethods)
+  ptr <- env_get(x, ".ptr")
+
+  if (identical(name, ".ptr")) {
+    ptr
+  } else if (name %in% method_names) {
+    PlRExprMethods[[name]](ptr)
+  } else {
+    NextMethod()
+  }
+}

R/zzz.R

@@ -36,6 +36,7 @@ assign_objects_to_env <- function(env, obj_name_pattern, ..., search_env = paren
 }
 
 POLARS_STORE_ENVS <- list(
+  "PlRExpr_" = PlRExprMethods,
   "pl__" = pl,
   "cs__" = cs,
   "pl_api_" = pl__api,