Add can_resolve function to prevent panic

Author: crazytonyli

wp_api/src/request.rs

@@ -10,7 +10,7 @@ use crate::{
 use base64::Engine;
 use chrono::{DateTime, Utc};
 use endpoint::{
-    ApiEndpointUrl, ApiUrlResolver,
+    ApiEndpointUrl, ApiUrlResolver, AsNamespace, WpNamespace,
     application_passwords_endpoint::{
         ApplicationPasswordsRequestBuilder,
         ApplicationPasswordsRequestRetrieveCurrentWithEditContextResponse,
@@ -943,6 +943,10 @@ pub async fn fetch_authentication_state(
     api_url_resolver: Arc<dyn ApiUrlResolver>,
     authentication_provider: Arc<WpAuthenticationProvider>,
 ) -> Result<AuthenticationState, WpApiError> {
+    if !api_url_resolver.can_resolve(WpNamespace::WpV2.namespace_value().to_string()) {
+        return Ok(AuthenticationState::Unauthorized);
+    }
+
     let request =
         ApplicationPasswordsRequestBuilder::new(api_url_resolver, authentication_provider)
             .retrieve_current_with_edit_context()

wp_api/src/request/endpoint.rs

@@ -118,6 +118,8 @@ impl AsNamespace for WpNamespace {
 
 #[uniffi::export(with_foreign)]
 pub trait ApiUrlResolver: Send + Sync {
+    fn can_resolve(&self, namespace: String) -> bool;
+
     fn resolve(&self, namespace: String, endpoint_segments: Vec<String>) -> Arc<ParsedUrl>;
 }
 
@@ -140,6 +142,10 @@ impl WpOrgSiteApiUrlResolver {
 
 #[uniffi::export]
 impl ApiUrlResolver for WpOrgSiteApiUrlResolver {
+    fn can_resolve(&self, _namespace: String) -> bool {
+        true
+    }
+
     fn resolve(&self, namespace: String, endpoint_segments: Vec<String>) -> Arc<ParsedUrl> {
         Arc::new(
             self.api_root_url

wp_api/src/wp_com/endpoint.rs

@@ -37,16 +37,17 @@ impl WpComDotOrgApiUrlResolver {
 
 #[uniffi::export]
 impl ApiUrlResolver for WpComDotOrgApiUrlResolver {
+    fn can_resolve(&self, namespace: String) -> bool {
+        WpNamespace::iter().any(|n| n.namespace_value() == namespace)
+    }
+
     fn resolve(&self, namespace: String, endpoint_segments: Vec<String>) -> Arc<ParsedUrl> {
-        {
-            if !WpNamespace::iter().any(|n| n.namespace_value() == namespace) {
-                panic!(
-                    "`WpComDotOrgApiUrlResolver` doesn't support the namespace `{}`. The supported namespaces are: {:?}",
-                    namespace,
-                    WpNamespace::iter()
-                );
-            }
-        }
+        assert!(
+            self.can_resolve(namespace.clone()),
+            "`WpComDotOrgApiUrlResolver` doesn't support the namespace `{}`. The supported namespaces are: {:?}",
+            namespace,
+            WpNamespace::iter()
+        );
 
         // The API root endpoint needs special handling for WordPress.com
         if namespace == WpNamespace::None.namespace_value() && endpoint_segments.is_empty() {
@@ -91,14 +92,15 @@ impl Default for WpComApiClientInternalUrlResolver {
 }
 
 impl ApiUrlResolver for WpComApiClientInternalUrlResolver {
+    fn can_resolve(&self, namespace: String) -> bool {
+        !WpNamespace::iter().any(|n| n.namespace_value() == namespace)
+    }
+
     fn resolve(&self, namespace: String, endpoint_segments: Vec<String>) -> Arc<ParsedUrl> {
-        {
-            if WpNamespace::iter().any(|n| n.namespace_value() == namespace) {
-                panic!(
-                    "`WpComApiClient` doesn't support the namespace `{namespace}`. Try using `WpApiClient` instead.",
-                );
-            }
-        }
+        assert!(
+            self.can_resolve(namespace.clone()),
+            "`WpComApiClient` doesn't support the namespace `{namespace}`. Try using `WpApiClient` instead.",
+        );
         Arc::new(
             self.base_url
                 .by_extending_and_splitting_by_forward_slash(