1use std::path::PathBuf;
 2
 3use gpui::{AppContext, ViewContext, WindowContext};
 4use modal::TasksModal;
 5use task::Task;
 6use util::ResultExt;
 7use workspace::Workspace;
 8
 9mod modal;
10
11pub fn init(cx: &mut AppContext) {
12    cx.observe_new_views(
13        |workspace: &mut Workspace, _: &mut ViewContext<Workspace>| {
14            workspace
15                .register_action(|workspace, _: &modal::Spawn, cx| {
16                    let inventory = workspace.project().read(cx).task_inventory().clone();
17                    let workspace_handle = workspace.weak_handle();
18                    workspace
19                        .toggle_modal(cx, |cx| TasksModal::new(inventory, workspace_handle, cx))
20                })
21                .register_action(move |workspace, _: &modal::Rerun, cx| {
22                    if let Some(task) = workspace.project().update(cx, |project, cx| {
23                        project
24                            .task_inventory()
25                            .update(cx, |inventory, cx| inventory.last_scheduled_task(cx))
26                    }) {
27                        schedule_task(workspace, task.as_ref(), cx)
28                    };
29                });
30        },
31    )
32    .detach();
33}
34
35fn schedule_task(workspace: &Workspace, task: &dyn Task, cx: &mut ViewContext<'_, Workspace>) {
36    let cwd = match task.cwd() {
37        Some(cwd) => Some(cwd.to_path_buf()),
38        None => task_cwd(workspace, cx).log_err().flatten(),
39    };
40    let spawn_in_terminal = task.exec(cwd);
41    if let Some(spawn_in_terminal) = spawn_in_terminal {
42        workspace.project().update(cx, |project, cx| {
43            project.task_inventory().update(cx, |inventory, _| {
44                inventory.last_scheduled_task = Some(task.id().clone());
45            })
46        });
47        cx.emit(workspace::Event::SpawnTask(spawn_in_terminal));
48    }
49}
50
51fn task_cwd(workspace: &Workspace, cx: &mut WindowContext) -> anyhow::Result<Option<PathBuf>> {
52    let project = workspace.project().read(cx);
53    let available_worktrees = project
54        .worktrees()
55        .filter(|worktree| {
56            let worktree = worktree.read(cx);
57            worktree.is_visible()
58                && worktree.is_local()
59                && worktree.root_entry().map_or(false, |e| e.is_dir())
60        })
61        .collect::<Vec<_>>();
62    let cwd = match available_worktrees.len() {
63        0 => None,
64        1 => Some(available_worktrees[0].read(cx).abs_path()),
65        _ => {
66            let cwd_for_active_entry = project.active_entry().and_then(|entry_id| {
67                available_worktrees.into_iter().find_map(|worktree| {
68                    let worktree = worktree.read(cx);
69                    if worktree.contains_entry(entry_id) {
70                        Some(worktree.abs_path())
71                    } else {
72                        None
73                    }
74                })
75            });
76            anyhow::ensure!(
77                cwd_for_active_entry.is_some(),
78                "Cannot determine task cwd for multiple worktrees"
79            );
80            cwd_for_active_entry
81        }
82    };
83    Ok(cwd.map(|path| path.to_path_buf()))
84}