feat!: rewrite the prql_compile() function (#317)

Author: eitsupi

DESCRIPTION

@@ -25,7 +25,8 @@ Suggests:
     dplyr,
     testthat (>= 3.2.0),
     patrick,
-    withr
+    withr,
+    cli
 License: MIT + file LICENSE
 Language: en-US
 Encoding: UTF-8
@@ -47,4 +48,4 @@ Config/Needs/dev:
     rhub
 Config/Needs/website:
     pkgdown
-Config/prqlr/LibVersion: 0.13.0
+Config/prqlr/LibVersion: 0.13.1

NEWS.md

@@ -1,5 +1,13 @@
 # prqlr (development version)
 
+## Breaking changes
+
+- For the `prql_compile()` funtion, arguments `format` and `singnature_comment` should be named arguments. (#317)
+
+## New features
+
+- New experimental argument `display` of `prql_compile()` function to colorize the error message from prqlc. (#317)
+
 # prqlr 0.9.0
 
 ## Breaking changes

R/000-wrappers.R

@@ -30,8 +30,8 @@ NULL
 #' @param signature_comment a logical flag. Whether to add a signature comment to the output SQL query.
 #' @return a list contains a SQL string or an error message.
 #' @noRd
-`compile` <- function(`prql_query`, `target`, `format`, `signature_comment`) {
-  .Call(savvy_compile__impl, `prql_query`, `target`, `format`, `signature_comment`)
+`compile` <- function(`prql_query`, `target`, `format`, `signature_comment`, `display`) {
+  .Call(savvy_compile__impl, `prql_query`, `target`, `format`, `signature_comment`, `display`)
 }
 
 #' @noRd

R/compile.R

@@ -1,13 +1,15 @@
 #' @title Compile a PRQL query into a SQL query
-#' @param prql_query a PRQL query string.
-#' @param target a compile target name to use. If not specified (`NULL`),
-#' the target contained in the query will be used.
-#' All available target names can be listed with the [prql_get_targets] function.
-#' @param format a logical flag (default: `TRUE`). Whether to format the SQL query.
+#' @param prql_query A character of PRQL query.
+#' @param target A character of the target name to use or `NULL`.
+#' If `NULL`, the target contained in the query will be used.
+#' All available target names can be listed with the [prql_get_targets()] function.
+#' @param ... Ignored.
+#' @param format A logical flag (default: `TRUE`). Whether to format the SQL query.
 #' @param signature_comment a logical flag. (default: `TRUE`).
 #' Whether to add a signature comment to the output SQL query.
-#' @return a SQL query string
-#' @seealso [prql_get_targets]
+#' @param display A character, one of `"plain"` (default) or `"ansi_color"`.
+#' If `"ansi_color"`, error will be displayed with ANSI color.
+#' @return A character of the compiled SQL query.
 #' @examples
 #' "from mtcars | filter cyl > 6 | select {cyl, mpg}" |>
 #'   prql_compile()
@@ -38,9 +40,17 @@
 prql_compile <- function(
     prql_query,
     target = getOption("prqlr.target", default = NULL),
+    ...,
     format = getOption("prqlr.format", default = TRUE),
-    signature_comment = getOption("prqlr.signature_comment", default = TRUE)) {
-  compile(prql_query, target %||% "sql.any", format, signature_comment)
+    signature_comment = getOption("prqlr.signature_comment", default = TRUE),
+    display = getOption("prqlr.display", default = "plain")) {
+  compile(
+    prql_query,
+    target = target %||% "sql.any",
+    format = format,
+    signature_comment = signature_comment,
+    display = display
+  )
 }
 
 #' @title prqlc's version

R/knitr-engine.R

@@ -1,3 +1,4 @@
+# TODO: support the `error=TRUE` option and support `ansi_color`ed error message with the `cli` package.
 #' @title PRQL knitr engine
 #' @description
 #' If options$connection is NULL, the output is SQL query.
@@ -27,7 +28,8 @@ eng_prql <- function(options) {
     prql_compile(
       target = options$engine.opts[["target"]] %||% getOption("prqlr.target"),
       format = TRUE,
-      signature_comment = options$engine.opts[["signature_comment"]] %||% getOption("prqlr.signature_comment", TRUE)
+      signature_comment = options$engine.opts[["signature_comment"]] %||% getOption("prqlr.signature_comment", TRUE),
+      display = "plain"
     )
 
   # Prints a SQL code block if there is no connection

man/prql_compile.Rd

@@ -34,8 +34,8 @@ SEXP handle_result(SEXP res_) {
     return (SEXP)res;
 }
 
-SEXP savvy_compile__impl(SEXP c_arg__prql_query, SEXP c_arg__target, SEXP c_arg__format, SEXP c_arg__signature_comment) {
-    SEXP res = savvy_compile__ffi(c_arg__prql_query, c_arg__target, c_arg__format, c_arg__signature_comment);
+SEXP savvy_compile__impl(SEXP c_arg__prql_query, SEXP c_arg__target, SEXP c_arg__format, SEXP c_arg__signature_comment, SEXP c_arg__display) {
+    SEXP res = savvy_compile__ffi(c_arg__prql_query, c_arg__target, c_arg__format, c_arg__signature_comment, c_arg__display);
     return handle_result(res);
 }
 
@@ -66,7 +66,7 @@ SEXP savvy_prql_get_targets__impl(void) {
 
 
 static const R_CallMethodDef CallEntries[] = {
-    {"savvy_compile__impl", (DL_FUNC) &savvy_compile__impl, 4},
+    {"savvy_compile__impl", (DL_FUNC) &savvy_compile__impl, 5},
     {"savvy_prql_to_pl__impl", (DL_FUNC) &savvy_prql_to_pl__impl, 1},
     {"savvy_pl_to_rq__impl", (DL_FUNC) &savvy_pl_to_rq__impl, 1},
     {"savvy_rq_to_sql__impl", (DL_FUNC) &savvy_rq_to_sql__impl, 1},

src/init.c

@@ -1,6 +1,6 @@
 [package]
 name = "prqlr"
-version = "0.13.0"
+version = "0.13.1"
 edition = "2021"
 rust-version = "1.69"
 publish = false
@@ -15,4 +15,3 @@ savvy = "0.6.8"
 # prqlc 0.13.0 is not compatible with Rust 1.69.0, so a slightly modified version is installed
 # See https://github.com/PRQL/prql/pull/4916
 prqlc = { git = "https://github.com/PRQL/prql", rev = "c7bd7a6fc73040394ffbbd85cea2ed6f986fd9dd", default-features = false }
-anstream = { version = "0", features = ["auto"] }

src/rust/Cargo.lock

@@ -1,4 +1,4 @@
-SEXP savvy_compile__ffi(SEXP c_arg__prql_query, SEXP c_arg__target, SEXP c_arg__format, SEXP c_arg__signature_comment);
+SEXP savvy_compile__ffi(SEXP c_arg__prql_query, SEXP c_arg__target, SEXP c_arg__format, SEXP c_arg__signature_comment, SEXP c_arg__display);
 SEXP savvy_prql_to_pl__ffi(SEXP c_arg__prql_query);
 SEXP savvy_pl_to_rq__ffi(SEXP c_arg__pl_json);
 SEXP savvy_rq_to_sql__ffi(SEXP c_arg__rq_json);