1use anyhow::{Context as _, bail};
   2use collections::{FxHashMap, HashMap, HashSet};
   3use language::{LanguageName, LanguageRegistry};
   4use std::{
   5    borrow::Cow,
   6    path::{Path, PathBuf},
   7    sync::Arc,
   8    usize,
   9};
  10use tasks_ui::{TaskOverrides, TasksModal};
  11
  12use dap::{
  13    DapRegistry, DebugRequest, TelemetrySpawnLocation, adapters::DebugAdapterName, send_telemetry,
  14};
  15use editor::{Editor, EditorElement, EditorStyle};
  16use fuzzy::{StringMatch, StringMatchCandidate};
  17use gpui::{
  18    Action, App, AppContext, DismissEvent, Entity, EventEmitter, FocusHandle, Focusable,
  19    KeyContext, Render, Subscription, Task, TextStyle, WeakEntity,
  20};
  21use itertools::Itertools as _;
  22use picker::{Picker, PickerDelegate, highlighted_match_with_paths::HighlightedMatch};
  23use project::{DebugScenarioContext, Project, TaskContexts, TaskSourceKind, task_store::TaskStore};
  24use settings::Settings;
  25use task::{DebugScenario, RevealTarget, VariableName, ZedDebugConfig};
  26use theme::ThemeSettings;
  27use ui::{
  28    ActiveTheme, Button, ButtonCommon, ButtonSize, CheckboxWithLabel, Clickable, Color, Context,
  29    ContextMenu, Disableable, DropdownMenu, FluentBuilder, Icon, IconName, IconSize,
  30    IconWithIndicator, Indicator, InteractiveElement, IntoElement, KeyBinding, Label,
  31    LabelCommon as _, LabelSize, ListItem, ListItemSpacing, ParentElement, RenderOnce,
  32    SharedString, Styled, StyledExt, ToggleButton, ToggleState, Toggleable, Tooltip, Window, div,
  33    h_flex, relative, rems, v_flex,
  34};
  35use util::{ResultExt, rel_path::RelPath};
  36use workspace::{ModalView, Workspace, notifications::DetachAndPromptErr, pane};
  37
  38use crate::{attach_modal::AttachModal, debugger_panel::DebugPanel};
  39
  40pub(super) struct NewProcessModal {
  41    workspace: WeakEntity<Workspace>,
  42    debug_panel: WeakEntity<DebugPanel>,
  43    mode: NewProcessMode,
  44    debug_picker: Entity<Picker<DebugDelegate>>,
  45    attach_mode: Entity<AttachMode>,
  46    configure_mode: Entity<ConfigureMode>,
  47    task_mode: TaskMode,
  48    debugger: Option<DebugAdapterName>,
  49    _subscriptions: [Subscription; 3],
  50}
  51
  52fn suggested_label(request: &DebugRequest, debugger: &str) -> SharedString {
  53    match request {
  54        DebugRequest::Launch(config) => {
  55            let last_path_component = Path::new(&config.program)
  56                .file_name()
  57                .map(|name| name.to_string_lossy())
  58                .unwrap_or_else(|| Cow::Borrowed(&config.program));
  59
  60            format!("{} ({debugger})", last_path_component).into()
  61        }
  62        DebugRequest::Attach(config) => format!(
  63            "pid: {} ({debugger})",
  64            config.process_id.unwrap_or(u32::MAX)
  65        )
  66        .into(),
  67    }
  68}
  69
  70impl NewProcessModal {
  71    pub(super) fn show(
  72        workspace: &mut Workspace,
  73        window: &mut Window,
  74        mode: NewProcessMode,
  75        reveal_target: Option<RevealTarget>,
  76        cx: &mut Context<Workspace>,
  77    ) {
  78        let Some(debug_panel) = workspace.panel::<DebugPanel>(cx) else {
  79            return;
  80        };
  81        let task_store = workspace.project().read(cx).task_store().clone();
  82        let languages = workspace.app_state().languages.clone();
  83
  84        cx.spawn_in(window, async move |workspace, cx| {
  85            let task_contexts = workspace.update_in(cx, |workspace, window, cx| {
  86                // todo(debugger): get the buffer here (if the active item is an editor) and store it so we can pass it to start_session later
  87                tasks_ui::task_contexts(workspace, window, cx)
  88            })?;
  89            workspace.update_in(cx, |workspace, window, cx| {
  90                let workspace_handle = workspace.weak_handle();
  91                let project = workspace.project().clone();
  92                workspace.toggle_modal(window, cx, |window, cx| {
  93                    let attach_mode =
  94                        AttachMode::new(None, workspace_handle.clone(), project, window, cx);
  95
  96                    let debug_picker = cx.new(|cx| {
  97                        let delegate =
  98                            DebugDelegate::new(debug_panel.downgrade(), task_store.clone());
  99                        Picker::list(delegate, window, cx)
 100                            .modal(false)
 101                            .list_measure_all()
 102                    });
 103
 104                    let configure_mode = ConfigureMode::new(window, cx);
 105
 106                    let task_overrides = Some(TaskOverrides { reveal_target });
 107
 108                    let task_mode = TaskMode {
 109                        task_modal: cx.new(|cx| {
 110                            TasksModal::new(
 111                                task_store.clone(),
 112                                Arc::new(TaskContexts::default()),
 113                                task_overrides,
 114                                false,
 115                                workspace_handle.clone(),
 116                                window,
 117                                cx,
 118                            )
 119                        }),
 120                    };
 121
 122                    let _subscriptions = [
 123                        cx.subscribe(&debug_picker, |_, _, _, cx| {
 124                            cx.emit(DismissEvent);
 125                        }),
 126                        cx.subscribe(
 127                            &attach_mode.read(cx).attach_picker.clone(),
 128                            |_, _, _, cx| {
 129                                cx.emit(DismissEvent);
 130                            },
 131                        ),
 132                        cx.subscribe(&task_mode.task_modal, |_, _, _: &DismissEvent, cx| {
 133                            cx.emit(DismissEvent)
 134                        }),
 135                    ];
 136
 137                    cx.spawn_in(window, {
 138                        let debug_picker = debug_picker.downgrade();
 139                        let configure_mode = configure_mode.downgrade();
 140                        let task_modal = task_mode.task_modal.downgrade();
 141                        let workspace = workspace_handle.clone();
 142
 143                        async move |this, cx| {
 144                            let task_contexts = task_contexts.await;
 145                            let task_contexts = Arc::new(task_contexts);
 146                            let lsp_task_sources = task_contexts.lsp_task_sources.clone();
 147                            let task_position = task_contexts.latest_selection;
 148                            // Get LSP tasks and filter out based on language vs lsp preference
 149                            let (lsp_tasks, prefer_lsp) =
 150                                workspace.update(cx, |workspace, cx| {
 151                                    let lsp_tasks = editor::lsp_tasks(
 152                                        workspace.project().clone(),
 153                                        &lsp_task_sources,
 154                                        task_position,
 155                                        cx,
 156                                    );
 157                                    let prefer_lsp = workspace
 158                                        .active_item(cx)
 159                                        .and_then(|item| item.downcast::<Editor>())
 160                                        .map(|editor| {
 161                                            editor
 162                                                .read(cx)
 163                                                .buffer()
 164                                                .read(cx)
 165                                                .language_settings(cx)
 166                                                .tasks
 167                                                .prefer_lsp
 168                                        })
 169                                        .unwrap_or(false);
 170                                    (lsp_tasks, prefer_lsp)
 171                                })?;
 172
 173                            let lsp_tasks = lsp_tasks.await;
 174                            let add_current_language_tasks = !prefer_lsp || lsp_tasks.is_empty();
 175
 176                            let lsp_tasks = lsp_tasks
 177                                .into_iter()
 178                                .flat_map(|(kind, tasks_with_locations)| {
 179                                    tasks_with_locations
 180                                        .into_iter()
 181                                        .sorted_by_key(|(location, task)| {
 182                                            (location.is_none(), task.resolved_label.clone())
 183                                        })
 184                                        .map(move |(_, task)| (kind.clone(), task))
 185                                })
 186                                .collect::<Vec<_>>();
 187
 188                            let Some(task_inventory) = task_store
 189                                .update(cx, |task_store, _| task_store.task_inventory().cloned())?
 190                            else {
 191                                return Ok(());
 192                            };
 193
 194                            let (used_tasks, current_resolved_tasks) = task_inventory
 195                                .update(cx, |task_inventory, cx| {
 196                                    task_inventory
 197                                        .used_and_current_resolved_tasks(task_contexts.clone(), cx)
 198                                })?
 199                                .await;
 200
 201                            if let Ok(task) = debug_picker.update(cx, |picker, cx| {
 202                                picker.delegate.tasks_loaded(
 203                                    task_contexts.clone(),
 204                                    languages,
 205                                    lsp_tasks.clone(),
 206                                    current_resolved_tasks.clone(),
 207                                    add_current_language_tasks,
 208                                    cx,
 209                                )
 210                            }) {
 211                                task.await;
 212                                debug_picker
 213                                    .update_in(cx, |picker, window, cx| {
 214                                        picker.refresh(window, cx);
 215                                        cx.notify();
 216                                    })
 217                                    .ok();
 218                            }
 219
 220                            if let Some(active_cwd) = task_contexts
 221                                .active_context()
 222                                .and_then(|context| context.cwd.clone())
 223                            {
 224                                configure_mode
 225                                    .update_in(cx, |configure_mode, window, cx| {
 226                                        configure_mode.load(active_cwd, window, cx);
 227                                    })
 228                                    .ok();
 229                            }
 230
 231                            task_modal
 232                                .update_in(cx, |task_modal, window, cx| {
 233                                    task_modal.tasks_loaded(
 234                                        task_contexts,
 235                                        lsp_tasks,
 236                                        used_tasks,
 237                                        current_resolved_tasks,
 238                                        add_current_language_tasks,
 239                                        window,
 240                                        cx,
 241                                    );
 242                                })
 243                                .ok();
 244
 245                            this.update(cx, |_, cx| {
 246                                cx.notify();
 247                            })
 248                            .ok();
 249
 250                            anyhow::Ok(())
 251                        }
 252                    })
 253                    .detach();
 254
 255                    Self {
 256                        debug_picker,
 257                        attach_mode,
 258                        configure_mode,
 259                        task_mode,
 260                        debugger: None,
 261                        mode,
 262                        debug_panel: debug_panel.downgrade(),
 263                        workspace: workspace_handle,
 264                        _subscriptions,
 265                    }
 266                });
 267            })?;
 268
 269            anyhow::Ok(())
 270        })
 271        .detach();
 272    }
 273
 274    fn render_mode(&mut self, window: &mut Window, cx: &mut Context<Self>) -> impl ui::IntoElement {
 275        let dap_menu = self.adapter_drop_down_menu(window, cx);
 276        match self.mode {
 277            NewProcessMode::Task => self
 278                .task_mode
 279                .task_modal
 280                .read(cx)
 281                .picker
 282                .clone()
 283                .into_any_element(),
 284            NewProcessMode::Attach => self.attach_mode.update(cx, |this, cx| {
 285                this.clone().render(window, cx).into_any_element()
 286            }),
 287            NewProcessMode::Launch => self.configure_mode.update(cx, |this, cx| {
 288                this.clone().render(dap_menu, window, cx).into_any_element()
 289            }),
 290            NewProcessMode::Debug => v_flex()
 291                .w(rems(34.))
 292                .child(self.debug_picker.clone())
 293                .into_any_element(),
 294        }
 295    }
 296
 297    fn mode_focus_handle(&self, cx: &App) -> FocusHandle {
 298        match self.mode {
 299            NewProcessMode::Task => self.task_mode.task_modal.focus_handle(cx),
 300            NewProcessMode::Attach => self.attach_mode.read(cx).attach_picker.focus_handle(cx),
 301            NewProcessMode::Launch => self.configure_mode.read(cx).program.focus_handle(cx),
 302            NewProcessMode::Debug => self.debug_picker.focus_handle(cx),
 303        }
 304    }
 305
 306    fn debug_scenario(&self, debugger: &str, cx: &App) -> Task<Option<DebugScenario>> {
 307        let request = match self.mode {
 308            NewProcessMode::Launch => {
 309                DebugRequest::Launch(self.configure_mode.read(cx).debug_request(cx))
 310            }
 311            NewProcessMode::Attach => {
 312                DebugRequest::Attach(self.attach_mode.read(cx).debug_request())
 313            }
 314            _ => return Task::ready(None),
 315        };
 316        let label = suggested_label(&request, debugger);
 317
 318        let stop_on_entry = if let NewProcessMode::Launch = &self.mode {
 319            Some(self.configure_mode.read(cx).stop_on_entry.selected())
 320        } else {
 321            None
 322        };
 323
 324        let session_scenario = ZedDebugConfig {
 325            adapter: debugger.to_owned().into(),
 326            label,
 327            request,
 328            stop_on_entry,
 329        };
 330
 331        let adapter = cx
 332            .global::<DapRegistry>()
 333            .adapter(&session_scenario.adapter);
 334
 335        cx.spawn(async move |_| adapter?.config_from_zed_format(session_scenario).await.ok())
 336    }
 337
 338    fn start_new_session(&mut self, window: &mut Window, cx: &mut Context<Self>) {
 339        if self.debugger.as_ref().is_none() {
 340            return;
 341        }
 342
 343        if let NewProcessMode::Debug = &self.mode {
 344            self.debug_picker.update(cx, |picker, cx| {
 345                picker.delegate.confirm(false, window, cx);
 346            });
 347            return;
 348        }
 349
 350        if let NewProcessMode::Launch = &self.mode
 351            && self.configure_mode.read(cx).save_to_debug_json.selected()
 352        {
 353            self.save_debug_scenario(window, cx);
 354        }
 355
 356        let Some(debugger) = self.debugger.clone() else {
 357            return;
 358        };
 359
 360        let debug_panel = self.debug_panel.clone();
 361        let Some(task_contexts) = self.task_contexts(cx) else {
 362            return;
 363        };
 364
 365        let task_context = task_contexts.active_context().cloned().unwrap_or_default();
 366        let worktree_id = task_contexts.worktree();
 367        let mode = self.mode;
 368        cx.spawn_in(window, async move |this, cx| {
 369            let Some(config) = this
 370                .update(cx, |this, cx| this.debug_scenario(&debugger, cx))?
 371                .await
 372            else {
 373                bail!("debug config not found in mode: {mode}");
 374            };
 375
 376            debug_panel.update_in(cx, |debug_panel, window, cx| {
 377                send_telemetry(&config, TelemetrySpawnLocation::Custom, cx);
 378                debug_panel.start_session(config, task_context, None, worktree_id, window, cx)
 379            })?;
 380            this.update(cx, |_, cx| {
 381                cx.emit(DismissEvent);
 382            })
 383            .ok();
 384            anyhow::Ok(())
 385        })
 386        .detach_and_log_err(cx);
 387    }
 388
 389    fn update_attach_picker(
 390        attach: &Entity<AttachMode>,
 391        adapter: &DebugAdapterName,
 392        window: &mut Window,
 393        cx: &mut App,
 394    ) {
 395        attach.update(cx, |this, cx| {
 396            if adapter.0 != this.definition.adapter {
 397                this.definition.adapter = adapter.0.clone();
 398
 399                this.attach_picker.update(cx, |this, cx| {
 400                    this.picker.update(cx, |this, cx| {
 401                        this.delegate.definition.adapter = adapter.0.clone();
 402                        this.focus(window, cx);
 403                    })
 404                });
 405            }
 406
 407            cx.notify();
 408        })
 409    }
 410
 411    fn task_contexts(&self, cx: &App) -> Option<Arc<TaskContexts>> {
 412        self.debug_picker.read(cx).delegate.task_contexts.clone()
 413    }
 414
 415    pub fn save_debug_scenario(&mut self, window: &mut Window, cx: &mut Context<Self>) {
 416        let task_contexts = self.task_contexts(cx);
 417        let Some(adapter) = self.debugger.as_ref() else {
 418            return;
 419        };
 420        let scenario = self.debug_scenario(adapter, cx);
 421        cx.spawn_in(window, async move |this, cx| {
 422            let scenario = scenario.await.context("no scenario to save")?;
 423            let worktree_id = task_contexts
 424                .context("no task contexts")?
 425                .worktree()
 426                .context("no active worktree")?;
 427            this.update_in(cx, |this, window, cx| {
 428                this.debug_panel.update(cx, |panel, cx| {
 429                    panel.save_scenario(scenario, worktree_id, window, cx)
 430                })
 431            })??
 432            .await?;
 433            this.update_in(cx, |_, _, cx| {
 434                cx.emit(DismissEvent);
 435            })
 436        })
 437        .detach_and_prompt_err("Failed to edit debug.json", window, cx, |_, _, _| None);
 438    }
 439
 440    fn adapter_drop_down_menu(
 441        &mut self,
 442        window: &mut Window,
 443        cx: &mut Context<Self>,
 444    ) -> ui::DropdownMenu {
 445        let workspace = self.workspace.clone();
 446        let weak = cx.weak_entity();
 447        let active_buffer = self.task_contexts(cx).and_then(|tc| {
 448            tc.active_item_context
 449                .as_ref()
 450                .and_then(|aic| aic.1.as_ref().map(|l| l.buffer.clone()))
 451        });
 452
 453        let active_buffer_language = active_buffer
 454            .and_then(|buffer| buffer.read(cx).language())
 455            .cloned();
 456
 457        let mut available_adapters: Vec<_> = workspace
 458            .update(cx, |_, cx| DapRegistry::global(cx).enumerate_adapters())
 459            .unwrap_or_default();
 460        if let Some(language) = active_buffer_language {
 461            available_adapters.sort_by_key(|adapter| {
 462                language
 463                    .config()
 464                    .debuggers
 465                    .get_index_of(adapter.0.as_ref())
 466                    .unwrap_or(usize::MAX)
 467            });
 468            if self.debugger.is_none() {
 469                self.debugger = available_adapters.first().cloned();
 470            }
 471        }
 472
 473        let label = self
 474            .debugger
 475            .as_ref()
 476            .map(|d| d.0.clone())
 477            .unwrap_or_else(|| SELECT_DEBUGGER_LABEL.clone());
 478
 479        DropdownMenu::new(
 480            "dap-adapter-picker",
 481            label,
 482            ContextMenu::build(window, cx, move |mut menu, _, _| {
 483                let setter_for_name = |name: DebugAdapterName| {
 484                    let weak = weak.clone();
 485                    move |window: &mut Window, cx: &mut App| {
 486                        weak.update(cx, |this, cx| {
 487                            this.debugger = Some(name.clone());
 488                            cx.notify();
 489                            if let NewProcessMode::Attach = &this.mode {
 490                                Self::update_attach_picker(&this.attach_mode, &name, window, cx);
 491                            }
 492                        })
 493                        .ok();
 494                    }
 495                };
 496
 497                for adapter in available_adapters.into_iter() {
 498                    menu = menu.entry(adapter.0.clone(), None, setter_for_name(adapter.clone()));
 499                }
 500
 501                menu
 502            }),
 503        )
 504    }
 505}
 506
 507static SELECT_DEBUGGER_LABEL: SharedString = SharedString::new_static("Select Debugger");
 508
 509#[derive(Clone, Copy)]
 510pub(crate) enum NewProcessMode {
 511    Task,
 512    Launch,
 513    Attach,
 514    Debug,
 515}
 516
 517impl std::fmt::Display for NewProcessMode {
 518    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
 519        let mode = match self {
 520            NewProcessMode::Task => "Run",
 521            NewProcessMode::Debug => "Debug",
 522            NewProcessMode::Attach => "Attach",
 523            NewProcessMode::Launch => "Launch",
 524        };
 525
 526        write!(f, "{}", mode)
 527    }
 528}
 529
 530impl Focusable for NewProcessMode {
 531    fn focus_handle(&self, cx: &App) -> FocusHandle {
 532        cx.focus_handle()
 533    }
 534}
 535
 536fn render_editor(editor: &Entity<Editor>, window: &mut Window, cx: &App) -> impl IntoElement {
 537    let settings = ThemeSettings::get_global(cx);
 538    let theme = cx.theme();
 539
 540    let text_style = TextStyle {
 541        color: cx.theme().colors().text,
 542        font_family: settings.buffer_font.family.clone(),
 543        font_features: settings.buffer_font.features.clone(),
 544        font_size: settings.buffer_font_size(cx).into(),
 545        font_weight: settings.buffer_font.weight,
 546        line_height: relative(settings.buffer_line_height.value()),
 547        background_color: Some(theme.colors().editor_background),
 548        ..Default::default()
 549    };
 550
 551    let element = EditorElement::new(
 552        editor,
 553        EditorStyle {
 554            background: theme.colors().editor_background,
 555            local_player: theme.players().local(),
 556            text: text_style,
 557            ..Default::default()
 558        },
 559    );
 560
 561    div()
 562        .rounded_md()
 563        .p_1()
 564        .border_1()
 565        .border_color(theme.colors().border_variant)
 566        .when(
 567            editor.focus_handle(cx).contains_focused(window, cx),
 568            |this| this.border_color(theme.colors().border_focused),
 569        )
 570        .child(element)
 571        .bg(theme.colors().editor_background)
 572}
 573
 574impl Render for NewProcessModal {
 575    fn render(
 576        &mut self,
 577        window: &mut ui::Window,
 578        cx: &mut ui::Context<Self>,
 579    ) -> impl ui::IntoElement {
 580        v_flex()
 581            .key_context({
 582                let mut key_context = KeyContext::new_with_defaults();
 583                key_context.add("Pane");
 584                key_context.add("RunModal");
 585                key_context
 586            })
 587            .size_full()
 588            .w(rems(34.))
 589            .elevation_3(cx)
 590            .overflow_hidden()
 591            .on_action(cx.listener(|_, _: &menu::Cancel, _, cx| {
 592                cx.emit(DismissEvent);
 593            }))
 594            .on_action(cx.listener(|this, _: &pane::ActivateNextItem, window, cx| {
 595                this.mode = match this.mode {
 596                    NewProcessMode::Task => NewProcessMode::Debug,
 597                    NewProcessMode::Debug => NewProcessMode::Attach,
 598                    NewProcessMode::Attach => NewProcessMode::Launch,
 599                    NewProcessMode::Launch => NewProcessMode::Task,
 600                };
 601
 602                this.mode_focus_handle(cx).focus(window);
 603            }))
 604            .on_action(
 605                cx.listener(|this, _: &pane::ActivatePreviousItem, window, cx| {
 606                    this.mode = match this.mode {
 607                        NewProcessMode::Task => NewProcessMode::Launch,
 608                        NewProcessMode::Debug => NewProcessMode::Task,
 609                        NewProcessMode::Attach => NewProcessMode::Debug,
 610                        NewProcessMode::Launch => NewProcessMode::Attach,
 611                    };
 612
 613                    this.mode_focus_handle(cx).focus(window);
 614                }),
 615            )
 616            .child(
 617                h_flex()
 618                    .p_2()
 619                    .w_full()
 620                    .border_b_1()
 621                    .border_color(cx.theme().colors().border_variant)
 622                    .child(
 623                        ToggleButton::new(
 624                            "debugger-session-ui-tasks-button",
 625                            NewProcessMode::Task.to_string(),
 626                        )
 627                        .size(ButtonSize::Default)
 628                        .toggle_state(matches!(self.mode, NewProcessMode::Task))
 629                        .style(ui::ButtonStyle::Subtle)
 630                        .on_click(cx.listener(|this, _, window, cx| {
 631                            this.mode = NewProcessMode::Task;
 632                            this.mode_focus_handle(cx).focus(window);
 633                            cx.notify();
 634                        }))
 635                        .tooltip(Tooltip::text("Run predefined task"))
 636                        .first(),
 637                    )
 638                    .child(
 639                        ToggleButton::new(
 640                            "debugger-session-ui-launch-button",
 641                            NewProcessMode::Debug.to_string(),
 642                        )
 643                        .size(ButtonSize::Default)
 644                        .style(ui::ButtonStyle::Subtle)
 645                        .toggle_state(matches!(self.mode, NewProcessMode::Debug))
 646                        .on_click(cx.listener(|this, _, window, cx| {
 647                            this.mode = NewProcessMode::Debug;
 648                            this.mode_focus_handle(cx).focus(window);
 649                            cx.notify();
 650                        }))
 651                        .tooltip(Tooltip::text("Start a predefined debug scenario"))
 652                        .middle(),
 653                    )
 654                    .child(
 655                        ToggleButton::new(
 656                            "debugger-session-ui-attach-button",
 657                            NewProcessMode::Attach.to_string(),
 658                        )
 659                        .size(ButtonSize::Default)
 660                        .toggle_state(matches!(self.mode, NewProcessMode::Attach))
 661                        .style(ui::ButtonStyle::Subtle)
 662                        .on_click(cx.listener(|this, _, window, cx| {
 663                            this.mode = NewProcessMode::Attach;
 664
 665                            if let Some(debugger) = this.debugger.as_ref() {
 666                                Self::update_attach_picker(&this.attach_mode, debugger, window, cx);
 667                            }
 668                            this.mode_focus_handle(cx).focus(window);
 669                            cx.notify();
 670                        }))
 671                        .tooltip(Tooltip::text("Attach the debugger to a running process"))
 672                        .middle(),
 673                    )
 674                    .child(
 675                        ToggleButton::new(
 676                            "debugger-session-ui-custom-button",
 677                            NewProcessMode::Launch.to_string(),
 678                        )
 679                        .size(ButtonSize::Default)
 680                        .toggle_state(matches!(self.mode, NewProcessMode::Launch))
 681                        .style(ui::ButtonStyle::Subtle)
 682                        .on_click(cx.listener(|this, _, window, cx| {
 683                            this.mode = NewProcessMode::Launch;
 684                            this.mode_focus_handle(cx).focus(window);
 685                            cx.notify();
 686                        }))
 687                        .tooltip(Tooltip::text("Launch a new process with a debugger"))
 688                        .last(),
 689                    ),
 690            )
 691            .child(v_flex().child(self.render_mode(window, cx)))
 692            .map(|el| {
 693                let container = h_flex()
 694                    .w_full()
 695                    .p_1p5()
 696                    .gap_2()
 697                    .justify_between()
 698                    .border_t_1()
 699                    .border_color(cx.theme().colors().border_variant);
 700                match self.mode {
 701                    NewProcessMode::Launch => el.child(
 702                        container
 703                            .child(
 704                                h_flex().child(
 705                                    Button::new("edit-custom-debug", "Edit in debug.json")
 706                                        .on_click(cx.listener(|this, _, window, cx| {
 707                                            this.save_debug_scenario(window, cx);
 708                                        }))
 709                                        .disabled(
 710                                            self.debugger.is_none()
 711                                                || self
 712                                                    .configure_mode
 713                                                    .read(cx)
 714                                                    .program
 715                                                    .read(cx)
 716                                                    .is_empty(cx),
 717                                        ),
 718                                ),
 719                            )
 720                            .child(
 721                                Button::new("debugger-spawn", "Start")
 722                                    .on_click(cx.listener(|this, _, window, cx| {
 723                                        this.start_new_session(window, cx)
 724                                    }))
 725                                    .disabled(
 726                                        self.debugger.is_none()
 727                                            || self
 728                                                .configure_mode
 729                                                .read(cx)
 730                                                .program
 731                                                .read(cx)
 732                                                .is_empty(cx),
 733                                    ),
 734                            ),
 735                    ),
 736                    NewProcessMode::Attach => el.child({
 737                        let disabled = self.debugger.is_none()
 738                            || self
 739                                .attach_mode
 740                                .read(cx)
 741                                .attach_picker
 742                                .read(cx)
 743                                .picker
 744                                .read(cx)
 745                                .delegate
 746                                .match_count()
 747                                == 0;
 748                        let secondary_action = menu::SecondaryConfirm.boxed_clone();
 749                        container
 750                            .child(div().child({
 751                                Button::new("edit-attach-task", "Edit in debug.json")
 752                                    .label_size(LabelSize::Small)
 753                                    .key_binding(KeyBinding::for_action(&*secondary_action, cx))
 754                                    .on_click(move |_, window, cx| {
 755                                        window.dispatch_action(secondary_action.boxed_clone(), cx)
 756                                    })
 757                                    .disabled(disabled)
 758                            }))
 759                            .child(
 760                                h_flex()
 761                                    .child(div().child(self.adapter_drop_down_menu(window, cx))),
 762                            )
 763                    }),
 764                    NewProcessMode::Debug => el,
 765                    NewProcessMode::Task => el,
 766                }
 767            })
 768    }
 769}
 770
 771impl EventEmitter<DismissEvent> for NewProcessModal {}
 772impl Focusable for NewProcessModal {
 773    fn focus_handle(&self, cx: &ui::App) -> gpui::FocusHandle {
 774        self.mode_focus_handle(cx)
 775    }
 776}
 777
 778impl ModalView for NewProcessModal {}
 779
 780impl RenderOnce for AttachMode {
 781    fn render(self, _: &mut Window, cx: &mut App) -> impl IntoElement {
 782        v_flex()
 783            .w_full()
 784            .track_focus(&self.attach_picker.focus_handle(cx))
 785            .child(self.attach_picker)
 786    }
 787}
 788
 789#[derive(Clone)]
 790pub(super) struct ConfigureMode {
 791    program: Entity<Editor>,
 792    cwd: Entity<Editor>,
 793    stop_on_entry: ToggleState,
 794    save_to_debug_json: ToggleState,
 795}
 796
 797impl ConfigureMode {
 798    pub(super) fn new(window: &mut Window, cx: &mut App) -> Entity<Self> {
 799        let program = cx.new(|cx| Editor::single_line(window, cx));
 800        program.update(cx, |this, cx| {
 801            this.set_placeholder_text("ENV=Zed ~/bin/program --option", window, cx);
 802        });
 803
 804        let cwd = cx.new(|cx| Editor::single_line(window, cx));
 805        cwd.update(cx, |this, cx| {
 806            this.set_placeholder_text("Ex: $ZED_WORKTREE_ROOT", window, cx);
 807        });
 808
 809        cx.new(|_| Self {
 810            program,
 811            cwd,
 812            stop_on_entry: ToggleState::Unselected,
 813            save_to_debug_json: ToggleState::Unselected,
 814        })
 815    }
 816
 817    fn load(&mut self, cwd: PathBuf, window: &mut Window, cx: &mut App) {
 818        self.cwd.update(cx, |editor, cx| {
 819            if editor.is_empty(cx) {
 820                editor.set_text(cwd.to_string_lossy(), window, cx);
 821            }
 822        });
 823    }
 824
 825    pub(super) fn debug_request(&self, cx: &App) -> task::LaunchRequest {
 826        let cwd_text = self.cwd.read(cx).text(cx);
 827        let cwd = if cwd_text.is_empty() {
 828            None
 829        } else {
 830            Some(PathBuf::from(cwd_text))
 831        };
 832
 833        if cfg!(windows) {
 834            return task::LaunchRequest {
 835                program: self.program.read(cx).text(cx),
 836                cwd,
 837                args: Default::default(),
 838                env: Default::default(),
 839            };
 840        }
 841        let command = self.program.read(cx).text(cx);
 842        let mut args = shlex::split(&command).into_iter().flatten().peekable();
 843        let mut env = FxHashMap::default();
 844        while args.peek().is_some_and(|arg| arg.contains('=')) {
 845            let arg = args.next().unwrap();
 846            let (lhs, rhs) = arg.split_once('=').unwrap();
 847            env.insert(lhs.to_string(), rhs.to_string());
 848        }
 849
 850        let program = if let Some(program) = args.next() {
 851            program
 852        } else {
 853            env = FxHashMap::default();
 854            command
 855        };
 856
 857        let args = args.collect::<Vec<_>>();
 858
 859        task::LaunchRequest {
 860            program,
 861            cwd,
 862            args,
 863            env,
 864        }
 865    }
 866
 867    fn render(
 868        &mut self,
 869        adapter_menu: DropdownMenu,
 870        window: &mut Window,
 871        cx: &mut ui::Context<Self>,
 872    ) -> impl IntoElement {
 873        v_flex()
 874            .p_2()
 875            .w_full()
 876            .gap_2()
 877            .track_focus(&self.program.focus_handle(cx))
 878            .child(
 879                h_flex()
 880                    .gap_2()
 881                    .child(
 882                        Label::new("Debugger")
 883                            .size(LabelSize::Small)
 884                            .color(Color::Muted),
 885                    )
 886                    .child(adapter_menu),
 887            )
 888            .child(
 889                v_flex()
 890                    .gap_0p5()
 891                    .child(
 892                        Label::new("Program")
 893                            .size(LabelSize::Small)
 894                            .color(Color::Muted),
 895                    )
 896                    .child(render_editor(&self.program, window, cx)),
 897            )
 898            .child(
 899                v_flex()
 900                    .gap_0p5()
 901                    .child(
 902                        Label::new("Working Directory")
 903                            .size(LabelSize::Small)
 904                            .color(Color::Muted),
 905                    )
 906                    .child(render_editor(&self.cwd, window, cx)),
 907            )
 908            .child(
 909                CheckboxWithLabel::new(
 910                    "debugger-stop-on-entry",
 911                    Label::new("Stop on Entry")
 912                        .size(LabelSize::Small)
 913                        .color(Color::Muted),
 914                    self.stop_on_entry,
 915                    {
 916                        let this = cx.weak_entity();
 917                        move |state, _, cx| {
 918                            this.update(cx, |this, _| {
 919                                this.stop_on_entry = *state;
 920                            })
 921                            .ok();
 922                        }
 923                    },
 924                )
 925                .checkbox_position(ui::IconPosition::End),
 926            )
 927    }
 928}
 929
 930#[derive(Clone)]
 931pub(super) struct AttachMode {
 932    pub(super) definition: ZedDebugConfig,
 933    pub(super) attach_picker: Entity<AttachModal>,
 934}
 935
 936impl AttachMode {
 937    pub(super) fn new(
 938        debugger: Option<DebugAdapterName>,
 939        workspace: WeakEntity<Workspace>,
 940        project: Entity<Project>,
 941        window: &mut Window,
 942        cx: &mut Context<NewProcessModal>,
 943    ) -> Entity<Self> {
 944        let definition = ZedDebugConfig {
 945            adapter: debugger.unwrap_or(DebugAdapterName("".into())).0,
 946            label: "Attach New Session Setup".into(),
 947            request: dap::DebugRequest::Attach(task::AttachRequest { process_id: None }),
 948            stop_on_entry: Some(false),
 949        };
 950        let attach_picker = cx.new(|cx| {
 951            let modal = AttachModal::new(definition.clone(), workspace, project, false, window, cx);
 952            window.focus(&modal.focus_handle(cx));
 953
 954            modal
 955        });
 956
 957        cx.new(|_| Self {
 958            definition,
 959            attach_picker,
 960        })
 961    }
 962    pub(super) fn debug_request(&self) -> task::AttachRequest {
 963        task::AttachRequest { process_id: None }
 964    }
 965}
 966
 967#[derive(Clone)]
 968pub(super) struct TaskMode {
 969    pub(super) task_modal: Entity<TasksModal>,
 970}
 971
 972pub(super) struct DebugDelegate {
 973    task_store: Entity<TaskStore>,
 974    candidates: Vec<(
 975        Option<TaskSourceKind>,
 976        Option<LanguageName>,
 977        DebugScenario,
 978        Option<DebugScenarioContext>,
 979    )>,
 980    selected_index: usize,
 981    matches: Vec<StringMatch>,
 982    prompt: String,
 983    debug_panel: WeakEntity<DebugPanel>,
 984    task_contexts: Option<Arc<TaskContexts>>,
 985    divider_index: Option<usize>,
 986    last_used_candidate_index: Option<usize>,
 987}
 988
 989impl DebugDelegate {
 990    pub(super) fn new(debug_panel: WeakEntity<DebugPanel>, task_store: Entity<TaskStore>) -> Self {
 991        Self {
 992            task_store,
 993            candidates: Vec::default(),
 994            selected_index: 0,
 995            matches: Vec::new(),
 996            prompt: String::new(),
 997            debug_panel,
 998            task_contexts: None,
 999            divider_index: None,
1000            last_used_candidate_index: None,
1001        }
1002    }
1003
1004    fn get_task_subtitle(
1005        &self,
1006        task_kind: &Option<TaskSourceKind>,
1007        context: &Option<DebugScenarioContext>,
1008        cx: &mut App,
1009    ) -> Option<String> {
1010        match task_kind {
1011            Some(TaskSourceKind::Worktree {
1012                id: worktree_id,
1013                directory_in_worktree,
1014                ..
1015            }) => self
1016                .debug_panel
1017                .update(cx, |debug_panel, cx| {
1018                    let project = debug_panel.project().read(cx);
1019                    let worktrees: Vec<_> = project.visible_worktrees(cx).collect();
1020
1021                    let mut path = if worktrees.len() > 1
1022                        && let Some(worktree) = project.worktree_for_id(*worktree_id, cx)
1023                    {
1024                        worktree
1025                            .read(cx)
1026                            .root_name()
1027                            .join(directory_in_worktree)
1028                            .to_rel_path_buf()
1029                    } else {
1030                        directory_in_worktree.to_rel_path_buf()
1031                    };
1032
1033                    match path.components().next_back() {
1034                        Some(".zed") => {
1035                            path.push(RelPath::unix("debug.json").unwrap());
1036                        }
1037                        Some(".vscode") => {
1038                            path.push(RelPath::unix("launch.json").unwrap());
1039                        }
1040                        _ => {}
1041                    }
1042                    path.display(project.path_style(cx)).to_string()
1043                })
1044                .ok(),
1045            Some(TaskSourceKind::AbsPath { abs_path, .. }) => {
1046                Some(abs_path.to_string_lossy().into_owned())
1047            }
1048            Some(TaskSourceKind::Lsp { language_name, .. }) => {
1049                Some(format!("LSP: {language_name}"))
1050            }
1051            Some(TaskSourceKind::Language { name }) => Some(format!("Lang: {name}")),
1052            _ => context.clone().and_then(|ctx| {
1053                ctx.task_context
1054                    .task_variables
1055                    .get(&VariableName::RelativeFile)
1056                    .map(|f| format!("in {f}"))
1057                    .or_else(|| {
1058                        ctx.task_context
1059                            .task_variables
1060                            .get(&VariableName::Dirname)
1061                            .map(|d| format!("in {d}/"))
1062                    })
1063            }),
1064        }
1065    }
1066
1067    fn get_scenario_language(
1068        languages: &Arc<LanguageRegistry>,
1069        dap_registry: &DapRegistry,
1070        scenario: DebugScenario,
1071    ) -> (Option<LanguageName>, DebugScenario) {
1072        let language_names = languages.language_names();
1073        let language_name = dap_registry.adapter_language(&scenario.adapter);
1074
1075        let language_name = language_name.or_else(|| {
1076            scenario.label.split_whitespace().find_map(|word| {
1077                language_names
1078                    .iter()
1079                    .find(|name| name.as_ref().eq_ignore_ascii_case(word))
1080                    .cloned()
1081            })
1082        });
1083
1084        (language_name, scenario)
1085    }
1086
1087    pub fn tasks_loaded(
1088        &mut self,
1089        task_contexts: Arc<TaskContexts>,
1090        languages: Arc<LanguageRegistry>,
1091        lsp_tasks: Vec<(TaskSourceKind, task::ResolvedTask)>,
1092        current_resolved_tasks: Vec<(TaskSourceKind, task::ResolvedTask)>,
1093        add_current_language_tasks: bool,
1094        cx: &mut Context<Picker<Self>>,
1095    ) -> Task<()> {
1096        self.task_contexts = Some(task_contexts.clone());
1097        let task = self.task_store.update(cx, |task_store, cx| {
1098            task_store.task_inventory().map(|inventory| {
1099                inventory.update(cx, |inventory, cx| {
1100                    inventory.list_debug_scenarios(
1101                        &task_contexts,
1102                        lsp_tasks,
1103                        current_resolved_tasks,
1104                        add_current_language_tasks,
1105                        cx,
1106                    )
1107                })
1108            })
1109        });
1110
1111        let valid_adapters: HashSet<_> = cx.global::<DapRegistry>().enumerate_adapters();
1112
1113        cx.spawn(async move |this, cx| {
1114            let (recent, scenarios) = if let Some(task) = task {
1115                task.await
1116            } else {
1117                (Vec::new(), Vec::new())
1118            };
1119
1120            this.update(cx, |this, cx| {
1121                if !recent.is_empty() {
1122                    this.delegate.last_used_candidate_index = Some(recent.len() - 1);
1123                }
1124
1125                let dap_registry = cx.global::<DapRegistry>();
1126                let hide_vscode = scenarios.iter().any(|(kind, _)| match kind {
1127                    TaskSourceKind::Worktree {
1128                        id: _,
1129                        directory_in_worktree: dir,
1130                        id_base: _,
1131                    } => dir.ends_with(RelPath::unix(".zed").unwrap()),
1132                    _ => false,
1133                });
1134
1135                this.delegate.candidates = recent
1136                    .into_iter()
1137                    .map(|(scenario, context)| {
1138                        let (language_name, scenario) =
1139                            Self::get_scenario_language(&languages, dap_registry, scenario);
1140                        (None, language_name, scenario, Some(context))
1141                    })
1142                    .chain(
1143                        scenarios
1144                            .into_iter()
1145                            .filter(|(kind, _)| match kind {
1146                                TaskSourceKind::Worktree {
1147                                    id: _,
1148                                    directory_in_worktree: dir,
1149                                    id_base: _,
1150                                } => {
1151                                    !(hide_vscode
1152                                        && dir.ends_with(RelPath::unix(".vscode").unwrap()))
1153                                }
1154                                _ => true,
1155                            })
1156                            .filter(|(_, scenario)| valid_adapters.contains(&scenario.adapter))
1157                            .map(|(kind, scenario)| {
1158                                let (language_name, scenario) =
1159                                    Self::get_scenario_language(&languages, dap_registry, scenario);
1160                                (Some(kind), language_name, scenario, None)
1161                            }),
1162                    )
1163                    .collect();
1164            })
1165            .ok();
1166        })
1167    }
1168}
1169
1170impl PickerDelegate for DebugDelegate {
1171    type ListItem = ui::ListItem;
1172
1173    fn match_count(&self) -> usize {
1174        self.matches.len()
1175    }
1176
1177    fn selected_index(&self) -> usize {
1178        self.selected_index
1179    }
1180
1181    fn set_selected_index(
1182        &mut self,
1183        ix: usize,
1184        _window: &mut Window,
1185        _cx: &mut Context<picker::Picker<Self>>,
1186    ) {
1187        self.selected_index = ix;
1188    }
1189
1190    fn placeholder_text(&self, _window: &mut Window, _cx: &mut App) -> std::sync::Arc<str> {
1191        "Find a debug task, or debug a command.".into()
1192    }
1193
1194    fn update_matches(
1195        &mut self,
1196        query: String,
1197        window: &mut Window,
1198        cx: &mut Context<picker::Picker<Self>>,
1199    ) -> gpui::Task<()> {
1200        let candidates = self.candidates.clone();
1201
1202        cx.spawn_in(window, async move |picker, cx| {
1203            let candidates: Vec<_> = candidates
1204                .into_iter()
1205                .enumerate()
1206                .map(|(index, (_, _, candidate, _))| {
1207                    StringMatchCandidate::new(index, candidate.label.as_ref())
1208                })
1209                .collect();
1210
1211            let matches = fuzzy::match_strings(
1212                &candidates,
1213                &query,
1214                true,
1215                true,
1216                1000,
1217                &Default::default(),
1218                cx.background_executor().clone(),
1219            )
1220            .await;
1221
1222            picker
1223                .update(cx, |picker, _| {
1224                    let delegate = &mut picker.delegate;
1225
1226                    delegate.matches = matches;
1227                    delegate.prompt = query;
1228
1229                    delegate.divider_index = delegate.last_used_candidate_index.and_then(|index| {
1230                        let index = delegate
1231                            .matches
1232                            .partition_point(|matching_task| matching_task.candidate_id <= index);
1233                        Some(index).and_then(|index| (index != 0).then(|| index - 1))
1234                    });
1235
1236                    if delegate.matches.is_empty() {
1237                        delegate.selected_index = 0;
1238                    } else {
1239                        delegate.selected_index =
1240                            delegate.selected_index.min(delegate.matches.len() - 1);
1241                    }
1242                })
1243                .log_err();
1244        })
1245    }
1246
1247    fn separators_after_indices(&self) -> Vec<usize> {
1248        if let Some(i) = self.divider_index {
1249            vec![i]
1250        } else {
1251            Vec::new()
1252        }
1253    }
1254
1255    fn confirm_input(&mut self, _: bool, window: &mut Window, cx: &mut Context<Picker<Self>>) {
1256        let text = self.prompt.clone();
1257        let (task_context, worktree_id) = self
1258            .task_contexts
1259            .as_ref()
1260            .and_then(|task_contexts| {
1261                Some((
1262                    task_contexts.active_context().cloned()?,
1263                    task_contexts.worktree(),
1264                ))
1265            })
1266            .unwrap_or_default();
1267
1268        let mut args = shlex::split(&text).into_iter().flatten().peekable();
1269        let mut env = HashMap::default();
1270        while args.peek().is_some_and(|arg| arg.contains('=')) {
1271            let arg = args.next().unwrap();
1272            let (lhs, rhs) = arg.split_once('=').unwrap();
1273            env.insert(lhs.to_string(), rhs.to_string());
1274        }
1275
1276        let program = if let Some(program) = args.next() {
1277            program
1278        } else {
1279            env = HashMap::default();
1280            text
1281        };
1282
1283        let args = args.collect::<Vec<_>>();
1284        let task = task::TaskTemplate {
1285            label: "one-off".to_owned(), // TODO: rename using command as label
1286            env,
1287            command: program,
1288            args,
1289            ..Default::default()
1290        };
1291
1292        let Some(location) = self
1293            .task_contexts
1294            .as_ref()
1295            .and_then(|cx| cx.location().cloned())
1296        else {
1297            return;
1298        };
1299        let file = location.buffer.read(cx).file();
1300        let language = location.buffer.read(cx).language();
1301        let language_name = language.as_ref().map(|l| l.name());
1302        let Some(adapter): Option<DebugAdapterName> =
1303            language::language_settings::language_settings(language_name, file, cx)
1304                .debuggers
1305                .first()
1306                .map(SharedString::from)
1307                .map(Into::into)
1308                .or_else(|| {
1309                    language.and_then(|l| {
1310                        l.config()
1311                            .debuggers
1312                            .first()
1313                            .map(SharedString::from)
1314                            .map(Into::into)
1315                    })
1316                })
1317        else {
1318            return;
1319        };
1320        let locators = cx.global::<DapRegistry>().locators();
1321        cx.spawn_in(window, async move |this, cx| {
1322            let Some(debug_scenario) = cx
1323                .background_spawn(async move {
1324                    for locator in locators {
1325                        if let Some(scenario) =
1326                            // TODO: use a more informative label than "one-off"
1327                            locator
1328                                .1
1329                                .create_scenario(&task, &task.label, &adapter)
1330                                .await
1331                        {
1332                            return Some(scenario);
1333                        }
1334                    }
1335                    None
1336                })
1337                .await
1338            else {
1339                return;
1340            };
1341
1342            this.update_in(cx, |this, window, cx| {
1343                send_telemetry(&debug_scenario, TelemetrySpawnLocation::ScenarioList, cx);
1344                this.delegate
1345                    .debug_panel
1346                    .update(cx, |panel, cx| {
1347                        panel.start_session(
1348                            debug_scenario,
1349                            task_context,
1350                            None,
1351                            worktree_id,
1352                            window,
1353                            cx,
1354                        );
1355                    })
1356                    .ok();
1357                cx.emit(DismissEvent);
1358            })
1359            .ok();
1360        })
1361        .detach();
1362    }
1363
1364    fn confirm(
1365        &mut self,
1366        secondary: bool,
1367        window: &mut Window,
1368        cx: &mut Context<picker::Picker<Self>>,
1369    ) {
1370        let debug_scenario = self
1371            .matches
1372            .get(self.selected_index())
1373            .and_then(|match_candidate| self.candidates.get(match_candidate.candidate_id).cloned());
1374
1375        let Some((kind, _, debug_scenario, context)) = debug_scenario else {
1376            return;
1377        };
1378
1379        let context = context.unwrap_or_else(|| {
1380            self.task_contexts
1381                .as_ref()
1382                .and_then(|task_contexts| {
1383                    Some(DebugScenarioContext {
1384                        task_context: task_contexts.active_context().cloned()?,
1385                        active_buffer: None,
1386                        worktree_id: task_contexts.worktree(),
1387                    })
1388                })
1389                .unwrap_or_default()
1390        });
1391        let DebugScenarioContext {
1392            task_context,
1393            active_buffer: _,
1394            worktree_id,
1395        } = context;
1396
1397        if secondary {
1398            let Some(kind) = kind else { return };
1399            let Some(id) = worktree_id else { return };
1400            let debug_panel = self.debug_panel.clone();
1401            cx.spawn_in(window, async move |_, cx| {
1402                debug_panel
1403                    .update_in(cx, |debug_panel, window, cx| {
1404                        debug_panel.go_to_scenario_definition(kind, debug_scenario, id, window, cx)
1405                    })?
1406                    .await?;
1407                anyhow::Ok(())
1408            })
1409            .detach();
1410        } else {
1411            send_telemetry(&debug_scenario, TelemetrySpawnLocation::ScenarioList, cx);
1412            self.debug_panel
1413                .update(cx, |panel, cx| {
1414                    panel.start_session(
1415                        debug_scenario,
1416                        task_context,
1417                        None,
1418                        worktree_id,
1419                        window,
1420                        cx,
1421                    );
1422                })
1423                .ok();
1424        }
1425
1426        cx.emit(DismissEvent);
1427    }
1428
1429    fn dismissed(&mut self, _: &mut Window, cx: &mut Context<picker::Picker<Self>>) {
1430        cx.emit(DismissEvent);
1431    }
1432
1433    fn render_footer(
1434        &self,
1435        window: &mut Window,
1436        cx: &mut Context<Picker<Self>>,
1437    ) -> Option<ui::AnyElement> {
1438        let current_modifiers = window.modifiers();
1439        let footer = h_flex()
1440            .w_full()
1441            .p_1p5()
1442            .justify_between()
1443            .border_t_1()
1444            .border_color(cx.theme().colors().border_variant)
1445            .child({
1446                let action = menu::SecondaryConfirm.boxed_clone();
1447                if self.matches.is_empty() {
1448                    Button::new("edit-debug-json", "Edit debug.json")
1449                        .label_size(LabelSize::Small)
1450                        .on_click(cx.listener(|_picker, _, window, cx| {
1451                            window.dispatch_action(
1452                                zed_actions::OpenProjectDebugTasks.boxed_clone(),
1453                                cx,
1454                            );
1455                            cx.emit(DismissEvent);
1456                        }))
1457                } else {
1458                    Button::new("edit-debug-task", "Edit in debug.json")
1459                        .label_size(LabelSize::Small)
1460                        .key_binding(KeyBinding::for_action(&*action, cx))
1461                        .on_click(move |_, window, cx| {
1462                            window.dispatch_action(action.boxed_clone(), cx)
1463                        })
1464                }
1465            })
1466            .map(|this| {
1467                if (current_modifiers.alt || self.matches.is_empty()) && !self.prompt.is_empty() {
1468                    let action = picker::ConfirmInput { secondary: false }.boxed_clone();
1469                    this.child({
1470                        Button::new("launch-custom", "Launch Custom")
1471                            .key_binding(KeyBinding::for_action(&*action, cx))
1472                            .on_click(move |_, window, cx| {
1473                                window.dispatch_action(action.boxed_clone(), cx)
1474                            })
1475                    })
1476                } else {
1477                    this.child({
1478                        let is_recent_selected = self.divider_index >= Some(self.selected_index);
1479                        let run_entry_label = if is_recent_selected { "Rerun" } else { "Spawn" };
1480
1481                        Button::new("spawn", run_entry_label)
1482                            .key_binding(KeyBinding::for_action(&menu::Confirm, cx))
1483                            .on_click(|_, window, cx| {
1484                                window.dispatch_action(menu::Confirm.boxed_clone(), cx);
1485                            })
1486                    })
1487                }
1488            });
1489        Some(footer.into_any_element())
1490    }
1491
1492    fn render_match(
1493        &self,
1494        ix: usize,
1495        selected: bool,
1496        window: &mut Window,
1497        cx: &mut Context<picker::Picker<Self>>,
1498    ) -> Option<Self::ListItem> {
1499        let hit = &self.matches.get(ix)?;
1500        let (task_kind, language_name, _scenario, context) = &self.candidates[hit.candidate_id];
1501
1502        let highlighted_location = HighlightedMatch {
1503            text: hit.string.clone(),
1504            highlight_positions: hit.positions.clone(),
1505            color: Color::Default,
1506        };
1507
1508        let subtitle = self.get_task_subtitle(task_kind, context, cx);
1509
1510        let language_icon = language_name.as_ref().and_then(|lang| {
1511            file_icons::FileIcons::get(cx)
1512                .get_icon_for_type(&lang.0.to_lowercase(), cx)
1513                .map(Icon::from_path)
1514        });
1515
1516        let (icon, indicator) = match task_kind {
1517            Some(TaskSourceKind::UserInput) => (Some(Icon::new(IconName::Terminal)), None),
1518            Some(TaskSourceKind::AbsPath { .. }) => (Some(Icon::new(IconName::Settings)), None),
1519            Some(TaskSourceKind::Worktree { .. }) => (Some(Icon::new(IconName::FileTree)), None),
1520            Some(TaskSourceKind::Lsp { language_name, .. }) => (
1521                file_icons::FileIcons::get(cx)
1522                    .get_icon_for_type(&language_name.to_lowercase(), cx)
1523                    .map(Icon::from_path),
1524                Some(Indicator::icon(
1525                    Icon::new(IconName::BoltFilled)
1526                        .color(Color::Muted)
1527                        .size(IconSize::Small),
1528                )),
1529            ),
1530            Some(TaskSourceKind::Language { name }) => (
1531                file_icons::FileIcons::get(cx)
1532                    .get_icon_for_type(&name.to_lowercase(), cx)
1533                    .map(Icon::from_path),
1534                None,
1535            ),
1536            None => (Some(Icon::new(IconName::HistoryRerun)), None),
1537        };
1538
1539        let icon = language_icon.or(icon).map(|icon| {
1540            IconWithIndicator::new(icon.color(Color::Muted).size(IconSize::Small), indicator)
1541                .indicator_border_color(Some(cx.theme().colors().border_transparent))
1542        });
1543
1544        Some(
1545            ListItem::new(SharedString::from(format!("debug-scenario-selection-{ix}")))
1546                .inset(true)
1547                .start_slot::<IconWithIndicator>(icon)
1548                .spacing(ListItemSpacing::Sparse)
1549                .toggle_state(selected)
1550                .child(
1551                    v_flex()
1552                        .items_start()
1553                        .child(highlighted_location.render(window, cx))
1554                        .when_some(subtitle, |this, subtitle_text| {
1555                            this.child(
1556                                Label::new(subtitle_text)
1557                                    .size(LabelSize::Small)
1558                                    .color(Color::Muted),
1559                            )
1560                        }),
1561                ),
1562        )
1563    }
1564}
1565
1566pub(crate) fn resolve_path(path: &mut String) {
1567    if path.starts_with('~') {
1568        let home = paths::home_dir().to_string_lossy().into_owned();
1569        let trimmed_path = path.trim().to_owned();
1570        *path = trimmed_path.replacen('~', &home, 1);
1571    } else if let Some(strip_path) = path.strip_prefix(&format!(".{}", std::path::MAIN_SEPARATOR)) {
1572        *path = format!(
1573            "$ZED_WORKTREE_ROOT{}{}",
1574            std::path::MAIN_SEPARATOR,
1575            &strip_path
1576        );
1577    };
1578}
1579
1580#[cfg(test)]
1581impl NewProcessModal {
1582    pub(crate) fn set_configure(
1583        &mut self,
1584        program: impl AsRef<str>,
1585        cwd: impl AsRef<str>,
1586        stop_on_entry: bool,
1587        window: &mut Window,
1588        cx: &mut Context<Self>,
1589    ) {
1590        self.mode = NewProcessMode::Launch;
1591        self.debugger = Some(dap::adapters::DebugAdapterName("fake-adapter".into()));
1592
1593        self.configure_mode.update(cx, |configure, cx| {
1594            configure.program.update(cx, |editor, cx| {
1595                editor.clear(window, cx);
1596                editor.set_text(program.as_ref(), window, cx);
1597            });
1598
1599            configure.cwd.update(cx, |editor, cx| {
1600                editor.clear(window, cx);
1601                editor.set_text(cwd.as_ref(), window, cx);
1602            });
1603
1604            configure.stop_on_entry = match stop_on_entry {
1605                true => ToggleState::Selected,
1606                _ => ToggleState::Unselected,
1607            }
1608        })
1609    }
1610
1611    pub(crate) fn debug_picker_candidate_subtitles(&self, cx: &mut App) -> Vec<String> {
1612        self.debug_picker.update(cx, |picker, cx| {
1613            picker
1614                .delegate
1615                .candidates
1616                .iter()
1617                .filter_map(|(task_kind, _, _, context)| {
1618                    picker.delegate.get_task_subtitle(task_kind, context, cx)
1619                })
1620                .collect()
1621        })
1622    }
1623}