Ensure all window translation get matching aggregate translation with…

… clear errro

Fixes #129

NEWS.md

@@ -1,5 +1,8 @@
 # dbplyr (development version)
 
+* Functions that are only available in a windowed (`mutate()`) query now
+  throw an error when called in a aggregate (`summarise()`) query (#129)
+
 * `na_if()` is translated to `NULLIF()` for all databases (#211).
 
 * SQL translation (via `partial_eval()`) now correctly interprets the 

R/translate-sql-base.r

@@ -217,6 +217,11 @@ base_agg <- sql_translator(
   sum        = sql_aggregate("sum"),
   min        = sql_aggregate("min"),
   max        = sql_aggregate("max"),
+
+  # first = sql_prefix("FIRST_VALUE", 1),
+  # last = sql_prefix("LAST_VALUE", 1),
+  # nth = sql_prefix("NTH_VALUE", 2),
+
   n_distinct = function(...) {
     vars <- sql_vector(list(...), parens = FALSE, collapse = ", ")
     build_sql("COUNT(DISTINCT ", vars, ")")

R/translate-sql-helpers.r

@@ -74,6 +74,11 @@ sql_variant <- function(scalar = sql_translator(),
     ))
   }
 
+  # An ensure that every window function is flagged in aggregate context
+  missing <- setdiff(ls(window), ls(aggregate))
+  missing_funs <- map(missing, sql_aggregate_win)
+  env_bind(aggregate, !!!set_names(missing_funs, missing))
+
   structure(
     list(scalar = scalar, aggregate = aggregate, window = window),
     class = "sql_variant"
@@ -187,6 +192,17 @@ sql_aggregate_2 <- function(f) {
   }
 }
 
+sql_aggregate_win <- function(f) {
+  force(f)
+
+  function(...) {
+    stop(
+      "`", f, "()` is only available in a windowed (`mutate()`) context",
+      call. = FALSE
+    )
+  }
+}
+
 
 check_na_rm <- function(f, na.rm) {
   if (identical(na.rm, TRUE)) {

tests/testthat/test-translate-sql-helpers.r

@@ -18,6 +18,15 @@ test_that("missing window functions create a warning", {
   )
 })
 
+test_that("missing aggregate functions filled in", {
+  sim_scalar <- sql_translator()
+  sim_agg <- sql_translator()
+  sim_win <- sql_translator(mean = function() {})
+
+  trans <- sql_variant(sim_scalar, sim_agg, sim_win)
+  expect_error(trans$aggregate$mean(), "only available in a window")
+})
+
 test_that("output of print method for sql_variant is correct", {
   sim_trans <- sql_translator(`+` = sql_infix("+"))
   expect_known_output(