Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Enable "duplicate prompt" button (#13588)

Browse source 
Release Notes:

- N/A
This commit is contained in:
Kirill Bulatov 2024-06-27 16:11:43 +03:00 • committed by GitHub
parent da22e0dd0b
commit b16075d00c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 55 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

73
crates/assistant/src/prompt_library.rs
View file

@ -42,7 +42,12 @@ use workspace::Workspace;
actions!( actions!(
prompt_library, prompt_library,
[NewPrompt, DeletePrompt, ToggleDefaultPrompt] [
NewPrompt,
DeletePrompt,
DuplicatePrompt,
ToggleDefaultPrompt
]
); );
/// Init starts loading the PromptStore in the background and assigns /// Init starts loading the PromptStore in the background and assigns
@ -411,6 +416,12 @@ impl PromptLibrary {
} }
} }
pub fn duplicate_active_prompt(&mut self, cx: &mut ViewContext<Self>) {
if let Some(active_prompt_id) = self.active_prompt_id {
self.duplicate_prompt(active_prompt_id, cx);
}
}
pub fn toggle_default_for_active_prompt(&mut self, cx: &mut ViewContext<Self>) { pub fn toggle_default_for_active_prompt(&mut self, cx: &mut ViewContext<Self>) {
if let Some(active_prompt_id) = self.active_prompt_id { if let Some(active_prompt_id) = self.active_prompt_id {
self.toggle_default_for_prompt(active_prompt_id, cx); self.toggle_default_for_prompt(active_prompt_id, cx);
@ -564,6 +575,25 @@ impl PromptLibrary {
} }
} }
pub fn duplicate_prompt(&mut self, prompt_id: PromptId, cx: &mut ViewContext<Self>) {
if let Some(prompt) = self.prompt_editors.get(&prompt_id) {
let new_id = PromptId::new();
let title = prompt.title_editor.read(cx).text(cx);
let body = prompt.body_editor.read(cx).text(cx);
let save = self
.store
.save(new_id, Some(title.into()), false, body.into());
self.picker.update(cx, |picker, cx| picker.refresh(cx));
cx.spawn(|this, mut cx| async move {
save.await?;
this.update(&mut cx, |prompt_library, cx| {
prompt_library.load_prompt(new_id, true, cx)
})
})
.detach_and_log_err(cx);
}
}
fn focus_active_prompt(&mut self, _: &Tab, cx: &mut ViewContext<Self>) { fn focus_active_prompt(&mut self, _: &Tab, cx: &mut ViewContext<Self>) {
if let Some(active_prompt) = self.active_prompt_id { if let Some(active_prompt) = self.active_prompt_id {
self.prompt_editors[&active_prompt] self.prompt_editors[&active_prompt]
@ -897,24 +927,28 @@ impl PromptLibrary {
cx.dispatch_action(Box::new(DeletePrompt)); cx.dispatch_action(Box::new(DeletePrompt));
}), }),
) )
// .child( .child(
// IconButton::new( IconButton::new(
// "duplicate-prompt", "duplicate-prompt",
// IconName::BookCopy, IconName::BookCopy,
// ) )
// .size(ButtonSize::Large) .size(ButtonSize::Large)
// .style(ButtonStyle::Transparent) .style(ButtonStyle::Transparent)
// .shape(IconButtonShape::Square) .shape(IconButtonShape::Square)
// .size(ButtonSize::Large) .size(ButtonSize::Large)
// .tooltip(move |cx| { .tooltip(move |cx| {
// Tooltip::for_action( Tooltip::for_action(
// "Duplicate Prompt", "Duplicate Prompt",
// &gpui::NoAction, &DuplicatePrompt,
// cx, cx,
// ) )
// }) })
// .disabled(true), .on_click(|_, cx| {
// ) cx.dispatch_action(Box::new(
DuplicatePrompt,
));
}),
)
.child( .child(
IconButton::new( IconButton::new(
"toggle-default-prompt", "toggle-default-prompt",
@ -973,6 +1007,7 @@ impl Render for PromptLibrary {
.key_context("PromptLibrary") .key_context("PromptLibrary")
.on_action(cx.listener(|this, &NewPrompt, cx| this.new_prompt(cx))) .on_action(cx.listener(|this, &NewPrompt, cx| this.new_prompt(cx)))
.on_action(cx.listener(|this, &DeletePrompt, cx| this.delete_active_prompt(cx))) .on_action(cx.listener(|this, &DeletePrompt, cx| this.delete_active_prompt(cx)))
.on_action(cx.listener(|this, &DuplicatePrompt, cx| this.duplicate_active_prompt(cx)))
.on_action(cx.listener(|this, &ToggleDefaultPrompt, cx| { .on_action(cx.listener(|this, &ToggleDefaultPrompt, cx| {
this.toggle_default_for_active_prompt(cx) this.toggle_default_for_active_prompt(cx)
})) }))

2
crates/editor/src/editor.rs
View file

@ -2131,7 +2131,7 @@ impl Editor {
self.refresh_inline_completion(false, cx); self.refresh_inline_completion(false, cx);
} }
pub fn placeholder_text(&self, _cx: &mut WindowContext) -> Option<&str> { pub fn placeholder_text(&self, _cx: &WindowContext) -> Option<&str> {
self.placeholder_text.as_deref() self.placeholder_text.as_deref()
} }