bctx_cloud_core/client/
upgrade.rs

1use serde::{Deserialize, Serialize};
2
3#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Serialize, Deserialize)]
4#[serde(rename_all = "lowercase")]
5pub enum Tier {
6    Free,
7    Beacon,
8    Studio,
9    Enterprise,
10}
11
12impl Tier {
13    pub fn parse_tier(s: &str) -> Self {
14        match s {
15            "beacon" => Tier::Beacon,
16            "studio" => Tier::Studio,
17            "enterprise" => Tier::Enterprise,
18            _ => Tier::Free,
19        }
20    }
21
22    pub fn vault_fact_limit(&self) -> Option<usize> {
23        match self {
24            Tier::Free => Some(500),
25            Tier::Beacon => Some(10_000),
26            Tier::Studio | Tier::Enterprise => None,
27        }
28    }
29
30    pub fn cloud_token_quota(&self) -> Option<u64> {
31        match self {
32            Tier::Free => None,
33            Tier::Beacon => Some(100_000),
34            Tier::Studio => Some(1_000_000),
35            Tier::Enterprise => None,
36        }
37    }
38
39    pub fn cloud_sync_enabled(&self) -> bool {
40        matches!(self, Tier::Beacon | Tier::Studio | Tier::Enterprise)
41    }
42
43    pub fn team_vault_enabled(&self) -> bool {
44        matches!(self, Tier::Studio | Tier::Enterprise)
45    }
46
47    pub fn upgrade_url(&self) -> &'static str {
48        "https://betterctx.com/upgrade"
49    }
50}
51
52/// Check whether the current tier allows an action and return an upgrade prompt if not.
53pub fn gate(tier: &Tier, required: &Tier, feature: &str) -> Result<(), String> {
54    if tier >= required {
55        Ok(())
56    } else {
57        Err(format!(
58            "{feature} requires the {required:?} tier. Upgrade at {}",
59            tier.upgrade_url()
60        ))
61    }
62}
bctx_cloud_core/client/upgrade.rs

bctx_cloud_core/client/
upgrade.rs
