chore: add from trait for subscription plan

2024-07-09 13:01:20 +08:00 · 2024-07-09 13:01:20 +08:00 · e1e6ad0e95
parent 23d1c2433b
commit e1e6ad0e95
1 changed files with 13 additions and 0 deletions
--- a/libs/shared-entity/src/dto/billing_dto.rs
+++ b/libs/shared-entity/src/dto/billing_dto.rs
@ -30,6 +30,19 @@ pub enum SubscriptionPlan {
  AiLocal = 4,
 }

+impl From<i16> for SubscriptionPlan {
+  fn from(value: i16) -> Self {
+    match value {
+      0 => SubscriptionPlan::Free,
+      1 => SubscriptionPlan::Pro,
+      2 => SubscriptionPlan::Team,
+      3 => SubscriptionPlan::AiMax,
+      4 => SubscriptionPlan::AiLocal,
+      _ => SubscriptionPlan::Unknown,
+    }
+  }
+}
+
 impl AsRef<str> for SubscriptionPlan {
  fn as_ref(&self) -> &str {
    match self {