add support of response status_code in rhai

apollo-router/src/plugins/rhai/engine.rs

@@ -16,6 +16,7 @@ use http::uri::Parts;
 use http::uri::PathAndQuery;
 use http::HeaderMap;
 use http::Method;
+use http::StatusCode;
 use http::Uri;
 use rhai::module_resolvers::FileModuleResolver;
 use rhai::plugin::*;
@@ -218,6 +219,34 @@ mod router_method {
     }
 }
 
+#[export_module]
+mod status_code {
+    pub(crate) type StatusCode = http::StatusCode;
+
+    #[rhai_fn(name = "to_string", pure)]
+    pub(crate) fn status_code_to_string(status_code: &mut StatusCode) -> String {
+        status_code.as_str().to_string()
+    }
+
+    #[rhai_fn(name = "to_number", pure)]
+    pub(crate) fn status_code_to_number(status_code: &mut StatusCode) -> i64 {
+        status_code.as_u16() as i64
+    }
+
+    #[rhai_fn(name = "==", pure)]
+    pub(crate) fn status_code_equal_comparator(status_code: &mut StatusCode, other: i64) -> bool {
+        status_code.as_u16() as i64 == other
+    }
+
+    #[rhai_fn(name = "!=", pure)]
+    pub(crate) fn status_code_not_equal_comparator(
+        status_code: &mut StatusCode,
+        other: i64,
+    ) -> bool {
+        status_code.as_u16() as i64 != other
+    }
+}
+
 #[export_module]
 mod router_header_map {
     pub(crate) type HeaderMap = http::HeaderMap;
@@ -1425,6 +1454,13 @@ macro_rules! register_rhai_router_interface {
                 }
             );
 
+            $engine.register_get(
+                "status_code",
+                |obj: &mut SharedMut<$base::Response>| -> Result<StatusCode, Box<EvalAltResult>> {
+                    Ok(obj.with_mut(|response| response.response.status()))
+                }
+            );
+
             $engine.register_set(
                 "uri",
                 |obj: &mut SharedMut<$base::Request>, uri: Uri| {
@@ -1460,6 +1496,13 @@ macro_rules! register_rhai_interface {
                 }
             );
 
+            $engine.register_get(
+                "status_code",
+                |obj: &mut SharedMut<$base::Response>| -> Result<StatusCode, Box<EvalAltResult>> {
+                    Ok(obj.with_mut(|response| response.response.status()))
+                }
+            );
+
             $engine.register_set(
                 "context",
                 |obj: &mut SharedMut<$base::Request>, context: Context| {
@@ -1632,6 +1675,7 @@ impl Rhai {
         let mut module = exported_module!(router_plugin);
         combine_with_exported_module!(&mut module, "header", router_header_map);
         combine_with_exported_module!(&mut module, "method", router_method);
+        combine_with_exported_module!(&mut module, "status_code", status_code);
         combine_with_exported_module!(&mut module, "context", router_context);
 
         let base64_module = exported_module!(router_base64);

apollo-router/src/plugins/rhai/tests.rs

@@ -486,7 +486,11 @@ async fn it_can_process_subgraph_response() {
 
     // We must wrap our canned response in Arc<Mutex<Option<>>> to keep the rhai runtime
     // happy
-    let response = Arc::new(Mutex::new(Some(subgraph::Response::fake_builder().build())));
+    let response = Arc::new(Mutex::new(Some(
+        subgraph::Response::fake_builder()
+            .status_code(StatusCode::OK)
+            .build(),
+    )));
 
     // Call our rhai test function. If it return an error, the test failed.
     let result: Result<(), Box<rhai::EvalAltResult>> = block.engine.call_fn(

apollo-router/tests/fixtures/request_response_test.rhai

@@ -123,6 +123,12 @@ fn process_execution_response(response) {
 
 fn process_subgraph_response(response) {
     process_common_response(response);
+    if type_of(response.status_code) != "http::status::StatusCode" {
+        throw(`status_code: expected: "i32", actual: ${type_of(response.status_code)}`);
+    }
+    if response.status_code.to_number() != 200 {
+        throw(`status_code: expected: 200, actual: ${response.status_code}`);
+    }
 }
 
 fn process_subgraph_response_om_forbidden(response) {

docs/source/customizations/rhai-api.mdx

@@ -540,6 +540,7 @@ The `response` object includes the following fields:
 ```
 response.context
 response.id
+response.status_code
 response.headers
 response.body.label
 response.body.data
@@ -618,3 +619,17 @@ let error_to_add = #{
 response.body.errors += error_to_add;
 print(`${response.body.errors}`); // logs the response errors
 ```
+
+### `response.status_code.to_string()`
+
+Convert response status code as a string
+
+### `response.status_code.to_number()`
+
+Convert response status code as a number in order to compare it
+
+```rhai
+if response.status_code.to_number() == 200 {
+    print(`ok`);
+}
+```