use super::{align_view, Align, Context, Editor}; use crate::{ commands, compositor::Compositor, job::Callback, ui::{self, FilePicker, Picker, Popup, Prompt, PromptEvent, Text}, }; use helix_core::{ syntax::{DebugArgumentValue, DebugConfigCompletion}, Selection, }; use helix_dap::{self as dap, Client, ThreadId}; use helix_lsp::block_on; use serde_json::{to_value, Value}; use tokio_stream::wrappers::UnboundedReceiverStream; use std::collections::HashMap; // general utils: pub fn dap_pos_to_pos(doc: &helix_core::Rope, line: usize, column: usize) -> Option { // 1-indexing to 0 indexing let line = doc.try_line_to_char(line - 1).ok()?; let pos = line + column; // TODO: this is probably utf-16 offsets Some(pos) } pub fn resume_application(debugger: &mut Client) { if let Some(thread_id) = debugger.thread_id { debugger .thread_states .insert(thread_id, "running".to_string()); debugger.stack_frames.remove(&thread_id); } debugger.active_frame = None; debugger.thread_id = None; } pub async fn select_thread_id(editor: &mut Editor, thread_id: ThreadId, force: bool) { let debugger = match &mut editor.debugger { Some(debugger) => debugger, None => return, }; if !force && debugger.thread_id.is_some() { return; } debugger.thread_id = Some(thread_id); fetch_stack_trace(debugger, thread_id).await; let frame = debugger.stack_frames[&thread_id].get(0).cloned(); if let Some(frame) = &frame { jump_to_stack_frame(editor, frame); } } pub async fn fetch_stack_trace(debugger: &mut Client, thread_id: ThreadId) { let (frames, _) = match debugger.stack_trace(thread_id).await { Ok(frames) => frames, Err(_) => return, }; debugger.stack_frames.insert(thread_id, frames); debugger.active_frame = Some(0); } pub fn jump_to_stack_frame(editor: &mut Editor, frame: &helix_dap::StackFrame) { let path = if let Some(helix_dap::Source { path: Some(ref path), .. }) = frame.source { path.clone() } else { return; }; editor .open(path, helix_view::editor::Action::Replace) .unwrap(); // TODO: there should be no unwrapping! let (view, doc) = current!(editor); let text_end = doc.text().len_chars().saturating_sub(1); let start = dap_pos_to_pos(doc.text(), frame.line, frame.column).unwrap_or(0); let end = frame .end_line .and_then(|end_line| dap_pos_to_pos(doc.text(), end_line, frame.end_column.unwrap_or(0))) .unwrap_or(start); let selection = Selection::single(start.min(text_end), end.min(text_end)); doc.set_selection(view.id, selection); align_view(doc, view, Align::Center); } fn thread_picker(cx: &mut Context, callback_fn: impl Fn(&mut Editor, &dap::Thread) + 'static) { let debugger = match &mut cx.editor.debugger { Some(debugger) => debugger, None => return, }; let threads = match block_on(debugger.threads()) { Ok(threads) => threads, Err(e) => { cx.editor .set_error(format!("Failed to retrieve threads: {:?}", e)); return; } }; if threads.len() == 1 { callback_fn(cx.editor, &threads[0]); return; } let thread_states = debugger.thread_states.clone(); let frames = debugger.stack_frames.clone(); let picker = FilePicker::new( threads, move |thread| { format!( "{} ({})", thread.name, thread_states .get(&thread.id) .unwrap_or(&"unknown".to_owned()) ) .into() }, move |cx, thread, _action| callback_fn(cx.editor, thread), move |_editor, thread| { if let Some(frame) = frames.get(&thread.id).and_then(|bt| bt.get(0)) { frame .source .as_ref() .and_then(|source| source.path.clone()) .map(|path| { ( path, Some(( frame.line.saturating_sub(1), frame.end_line.unwrap_or(frame.line).saturating_sub(1), )), ) }) } else { None } }, ); cx.push_layer(Box::new(picker)) } // -- DAP pub fn dap_start_impl( editor: &mut Editor, name: Option<&str>, socket: Option, params: Option>, ) { let (_, doc) = current!(editor); let path = match doc.path() { Some(path) => path.to_path_buf(), None => { editor.set_error("Can't start debug: document has no path".to_string()); return; } }; let language_config = editor.syn_loader.language_config_for_file_name(&path); let config = match language_config .as_deref() .and_then(|config| config.debugger.as_ref()) { Some(c) => c, None => { editor.set_error( "Can't start debug: no debug adapter available for language".to_string(), ); return; } }; let result = match socket { Some(socket) => block_on(Client::tcp(socket, 0)), None => block_on(Client::process( &config.transport, &config.command, config.args.iter().map(|arg| arg.as_str()).collect(), config.port_arg.as_deref(), 0, )), }; let (mut debugger, events) = match result { Ok(r) => r, Err(e) => { editor.set_error(format!("Failed to start debug session: {:?}", e)); return; } }; let request = debugger.initialize(config.name.clone()); if let Err(e) = block_on(request) { editor.set_error(format!("Failed to initialize debug adapter: {:?}", e)); return; } debugger.quirks = config.quirks.clone(); let start_config = match name { Some(name) => config.templates.iter().find(|t| t.name == name), None => config.templates.get(0), }; let start_config = match start_config { Some(c) => c, None => { editor.set_error("Can't start debug: no debug config with given name".to_string()); return; } }; let template = start_config.args.clone(); let mut args: HashMap = HashMap::new(); if let Some(params) = params { for (k, t) in template { let mut value = t; for (i, x) in params.iter().enumerate() { let mut param = x.to_string(); if let Some(DebugConfigCompletion::Advanced(cfg)) = start_config.completion.get(i) { if cfg.completion == Some("filename".to_owned()) || cfg.completion == Some("directory".to_owned()) { param = std::fs::canonicalize(x) .ok() .and_then(|pb| pb.into_os_string().into_string().ok()) .unwrap_or_else(|| x.to_string()); } } // For param #0 replace {0} in args value = match value { DebugArgumentValue::String(v) => { DebugArgumentValue::String(v.replace(format!("{{{}}}", i).as_str(), ¶m)) } DebugArgumentValue::Array(arr) => DebugArgumentValue::Array( arr.iter() .map(|v| v.replace(format!("{{{}}}", i).as_str(), ¶m)) .collect(), ), }; } if let DebugArgumentValue::String(string) = value { if let Ok(integer) = string.parse::() { args.insert(k, to_value(integer).unwrap()); } else { args.insert(k, to_value(string).unwrap()); } } else if let DebugArgumentValue::Array(arr) = value { args.insert(k, to_value(arr).unwrap()); } } } let args = to_value(args).unwrap(); let result = match &start_config.request[..] { "launch" => block_on(debugger.launch(args)), "attach" => block_on(debugger.attach(args)), _ => { editor.set_error("Unsupported request".to_string()); return; } }; if let Err(e) = result { editor.set_error(format!("Failed {} target: {:?}", start_config.request, e)); return; } // TODO: either await "initialized" or buffer commands until event is received editor.debugger = Some(debugger); let stream = UnboundedReceiverStream::new(events); editor.debugger_events.push(stream); } pub fn dap_launch(cx: &mut Context) { if cx.editor.debugger.is_some() { cx.editor .set_error("Can't start debug: debugger is running".to_string()); return; } let (_, doc) = current!(cx.editor); let path = match doc.path() { Some(path) => path, None => { cx.editor .set_error("Can't start debug: document has no path".to_string()); return; } }; let language_config = cx.editor.syn_loader.language_config_for_file_name(path); let config = match language_config .as_deref() .and_then(|config| config.debugger.as_ref()) { Some(c) => c, None => { cx.editor.set_error( "Can't start debug: no debug adapter available for language".to_string(), ); return; } }; cx.push_layer(Box::new(Picker::new( true, config.templates.clone(), |template| template.name.as_str().into(), |cx, template, _action| { let completions = template.completion.clone(); let name = template.name.clone(); let callback = Box::pin(async move { let call: Callback = Box::new(move |_editor: &mut Editor, compositor: &mut Compositor| { let prompt = debug_parameter_prompt(completions, name, Vec::new()); compositor.push(Box::new(prompt)); }); Ok(call) }); cx.jobs.callback(callback); }, ))); // TODO: wrap in popup with fixed size } fn debug_parameter_prompt( completions: Vec, config_name: String, mut params: Vec, ) -> Prompt { let i = params.len(); let completion = completions.get(i).unwrap(); let field_type = if let DebugConfigCompletion::Advanced(cfg) = completion { cfg.completion.clone().unwrap_or_else(|| "".to_owned()) } else { "".to_owned() }; let name = match completion { DebugConfigCompletion::Advanced(cfg) => { cfg.name.clone().unwrap_or_else(|| field_type.to_owned()) } DebugConfigCompletion::Named(name) => name.clone(), }; let default_val = match completion { DebugConfigCompletion::Advanced(cfg) => { cfg.default.clone().unwrap_or_else(|| "".to_owned()) } _ => "".to_owned(), }; let completer = match &field_type[..] { "filename" => ui::completers::filename, "directory" => ui::completers::directory, _ => |_input: &str| Vec::new(), }; Prompt::new( format!("{}: ", name).into(), None, completer, move |cx: &mut crate::compositor::Context, input: &str, event: PromptEvent| { if event != PromptEvent::Validate { return; } let mut value = input.to_owned(); if value.is_empty() { value = default_val.clone(); } params.push(value); if params.len() < completions.len() { let completions = completions.clone(); let config_name = config_name.clone(); let params = params.clone(); let callback = Box::pin(async move { let call: Callback = Box::new(move |_editor: &mut Editor, compositor: &mut Compositor| { let prompt = debug_parameter_prompt(completions, config_name, params); compositor.push(Box::new(prompt)); }); Ok(call) }); cx.jobs.callback(callback); } else { commands::dap_start_impl( cx.editor, Some(&config_name), None, Some(params.iter().map(|x| x.as_str()).collect()), ); } }, None, ) } pub fn dap_toggle_breakpoint(cx: &mut Context) { let (view, doc) = current!(cx.editor); let text = doc.text().slice(..); let pos = doc.selection(view.id).primary().cursor(text); let breakpoint = helix_dap::SourceBreakpoint { line: text.char_to_line(pos) + 1, // convert from 0-indexing to 1-indexing (TODO: could set debugger to 0-indexing on init) ..Default::default() }; let path = match doc.path() { Some(path) => path.to_path_buf(), None => { cx.editor .set_error("Can't set breakpoint: document has no path".to_string()); return; } }; // TODO: need to map breakpoints over edits and update them? // we shouldn't really allow editing while debug is running though let breakpoints = cx.editor.breakpoints.entry(path.clone()).or_default(); if let Some(pos) = breakpoints.iter().position(|b| b.line == breakpoint.line) { breakpoints.remove(pos); } else { breakpoints.push(breakpoint); } let breakpoints = breakpoints.clone(); let debugger = match &mut cx.editor.debugger { Some(debugger) => debugger, None => return, }; let request = debugger.set_breakpoints(path, breakpoints); match block_on(request) { Ok(Some(breakpoints)) => { let old_breakpoints = debugger.breakpoints.clone(); debugger.breakpoints = breakpoints.clone(); for bp in breakpoints { if !old_breakpoints.iter().any(|b| b.message == bp.message) { if let Some(msg) = &bp.message { cx.editor.set_status(format!("Breakpoint set: {}", msg)); break; } } } } Err(e) => cx .editor .set_error(format!("Failed to set breakpoints: {:?}", e)), _ => {} }; } pub fn dap_continue(cx: &mut Context) { let debugger = match &mut cx.editor.debugger { Some(debugger) => debugger, None => return, }; if let Some(thread_id) = debugger.thread_id { let request = debugger.continue_thread(thread_id); if let Err(e) = block_on(request) { cx.editor.set_error(format!("Failed to continue: {:?}", e)); return; } resume_application(debugger); } else { cx.editor .set_error("Currently active thread is not stopped. Switch the thread.".into()); } } pub fn dap_pause(cx: &mut Context) { thread_picker(cx, |editor, thread| { let debugger = match &mut editor.debugger { Some(debugger) => debugger, None => return, }; let request = debugger.pause(thread.id); // NOTE: we don't need to set active thread id here because DAP will emit a "stopped" event if let Err(e) = block_on(request) { editor.set_error(format!("Failed to pause: {:?}", e)); } }) } pub fn dap_step_in(cx: &mut Context) { let debugger = match &mut cx.editor.debugger { Some(debugger) => debugger, None => return, }; if let Some(thread_id) = debugger.thread_id { let request = debugger.step_in(thread_id); if let Err(e) = block_on(request) { cx.editor.set_error(format!("Failed to continue: {:?}", e)); return; } resume_application(debugger); } else { cx.editor .set_error("Currently active thread is not stopped. Switch the thread.".into()); } } pub fn dap_step_out(cx: &mut Context) { let debugger = match &mut cx.editor.debugger { Some(debugger) => debugger, None => return, }; if let Some(thread_id) = debugger.thread_id { let request = debugger.step_out(thread_id); if let Err(e) = block_on(request) { cx.editor.set_error(format!("Failed to continue: {:?}", e)); return; } resume_application(debugger); } else { cx.editor .set_error("Currently active thread is not stopped. Switch the thread.".into()); } } pub fn dap_next(cx: &mut Context) { let debugger = match &mut cx.editor.debugger { Some(debugger) => debugger, None => return, }; if let Some(thread_id) = debugger.thread_id { let request = debugger.next(thread_id); if let Err(e) = block_on(request) { cx.editor.set_error(format!("Failed to continue: {:?}", e)); return; } resume_application(debugger); } else { cx.editor .set_error("Currently active thread is not stopped. Switch the thread.".into()); } } pub fn dap_variables(cx: &mut Context) { let debugger = match &mut cx.editor.debugger { Some(debugger) => debugger, None => return, }; if debugger.thread_id.is_none() { cx.editor .set_status("Cannot access variables while target is running".to_owned()); return; } let (frame, thread_id) = match (debugger.active_frame, debugger.thread_id) { (Some(frame), Some(thread_id)) => (frame, thread_id), _ => { cx.editor .set_status("Cannot find current stack frame to access variables".to_owned()); return; } }; let frame_id = debugger.stack_frames[&thread_id][frame].id; let scopes = match block_on(debugger.scopes(frame_id)) { Ok(s) => s, Err(e) => { cx.editor .set_error(format!("Failed to get scopes: {:?}", e)); return; } }; let mut variables = Vec::new(); for scope in scopes.iter() { let response = block_on(debugger.variables(scope.variables_reference)); if let Ok(vars) = response { variables.reserve(vars.len()); for var in vars { let prefix = match var.data_type { Some(data_type) => format!("{} ", data_type), None => "".to_owned(), }; variables.push(format!("{}{} = {}", prefix, var.name, var.value)); } } } let contents = Text::new(variables.join("\n")); let popup = Popup::new(contents); cx.push_layer(Box::new(popup)); } pub fn dap_terminate(cx: &mut Context) { let debugger = match &mut cx.editor.debugger { Some(debugger) => debugger, None => return, }; let request = debugger.disconnect(); if let Err(e) = block_on(request) { cx.editor .set_error(format!("Failed to disconnect: {:?}", e)); return; } cx.editor.debugger = None; } pub fn dap_enable_exceptions(cx: &mut Context) { let debugger = match &mut cx.editor.debugger { Some(debugger) => debugger, None => return, }; let filters = match &debugger.capabilities().exception_breakpoint_filters { Some(filters) => filters.clone(), None => return, }; if let Err(e) = block_on( debugger.set_exception_breakpoints(filters.iter().map(|f| f.filter.clone()).collect()), ) { cx.editor .set_error(format!("Failed to set up exception breakpoints: {:?}", e)); } } pub fn dap_disable_exceptions(cx: &mut Context) { let debugger = match &mut cx.editor.debugger { Some(debugger) => debugger, None => return, }; if let Err(e) = block_on(debugger.set_exception_breakpoints(vec![])) { cx.editor .set_error(format!("Failed to set up exception breakpoints: {:?}", e)); } } pub fn dap_edit_condition(cx: &mut Context) { if let Some((pos, mut bp)) = commands::cmd::get_breakpoint_at_current_line(cx.editor) { let callback = Box::pin(async move { let call: Callback = Box::new(move |_editor: &mut Editor, compositor: &mut Compositor| { let condition = bp.condition.clone(); let prompt = Prompt::new( "condition: ".into(), None, |_input: &str| Vec::new(), move |cx: &mut crate::compositor::Context, input: &str, event: PromptEvent| { if event != PromptEvent::Validate { return; } let (_, doc) = current!(cx.editor); let path = match doc.path() { Some(path) => path.to_path_buf(), None => { cx.editor.set_status( "Can't edit breakpoint: document has no path".to_owned(), ); return; } }; let breakpoints = cx.editor.breakpoints.entry(path.clone()).or_default(); breakpoints.remove(pos); bp.condition = match input { "" => None, input => Some(input.to_owned()), }; breakpoints.push(bp.clone()); if let Some(debugger) = &mut cx.editor.debugger { // TODO: handle capabilities correctly again, by filterin breakpoints when emitting // if breakpoint.condition.is_some() // && !debugger // .caps // .as_ref() // .unwrap() // .supports_conditional_breakpoints // .unwrap_or_default() // { // bail!( // "Can't edit breakpoint: debugger does not support conditional breakpoints" // ) // } // if breakpoint.log_message.is_some() // && !debugger // .caps // .as_ref() // .unwrap() // .supports_log_points // .unwrap_or_default() // { // bail!("Can't edit breakpoint: debugger does not support logpoints") // } let request = debugger.set_breakpoints(path, breakpoints.clone()); if let Err(e) = block_on(request) { cx.editor .set_status(format!("Failed to set breakpoints: {:?}", e)) } } }, condition, ); compositor.push(Box::new(prompt)); }); Ok(call) }); cx.jobs.callback(callback); } } pub fn dap_edit_log(cx: &mut Context) { if let Some((pos, mut bp)) = commands::cmd::get_breakpoint_at_current_line(cx.editor) { let callback = Box::pin(async move { let call: Callback = Box::new(move |_editor: &mut Editor, compositor: &mut Compositor| { let log_message = bp.log_message.clone(); let prompt = Prompt::new( "log message: ".into(), None, |_input: &str| Vec::new(), move |cx: &mut crate::compositor::Context, input: &str, event: PromptEvent| { if event != PromptEvent::Validate { return; } let (_, doc) = current!(cx.editor); let path = match doc.path() { Some(path) => path.to_path_buf(), None => { cx.editor.set_status( "Can't edit breakpoint: document has no path".to_owned(), ); return; } }; let breakpoints = cx.editor.breakpoints.entry(path.clone()).or_default(); breakpoints.remove(pos); bp.log_message = match input { "" => None, input => Some(input.to_owned()), }; breakpoints.push(bp.clone()); if let Some(debugger) = &mut cx.editor.debugger { // TODO: handle capabilities correctly again, by filterin breakpoints when emitting // if breakpoint.condition.is_some() // && !debugger // .caps // .as_ref() // .unwrap() // .supports_conditional_breakpoints // .unwrap_or_default() // { // bail!( // "Can't edit breakpoint: debugger does not support conditional breakpoints" // ) // } // if breakpoint.log_message.is_some() // && !debugger // .caps // .as_ref() // .unwrap() // .supports_log_points // .unwrap_or_default() // { // bail!("Can't edit breakpoint: debugger does not support logpoints") // } let request = debugger.set_breakpoints(path, breakpoints.clone()); if let Err(e) = block_on(request) { cx.editor .set_status(format!("Failed to set breakpoints: {:?}", e)) } } }, log_message, ); compositor.push(Box::new(prompt)); }); Ok(call) }); cx.jobs.callback(callback); } } pub fn dap_switch_thread(cx: &mut Context) { thread_picker(cx, |editor, thread| { block_on(select_thread_id(editor, thread.id, true)); }) } pub fn dap_switch_stack_frame(cx: &mut Context) { let debugger = match &mut cx.editor.debugger { Some(debugger) => debugger, None => return, }; let thread_id = match debugger.thread_id { Some(thread_id) => thread_id, None => { cx.editor .set_error("No thread is currently active".to_owned()); return; } }; let frames = debugger.stack_frames[&thread_id].clone(); let picker = FilePicker::new( frames, |frame| frame.name.clone().into(), // TODO: include thread_states in the label move |cx, frame, _action| { let debugger = match &mut cx.editor.debugger { Some(debugger) => debugger, None => return, }; // TODO: this should be simpler to find let pos = debugger.stack_frames[&thread_id] .iter() .position(|f| f.id == frame.id); debugger.active_frame = pos; let frame = debugger.stack_frames[&thread_id] .get(pos.unwrap_or(0)) .cloned(); if let Some(frame) = &frame { jump_to_stack_frame(cx.editor, frame); } }, move |_editor, frame| { frame .source .as_ref() .and_then(|source| source.path.clone()) .map(|path| { ( path, Some(( frame.line.saturating_sub(1), frame.end_line.unwrap_or(frame.line).saturating_sub(1), )), ) }) }, ); cx.push_layer(Box::new(picker)) }