b1872e3afd
Done automatically with
> ast-grep -p '$A.background_executor().spawn($B)' -r
'$A.background_spawn($B)' --update-all --globs "\!crates/gpui"
Followed by:
* `cargo fmt`
* Unexpected need to remove some trailing whitespace.
* Manually adding imports of `gpui::{AppContext as _}` which provides
`background_spawn`
* Added `AppContext as _` to existing use of `AppContext`
Release Notes:
- N/A
118 lines
3.7 KiB
Rust
118 lines
3.7 KiB
Rust
use std::sync::atomic::AtomicBool;
|
|
use std::sync::Arc;
|
|
use std::time::Duration;
|
|
|
|
use anyhow::Result;
|
|
use assistant_slash_command::{
|
|
ArgumentCompletion, SlashCommand, SlashCommandContent, SlashCommandEvent,
|
|
SlashCommandOutputSection, SlashCommandResult,
|
|
};
|
|
use feature_flags::FeatureFlag;
|
|
use futures::channel::mpsc;
|
|
use gpui::{Task, WeakEntity};
|
|
use language::{BufferSnapshot, LspAdapterDelegate};
|
|
use smol::stream::StreamExt;
|
|
use smol::Timer;
|
|
use ui::prelude::*;
|
|
use workspace::Workspace;
|
|
|
|
pub struct StreamingExampleSlashCommandFeatureFlag;
|
|
|
|
impl FeatureFlag for StreamingExampleSlashCommandFeatureFlag {
|
|
const NAME: &'static str = "streaming-example-slash-command";
|
|
}
|
|
|
|
pub struct StreamingExampleSlashCommand;
|
|
|
|
impl SlashCommand for StreamingExampleSlashCommand {
|
|
fn name(&self) -> String {
|
|
"streaming-example".into()
|
|
}
|
|
|
|
fn description(&self) -> String {
|
|
"An example slash command that showcases streaming.".into()
|
|
}
|
|
|
|
fn menu_text(&self) -> String {
|
|
self.description()
|
|
}
|
|
|
|
fn requires_argument(&self) -> bool {
|
|
false
|
|
}
|
|
|
|
fn complete_argument(
|
|
self: Arc<Self>,
|
|
_arguments: &[String],
|
|
_cancel: Arc<AtomicBool>,
|
|
_workspace: Option<WeakEntity<Workspace>>,
|
|
_window: &mut Window,
|
|
_cx: &mut App,
|
|
) -> Task<Result<Vec<ArgumentCompletion>>> {
|
|
Task::ready(Ok(Vec::new()))
|
|
}
|
|
|
|
fn run(
|
|
self: Arc<Self>,
|
|
_arguments: &[String],
|
|
_context_slash_command_output_sections: &[SlashCommandOutputSection<language::Anchor>],
|
|
_context_buffer: BufferSnapshot,
|
|
_workspace: WeakEntity<Workspace>,
|
|
_delegate: Option<Arc<dyn LspAdapterDelegate>>,
|
|
_: &mut Window,
|
|
cx: &mut App,
|
|
) -> Task<SlashCommandResult> {
|
|
let (events_tx, events_rx) = mpsc::unbounded();
|
|
cx.background_spawn(async move {
|
|
events_tx.unbounded_send(Ok(SlashCommandEvent::StartSection {
|
|
icon: IconName::FileRust,
|
|
label: "Section 1".into(),
|
|
metadata: None,
|
|
}))?;
|
|
events_tx.unbounded_send(Ok(SlashCommandEvent::Content(
|
|
SlashCommandContent::Text {
|
|
text: "Hello".into(),
|
|
run_commands_in_text: false,
|
|
},
|
|
)))?;
|
|
events_tx.unbounded_send(Ok(SlashCommandEvent::EndSection))?;
|
|
|
|
Timer::after(Duration::from_secs(1)).await;
|
|
|
|
events_tx.unbounded_send(Ok(SlashCommandEvent::StartSection {
|
|
icon: IconName::FileRust,
|
|
label: "Section 2".into(),
|
|
metadata: None,
|
|
}))?;
|
|
events_tx.unbounded_send(Ok(SlashCommandEvent::Content(
|
|
SlashCommandContent::Text {
|
|
text: "World".into(),
|
|
run_commands_in_text: false,
|
|
},
|
|
)))?;
|
|
events_tx.unbounded_send(Ok(SlashCommandEvent::EndSection))?;
|
|
|
|
for n in 1..=10 {
|
|
Timer::after(Duration::from_secs(1)).await;
|
|
|
|
events_tx.unbounded_send(Ok(SlashCommandEvent::StartSection {
|
|
icon: IconName::StarFilled,
|
|
label: format!("Section {n}").into(),
|
|
metadata: None,
|
|
}))?;
|
|
events_tx.unbounded_send(Ok(SlashCommandEvent::Content(
|
|
SlashCommandContent::Text {
|
|
text: "lorem ipsum ".repeat(n).trim().into(),
|
|
run_commands_in_text: false,
|
|
},
|
|
)))?;
|
|
events_tx.unbounded_send(Ok(SlashCommandEvent::EndSection))?;
|
|
}
|
|
|
|
anyhow::Ok(())
|
|
})
|
|
.detach_and_log_err(cx);
|
|
|
|
Task::ready(Ok(events_rx.boxed()))
|
|
}
|
|
}
|