chore: Use built-in rfuns extension

Author: krlmlr

Diff for: R/meta.R

@@ -1,12 +1,14 @@
 pre_code_cache <- collections::queue()
 code_cache <- collections::queue()
+ext_cache <- collections::dict()
 macro_cache <- collections::dict()
 df_cache <- collections::dict()
 rel_cache <- collections::dict()
 
 meta_clear <- function() {
   pre_code_cache$clear()
   code_cache$clear()
+  ext_cache$clear()
   macro_cache$clear()
   df_cache$clear()
   rel_cache$clear()
@@ -26,7 +28,8 @@ meta_replay <- function(add_pre_code = TRUE) {
   if (add_pre_code) {
     con_exprs <- list(
       expr(duckdb <- asNamespace("duckdb")),
-      expr(con <- DBI::dbConnect(duckdb::duckdb())),
+      expr(drv <- duckdb::duckdb()),
+      expr(con <- DBI::dbConnect(drv)),
       expr(experimental <- !!(Sys.getenv("DUCKPLYR_EXPERIMENTAL") == "TRUE"))
     )
     con_code <- map(con_exprs, constructive::deparse_call)
@@ -100,6 +103,22 @@ meta_eval <- function() {
   eval(parse(text = code))
 }
 
+meta_ext_register <- function(name = "rfuns") {
+  if (ext_cache$has(name)) {
+    return(invisible())
+  }
+
+  stopifnot(identical(name, "rfuns"))
+
+  ext_install_expr <- expr(invisible(
+    duckdb$rapi_load_rfuns(drv@database_ref)
+  ))
+  meta_pre_record(constructive::deparse_call(ext_install_expr))
+
+  ext_cache$set(name, TRUE)
+  invisible()
+}
+
 meta_macro_register <- function(name) {
   macro <- duckplyr_macros[name]
   if (is.na(macro)) {
@@ -110,6 +129,12 @@ meta_macro_register <- function(name) {
     return(invisible())
   }
 
+  # Register functions from the rfuns extension
+  # Can't use '^"r_' because of the way the macro is defined
+  if (grepl('"r_', macro)) {
+    meta_ext_register()
+  }
+
   macro_expr <- expr(invisible(
     DBI::dbExecute(con, !!paste0('CREATE MACRO "', names(macro), '"', macro))
   ))

Diff for: R/relational-duckdb.R

@@ -15,12 +15,12 @@ get_default_duckdb_connection <- function() {
 }
 
 duckplyr_macros <- c(
-  "<" = "(x, y) AS x < y",
-  "<=" = "(x, y) AS x <= y",
-  ">" = "(x, y) AS x > y",
-  ">=" = "(x, y) AS x >= y",
-  "==" = "(x, y) AS x = y",
-  "!=" = "(x, y) AS x <> y",
+  "<" = '(x, y) AS "r_base::<"(x, y)',
+  "<=" = '(x, y) AS "r_base::<="(x, y)',
+  ">" = '(x, y) AS "r_base::>"(x, y)',
+  ">=" = '(x, y) AS "r_base::>="(x, y)',
+  "==" = '(x, y) AS "r_base::=="(x, y)',
+  "!=" = '(x, y) AS "r_base::!="(x, y)',
   #
   "___divide" = "(x, y) AS CASE WHEN y = 0 THEN CASE WHEN x = 0 THEN CAST('NaN' AS double) WHEN x > 0 THEN CAST('+Infinity' AS double) ELSE CAST('-Infinity' AS double) END ELSE CAST(x AS double) / y END",
   #
@@ -56,11 +56,14 @@ duckplyr_macros <- c(
 )
 
 create_default_duckdb_connection <- function() {
-  con <- DBI::dbConnect(duckdb::duckdb())
+  drv <- duckdb::duckdb()
+  con <- DBI::dbConnect(drv)
 
   DBI::dbExecute(con, "set memory_limit='1GB'")
   DBI::dbExecute(con, paste0("pragma temp_directory='", tempdir(), "'"))
 
+  duckdb$rapi_load_rfuns(drv@database_ref)
+
   for (i in seq_along(duckplyr_macros)) {
     sql <- paste0('CREATE MACRO "', names(duckplyr_macros)[[i]], '"', duckplyr_macros[[i]])
     DBI::dbExecute(con, sql)