1use std::{ops::Range, path::PathBuf, sync::Arc};
2
3use crate::{File, LanguageToolchainStore, Location, Runnable};
4
5use anyhow::Result;
6use collections::HashMap;
7use fs::Fs;
8use gpui::{App, Task};
9use lsp::LanguageServerName;
10use task::{TaskTemplates, TaskVariables};
11use text::BufferId;
12
13pub struct RunnableRange {
14 pub buffer_id: BufferId,
15 pub run_range: Range<usize>,
16 pub full_range: Range<usize>,
17 pub runnable: Runnable,
18 pub extra_captures: HashMap<String, String>,
19}
20
21/// Language Contexts are used by Zed tasks to extract information about the source file where the tasks are supposed to be scheduled from.
22/// Multiple context providers may be used together: by default, Zed provides a base [`BasicContextProvider`] context that fills all non-custom [`VariableName`] variants.
23///
24/// The context will be used to fill data for the tasks, and filter out the ones that do not have the variables required.
25pub trait ContextProvider: Send + Sync {
26 /// Builds a specific context to be placed on top of the basic one (replacing all conflicting entries) and to be used for task resolving later.
27 fn build_context(
28 &self,
29 _variables: &TaskVariables,
30 _location: ContextLocation<'_>,
31 _project_env: Option<HashMap<String, String>>,
32 _toolchains: Arc<dyn LanguageToolchainStore>,
33 _cx: &mut App,
34 ) -> Task<Result<TaskVariables>> {
35 let _ = _location;
36 Task::ready(Ok(TaskVariables::default()))
37 }
38
39 /// Provides all tasks, associated with the current language.
40 fn associated_tasks(
41 &self,
42 _: Arc<dyn Fs>,
43 _: Option<Arc<dyn File>>,
44 _: &App,
45 ) -> Task<Option<TaskTemplates>> {
46 Task::ready(None)
47 }
48
49 /// A language server name, that can return tasks using LSP (ext) for this language.
50 fn lsp_task_source(&self) -> Option<LanguageServerName> {
51 None
52 }
53}
54
55/// Metadata about the place in the project we gather the context for.
56pub struct ContextLocation<'a> {
57 pub fs: Option<Arc<dyn Fs>>,
58 pub worktree_root: Option<PathBuf>,
59 pub file_location: &'a Location,
60}