language.rs

  1mod buffer;
  2mod diagnostic_set;
  3mod highlight_map;
  4mod outline;
  5pub mod proto;
  6#[cfg(test)]
  7mod tests;
  8
  9use anyhow::{anyhow, Context, Result};
 10use async_trait::async_trait;
 11use client::http::HttpClient;
 12use collections::HashMap;
 13use futures::{
 14    future::{BoxFuture, Shared},
 15    FutureExt, TryFutureExt,
 16};
 17use gpui::{MutableAppContext, Task};
 18use highlight_map::HighlightMap;
 19use lazy_static::lazy_static;
 20use parking_lot::{Mutex, RwLock};
 21use postage::watch;
 22use regex::Regex;
 23use serde::{de, Deserialize, Deserializer};
 24use serde_json::Value;
 25use std::{
 26    any::Any,
 27    cell::RefCell,
 28    mem,
 29    ops::Range,
 30    path::{Path, PathBuf},
 31    str,
 32    sync::Arc,
 33};
 34use theme::SyntaxTheme;
 35use tree_sitter::{self, Query};
 36use util::ResultExt;
 37
 38#[cfg(any(test, feature = "test-support"))]
 39use futures::channel::mpsc;
 40
 41pub use buffer::Operation;
 42pub use buffer::*;
 43pub use diagnostic_set::DiagnosticEntry;
 44pub use outline::{Outline, OutlineItem};
 45pub use tree_sitter::{Parser, Tree};
 46
 47thread_local! {
 48    static PARSER: RefCell<Parser> = RefCell::new(Parser::new());
 49}
 50
 51lazy_static! {
 52    pub static ref PLAIN_TEXT: Arc<Language> = Arc::new(Language::new(
 53        LanguageConfig {
 54            name: "Plain Text".into(),
 55            ..Default::default()
 56        },
 57        None,
 58    ));
 59}
 60
 61pub trait ToLspPosition {
 62    fn to_lsp_position(self) -> lsp::Position;
 63}
 64
 65#[derive(Clone, Debug, PartialEq, Eq, Hash)]
 66pub struct LanguageServerName(pub Arc<str>);
 67
 68/// Represents a Language Server, with certain cached sync properties.
 69/// Uses [`LspAdapterTrait`] under the hood, but calls all 'static' methods
 70/// once at startup, and caches the results.
 71pub struct LspAdapter {
 72    pub name: LanguageServerName,
 73    pub server_args: Vec<String>,
 74    pub initialization_options: Option<Value>,
 75    pub disk_based_diagnostic_sources: Vec<String>,
 76    pub disk_based_diagnostics_progress_token: Option<String>,
 77    pub id_for_language: Option<String>,
 78    pub adapter: Box<dyn LspAdapterTrait>,
 79}
 80
 81impl LspAdapter {
 82    pub async fn new<T: LspAdapterTrait>(adapter: T) -> Arc<Self> {
 83        let adapter = Box::new(adapter);
 84        let name = adapter.name().await;
 85        let server_args = adapter.server_args().await;
 86        let initialization_options = adapter.initialization_options().await;
 87        let disk_based_diagnostic_sources = adapter.disk_based_diagnostic_sources().await;
 88        let disk_based_diagnostics_progress_token =
 89            adapter.disk_based_diagnostics_progress_token().await;
 90        let id_for_language = adapter.id_for_language(name.0.as_ref()).await;
 91
 92        Arc::new(LspAdapter {
 93            name,
 94            server_args,
 95            initialization_options,
 96            disk_based_diagnostic_sources,
 97            disk_based_diagnostics_progress_token,
 98            id_for_language,
 99            adapter,
100        })
101    }
102
103    pub async fn fetch_latest_server_version(
104        &self,
105        http: Arc<dyn HttpClient>,
106    ) -> Result<Box<dyn 'static + Send + Any>> {
107        self.adapter.fetch_latest_server_version(http).await
108    }
109
110    pub async fn fetch_server_binary(
111        &self,
112        version: Box<dyn 'static + Send + Any>,
113        http: Arc<dyn HttpClient>,
114        container_dir: PathBuf,
115    ) -> Result<PathBuf> {
116        self.adapter
117            .fetch_server_binary(version, http, container_dir)
118            .await
119    }
120
121    pub async fn cached_server_binary(&self, container_dir: PathBuf) -> Option<PathBuf> {
122        self.adapter.cached_server_binary(container_dir).await
123    }
124
125    pub async fn process_diagnostics(&self, params: &mut lsp::PublishDiagnosticsParams) {
126        self.adapter.process_diagnostics(params).await
127    }
128
129    pub async fn label_for_completion(
130        &self,
131        completion_item: &lsp::CompletionItem,
132        language: &Language,
133    ) -> Option<CodeLabel> {
134        self.adapter
135            .label_for_completion(completion_item, language)
136            .await
137    }
138
139    pub async fn label_for_symbol(
140        &self,
141        name: &str,
142        kind: lsp::SymbolKind,
143        language: &Language,
144    ) -> Option<CodeLabel> {
145        self.adapter.label_for_symbol(name, kind, language).await
146    }
147}
148
149#[async_trait]
150pub trait LspAdapterTrait: 'static + Send + Sync {
151    async fn name(&self) -> LanguageServerName;
152
153    async fn fetch_latest_server_version(
154        &self,
155        http: Arc<dyn HttpClient>,
156    ) -> Result<Box<dyn 'static + Send + Any>>;
157
158    async fn fetch_server_binary(
159        &self,
160        version: Box<dyn 'static + Send + Any>,
161        http: Arc<dyn HttpClient>,
162        container_dir: PathBuf,
163    ) -> Result<PathBuf>;
164
165    async fn cached_server_binary(&self, container_dir: PathBuf) -> Option<PathBuf>;
166
167    async fn process_diagnostics(&self, _: &mut lsp::PublishDiagnosticsParams) {}
168
169    async fn label_for_completion(
170        &self,
171        _: &lsp::CompletionItem,
172        _: &Language,
173    ) -> Option<CodeLabel> {
174        None
175    }
176
177    async fn label_for_symbol(
178        &self,
179        _: &str,
180        _: lsp::SymbolKind,
181        _: &Language,
182    ) -> Option<CodeLabel> {
183        None
184    }
185
186    async fn server_args(&self) -> Vec<String> {
187        Vec::new()
188    }
189
190    async fn initialization_options(&self) -> Option<Value> {
191        None
192    }
193
194    async fn disk_based_diagnostic_sources(&self) -> Vec<String> {
195        Default::default()
196    }
197
198    async fn disk_based_diagnostics_progress_token(&self) -> Option<String> {
199        None
200    }
201
202    async fn id_for_language(&self, _name: &str) -> Option<String> {
203        None
204    }
205}
206
207#[derive(Clone, Debug, PartialEq, Eq)]
208pub struct CodeLabel {
209    pub text: String,
210    pub runs: Vec<(Range<usize>, HighlightId)>,
211    pub filter_range: Range<usize>,
212}
213
214#[derive(Deserialize)]
215pub struct LanguageConfig {
216    pub name: Arc<str>,
217    pub path_suffixes: Vec<String>,
218    pub brackets: Vec<BracketPair>,
219    #[serde(default = "auto_indent_using_last_non_empty_line_default")]
220    pub auto_indent_using_last_non_empty_line: bool,
221    #[serde(default, deserialize_with = "deserialize_regex")]
222    pub increase_indent_pattern: Option<Regex>,
223    #[serde(default, deserialize_with = "deserialize_regex")]
224    pub decrease_indent_pattern: Option<Regex>,
225    #[serde(default)]
226    pub autoclose_before: String,
227    pub line_comment: Option<String>,
228}
229
230impl Default for LanguageConfig {
231    fn default() -> Self {
232        Self {
233            name: "".into(),
234            path_suffixes: Default::default(),
235            brackets: Default::default(),
236            auto_indent_using_last_non_empty_line: auto_indent_using_last_non_empty_line_default(),
237            increase_indent_pattern: Default::default(),
238            decrease_indent_pattern: Default::default(),
239            autoclose_before: Default::default(),
240            line_comment: Default::default(),
241        }
242    }
243}
244
245fn auto_indent_using_last_non_empty_line_default() -> bool {
246    true
247}
248
249fn deserialize_regex<'de, D: Deserializer<'de>>(d: D) -> Result<Option<Regex>, D::Error> {
250    let source = Option::<String>::deserialize(d)?;
251    if let Some(source) = source {
252        Ok(Some(regex::Regex::new(&source).map_err(de::Error::custom)?))
253    } else {
254        Ok(None)
255    }
256}
257
258// #[cfg(any(test, feature = "test-support"))]
259// pub type FakeLspAdapter = Arc<FakeLspAdapterInner>;
260
261#[cfg(any(test, feature = "test-support"))]
262pub struct FakeLspAdapter {
263    pub name: &'static str,
264    pub capabilities: lsp::ServerCapabilities,
265    pub initializer: Option<Box<dyn 'static + Send + Sync + Fn(&mut lsp::FakeLanguageServer)>>,
266    pub disk_based_diagnostics_progress_token: Option<String>,
267    pub disk_based_diagnostics_sources: Vec<String>,
268}
269
270#[derive(Clone, Debug, Deserialize)]
271pub struct BracketPair {
272    pub start: String,
273    pub end: String,
274    pub close: bool,
275    pub newline: bool,
276}
277
278pub struct Language {
279    pub(crate) config: LanguageConfig,
280    pub(crate) grammar: Option<Arc<Grammar>>,
281    pub(crate) adapter: Option<Arc<LspAdapter>>,
282
283    #[cfg(any(test, feature = "test-support"))]
284    fake_adapter: Option<(
285        mpsc::UnboundedSender<lsp::FakeLanguageServer>,
286        Arc<FakeLspAdapter>,
287    )>,
288}
289
290pub struct Grammar {
291    pub(crate) ts_language: tree_sitter::Language,
292    pub(crate) highlights_query: Option<Query>,
293    pub(crate) brackets_query: Option<Query>,
294    pub(crate) indents_query: Option<Query>,
295    pub(crate) outline_query: Option<Query>,
296    pub(crate) highlight_map: Mutex<HighlightMap>,
297}
298
299#[derive(Clone)]
300pub enum LanguageServerBinaryStatus {
301    CheckingForUpdate,
302    Downloading,
303    Downloaded,
304    Cached,
305    Failed { error: String },
306}
307
308pub struct LanguageRegistry {
309    languages: RwLock<Vec<Arc<Language>>>,
310    language_server_download_dir: Option<Arc<Path>>,
311    lsp_binary_statuses_tx: async_broadcast::Sender<(Arc<Language>, LanguageServerBinaryStatus)>,
312    lsp_binary_statuses_rx: async_broadcast::Receiver<(Arc<Language>, LanguageServerBinaryStatus)>,
313    login_shell_env_loaded: Shared<Task<()>>,
314    lsp_binary_paths: Mutex<
315        HashMap<
316            LanguageServerName,
317            Shared<BoxFuture<'static, Result<PathBuf, Arc<anyhow::Error>>>>,
318        >,
319    >,
320    subscription: RwLock<(watch::Sender<()>, watch::Receiver<()>)>,
321}
322
323impl LanguageRegistry {
324    pub fn new(login_shell_env_loaded: Task<()>) -> Self {
325        let (lsp_binary_statuses_tx, lsp_binary_statuses_rx) = async_broadcast::broadcast(16);
326        Self {
327            language_server_download_dir: None,
328            languages: Default::default(),
329            lsp_binary_statuses_tx,
330            lsp_binary_statuses_rx,
331            login_shell_env_loaded: login_shell_env_loaded.shared(),
332            lsp_binary_paths: Default::default(),
333            subscription: RwLock::new(watch::channel()),
334        }
335    }
336
337    #[cfg(any(test, feature = "test-support"))]
338    pub fn test() -> Self {
339        Self::new(Task::ready(()))
340    }
341
342    pub fn add(&self, language: Arc<Language>) {
343        self.languages.write().push(language.clone());
344        *self.subscription.write().0.borrow_mut() = ();
345    }
346
347    pub fn subscribe(&self) -> watch::Receiver<()> {
348        self.subscription.read().1.clone()
349    }
350
351    pub fn set_theme(&self, theme: &SyntaxTheme) {
352        for language in self.languages.read().iter() {
353            language.set_theme(theme);
354        }
355    }
356
357    pub fn set_language_server_download_dir(&mut self, path: impl Into<Arc<Path>>) {
358        self.language_server_download_dir = Some(path.into());
359    }
360
361    pub fn get_language(&self, name: &str) -> Option<Arc<Language>> {
362        self.languages
363            .read()
364            .iter()
365            .find(|language| language.name().to_lowercase() == name.to_lowercase())
366            .cloned()
367    }
368
369    pub fn to_vec(&self) -> Vec<Arc<Language>> {
370        self.languages.read().iter().cloned().collect()
371    }
372
373    pub fn language_names(&self) -> Vec<String> {
374        self.languages
375            .read()
376            .iter()
377            .map(|language| language.name().to_string())
378            .collect()
379    }
380
381    pub fn select_language(&self, path: impl AsRef<Path>) -> Option<Arc<Language>> {
382        let path = path.as_ref();
383        let filename = path.file_name().and_then(|name| name.to_str());
384        let extension = path.extension().and_then(|name| name.to_str());
385        let path_suffixes = [extension, filename];
386        self.languages
387            .read()
388            .iter()
389            .find(|language| {
390                language
391                    .config
392                    .path_suffixes
393                    .iter()
394                    .any(|suffix| dbg!(path_suffixes.contains(&Some(dbg!(suffix.as_str())))))
395            })
396            .cloned()
397    }
398
399    pub fn start_language_server(
400        self: &Arc<Self>,
401        server_id: usize,
402        language: Arc<Language>,
403        root_path: Arc<Path>,
404        http_client: Arc<dyn HttpClient>,
405        cx: &mut MutableAppContext,
406    ) -> Option<Task<Result<lsp::LanguageServer>>> {
407        #[cfg(any(test, feature = "test-support"))]
408        if language.fake_adapter.is_some() {
409            let language = language.clone();
410            return Some(cx.spawn(|cx| async move {
411                let (servers_tx, fake_adapter) = language.fake_adapter.as_ref().unwrap();
412                let (server, mut fake_server) = lsp::LanguageServer::fake(
413                    fake_adapter.name.to_string(),
414                    fake_adapter.capabilities.clone(),
415                    cx.clone(),
416                );
417
418                if let Some(initializer) = &fake_adapter.initializer {
419                    initializer(&mut fake_server);
420                }
421
422                let servers_tx = servers_tx.clone();
423                cx.background()
424                    .spawn(async move {
425                        if fake_server
426                            .try_receive_notification::<lsp::notification::Initialized>()
427                            .await
428                            .is_some()
429                        {
430                            servers_tx.unbounded_send(fake_server).ok();
431                        }
432                    })
433                    .detach();
434                Ok(server)
435            }));
436        }
437
438        let download_dir = self
439            .language_server_download_dir
440            .clone()
441            .ok_or_else(|| anyhow!("language server download directory has not been assigned"))
442            .log_err()?;
443
444        let this = self.clone();
445        let adapter = language.adapter.clone()?;
446        let lsp_binary_statuses = self.lsp_binary_statuses_tx.clone();
447        let login_shell_env_loaded = self.login_shell_env_loaded.clone();
448        Some(cx.spawn(|cx| async move {
449            login_shell_env_loaded.await;
450            let server_binary_path = this
451                .lsp_binary_paths
452                .lock()
453                .entry(adapter.name.clone())
454                .or_insert_with(|| {
455                    get_server_binary_path(
456                        adapter.clone(),
457                        language.clone(),
458                        http_client,
459                        download_dir,
460                        lsp_binary_statuses,
461                    )
462                    .map_err(Arc::new)
463                    .boxed()
464                    .shared()
465                })
466                .clone()
467                .map_err(|e| anyhow!(e));
468
469            let server_binary_path = server_binary_path.await?;
470            let server_args = &adapter.server_args;
471            let server = lsp::LanguageServer::new(
472                server_id,
473                &server_binary_path,
474                &server_args,
475                &root_path,
476                cx,
477            )?;
478            Ok(server)
479        }))
480    }
481
482    pub fn language_server_binary_statuses(
483        &self,
484    ) -> async_broadcast::Receiver<(Arc<Language>, LanguageServerBinaryStatus)> {
485        self.lsp_binary_statuses_rx.clone()
486    }
487}
488
489async fn get_server_binary_path(
490    adapter: Arc<LspAdapter>,
491    language: Arc<Language>,
492    http_client: Arc<dyn HttpClient>,
493    download_dir: Arc<Path>,
494    statuses: async_broadcast::Sender<(Arc<Language>, LanguageServerBinaryStatus)>,
495) -> Result<PathBuf> {
496    let container_dir = download_dir.join(adapter.name.0.as_ref());
497    if !container_dir.exists() {
498        smol::fs::create_dir_all(&container_dir)
499            .await
500            .context("failed to create container directory")?;
501    }
502
503    let path = fetch_latest_server_binary_path(
504        adapter.clone(),
505        language.clone(),
506        http_client,
507        &container_dir,
508        statuses.clone(),
509    )
510    .await;
511    if let Err(error) = path.as_ref() {
512        if let Some(cached_path) = adapter.cached_server_binary(container_dir).await {
513            statuses
514                .broadcast((language.clone(), LanguageServerBinaryStatus::Cached))
515                .await?;
516            return Ok(cached_path);
517        } else {
518            statuses
519                .broadcast((
520                    language.clone(),
521                    LanguageServerBinaryStatus::Failed {
522                        error: format!("{:?}", error),
523                    },
524                ))
525                .await?;
526        }
527    }
528    path
529}
530
531async fn fetch_latest_server_binary_path(
532    adapter: Arc<LspAdapter>,
533    language: Arc<Language>,
534    http_client: Arc<dyn HttpClient>,
535    container_dir: &Path,
536    lsp_binary_statuses_tx: async_broadcast::Sender<(Arc<Language>, LanguageServerBinaryStatus)>,
537) -> Result<PathBuf> {
538    let container_dir: Arc<Path> = container_dir.into();
539    lsp_binary_statuses_tx
540        .broadcast((
541            language.clone(),
542            LanguageServerBinaryStatus::CheckingForUpdate,
543        ))
544        .await?;
545    let version_info = adapter
546        .fetch_latest_server_version(http_client.clone())
547        .await?;
548    lsp_binary_statuses_tx
549        .broadcast((language.clone(), LanguageServerBinaryStatus::Downloading))
550        .await?;
551    let path = adapter
552        .fetch_server_binary(version_info, http_client, container_dir.to_path_buf())
553        .await?;
554    lsp_binary_statuses_tx
555        .broadcast((language.clone(), LanguageServerBinaryStatus::Downloaded))
556        .await?;
557    Ok(path)
558}
559
560impl Language {
561    pub fn new(config: LanguageConfig, ts_language: Option<tree_sitter::Language>) -> Self {
562        Self {
563            config,
564            grammar: ts_language.map(|ts_language| {
565                Arc::new(Grammar {
566                    highlights_query: None,
567                    brackets_query: None,
568                    indents_query: None,
569                    outline_query: None,
570                    ts_language,
571                    highlight_map: Default::default(),
572                })
573            }),
574            adapter: None,
575
576            #[cfg(any(test, feature = "test-support"))]
577            fake_adapter: None,
578        }
579    }
580
581    pub fn lsp_adapter(&self) -> Option<Arc<LspAdapter>> {
582        self.adapter.clone()
583    }
584
585    pub fn with_highlights_query(mut self, source: &str) -> Result<Self> {
586        let grammar = self.grammar_mut();
587        grammar.highlights_query = Some(Query::new(grammar.ts_language, source)?);
588        Ok(self)
589    }
590
591    pub fn with_brackets_query(mut self, source: &str) -> Result<Self> {
592        let grammar = self.grammar_mut();
593        grammar.brackets_query = Some(Query::new(grammar.ts_language, source)?);
594        Ok(self)
595    }
596
597    pub fn with_indents_query(mut self, source: &str) -> Result<Self> {
598        let grammar = self.grammar_mut();
599        grammar.indents_query = Some(Query::new(grammar.ts_language, source)?);
600        Ok(self)
601    }
602
603    pub fn with_outline_query(mut self, source: &str) -> Result<Self> {
604        let grammar = self.grammar_mut();
605        grammar.outline_query = Some(Query::new(grammar.ts_language, source)?);
606        Ok(self)
607    }
608
609    fn grammar_mut(&mut self) -> &mut Grammar {
610        Arc::get_mut(self.grammar.as_mut().unwrap()).unwrap()
611    }
612
613    pub fn with_lsp_adapter(mut self, lsp_adapter: Arc<LspAdapter>) -> Self {
614        self.adapter = Some(lsp_adapter);
615        self
616    }
617
618    #[cfg(any(test, feature = "test-support"))]
619    pub fn set_fake_lsp_adapter(
620        &mut self,
621        fake_lsp_adapter: Arc<FakeLspAdapter>,
622    ) -> mpsc::UnboundedReceiver<lsp::FakeLanguageServer> {
623        let (servers_tx, servers_rx) = mpsc::unbounded();
624        self.fake_adapter = Some((servers_tx, fake_lsp_adapter.clone()));
625        let adapter = smol::block_on(LspAdapter::new(fake_lsp_adapter));
626        self.adapter = Some(adapter);
627        servers_rx
628    }
629
630    pub fn name(&self) -> Arc<str> {
631        self.config.name.clone()
632    }
633
634    pub fn line_comment_prefix(&self) -> Option<&str> {
635        self.config.line_comment.as_deref()
636    }
637
638    pub async fn disk_based_diagnostic_sources(&self) -> &[String] {
639        match self.adapter.as_ref() {
640            Some(adapter) => &adapter.disk_based_diagnostic_sources,
641            None => &[],
642        }
643    }
644
645    pub async fn disk_based_diagnostics_progress_token(&self) -> Option<&str> {
646        if let Some(adapter) = self.adapter.as_ref() {
647            adapter.disk_based_diagnostics_progress_token.as_deref()
648        } else {
649            None
650        }
651    }
652
653    pub async fn process_diagnostics(&self, diagnostics: &mut lsp::PublishDiagnosticsParams) {
654        if let Some(processor) = self.adapter.as_ref() {
655            processor.process_diagnostics(diagnostics).await;
656        }
657    }
658
659    pub async fn label_for_completion(
660        &self,
661        completion: &lsp::CompletionItem,
662    ) -> Option<CodeLabel> {
663        self.adapter
664            .as_ref()?
665            .label_for_completion(completion, self)
666            .await
667    }
668
669    pub async fn label_for_symbol(&self, name: &str, kind: lsp::SymbolKind) -> Option<CodeLabel> {
670        self.adapter
671            .as_ref()?
672            .label_for_symbol(name, kind, self)
673            .await
674    }
675
676    pub fn highlight_text<'a>(
677        &'a self,
678        text: &'a Rope,
679        range: Range<usize>,
680    ) -> Vec<(Range<usize>, HighlightId)> {
681        let mut result = Vec::new();
682        if let Some(grammar) = &self.grammar {
683            let tree = grammar.parse_text(text, None);
684            let mut offset = 0;
685            for chunk in BufferChunks::new(text, range, Some(&tree), self.grammar.as_ref(), vec![])
686            {
687                let end_offset = offset + chunk.text.len();
688                if let Some(highlight_id) = chunk.syntax_highlight_id {
689                    if !highlight_id.is_default() {
690                        result.push((offset..end_offset, highlight_id));
691                    }
692                }
693                offset = end_offset;
694            }
695        }
696        result
697    }
698
699    pub fn brackets(&self) -> &[BracketPair] {
700        &self.config.brackets
701    }
702
703    pub fn path_suffixes(&self) -> &[String] {
704        &self.config.path_suffixes
705    }
706
707    pub fn should_autoclose_before(&self, c: char) -> bool {
708        c.is_whitespace() || self.config.autoclose_before.contains(c)
709    }
710
711    pub fn set_theme(&self, theme: &SyntaxTheme) {
712        if let Some(grammar) = self.grammar.as_ref() {
713            if let Some(highlights_query) = &grammar.highlights_query {
714                *grammar.highlight_map.lock() =
715                    HighlightMap::new(highlights_query.capture_names(), theme);
716            }
717        }
718    }
719
720    pub fn grammar(&self) -> Option<&Arc<Grammar>> {
721        self.grammar.as_ref()
722    }
723}
724
725impl Grammar {
726    fn parse_text(&self, text: &Rope, old_tree: Option<Tree>) -> Tree {
727        PARSER.with(|parser| {
728            let mut parser = parser.borrow_mut();
729            parser
730                .set_language(self.ts_language)
731                .expect("incompatible grammar");
732            let mut chunks = text.chunks_in_range(0..text.len());
733            parser
734                .parse_with(
735                    &mut move |offset, _| {
736                        chunks.seek(offset);
737                        chunks.next().unwrap_or("").as_bytes()
738                    },
739                    old_tree.as_ref(),
740                )
741                .unwrap()
742        })
743    }
744
745    pub fn highlight_map(&self) -> HighlightMap {
746        self.highlight_map.lock().clone()
747    }
748
749    pub fn highlight_id_for_name(&self, name: &str) -> Option<HighlightId> {
750        let capture_id = self
751            .highlights_query
752            .as_ref()?
753            .capture_index_for_name(name)?;
754        Some(self.highlight_map.lock().get(capture_id))
755    }
756}
757
758impl CodeLabel {
759    pub fn plain(text: String, filter_text: Option<&str>) -> Self {
760        let mut result = Self {
761            runs: Vec::new(),
762            filter_range: 0..text.len(),
763            text,
764        };
765        if let Some(filter_text) = filter_text {
766            if let Some(ix) = result.text.find(filter_text) {
767                result.filter_range = ix..ix + filter_text.len();
768            }
769        }
770        result
771    }
772}
773
774#[cfg(any(test, feature = "test-support"))]
775impl Default for FakeLspAdapter {
776    fn default() -> Self {
777        Self {
778            name: "the-fake-language-server",
779            capabilities: lsp::LanguageServer::full_capabilities(),
780            initializer: None,
781            disk_based_diagnostics_progress_token: None,
782            disk_based_diagnostics_sources: Vec::new(),
783        }
784    }
785}
786
787#[cfg(any(test, feature = "test-support"))]
788#[async_trait]
789impl LspAdapterTrait for Arc<FakeLspAdapter> {
790    async fn name(&self) -> LanguageServerName {
791        LanguageServerName(self.name.into())
792    }
793
794    async fn fetch_latest_server_version(
795        &self,
796        _: Arc<dyn HttpClient>,
797    ) -> Result<Box<dyn 'static + Send + Any>> {
798        unreachable!();
799    }
800
801    async fn fetch_server_binary(
802        &self,
803        _: Box<dyn 'static + Send + Any>,
804        _: Arc<dyn HttpClient>,
805        _: PathBuf,
806    ) -> Result<PathBuf> {
807        unreachable!();
808    }
809
810    async fn cached_server_binary(&self, _: PathBuf) -> Option<PathBuf> {
811        unreachable!();
812    }
813
814    async fn process_diagnostics(&self, _: &mut lsp::PublishDiagnosticsParams) {}
815
816    async fn disk_based_diagnostic_sources(&self) -> Vec<String> {
817        self.disk_based_diagnostics_sources.clone()
818    }
819
820    async fn disk_based_diagnostics_progress_token(&self) -> Option<String> {
821        self.disk_based_diagnostics_progress_token.clone()
822    }
823}
824
825pub fn point_to_lsp(point: PointUtf16) -> lsp::Position {
826    lsp::Position::new(point.row, point.column)
827}
828
829pub fn point_from_lsp(point: lsp::Position) -> PointUtf16 {
830    PointUtf16::new(point.line, point.character)
831}
832
833pub fn range_to_lsp(range: Range<PointUtf16>) -> lsp::Range {
834    lsp::Range {
835        start: point_to_lsp(range.start),
836        end: point_to_lsp(range.end),
837    }
838}
839
840pub fn range_from_lsp(range: lsp::Range) -> Range<PointUtf16> {
841    let mut start = point_from_lsp(range.start);
842    let mut end = point_from_lsp(range.end);
843    if start > end {
844        mem::swap(&mut start, &mut end);
845    }
846    start..end
847}