Initial public release of Warp.

Repo-Sync-Origin: warpdotdev/warp-internal@12af1d983b

app/src/pricing/mod.rs

@@ -0,0 +1,76 @@
+use warp_graphql::billing::{
+    AddonCreditsOption, OveragesPricing, PlanPricing, PricingInfo, StripeSubscriptionPlan,
+};
+use warpui::{Entity, ModelContext, SingletonEntity};
+
+/// A global model for maintaining pricing information from the server.
+#[derive(Debug)]
+pub struct PricingInfoModel {
+    /// The latest-known pricing information from the server.
+    pricing_info: Option<PricingInfo>,
+}
+
+impl PricingInfoModel {
+    pub fn new() -> Self {
+        Self { pricing_info: None }
+    }
+
+    /// Updates the model with the latest pricing information from the server.
+    pub fn update_pricing_info(&mut self, pricing_info: PricingInfo, ctx: &mut ModelContext<Self>) {
+        self.pricing_info = Some(pricing_info);
+        ctx.emit(PricingInfoModelEvent::PricingInfoUpdated);
+    }
+
+    /// Returns the current overage pricing information.
+    #[allow(dead_code)]
+    fn overage_pricing(&self) -> Option<&OveragesPricing> {
+        self.pricing_info.as_ref().map(|info| &info.overages)
+    }
+
+    /// Returns the pricing for a specific plan.
+    #[allow(dead_code)]
+    pub fn plan_pricing(&self, plan: &StripeSubscriptionPlan) -> Option<&PlanPricing> {
+        self.pricing_info
+            .as_ref()?
+            .plans
+            .iter()
+            .find(|p| &p.plan == plan)
+    }
+
+    /// Returns the overage cost in dollars (converted from cents).
+    #[allow(dead_code)]
+    pub fn overage_cost_dollars(&self) -> Option<f64> {
+        self.overage_pricing()
+            .map(|overages| overages.price_per_request_usd_cents as f64 / 100.0)
+    }
+
+    /// Returns the monthly cost for a plan in dollars (converted from cents).
+    #[allow(dead_code)]
+    pub fn monthly_plan_cost_dollars(&self, plan: &StripeSubscriptionPlan) -> Option<f64> {
+        self.plan_pricing(plan)
+            .map(|pricing| pricing.monthly_plan_price_per_month_usd_cents as f64 / 100.0)
+    }
+
+    pub fn addon_credits_options(&self) -> Option<&[AddonCreditsOption]> {
+        self.pricing_info
+            .as_ref()
+            .map(|info| info.addon_credits_options.as_slice())
+    }
+}
+
+impl Default for PricingInfoModel {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+#[derive(Debug, Clone)]
+pub enum PricingInfoModelEvent {
+    PricingInfoUpdated,
+}
+
+impl Entity for PricingInfoModel {
+    type Event = PricingInfoModelEvent;
+}
+
+impl SingletonEntity for PricingInfoModel {}