use collections::HashMap;use editor::Editor;use gpui::{ actions, prelude::*, AnyElement, AppContext, EventEmitter, FocusHandle, FocusableView, FontWeight, Subscription, View,};use ui::{prelude::*, ButtonLike, ElevationIndex, KeyBinding, ListItem, Tooltip};use util::ResultExt as _;use workspace::item::ItemEvent;use workspace::WorkspaceId;use workspace::{item::Item, Workspace}; use crate::jupyter_settings::JupyterSettings;use crate::repl_store::ReplStore;use crate::KernelSpecification; actions!( repl, [ Run, RunInPlace, ClearOutputs, Sessions, Interrupt, Shutdown, RefreshKernelspecs ]); pub fn init(cx: &mut AppContext) { cx.observe_new_views( |workspace: &mut Workspace, _cx: &mut ViewContext<Workspace>| { workspace.register_action(|workspace, _: &Sessions, cx| { let existing = workspace .active_pane() .read(cx) .items() .find_map(|item| item.downcast::<ReplSessionsPage>()); if let Some(existing) = existing { workspace.activate_item(&existing, true, true, cx); } else { let repl_sessions_page = ReplSessionsPage::new(cx); workspace.add_item_to_active_pane(Box::new(repl_sessions_page), None, true, cx) } }); workspace.register_action(|_workspace, _: &RefreshKernelspecs, cx| { let store = ReplStore::global(cx); store.update(cx, |store, cx| { store.refresh_kernelspecs(cx).detach(); }); }); }, ) .detach(); cx.observe_new_views(move |editor: &mut Editor, cx: &mut ViewContext<Editor>| { if !editor.use_modal_editing() || !editor.buffer().read(cx).is_singleton() { return; } let editor_handle = cx.view().downgrade(); editor .register_action({ let editor_handle = editor_handle.clone(); move |_: &Run, cx| { if !JupyterSettings::enabled(cx) { return; } crate::run(editor_handle.clone(), true, cx).log_err(); } }) .detach(); editor .register_action({ let editor_handle = editor_handle.clone(); move |_: &RunInPlace, cx| { if !JupyterSettings::enabled(cx) { return; } crate::run(editor_handle.clone(), false, cx).log_err(); } }) .detach(); editor .register_action({ let editor_handle = editor_handle.clone(); move |_: &ClearOutputs, cx| { if !JupyterSettings::enabled(cx) { return; } crate::clear_outputs(editor_handle.clone(), cx); } }) .detach(); editor .register_action({ let editor_handle = editor_handle.clone(); move |_: &Interrupt, cx| { if !JupyterSettings::enabled(cx) { return; } crate::interrupt(editor_handle.clone(), cx); } }) .detach(); editor .register_action({ let editor_handle = editor_handle.clone(); move |_: &Shutdown, cx| { if !JupyterSettings::enabled(cx) { return; } crate::shutdown(editor_handle.clone(), cx); } }) .detach(); }) .detach();} pub struct ReplSessionsPage { focus_handle: FocusHandle, _subscriptions: Vec<Subscription>,} impl ReplSessionsPage { pub fn new(cx: &mut ViewContext<Workspace>) -> View<Self> { cx.new_view(|cx: &mut ViewContext<Self>| { let focus_handle = cx.focus_handle(); let subscriptions = vec![ cx.on_focus_in(&focus_handle, |_this, cx| cx.notify()), cx.on_focus_out(&focus_handle, |_this, _event, cx| cx.notify()), ]; Self { focus_handle, _subscriptions: subscriptions, } }) }} impl EventEmitter<ItemEvent> for ReplSessionsPage {} impl FocusableView for ReplSessionsPage { fn focus_handle(&self, _cx: &AppContext) -> FocusHandle { self.focus_handle.clone() }} impl Item for ReplSessionsPage { type Event = ItemEvent; fn tab_content_text(&self, _cx: &WindowContext) -> Option<SharedString> { Some("REPL Sessions".into()) } fn telemetry_event_text(&self) -> Option<&'static str> { Some("repl sessions") } fn show_toolbar(&self) -> bool { false } fn clone_on_split( &self, _workspace_id: Option<WorkspaceId>, _: &mut ViewContext<Self>, ) -> Option<View<Self>> { None } fn to_item_events(event: &Self::Event, mut f: impl FnMut(workspace::item::ItemEvent)) { f(*event) }} impl Render for ReplSessionsPage { fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement { let store = ReplStore::global(cx); let (kernel_specifications, sessions) = store.update(cx, |store, _cx| { ( store.kernel_specifications().cloned().collect::<Vec<_>>(), store.sessions().cloned().collect::<Vec<_>>(), ) }); // When there are no kernel specifications, show a link to the Zed docs explaining how to // install kernels. It can be assumed they don't have a running kernel if we have no // specifications. if kernel_specifications.is_empty() { let instructions = "To start interactively running code in your editor, you need to install and configure Jupyter kernels."; return ReplSessionsContainer::new("No Jupyter Kernels Available") .child(Label::new(instructions)) .child( h_flex().w_full().p_4().justify_center().gap_2().child( ButtonLike::new("install-kernels") .style(ButtonStyle::Filled) .size(ButtonSize::Large) .layer(ElevationIndex::ModalSurface) .child(Label::new("Install Kernels")) .on_click(move |_, cx| { cx.open_url( "https://zed.dev/docs/repl#language-specific-instructions", ) }), ), ); } let mut kernels_by_language: HashMap<String, Vec<KernelSpecification>> = HashMap::default(); for spec in kernel_specifications { kernels_by_language .entry(spec.kernelspec.language.clone()) .or_insert_with(Vec::new) .push(spec); } let kernels_available = v_flex() .child(Label::new("Kernels available").size(LabelSize::Large)) .gap_2() .child( h_flex() .child(Label::new( "Defaults indicated with a checkmark. Learn how to change your default kernel in the ", )) .child( ButtonLike::new("configure-kernels") .style(ButtonStyle::Filled) // .size(ButtonSize::Compact) .layer(ElevationIndex::Surface) .child(Label::new("REPL documentation")) .child(Icon::new(IconName::Link)) .on_click(move |_, cx| { cx.open_url("https://zed.dev/docs/repl#changing-kernels") }), ), ) .children(kernels_by_language.into_iter().map(|(language, specs)| { let chosen_kernel = store.read(cx).kernelspec(&language, cx); v_flex() .gap_1() .child(Label::new(language.clone()).weight(FontWeight::BOLD)) .children(specs.into_iter().map(|spec| { let is_choice = if let Some(chosen_kernel) = &chosen_kernel { chosen_kernel.name.to_lowercase() == spec.name.to_lowercase() && chosen_kernel.path == spec.path } else { false }; let path = SharedString::from(spec.path.to_string_lossy().to_string()); ListItem::new(path.clone()) .selectable(false) .tooltip({ let path = path.clone(); move |cx| Tooltip::text(path.clone(), cx)}) .child( h_flex() .gap_1() .child(div().id(path.clone()).child(Label::new(spec.name.clone()))) .when(is_choice, |el| { let language = language.clone(); el.child( div().id("check").tooltip(move |cx| Tooltip::text(format!("Default Kernel for {language}"), cx)) .child(Icon::new(IconName::Check)))}), ) })) })); // When there are no sessions, show the command to run code in an editor if sessions.is_empty() { let instructions = "To run code in a Jupyter kernel, select some code and use the 'repl::Run' command."; return ReplSessionsContainer::new("No Jupyter Kernel Sessions") .child( v_flex() .child(Label::new(instructions)) .children(KeyBinding::for_action(&Run, cx)), ) .child(div().pt_3().child(kernels_available)); } ReplSessionsContainer::new("Jupyter Kernel Sessions") .children(sessions) .child(kernels_available) }} #[derive(IntoElement)]struct ReplSessionsContainer { title: SharedString, children: Vec<AnyElement>,} impl ReplSessionsContainer { pub fn new(title: impl Into<SharedString>) -> Self { Self { title: title.into(), children: Vec::new(), } }} impl ParentElement for ReplSessionsContainer { fn extend(&mut self, elements: impl IntoIterator<Item = AnyElement>) { self.children.extend(elements) }} impl RenderOnce for ReplSessionsContainer { fn render(self, _cx: &mut WindowContext) -> impl IntoElement { v_flex() .p_4() .gap_2() .size_full() .child(Label::new(self.title).size(LabelSize::Large)) .children(self.children) }}
use collections::HashMap;
use editor::Editor;
use gpui::{
actions, prelude::*, AnyElement, AppContext, EventEmitter, FocusHandle, FocusableView,
FontWeight, Subscription, View,
};
use ui::{prelude::*, ButtonLike, ElevationIndex, KeyBinding, ListItem, Tooltip};
use util::ResultExt as _;
use workspace::item::ItemEvent;
use workspace::WorkspaceId;
use workspace::{item::Item, Workspace};
use crate::jupyter_settings::JupyterSettings;
use crate::repl_store::ReplStore;
use crate::KernelSpecification;
actions!(
repl,
[
Run,
RunInPlace,
ClearOutputs,
Sessions,
Interrupt,
Shutdown,
RefreshKernelspecs
]
);
pub fn init(cx: &mut AppContext) {
cx.observe_new_views(
|workspace: &mut Workspace, _cx: &mut ViewContext<Workspace>| {
workspace.register_action(|workspace, _: &Sessions, cx| {
let existing = workspace
.active_pane()
.read(cx)
.items()
.find_map(|item| item.downcast::<ReplSessionsPage>());
if let Some(existing) = existing {
workspace.activate_item(&existing, true, true, cx);
} else {
let repl_sessions_page = ReplSessionsPage::new(cx);
workspace.add_item_to_active_pane(Box::new(repl_sessions_page), None, true, cx)
}
});
workspace.register_action(|_workspace, _: &RefreshKernelspecs, cx| {
let store = ReplStore::global(cx);
store.update(cx, |store, cx| {
store.refresh_kernelspecs(cx).detach();
});
});
},
)
.detach();
cx.observe_new_views(move |editor: &mut Editor, cx: &mut ViewContext<Editor>| {
if !editor.use_modal_editing() || !editor.buffer().read(cx).is_singleton() {
return;
}
let editor_handle = cx.view().downgrade();
editor
.register_action({
let editor_handle = editor_handle.clone();
move |_: &Run, cx| {
if !JupyterSettings::enabled(cx) {
return;
}
crate::run(editor_handle.clone(), true, cx).log_err();
}
})
.detach();
editor
.register_action({
let editor_handle = editor_handle.clone();
move |_: &RunInPlace, cx| {
if !JupyterSettings::enabled(cx) {
return;
}
crate::run(editor_handle.clone(), false, cx).log_err();
}
})
.detach();
editor
.register_action({
let editor_handle = editor_handle.clone();
move |_: &ClearOutputs, cx| {
if !JupyterSettings::enabled(cx) {
return;
}
crate::clear_outputs(editor_handle.clone(), cx);
}
})
.detach();
editor
.register_action({
let editor_handle = editor_handle.clone();
move |_: &Interrupt, cx| {
if !JupyterSettings::enabled(cx) {
return;
}
crate::interrupt(editor_handle.clone(), cx);
}
})
.detach();
editor
.register_action({
let editor_handle = editor_handle.clone();
move |_: &Shutdown, cx| {
if !JupyterSettings::enabled(cx) {
return;
}
crate::shutdown(editor_handle.clone(), cx);
}
})
.detach();
})
.detach();
}
pub struct ReplSessionsPage {
focus_handle: FocusHandle,
_subscriptions: Vec<Subscription>,
}
impl ReplSessionsPage {
pub fn new(cx: &mut ViewContext<Workspace>) -> View<Self> {
cx.new_view(|cx: &mut ViewContext<Self>| {
let focus_handle = cx.focus_handle();
let subscriptions = vec![
cx.on_focus_in(&focus_handle, |_this, cx| cx.notify()),
cx.on_focus_out(&focus_handle, |_this, _event, cx| cx.notify()),
];
Self {
focus_handle,
_subscriptions: subscriptions,
}
})
}
}
impl EventEmitter<ItemEvent> for ReplSessionsPage {}
impl FocusableView for ReplSessionsPage {
fn focus_handle(&self, _cx: &AppContext) -> FocusHandle {
self.focus_handle.clone()
}
}
impl Item for ReplSessionsPage {
type Event = ItemEvent;
fn tab_content_text(&self, _cx: &WindowContext) -> Option<SharedString> {
Some("REPL Sessions".into())
}
fn telemetry_event_text(&self) -> Option<&'static str> {
Some("repl sessions")
}
fn show_toolbar(&self) -> bool {
false
}
fn clone_on_split(
&self,
_workspace_id: Option<WorkspaceId>,
_: &mut ViewContext<Self>,
) -> Option<View<Self>> {
None
}
fn to_item_events(event: &Self::Event, mut f: impl FnMut(workspace::item::ItemEvent)) {
f(*event)
}
}
impl Render for ReplSessionsPage {
fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement {
let store = ReplStore::global(cx);
let (kernel_specifications, sessions) = store.update(cx, |store, _cx| {
(
store.kernel_specifications().cloned().collect::<Vec<_>>(),
store.sessions().cloned().collect::<Vec<_>>(),
)
});
// When there are no kernel specifications, show a link to the Zed docs explaining how to
// install kernels. It can be assumed they don't have a running kernel if we have no
// specifications.
if kernel_specifications.is_empty() {
let instructions = "To start interactively running code in your editor, you need to install and configure Jupyter kernels.";
return ReplSessionsContainer::new("No Jupyter Kernels Available")
.child(Label::new(instructions))
.child(
h_flex().w_full().p_4().justify_center().gap_2().child(
ButtonLike::new("install-kernels")
.style(ButtonStyle::Filled)
.size(ButtonSize::Large)
.layer(ElevationIndex::ModalSurface)
.child(Label::new("Install Kernels"))
.on_click(move |_, cx| {
cx.open_url(
"https://zed.dev/docs/repl#language-specific-instructions",
)
}),
),
);
}
let mut kernels_by_language: HashMap<String, Vec<KernelSpecification>> = HashMap::default();
for spec in kernel_specifications {
kernels_by_language
.entry(spec.kernelspec.language.clone())
.or_insert_with(Vec::new)
.push(spec);
}
let kernels_available = v_flex()
.child(Label::new("Kernels available").size(LabelSize::Large))
.gap_2()
.child(
h_flex()
.child(Label::new(
"Defaults indicated with a checkmark. Learn how to change your default kernel in the ",
))
.child(
ButtonLike::new("configure-kernels")
.style(ButtonStyle::Filled)
// .size(ButtonSize::Compact)
.layer(ElevationIndex::Surface)
.child(Label::new("REPL documentation"))
.child(Icon::new(IconName::Link))
.on_click(move |_, cx| {
cx.open_url("https://zed.dev/docs/repl#changing-kernels")
}),
),
)
.children(kernels_by_language.into_iter().map(|(language, specs)| {
let chosen_kernel = store.read(cx).kernelspec(&language, cx);
v_flex()
.gap_1()
.child(Label::new(language.clone()).weight(FontWeight::BOLD))
.children(specs.into_iter().map(|spec| {
let is_choice = if let Some(chosen_kernel) = &chosen_kernel {
chosen_kernel.name.to_lowercase() == spec.name.to_lowercase()
&& chosen_kernel.path == spec.path
} else {
false
};
let path = SharedString::from(spec.path.to_string_lossy().to_string());
ListItem::new(path.clone())
.selectable(false)
.tooltip({
let path = path.clone();
move |cx| Tooltip::text(path.clone(), cx)})
.child(
h_flex()
.gap_1()
.child(div().id(path.clone()).child(Label::new(spec.name.clone())))
.when(is_choice, |el| {
let language = language.clone();
el.child(
div().id("check").tooltip(move |cx| Tooltip::text(format!("Default Kernel for {language}"), cx))
.child(Icon::new(IconName::Check)))}),
)
}))
}));
// When there are no sessions, show the command to run code in an editor
if sessions.is_empty() {
let instructions = "To run code in a Jupyter kernel, select some code and use the 'repl::Run' command.";
return ReplSessionsContainer::new("No Jupyter Kernel Sessions")
.child(
v_flex()
.child(Label::new(instructions))
.children(KeyBinding::for_action(&Run, cx)),
)
.child(div().pt_3().child(kernels_available));
}
ReplSessionsContainer::new("Jupyter Kernel Sessions")
.children(sessions)
.child(kernels_available)
}
}
#[derive(IntoElement)]
struct ReplSessionsContainer {
title: SharedString,
children: Vec<AnyElement>,
}
impl ReplSessionsContainer {
pub fn new(title: impl Into<SharedString>) -> Self {
Self {
title: title.into(),
children: Vec::new(),
}
}
}
impl ParentElement for ReplSessionsContainer {
fn extend(&mut self, elements: impl IntoIterator<Item = AnyElement>) {
self.children.extend(elements)
}
}
impl RenderOnce for ReplSessionsContainer {
fn render(self, _cx: &mut WindowContext) -> impl IntoElement {
v_flex()
.p_4()
.gap_2()
.size_full()
.child(Label::new(self.title).size(LabelSize::Large))
.children(self.children)
}
}