1use gpui::{AppContext, Subscription, ViewContext};
 2
 3#[derive(Default)]
 4struct FeatureFlags {
 5    flags: Vec<String>,
 6    staff: bool,
 7}
 8
 9impl FeatureFlags {
10    fn has_flag(&self, flag: &str) -> bool {
11        self.staff || self.flags.iter().find(|f| f.as_str() == flag).is_some()
12    }
13}
14
15pub trait FeatureFlag {
16    const NAME: &'static str;
17}
18
19pub trait FeatureFlagViewExt<V: 'static> {
20    fn observe_flag<T: FeatureFlag, F>(&mut self, callback: F) -> Subscription
21    where
22        F: Fn(bool, &mut V, &mut ViewContext<V>) + Send + Sync + 'static;
23}
24
25impl<V> FeatureFlagViewExt<V> for ViewContext<'_, V>
26where
27    V: 'static,
28{
29    fn observe_flag<T: FeatureFlag, F>(&mut self, callback: F) -> Subscription
30    where
31        F: Fn(bool, &mut V, &mut ViewContext<V>) + 'static,
32    {
33        self.observe_global::<FeatureFlags>(move |v, cx| {
34            let feature_flags = cx.global::<FeatureFlags>();
35            callback(feature_flags.has_flag(<T as FeatureFlag>::NAME), v, cx);
36        })
37    }
38}
39
40pub trait FeatureFlagAppExt {
41    fn update_flags(&mut self, staff: bool, flags: Vec<String>);
42    fn set_staff(&mut self, staff: bool);
43    fn has_flag<T: FeatureFlag>(&self) -> bool;
44    fn is_staff(&self) -> bool;
45}
46
47impl FeatureFlagAppExt for AppContext {
48    fn update_flags(&mut self, staff: bool, flags: Vec<String>) {
49        let feature_flags = self.default_global::<FeatureFlags>();
50        feature_flags.staff = staff;
51        feature_flags.flags = flags;
52    }
53
54    fn set_staff(&mut self, staff: bool) {
55        let feature_flags = self.default_global::<FeatureFlags>();
56        feature_flags.staff = staff;
57    }
58
59    fn has_flag<T: FeatureFlag>(&self) -> bool {
60        self.try_global::<FeatureFlags>()
61            .map(|flags| flags.has_flag(T::NAME))
62            .unwrap_or(false)
63    }
64
65    fn is_staff(&self) -> bool {
66        self.try_global::<FeatureFlags>()
67            .map(|flags| flags.staff)
68            .unwrap_or(false)
69    }
70}