feat(filter): add gzip in filter chain, update ffi api to add response headers when creating body filter

Author: joelwurtz

Cargo.toml

@@ -25,6 +25,7 @@ cfg-if = "1.0.0"
 chrono = { version = "0.4.22", features = ["serde"] }
 cidr = { version = "0.2.1", features = ["serde"] }
 console_log = "0.2.0"
+flate2 = "1.0.24"
 heck = "0.4.0"
 http = "0.2.8"
 lazy_static = "1.4.0"

src/action/ffi.rs

@@ -93,14 +93,16 @@ pub unsafe extern "C" fn redirectionio_action_header_filter_filter(
 pub unsafe extern "C" fn redirectionio_action_body_filter_create(
     _action: *mut Action,
     response_status_code: u16,
+    response_header_map: *const HeaderMap,
 ) -> *const FilterBodyAction {
     if _action.is_null() {
         return null() as *const FilterBodyAction;
     }
 
     let action = &mut *_action;
+    let headers = header_map_to_http_headers(response_header_map);
 
-    match action.create_filter_body(response_status_code) {
+    match action.create_filter_body(response_status_code, &headers) {
         None => null(),
         Some(filter_body) => Box::into_raw(Box::new(filter_body)),
     }
@@ -115,14 +117,9 @@ pub unsafe extern "C" fn redirectionio_action_body_filter_filter(_filter: *mut F
     let filter = &mut *_filter;
     let bytes = buffer.into_vec();
 
-    let body = match String::from_utf8(bytes) {
-        Err(error) => return Buffer::from_vec(error.into_bytes()),
-        Ok(body) => body,
-    };
-
-    let new_body = filter.filter(body);
+    let new_body = filter.filter(bytes);
 
-    Buffer::from_string(new_body)
+    Buffer::from_vec(new_body)
 }
 
 #[no_mangle]
@@ -134,7 +131,7 @@ pub unsafe extern "C" fn redirectionio_action_body_filter_close(_filter: *mut Fi
     let mut filter = Box::from_raw(_filter);
     let end_body = filter.end();
 
-    Buffer::from_string(end_body)
+    Buffer::from_vec(end_body)
 }
 
 #[no_mangle]

src/action/mod.rs

@@ -396,7 +396,7 @@ impl Action {
         new_headers
     }
 
-    pub fn create_filter_body(&mut self, response_status_code: u16) -> Option<FilterBodyAction> {
+    pub fn create_filter_body(&mut self, response_status_code: u16, headers: &[Header]) -> Option<FilterBodyAction> {
         let mut filters = Vec::new();
         let rule_applied = self.rule_traces.is_some();
 
@@ -412,7 +412,7 @@ impl Action {
             filters.push(filter.filter);
         }
 
-        let body_filter = FilterBodyAction::new(filters);
+        let body_filter = FilterBodyAction::new(filters, headers);
 
         if body_filter.is_empty() {
             None