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 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,
34{
35    fn observe_flag<T: FeatureFlag, F>(&mut self, callback: F) -> Subscription
36    where
37        F: Fn(bool, &mut V, &mut ViewContext<V>) + 'static,
38    {
39        self.observe_global::<FeatureFlags>(move |v, cx| {
40            let feature_flags = cx.global::<FeatureFlags>();
41            callback(feature_flags.has_flag(<T as FeatureFlag>::NAME), v, cx);
42        })
43    }
44}
45
46pub trait FeatureFlagAppExt {
47    fn update_flags(&mut self, staff: bool, flags: Vec<String>);
48    fn set_staff(&mut self, staff: bool);
49    fn has_flag<T: FeatureFlag>(&self) -> bool;
50    fn is_staff(&self) -> bool;
51}
52
53impl FeatureFlagAppExt for AppContext {
54    fn update_flags(&mut self, staff: bool, flags: Vec<String>) {
55        let feature_flags = self.default_global::<FeatureFlags>();
56        feature_flags.staff = staff;
57        feature_flags.flags = flags;
58    }
59
60    fn set_staff(&mut self, staff: bool) {
61        let feature_flags = self.default_global::<FeatureFlags>();
62        feature_flags.staff = staff;
63    }
64
65    fn has_flag<T: FeatureFlag>(&self) -> bool {
66        if self.has_global::<FeatureFlags>() {
67            self.global::<FeatureFlags>().has_flag(T::NAME)
68        } else {
69            false
70        }
71    }
72
73    fn is_staff(&self) -> bool {
74        if self.has_global::<FeatureFlags>() {
75            return self.global::<FeatureFlags>().staff;
76        } else {
77            false
78        }
79    }
80}