add a native sqlpage implementation of json_array, which allows passing json arrays to nested sqlpage functions

Author: lovasoa

src/webserver/database/sql.rs

@@ -609,7 +609,10 @@ fn expr_to_stmt_param(arg: &mut Expr) -> Option<StmtParam> {
             let right = expr_to_stmt_param(right)?;
             Some(StmtParam::Concat(vec![left, right]))
         }
+        // SQLPage can evaluate some functions natively without sending them to the database:
         // CONCAT('str1', 'str2', ...)
+        // json_object('key1', 'value1', 'key2', 'value2', ...)
+        // json_array('value1', 'value2', ...)
         Expr::Function(Function {
             name: ObjectName(func_name_parts),
             args:
@@ -628,13 +631,25 @@ fn expr_to_stmt_param(arg: &mut Expr) -> Option<StmtParam> {
                 }
                 Some(StmtParam::Concat(concat_args))
             } else if func_name.eq_ignore_ascii_case("json_object")
+                || func_name.eq_ignore_ascii_case("jsonb_object")
                 || func_name.eq_ignore_ascii_case("json_build_object")
+                || func_name.eq_ignore_ascii_case("jsonb_build_object")
             {
                 let mut json_obj_args = Vec::with_capacity(args.len());
                 for arg in args {
                     json_obj_args.push(function_arg_to_stmt_param(arg)?);
                 }
                 Some(StmtParam::JsonObject(json_obj_args))
+            } else if func_name.eq_ignore_ascii_case("json_array")
+                || func_name.eq_ignore_ascii_case("jsonb_array")
+                || func_name.eq_ignore_ascii_case("json_build_array")
+                || func_name.eq_ignore_ascii_case("jsonb_build_array")
+            {
+                let mut json_obj_args = Vec::with_capacity(args.len());
+                for arg in args {
+                    json_obj_args.push(function_arg_to_stmt_param(arg)?);
+                }
+                Some(StmtParam::JsonArray(json_obj_args))
             } else {
                 log::warn!("SQLPage cannot emulate the following function: {func_name}");
                 None

src/webserver/database/syntax_tree.rs

@@ -39,6 +39,7 @@ pub(crate) enum StmtParam {
     Null,
     Concat(Vec<StmtParam>),
     JsonObject(Vec<StmtParam>),
+    JsonArray(Vec<StmtParam>),
     FunctionCall(SqlPageFunctionCall),
 }
 
@@ -64,6 +65,13 @@ impl std::fmt::Display for StmtParam {
                 }
                 write!(f, ")")
             }
+            StmtParam::JsonArray(items) => {
+                write!(f, "JSON_ARRAY(")?;
+                for item in items {
+                    write!(f, "{item}, ")?;
+                }
+                write!(f, ")")
+            }
             StmtParam::FunctionCall(call) => write!(f, "{call}"),
             StmtParam::Error(x) => {
                 if let Some((i, _)) = x.char_indices().nth(21) {
@@ -154,6 +162,7 @@ pub(super) async fn extract_req_param<'a, 'b>(
         StmtParam::Null => None,
         StmtParam::Concat(args) => concat_params(&args[..], request, db_connection).await?,
         StmtParam::JsonObject(args) => json_object_params(&args[..], request, db_connection).await?,
+        StmtParam::JsonArray(args) => json_array_params(&args[..], request, db_connection).await?,
         StmtParam::FunctionCall(func) => func.evaluate(request, db_connection).await.with_context(|| {
             format!(
                 "Error in function call {func}.\nExpected {:#}",
@@ -200,3 +209,20 @@ async fn json_object_params<'a, 'b>(
     map_ser.end()?;
     Ok(Some(Cow::Owned(String::from_utf8(result)?)))
 }
+
+async fn json_array_params<'a, 'b>(
+    args: &[StmtParam],
+    request: &'a RequestInfo,
+    db_connection: &'b mut DbConn,
+) -> anyhow::Result<Option<Cow<'a, str>>> {
+    use serde::{ser::SerializeSeq, Serializer};
+    let mut result = Vec::new();
+    let mut ser = serde_json::Serializer::new(&mut result);
+    let mut seq_ser = ser.serialize_seq(Some(args.len()))?;
+    for element in args {
+        let element = Box::pin(extract_req_param(element, request, db_connection)).await?;
+        seq_ser.serialize_element(&element)?;
+    }
+    seq_ser.end()?;
+    Ok(Some(Cow::Owned(String::from_utf8(result)?)))
+}

tests/sql_test_files/it_works_native_json_array_impl.sql

@@ -0,0 +1,15 @@
+set res = sqlpage.fetch(json_object(
+    'method', 'POST',
+    'url', 'http://localhost:62802/post',
+    'headers', json_object('x-custom', '1'),
+    'body', json_array('hello', 'world')
+));
+set expected = 'POST /post|accept-encoding: br, gzip, deflate, zstd|content-length: 18|content-type: application/json|host: localhost:62802|user-agent: sqlpage|x-custom: 1|["hello", "world"]';
+select 'text' as component,
+    case $res
+        when $expected then 'It works !'
+        else 'It failed ! Expected: 
+' || $expected || '
+Got: 
+' || $res
+    end as contents;