Parse additional data in WP.com comments API

Author: crazytonyli

wp_api/src/comments.rs

@@ -1,5 +1,5 @@
 use crate::{
-    JsonValue, UserAvatarSize, UserId, WpApiParamOrder, WpResponseString,
+    AnyJson, UserAvatarSize, UserId, WpApiParamOrder, WpResponseString,
     date::WpGmtDateTime,
     impl_as_query_value_for_new_type, impl_as_query_value_from_to_string,
     posts::PostId,
@@ -8,7 +8,7 @@ use crate::{
     },
 };
 use serde::{Deserialize, Serialize};
-use std::{collections::HashMap, num::ParseIntError, str::FromStr};
+use std::{collections::HashMap, num::ParseIntError, str::FromStr, sync::Arc};
 use strum_macros::IntoStaticStr;
 use wp_contextual::WpContextual;
 
@@ -530,7 +530,7 @@ pub struct SparseComment {
     pub author_avatar_urls: Option<HashMap<UserAvatarSize, WpResponseString>>,
     #[serde(flatten)]
     #[WpContext(edit, embed, view)]
-    pub extras: Option<JsonValue>,
+    pub additional_fields: Option<Arc<AnyJson>>,
     // meta field is omitted for now: https://github.com/Automattic/wordpress-rs/issues/422
 }
 

wp_api/src/lib.rs

@@ -1,5 +1,6 @@
 use plugins::*;
 use serde::{Deserialize, Serialize};
+use serde_json::Value;
 use std::collections::HashMap;
 use users::*;
 use wp_localization::{MessageBundle, WpMessages, WpSupportsLocalization};
@@ -118,6 +119,14 @@ pub enum JsonValue {
     Object(HashMap<String, JsonValue>),
 }
 
+/// Similar to `JsonValue`, but exported as a Uniffi object.
+#[derive(Debug, Clone, Eq, PartialEq, Hash, Serialize, Deserialize, uniffi::Object)]
+#[uniffi::export(Eq, Hash)]
+pub struct AnyJson {
+    #[serde(flatten)]
+    pub raw: Value,
+}
+
 uniffi::custom_newtype!(WpResponseString, Option<String>);
 #[derive(Debug, Serialize, Deserialize)]
 #[serde(try_from = "BoolOrString")]
@@ -235,4 +244,30 @@ mod tests {
     fn test_orderby_string_conversion(#[case] orderby: WpApiParamOrder) {
         assert_eq!(orderby, orderby.to_string().parse().unwrap());
     }
+
+    #[derive(Deserialize, Debug)]
+    struct Person {
+        name: String,
+        #[serde(flatten)]
+        other_fields: AnyJson,
+    }
+
+    #[test]
+    fn test_parse_any_json() {
+        let json = r#"{"name": "Alice", "age": 30, "city": "Wonderland"}"#;
+        let person: Person = serde_json::from_str(json).unwrap();
+        assert_eq!(person.name, "Alice");
+        assert_eq!(
+            person.other_fields.raw,
+            serde_json::json!({"age": 30, "city": "Wonderland"})
+        );
+    }
+
+    #[test]
+    fn test_parse_empty_any_json() {
+        let json = r#"{"name": "Alice"}"#;
+        let person: Person = serde_json::from_str(json).unwrap();
+        assert_eq!(person.name, "Alice");
+        assert_eq!(person.other_fields.raw, serde_json::json!({}));
+    }
 }

wp_api/src/uniffi_serde.rs

@@ -2,7 +2,7 @@ use wp_localization::{MessageBundle, WpMessages, WpSupportsLocalization};
 use wp_localization_macro::WpDeriveLocalizable;
 
 #[derive(Debug, thiserror::Error, uniffi::Error, WpDeriveLocalizable)]
-pub(crate) enum UniffiSerializationError {
+pub enum UniffiSerializationError {
     Serde { reason: String },
 }
 

wp_api/src/wp_com/endpoint.rs

@@ -6,6 +6,7 @@ use crate::{
 use std::sync::Arc;
 use strum::IntoEnumIterator;
 
+pub mod extensions;
 pub mod followers_endpoint;
 pub mod jetpack_connection_endpoint;
 pub mod oauth2;

wp_api/src/wp_com/endpoint/extensions/comments.rs

@@ -0,0 +1,38 @@
+use serde::Deserialize;
+
+use crate::{AnyJson, uniffi_serde::UniffiSerializationError};
+
+#[derive(Debug, Deserialize, uniffi::Record)]
+pub struct WpComCommentExtension {
+    #[serde(rename = "extended_post")]
+    pub post: Option<WpComCommentExtensionPostInfo>,
+    #[serde(rename = "extended_i_replied")]
+    pub i_replied: bool,
+    #[serde(rename = "extended_like_count")]
+    pub like_count: u32,
+    #[serde(rename = "extended_i_like")]
+    pub i_like: bool,
+}
+
+#[derive(Debug, Deserialize, uniffi::Record)]
+pub struct WpComCommentExtensionPostInfo {
+    pub id: u64,
+    pub title: String,
+    #[serde(rename = "type")]
+    pub kind: String,
+    pub link: String,
+}
+
+#[uniffi::export(with_foreign)]
+pub trait WpComCommentExtensionProvider: Send + Sync {
+    fn parse_extension(&self) -> Result<WpComCommentExtension, UniffiSerializationError>;
+}
+
+#[uniffi::export]
+impl WpComCommentExtensionProvider for AnyJson {
+    fn parse_extension(&self) -> Result<WpComCommentExtension, UniffiSerializationError> {
+        serde_json::to_string(&self.raw)
+            .and_then(|json| serde_json::from_str(&json))
+            .map_err(Into::into)
+    }
+}

wp_api/src/wp_com/endpoint/extensions/mod.rs

@@ -0,0 +1 @@
+pub mod comments;

wp_api_integration_tests/tests/test_comments_immut.rs

@@ -1,14 +1,15 @@
+use serde_json::Value;
 use wp_api::{
-    JsonValue,
     comments::{
         CommentId, CommentListParams, CommentRetrieveParams, CommentStatus, CommentType,
         SparseCommentFieldWithEditContext, SparseCommentFieldWithEmbedContext,
         SparseCommentFieldWithViewContext, WpApiParamCommentsOrderBy,
     },
     posts::PostId,
     users::UserAvatarSize,
+    wp_com::{WpComBaseUrl, endpoint::WpComDotOrgApiUrlResolver},
 };
-use wp_api_integration_tests::prelude::*;
+use wp_api_integration_tests::{WpComTestCredentials, prelude::*};
 
 #[tokio::test]
 #[apply(list_cases)]
@@ -217,14 +218,64 @@ async fn parse_extras() {
         .await
         .assert_response()
         .data;
-    match comment.extras {
-        JsonValue::Object(ref map) => {
+    match comment.additional_fields.raw {
+        Value::Object(ref map) => {
             assert!(map.contains_key("_links"));
         }
         _ => panic!("Expected extras to be an object"),
     }
 }
 
+#[tokio::test]
+#[parallel]
+#[ignore]
+async fn wpcom_comment_extension() {
+    // You'll need to replace the site ID and comment ID with valid ones.
+    let site_id = "site_id".to_string();
+    let comment_id = CommentId(2);
+    let client = WpApiClient::new(
+        Arc::new(WpComDotOrgApiUrlResolver::new(
+            site_id,
+            WpComBaseUrl::Production,
+        )),
+        WpApiClientDelegate {
+            auth_provider: Arc::new(WpAuthenticationProvider::static_with_auth(
+                WpAuthentication::Bearer {
+                    token: WpComTestCredentials::instance().bearer_token.to_string(),
+                },
+            )),
+            request_executor: Arc::new(ReqwestRequestExecutor::default()),
+            middleware_pipeline: Arc::new(WpApiMiddlewarePipeline::default()),
+            app_notifier: Arc::new(EmptyAppNotifier),
+        },
+    );
+    use wp_api::wp_com::endpoint::extensions::comments::WpComCommentExtensionProvider;
+
+    let comment = client
+        .comments()
+        .retrieve_with_view_context(&comment_id, &CommentRetrieveParams::default())
+        .await
+        .assert_response()
+        .data;
+    assert!(comment.additional_fields.parse_extension().is_ok());
+
+    let comment = client
+        .comments()
+        .retrieve_with_edit_context(&comment_id, &CommentRetrieveParams::default())
+        .await
+        .assert_response()
+        .data;
+    assert!(comment.additional_fields.parse_extension().is_ok());
+
+    let comment = client
+        .comments()
+        .retrieve_with_embed_context(&comment_id, &CommentRetrieveParams::default())
+        .await
+        .assert_response()
+        .data;
+    assert!(comment.additional_fields.parse_extension().is_ok());
+}
+
 #[template]
 #[rstest]
 #[case::default(CommentListParams::default())]