feat(WidgetDriver): Pass matrix api errors to the widget

toger5 · toger5 · commit 1b1a0fb93136 · 2024-11-13T00:00:15.000+01:00
diff --git a/crates/matrix-sdk/src/widget/machine/driver_req.rs b/crates/matrix-sdk/src/widget/machine/driver_req.rs
@@ -76,7 +76,7 @@ where
 
     pub(crate) fn then(
         self,
-        response_handler: impl FnOnce(Result<T, String>, &mut WidgetMachine) -> Vec<Action>
+        response_handler: impl FnOnce(Result<T, crate::Error>, &mut WidgetMachine) -> Vec<Action>
             + Send
             + 'static,
     ) {
diff --git a/crates/matrix-sdk/src/widget/machine/from_widget.rs b/crates/matrix-sdk/src/widget/machine/from_widget.rs
@@ -12,20 +12,22 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-use std::fmt;
+use std::collections::HashMap;
 
 use ruma::{
-    api::client::delayed_events::{
-        delayed_message_event, delayed_state_event, update_delayed_event,
+    api::client::{
+        delayed_events::{delayed_message_event, delayed_state_event, update_delayed_event},
+        error::ErrorBody,
     },
     events::{AnyTimelineEvent, MessageLikeEventType, StateEventType},
     serde::Raw,
     OwnedEventId, OwnedRoomId,
 };
 use serde::{Deserialize, Serialize};
+use serde_json::{json, Value};
 
 use super::{SendEventRequest, UpdateDelayedEventRequest};
-use crate::widget::StateKeySelector;
+use crate::{widget::StateKeySelector, Error, HttpError};
 
 #[derive(Deserialize, Debug)]
 #[serde(tag = "action", rename_all = "snake_case", content = "data")]
@@ -46,17 +48,63 @@ pub(super) struct FromWidgetErrorResponse {
     error: FromWidgetError,
 }
 
-impl FromWidgetErrorResponse {
-    pub(super) fn new(e: impl fmt::Display) -> Self {
-        Self { error: FromWidgetError { message: e.to_string() } }
+impl From<&HttpError> for FromWidgetErrorResponse {
+    fn from(error: &HttpError) -> Self {
+        Self {
+            error: FromWidgetError {
+                message: error.to_string(),
+                matrix_api_error: error.as_client_api_error().and_then(
+                    |api_error| match &api_error.body {
+                        ErrorBody::Standard { kind, message } => Some(FromWidgetMatrixErrorBody {
+                            http_status: api_error.status_code.as_u16().into(),
+                            http_headers: HashMap::new(),
+                            url: "".to_owned(),
+                            response: json!({"errcode": kind.to_string(), "error": message }),
+                        }),
+                        _ => None,
+                    },
+                ),
+            },
+        }
+    }
+}
+
+impl From<&Error> for FromWidgetErrorResponse {
+    fn from(error: &Error) -> Self {
+        match &error {
+            Error::Http(e) => e.into(),
+            Error::UnknownError(e) => e.to_string().into(),
+            _ => error.to_string().into(),
+        }
+    }
+}
+
+impl From<String> for FromWidgetErrorResponse {
+    fn from(error: String) -> Self {
+        Self { error: FromWidgetError { message: error, matrix_api_error: None } }
+    }
+}
+
+impl From<&str> for FromWidgetErrorResponse {
+    fn from(error: &str) -> Self {
+        error.to_string().into()
     }
 }
 
 #[derive(Serialize)]
 struct FromWidgetError {
     message: String,
+    matrix_api_error: Option<FromWidgetMatrixErrorBody>,
 }
 
+#[derive(Serialize)]
+struct FromWidgetMatrixErrorBody {
+    http_status: u32,
+    // TODO: figure out the which type to use here.
+    http_headers: HashMap<String, String>,
+    url: String,
+    response: Value,
+}
 #[derive(Serialize)]
 pub(super) struct SupportedApiVersionsResponse {
     supported_versions: Vec<ApiVersion>,
diff --git a/crates/matrix-sdk/src/widget/machine/incoming.rs b/crates/matrix-sdk/src/widget/machine/incoming.rs
@@ -38,7 +38,7 @@ pub(crate) enum IncomingMessage {
         request_id: Uuid,
 
         /// The result of the request: response data or error message.
-        response: Result<MatrixDriverResponse, String>,
+        response: Result<MatrixDriverResponse, crate::Error>,
     },
 
     /// The `MatrixDriver` notified the `WidgetMachine` of a new matrix event.
diff --git a/crates/matrix-sdk/src/widget/machine/mod.rs b/crates/matrix-sdk/src/widget/machine/mod.rs
@@ -16,7 +16,7 @@
 
 #![warn(unreachable_pub)]
 
-use std::{fmt, iter, time::Duration};
+use std::{iter, time::Duration};
 
 use driver_req::UpdateDelayedEventRequest;
 use from_widget::UpdateDelayedEventResponse;
@@ -50,11 +50,11 @@ use self::{
 #[cfg(doc)]
 use super::WidgetDriver;
 use super::{
-    capabilities,
+    capabilities::{self, SEND_DELAYED_EVENT, UPDATE_DELAYED_EVENT},
     filter::{MatrixEventContent, MatrixEventFilterInput},
     Capabilities, StateKeySelector,
 };
-use crate::widget::EventFilter;
+use crate::{widget::EventFilter, Error};
 
 mod driver_req;
 mod from_widget;
@@ -203,7 +203,9 @@ impl WidgetMachine {
     ) -> Vec<Action> {
         let request = match raw_request.deserialize() {
             Ok(r) => r,
-            Err(e) => return vec![self.send_from_widget_error_response(raw_request, e)],
+            Err(e) => {
+                return vec![self.send_from_widget_error_response(raw_request, Error::SerdeJson(e))]
+            }
         };
 
         match request {
@@ -252,28 +254,29 @@ impl WidgetMachine {
                 let CapabilitiesState::Negotiated(capabilities) = &self.capabilities else {
                     let text =
                         "Received send update delayed event request before capabilities were negotiated";
-                    return vec![self.send_from_widget_error_response(raw_request, text)];
+                    return vec![self.send_from_widget_error_response(raw_request, text.into())];
                 };
                 if !capabilities.update_delayed_event {
                     return vec![self.send_from_widget_error_response(
                         raw_request,
-                        format!(
-                            "Not allowed: missing the {} capability.",
-                            capabilities::UPDATE_DELAYED_EVENT
-                        ),
+                        format!("Not allowed: missing the {UPDATE_DELAYED_EVENT} capability.")
+                            .into(),
                     )];
                 }
                 let (request, request_action) =
                     self.send_matrix_driver_request(UpdateDelayedEventRequest {
                         action: req.action,
                         delay_id: req.delay_id,
                     });
-                request.then(|res, machine| {
+                request.then(|result, machine| {
                     vec![machine.send_from_widget_result_response(
                         raw_request,
                         // This is mapped to another type because the update_delay_event::Response
                         // does not impl Serialize
-                        res.map(Into::<UpdateDelayedEventResponse>::into),
+                        match result {
+                            Ok(response) => Ok(Into::<UpdateDelayedEventResponse>::into(response)),
+                            Err(e) => Err((&e).into()),
+                        },
                     )]
                 });
                 request_action.map(|a| vec![a]).unwrap_or_default()
@@ -288,7 +291,7 @@ impl WidgetMachine {
     ) -> Option<Action> {
         let CapabilitiesState::Negotiated(capabilities) = &self.capabilities else {
             let text = "Received read event request before capabilities were negotiated";
-            return Some(self.send_from_widget_error_response(raw_request, text));
+            return Some(self.send_from_widget_error_response(raw_request, text.into()));
         };
 
         match request {
@@ -297,7 +300,7 @@ impl WidgetMachine {
                 if !capabilities.read.iter().any(filter_fn) {
                     return Some(self.send_from_widget_error_response(
                         raw_request,
-                        "Not allowed to read message like event",
+                        "Not allowed to read message like event".into(),
                     ));
                 }
 
@@ -306,16 +309,32 @@ impl WidgetMachine {
                 let request = ReadMessageLikeEventRequest { event_type, limit };
                 let (request, action) = self.send_matrix_driver_request(request);
                 request.then(|result, machine| {
-                    let response = result.and_then(|mut events| {
-                        let CapabilitiesState::Negotiated(capabilities) = &machine.capabilities
-                        else {
-                            let err = "Received read event request before capabilities negotiation";
-                            return Err(err.into());
-                        };
+                    let response = match (result, &machine.capabilities) {
+                        (Ok(mut events), CapabilitiesState::Negotiated(capabilities)) => {
+                            events.retain(|e| capabilities.raw_event_matches_read_filter(e));
+                            Ok(ReadEventResponse { events })
+                        }
+                        (Ok(_), _) => {
+                            Err("Received read event request before capabilities negotiation"
+                                .into())
+                        }
+                        (Err(e), _) => Err((&e).into()),
+                    };
 
-                        events.retain(|e| capabilities.raw_event_matches_read_filter(e));
-                        Ok(ReadEventResponse { events })
-                    });
+                    // let response = result
+                    //     .and_then(|mut events| {
+                    //         let CapabilitiesState::Negotiated(capabilities) = &machine.capabilities
+                    //         else {
+                    //             return Err(Error::UnknownError(
+                    //                 "Received read event request before capabilities negotiation"
+                    //                     .into(),
+                    //             ));
+                    //         };
+
+                    //         events.retain(|e| capabilities.raw_event_matches_read_filter(e));
+                    //         Ok(ReadEventResponse { events })
+                    //     })
+                    //     .map_err(|e| (&e).into());
                     vec![machine.send_from_widget_result_response(raw_request, response)]
                 });
                 action
@@ -343,14 +362,16 @@ impl WidgetMachine {
                     let request = ReadStateEventRequest { event_type, state_key };
                     let (request, action) = self.send_matrix_driver_request(request);
                     request.then(|result, machine| {
-                        let response = result.map(|events| ReadEventResponse { events });
+                        let response = result
+                            .map(|events| ReadEventResponse { events })
+                            .map_err(|e| (&e).into());
                         vec![machine.send_from_widget_result_response(raw_request, response)]
                     });
                     action
                 } else {
                     Some(self.send_from_widget_error_response(
                         raw_request,
-                        "Not allowed to read state event",
+                        "Not allowed to read state event".into(),
                     ))
                 }
             }
@@ -380,15 +401,15 @@ impl WidgetMachine {
         if !capabilities.send_delayed_event && request.delay.is_some() {
             return Some(self.send_from_widget_error_response(
                 raw_request,
-                format!(
-                    "Not allowed: missing the {} capability.",
-                    capabilities::SEND_DELAYED_EVENT
-                ),
+                format!("Not allowed: missing the {SEND_DELAYED_EVENT} capability.").into(),
             ));
         }
         if !capabilities.send.iter().any(|filter| filter.matches(&filter_in)) {
             return Some(
-                self.send_from_widget_error_response(raw_request, "Not allowed to send event"),
+                self.send_from_widget_error_response(
+                    raw_request,
+                    "Not allowed to send event".into(),
+                ),
             );
         }
 
@@ -397,7 +418,8 @@ impl WidgetMachine {
             if let Ok(r) = result.as_mut() {
                 r.set_room_id(machine.room_id.clone());
             }
-            vec![machine.send_from_widget_result_response(raw_request, result)]
+            vec![machine
+                .send_from_widget_result_response(raw_request, result.map_err(|e| (&e).into()))]
         });
         action
     }
@@ -439,7 +461,7 @@ impl WidgetMachine {
     fn process_matrix_driver_response(
         &mut self,
         request_id: Uuid,
-        response: Result<MatrixDriverResponse, String>,
+        response: Result<MatrixDriverResponse, Error>,
     ) -> Vec<Action> {
         match self.pending_matrix_driver_requests.extract(&request_id) {
             Ok(request) => request
@@ -472,15 +494,15 @@ impl WidgetMachine {
     fn send_from_widget_error_response(
         &self,
         raw_request: Raw<FromWidgetRequest>,
-        error: impl fmt::Display,
+        error: FromWidgetErrorResponse,
     ) -> Action {
-        self.send_from_widget_response(raw_request, FromWidgetErrorResponse::new(error))
+        self.send_from_widget_response(raw_request, error)
     }
 
     fn send_from_widget_result_response(
         &self,
         raw_request: Raw<FromWidgetRequest>,
-        result: Result<impl Serialize, impl fmt::Display>,
+        result: Result<impl Serialize, FromWidgetErrorResponse>,
     ) -> Action {
         match result {
             Ok(res) => self.send_from_widget_response(raw_request, res),
@@ -586,7 +608,7 @@ impl ToWidgetRequestMeta {
 }
 
 type MatrixDriverResponseFn =
-    Box<dyn FnOnce(Result<MatrixDriverResponse, String>, &mut WidgetMachine) -> Vec<Action> + Send>;
+    Box<dyn FnOnce(Result<MatrixDriverResponse, Error>, &mut WidgetMachine) -> Vec<Action> + Send>;
 
 pub(crate) struct MatrixDriverRequestMeta {
     response_fn: Option<MatrixDriverResponseFn>,
diff --git a/crates/matrix-sdk/src/widget/machine/tests/capabilities.rs b/crates/matrix-sdk/src/widget/machine/tests/capabilities.rs
@@ -114,7 +114,7 @@ fn capabilities_failure_results_into_empty_capabilities() {
 
         machine.process(IncomingMessage::MatrixDriverResponse {
             request_id,
-            response: Err("OHMG!".into()),
+            response: Err(crate::Error::UnknownError("OHMG!".into())),
         })
     };
 
diff --git a/crates/matrix-sdk/src/widget/machine/tests/openid.rs b/crates/matrix-sdk/src/widget/machine/tests/openid.rs
@@ -162,7 +162,7 @@ fn openid_fail_results_in_response_blocked() {
 
         machine.process(IncomingMessage::MatrixDriverResponse {
             request_id,
-            response: Err("Unlucky one".into()),
+            response: Err(crate::Error::UnknownError("Unlucky one".into())),
         })
     };
 
diff --git a/crates/matrix-sdk/src/widget/matrix.rs b/crates/matrix-sdk/src/widget/matrix.rs
diff --git a/crates/matrix-sdk/src/widget/mod.rs b/crates/matrix-sdk/src/widget/mod.rs
diff --git a/crates/matrix-sdk/tests/integration/widget.rs b/crates/matrix-sdk/tests/integration/widget.rs
