fix: add meta to jsonb_array_elements

Author: tobyhede

src/encrypted/functions.sql

@@ -18,7 +18,6 @@ AS $$
 $$ LANGUAGE plpgsql;
 
 
-
 CREATE FUNCTION eql_v2.ciphertext(val eql_v2_encrypted)
   RETURNS text
   IMMUTABLE STRICT PARALLEL SAFE
@@ -29,7 +28,6 @@ AS $$
 $$ LANGUAGE plpgsql;
 
 
-
 CREATE FUNCTION eql_v2._first_grouped_value(jsonb, jsonb)
 RETURNS jsonb AS $$
   SELECT COALESCE($1, $2);

src/jsonb/functions.sql

@@ -246,17 +246,22 @@ CREATE FUNCTION eql_v2.jsonb_array_elements(val jsonb)
 AS $$
   DECLARE
     sv eql_v2_encrypted[];
-    found eql_v2_encrypted[];
+    meta jsonb;
+    item jsonb;
   BEGIN
 
     IF NOT eql_v2.is_ste_vec_array(val) THEN
       RAISE 'cannot extract elements from non-array';
     END IF;
 
+    -- Column identifier and version
+    meta := eql_v2.meta_data(val);
+
     sv := eql_v2.ste_vec(val);
 
     FOR idx IN 1..array_length(sv, 1) LOOP
-      RETURN NEXT sv[idx];
+      item = sv[idx];
+      RETURN NEXT (meta || item)::eql_v2_encrypted;
     END LOOP;
 
     RETURN;

src/jsonb/functions_test.sql

@@ -208,6 +208,21 @@ $$ LANGUAGE plpgsql;
 
 
 
+DO $$
+  DECLARE
+    result jsonb;
+  BEGIN
+    PERFORM seed_encrypted_json();
+    PERFORM seed_encrypted(get_array_ste_vec()::eql_v2_encrypted);
+
+    SELECT eql_v2.jsonb_array_elements(eql_v2.jsonb_path_query(e, 'f510853730e1c3dbd31b86963f029dd5')::jsonb) FROM encrypted INTO result;
+
+    ASSERT result ? 'i';
+    ASSERT result ? 'v';
+  END;
+$$ LANGUAGE plpgsql;
+
+
 DO $$
   BEGIN