1use std::path::PathBuf;
 2
 3use project::TaskSourceKind;
 4use task::{ResolvedTask, TaskContext, TaskTemplate};
 5use ui::{ViewContext, WindowContext};
 6
 7use crate::Workspace;
 8
 9pub fn task_cwd(workspace: &Workspace, cx: &mut WindowContext) -> anyhow::Result<Option<PathBuf>> {
10    let project = workspace.project().read(cx);
11    let available_worktrees = project
12        .worktrees()
13        .filter(|worktree| {
14            let worktree = worktree.read(cx);
15            worktree.is_visible()
16                && worktree.is_local()
17                && worktree.root_entry().map_or(false, |e| e.is_dir())
18        })
19        .collect::<Vec<_>>();
20    let cwd = match available_worktrees.len() {
21        0 => None,
22        1 => Some(available_worktrees[0].read(cx).abs_path()),
23        _ => {
24            let cwd_for_active_entry = project.active_entry().and_then(|entry_id| {
25                available_worktrees.into_iter().find_map(|worktree| {
26                    let worktree = worktree.read(cx);
27                    if worktree.contains_entry(entry_id) {
28                        Some(worktree.abs_path())
29                    } else {
30                        None
31                    }
32                })
33            });
34            anyhow::ensure!(
35                cwd_for_active_entry.is_some(),
36                "Cannot determine task cwd for multiple worktrees"
37            );
38            cwd_for_active_entry
39        }
40    };
41    Ok(cwd.map(|path| path.to_path_buf()))
42}
43
44pub fn schedule_task(
45    workspace: &Workspace,
46    task_source_kind: TaskSourceKind,
47    task_to_resolve: &TaskTemplate,
48    task_cx: &TaskContext,
49    omit_history: bool,
50    cx: &mut ViewContext<'_, Workspace>,
51) {
52    if let Some(spawn_in_terminal) =
53        task_to_resolve.resolve_task(&task_source_kind.to_id_base(), task_cx)
54    {
55        schedule_resolved_task(
56            workspace,
57            task_source_kind,
58            spawn_in_terminal,
59            omit_history,
60            cx,
61        );
62    }
63}
64
65pub fn schedule_resolved_task(
66    workspace: &Workspace,
67    task_source_kind: TaskSourceKind,
68    mut resolved_task: ResolvedTask,
69    omit_history: bool,
70    cx: &mut ViewContext<'_, Workspace>,
71) {
72    if let Some(spawn_in_terminal) = resolved_task.resolved.take() {
73        if !omit_history {
74            resolved_task.resolved = Some(spawn_in_terminal.clone());
75            workspace.project().update(cx, |project, cx| {
76                project.task_inventory().update(cx, |inventory, _| {
77                    inventory.task_scheduled(task_source_kind, resolved_task);
78                })
79            });
80        }
81        cx.emit(crate::Event::SpawnTask(spawn_in_terminal));
82    }
83}