diff --git a/crates/repl/src/repl_sessions_ui.rs b/crates/repl/src/repl_sessions_ui.rs index ae30d8cb9b5f1317b7f1c7e2c8bb4c4980c1a4d7..5b01b971baa957588816edd48db1c814aaae4a0a 100644 --- a/crates/repl/src/repl_sessions_ui.rs +++ b/crates/repl/src/repl_sessions_ui.rs @@ -61,42 +61,45 @@ pub fn init(cx: &mut AppContext) { return; } - let is_local_project = editor - .workspace() - .map(|workspace| workspace.read(cx).project().read(cx).is_local()) - .unwrap_or(false); + cx.defer(|editor, cx| { + let workspace = Workspace::for_window(cx); - if !is_local_project { - return; - } + let is_local_project = workspace + .map(|workspace| workspace.read(cx).project().read(cx).is_local()) + .unwrap_or(false); - let editor_handle = cx.view().downgrade(); + if !is_local_project { + return; + } - editor - .register_action({ - let editor_handle = editor_handle.clone(); - move |_: &Run, cx| { - if !JupyterSettings::enabled(cx) { - return; - } + let editor_handle = cx.view().downgrade(); - 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; - } + editor + .register_action({ + let editor_handle = editor_handle.clone(); + move |_: &Run, cx| { + if !JupyterSettings::enabled(cx) { + return; + } - crate::run(editor_handle.clone(), false, cx).log_err(); - } - }) - .detach(); + 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(); + }); }) .detach(); } diff --git a/crates/workspace/src/workspace.rs b/crates/workspace/src/workspace.rs index 43c8916593eaf2c3576335fa1779f4b61c0751b3..02926e75729b528caab9f9dc1d586acc268c9739 100644 --- a/crates/workspace/src/workspace.rs +++ b/crates/workspace/src/workspace.rs @@ -4547,6 +4547,11 @@ impl Workspace { .children(leader_border), ) } + + pub fn for_window(cx: &mut WindowContext) -> Option> { + let window = cx.window_handle().downcast::()?; + cx.read_window(&window, |workspace, _| workspace).ok() + } } fn leader_border_for_pane(