1use gpui2::{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 enum ChannelsAlpha {}
20
21impl FeatureFlag for ChannelsAlpha {
22    const NAME: &'static str = "channels_alpha";
23}
24
25pub trait FeatureFlagViewExt<V: 'static> {
26    fn observe_flag<T: FeatureFlag, F>(&mut self, callback: F) -> Subscription
27    where
28        F: Fn(bool, &mut V, &mut ViewContext<V>) + Send + Sync + 'static;
29}
30
31impl<V> FeatureFlagViewExt<V> for ViewContext<'_, '_, V>
32where
33    V: 'static + Send + Sync,
34{
35    fn observe_flag<T: FeatureFlag, F>(&mut self, callback: F) -> Subscription
36    where
37        F: Fn(bool, &mut V, &mut ViewContext<V>) + Send + Sync + 'static,
38    {
39        self.observe_global::<FeatureFlags>(move |v, feature_flags, cx| {
40            callback(feature_flags.has_flag(<T as FeatureFlag>::NAME), v, cx);
41        })
42    }
43}
44
45pub trait FeatureFlagAppExt {
46    fn update_flags(&mut self, staff: bool, flags: Vec<String>);
47    fn set_staff(&mut self, staff: bool);
48    fn has_flag<T: FeatureFlag>(&self) -> bool;
49    fn is_staff(&self) -> bool;
50}
51
52impl FeatureFlagAppExt for AppContext {
53    fn update_flags(&mut self, staff: bool, flags: Vec<String>) {
54        let feature_flags = self.default_global_mut::<FeatureFlags>();
55        feature_flags.staff = staff;
56        feature_flags.flags = flags;
57    }
58
59    fn set_staff(&mut self, staff: bool) {
60        let feature_flags = self.default_global_mut::<FeatureFlags>();
61        feature_flags.staff = staff;
62    }
63
64    fn has_flag<T: FeatureFlag>(&self) -> bool {
65        if self.has_global::<FeatureFlags>() {
66            self.global::<FeatureFlags>().has_flag(T::NAME)
67        } else {
68            false
69        }
70    }
71
72    fn is_staff(&self) -> bool {
73        if self.has_global::<FeatureFlags>() {
74            return self.global::<FeatureFlags>().staff;
75        } else {
76            false
77        }
78    }
79}