Fix the issue of failing to read the JWTFrom type configuration.

bittcrafter · bittcrafter · commit ae3eeda95dee · 2025-04-04T22:20:31.000+08:00
diff --git a/rmqtt-plugins/rmqtt-auth-jwt/src/config.rs b/rmqtt-plugins/rmqtt-auth-jwt/src/config.rs
@@ -26,7 +26,11 @@ pub struct PluginConfig {
     #[serde(default = "PluginConfig::priority_default")]
     pub priority: Priority,
 
-    #[serde(default = "PluginConfig::from_default")]
+    #[serde(
+        default = "PluginConfig::from_default",
+        serialize_with = "PluginConfig::serialize_from",
+        deserialize_with = "PluginConfig::deserialize_from"
+    )]
     pub from: JWTFrom,
 
     #[serde(
@@ -142,6 +146,36 @@ impl PluginConfig {
         Ok(enc)
     }
 
+    #[inline]
+    fn serialize_from<S>(enc: &JWTFrom, s: S) -> std::result::Result<S::Ok, S::Error>
+    where
+        S: ser::Serializer,
+    {
+        let enc = match enc {
+            JWTFrom::Username => "username",
+            JWTFrom::Password => "password",
+        };
+        enc.serialize(s)
+    }
+
+    #[inline]
+    fn deserialize_from<'de, D>(deserializer: D) -> std::result::Result<JWTFrom, D::Error>
+    where
+        D: Deserializer<'de>,
+    {
+        let enc: String = String::deserialize(deserializer)?;
+        let enc = match enc.as_str() {
+            "username" => JWTFrom::Username,
+            "password" => JWTFrom::Password,
+            _ => {
+                return Err(de::Error::custom(
+                    "Invalid jwt from, only 'username' and 'password' are supported.",
+                ))
+            }
+        };
+        Ok(enc)
+    }
+
     #[inline]
     fn serialize_validate_claims<S>(claims: &ValidateClaims, s: S) -> std::result::Result<S::Ok, S::Error>
     where
@@ -214,7 +248,7 @@ impl PluginConfig {
     }
 }
 
-#[derive(Debug, Clone, Copy, Deserialize, Serialize)]
+#[derive(Debug, Clone, Copy)]
 pub(crate) enum JWTFrom {
     Username,
     Password,
