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