diff --git a/Cargo.lock b/Cargo.lock index 97a2ad0f9a394240b5a63c077d16076cf4a8b117..3da4ed86879c02df669fcb19d785260e995e4d2c 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -346,6 +346,7 @@ dependencies = [ "ctor", "editor", "env_logger", + "feature_flags", "file_icons", "fs", "futures 0.3.28", diff --git a/crates/assistant/Cargo.toml b/crates/assistant/Cargo.toml index 6e41f514f94a29582a57453d030aab2314b6a5d4..9d4cc59e75454f113a682bdb5d2f6d04a752e434 100644 --- a/crates/assistant/Cargo.toml +++ b/crates/assistant/Cargo.toml @@ -22,6 +22,7 @@ client.workspace = true collections.workspace = true command_palette_hooks.workspace = true editor.workspace = true +feature_flags.workspace = true file_icons.workspace = true fs.workspace = true futures.workspace = true diff --git a/crates/assistant/src/assistant_panel.rs b/crates/assistant/src/assistant_panel.rs index 0cb202ecc6e48f143f533118355bb1003312c9a7..ad184694a6397dd7828a89418b16e625df0d1e73 100644 --- a/crates/assistant/src/assistant_panel.rs +++ b/crates/assistant/src/assistant_panel.rs @@ -28,6 +28,7 @@ use editor::{ ToOffset as _, ToPoint, }; use editor::{display_map::FlapId, FoldPlaceholder}; +use feature_flags::{FeatureFlag, FeatureFlagAppExt, FeatureFlagViewExt}; use file_icons::FileIcons; use fs::Fs; use futures::future::Shared; @@ -127,6 +128,12 @@ struct ActiveConversationEditor { _subscriptions: Vec, } +struct PromptLibraryFeatureFlag; + +impl FeatureFlag for PromptLibraryFeatureFlag { + const NAME: &'static str = "prompt-library"; +} + impl AssistantPanel { const INLINE_PROMPT_HISTORY_MAX_LEN: usize = 20; @@ -152,6 +159,9 @@ impl AssistantPanel { let workspace_handle = workspace.clone(); workspace.update(&mut cx, |workspace, cx| { cx.new_view::(|cx| { + cx.observe_flag::(|_, _, cx| cx.notify()) + .detach(); + const CONVERSATION_WATCH_DURATION: Duration = Duration::from_millis(100); let _watch_saved_conversations = cx.spawn(move |this, mut cx| async move { let mut events = fs @@ -1178,38 +1188,38 @@ impl AssistantPanel { } fn render_signed_in(&mut self, cx: &mut ViewContext) -> impl IntoElement { - let header = - TabBar::new("assistant_header") - .start_child(h_flex().gap_1().child(self.render_popover_button(cx))) - .children(self.active_conversation_editor().map(|editor| { - h_flex() - .h(rems(Tab::CONTAINER_HEIGHT_IN_REMS)) - .flex_1() - .px_2() - .child(Label::new(editor.read(cx).title(cx)).into_element()) - })) - .end_child( - h_flex() - .gap_2() - .when_some(self.active_conversation_editor(), |this, editor| { - let conversation = editor.read(cx).conversation.clone(); - this.child( - h_flex() - .gap_1() - .child(self.render_model(&conversation, cx)) - .children(self.render_remaining_tokens(&conversation, cx)), - ) - .child( - ui::Divider::vertical() - .inset() - .color(ui::DividerColor::Border), - ) - }) - .child( + let header = TabBar::new("assistant_header") + .start_child(h_flex().gap_1().child(self.render_popover_button(cx))) + .children(self.active_conversation_editor().map(|editor| { + h_flex() + .h(rems(Tab::CONTAINER_HEIGHT_IN_REMS)) + .flex_1() + .px_2() + .child(Label::new(editor.read(cx).title(cx)).into_element()) + })) + .end_child( + h_flex() + .gap_2() + .when_some(self.active_conversation_editor(), |this, editor| { + let conversation = editor.read(cx).conversation.clone(); + this.child( h_flex() .gap_1() - .child(self.render_inject_context_menu(cx)) - .child( + .child(self.render_model(&conversation, cx)) + .children(self.render_remaining_tokens(&conversation, cx)), + ) + .child( + ui::Divider::vertical() + .inset() + .color(ui::DividerColor::Border), + ) + }) + .child( + h_flex() + .gap_1() + .child(self.render_inject_context_menu(cx)) + .children( + cx.has_flag::().then_some( IconButton::new("show_prompt_manager", IconName::Library) .icon_size(IconSize::Small) .on_click(cx.listener(|this, _event, cx| { @@ -1217,8 +1227,9 @@ impl AssistantPanel { })) .tooltip(|cx| Tooltip::text("Prompt Library…", cx)), ), - ), - ); + ), + ), + ); let contents = if self.active_conversation_editor().is_some() { let mut registrar = DivRegistrar::new( diff --git a/crates/client/src/user.rs b/crates/client/src/user.rs index f97f45abe98bdaadea778f3802f4809a77c4fd38..5f9889e3e578376b3c3f30a635662ffc80f822e3 100644 --- a/crates/client/src/user.rs +++ b/crates/client/src/user.rs @@ -192,10 +192,13 @@ impl UserStore { cx.update(|cx| { if let Some(info) = info { - cx.update_flags(info.staff, info.flags); + let disable_staff = std::env::var("ZED_DISABLE_STAFF") + .map_or(false, |v| v != "" && v != "0"); + let staff = info.staff && !disable_staff; + cx.update_flags(staff, info.flags); client.telemetry.set_authenticated_user_info( Some(info.metrics_id.clone()), - info.staff, + staff, ) } })?; diff --git a/crates/feature_flags/src/feature_flags.rs b/crates/feature_flags/src/feature_flags.rs index 0823e2f460c7c1326b9f301cc668cd78565d6ee8..200468d6f32e158542d02cde116f7330c5b3f32b 100644 --- a/crates/feature_flags/src/feature_flags.rs +++ b/crates/feature_flags/src/feature_flags.rs @@ -14,6 +14,12 @@ impl FeatureFlags { impl Global for FeatureFlags {} +/// To create a feature flag, implement this trait on a trivial type and use it as +/// a generic parameter when called [`FeatureFlagAppExt::has_flag`]. +/// +/// Feature flags are always enabled for members of Zed staff. To disable this behavior +/// so you can test flags being disabled, set ZED_DISABLE_STAFF=1 in your environment, +/// which will force Zed to treat the current user as non-staff. pub trait FeatureFlag { const NAME: &'static str; }