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, Theme};
 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 [`LspAdapter`] under the hood, but calls all 'static' methods
 70/// once at startup, and caches the results.
 71pub struct CachedLspAdapter {
 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 LspAdapter>,
 79}
 80
 81impl CachedLspAdapter {
 82    pub async fn new<T: LspAdapter>(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(CachedLspAdapter {
 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 LspAdapter: '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"))]
259pub struct FakeLspAdapter {
260    pub name: &'static str,
261    pub capabilities: lsp::ServerCapabilities,
262    pub initializer: Option<Box<dyn 'static + Send + Sync + Fn(&mut lsp::FakeLanguageServer)>>,
263    pub disk_based_diagnostics_progress_token: Option<String>,
264    pub disk_based_diagnostics_sources: Vec<String>,
265}
266
267#[derive(Clone, Debug, Deserialize)]
268pub struct BracketPair {
269    pub start: String,
270    pub end: String,
271    pub close: bool,
272    pub newline: bool,
273}
274
275pub struct Language {
276    pub(crate) config: LanguageConfig,
277    pub(crate) grammar: Option<Arc<Grammar>>,
278    pub(crate) adapter: Option<Arc<CachedLspAdapter>>,
279
280    #[cfg(any(test, feature = "test-support"))]
281    fake_adapter: Option<(
282        mpsc::UnboundedSender<lsp::FakeLanguageServer>,
283        Arc<FakeLspAdapter>,
284    )>,
285}
286
287pub struct Grammar {
288    pub(crate) ts_language: tree_sitter::Language,
289    pub(crate) highlights_query: Option<Query>,
290    pub(crate) brackets_query: Option<Query>,
291    pub(crate) indents_query: Option<Query>,
292    pub(crate) outline_query: Option<Query>,
293    pub(crate) highlight_map: Mutex<HighlightMap>,
294}
295
296#[derive(Clone)]
297pub enum LanguageServerBinaryStatus {
298    CheckingForUpdate,
299    Downloading,
300    Downloaded,
301    Cached,
302    Failed { error: String },
303}
304
305pub struct LanguageRegistry {
306    languages: RwLock<Vec<Arc<Language>>>,
307    language_server_download_dir: Option<Arc<Path>>,
308    lsp_binary_statuses_tx: async_broadcast::Sender<(Arc<Language>, LanguageServerBinaryStatus)>,
309    lsp_binary_statuses_rx: async_broadcast::Receiver<(Arc<Language>, LanguageServerBinaryStatus)>,
310    login_shell_env_loaded: Shared<Task<()>>,
311    lsp_binary_paths: Mutex<
312        HashMap<
313            LanguageServerName,
314            Shared<BoxFuture<'static, Result<PathBuf, Arc<anyhow::Error>>>>,
315        >,
316    >,
317    subscription: RwLock<(watch::Sender<()>, watch::Receiver<()>)>,
318    theme: RwLock<Option<Arc<Theme>>>,
319}
320
321impl LanguageRegistry {
322    pub fn new(login_shell_env_loaded: Task<()>) -> Self {
323        let (lsp_binary_statuses_tx, lsp_binary_statuses_rx) = async_broadcast::broadcast(16);
324        Self {
325            language_server_download_dir: None,
326            languages: Default::default(),
327            lsp_binary_statuses_tx,
328            lsp_binary_statuses_rx,
329            login_shell_env_loaded: login_shell_env_loaded.shared(),
330            lsp_binary_paths: Default::default(),
331            subscription: RwLock::new(watch::channel()),
332            theme: Default::default(),
333        }
334    }
335
336    #[cfg(any(test, feature = "test-support"))]
337    pub fn test() -> Self {
338        Self::new(Task::ready(()))
339    }
340
341    pub fn add(&self, language: Arc<Language>) {
342        if let Some(theme) = self.theme.read().clone() {
343            language.set_theme(&theme.editor.syntax);
344        }
345        self.languages.write().push(language.clone());
346        *self.subscription.write().0.borrow_mut() = ();
347    }
348
349    pub fn subscribe(&self) -> watch::Receiver<()> {
350        self.subscription.read().1.clone()
351    }
352
353    pub fn set_theme(&self, theme: Arc<Theme>) {
354        *self.theme.write() = Some(theme.clone());
355        for language in self.languages.read().iter() {
356            language.set_theme(&theme.editor.syntax);
357        }
358    }
359
360    pub fn set_language_server_download_dir(&mut self, path: impl Into<Arc<Path>>) {
361        self.language_server_download_dir = Some(path.into());
362    }
363
364    pub fn get_language(&self, name: &str) -> Option<Arc<Language>> {
365        self.languages
366            .read()
367            .iter()
368            .find(|language| language.name().to_lowercase() == name.to_lowercase())
369            .cloned()
370    }
371
372    pub fn to_vec(&self) -> Vec<Arc<Language>> {
373        self.languages.read().iter().cloned().collect()
374    }
375
376    pub fn language_names(&self) -> Vec<String> {
377        self.languages
378            .read()
379            .iter()
380            .map(|language| language.name().to_string())
381            .collect()
382    }
383
384    pub fn select_language(&self, path: impl AsRef<Path>) -> Option<Arc<Language>> {
385        let path = path.as_ref();
386        let filename = path.file_name().and_then(|name| name.to_str());
387        let extension = path.extension().and_then(|name| name.to_str());
388        let path_suffixes = [extension, filename];
389        self.languages
390            .read()
391            .iter()
392            .find(|language| {
393                language
394                    .config
395                    .path_suffixes
396                    .iter()
397                    .any(|suffix| path_suffixes.contains(&Some(suffix.as_str())))
398            })
399            .cloned()
400    }
401
402    pub fn start_language_server(
403        self: &Arc<Self>,
404        server_id: usize,
405        language: Arc<Language>,
406        root_path: Arc<Path>,
407        http_client: Arc<dyn HttpClient>,
408        cx: &mut MutableAppContext,
409    ) -> Option<Task<Result<lsp::LanguageServer>>> {
410        #[cfg(any(test, feature = "test-support"))]
411        if language.fake_adapter.is_some() {
412            let language = language.clone();
413            return Some(cx.spawn(|cx| async move {
414                let (servers_tx, fake_adapter) = language.fake_adapter.as_ref().unwrap();
415                let (server, mut fake_server) = lsp::LanguageServer::fake(
416                    fake_adapter.name.to_string(),
417                    fake_adapter.capabilities.clone(),
418                    cx.clone(),
419                );
420
421                if let Some(initializer) = &fake_adapter.initializer {
422                    initializer(&mut fake_server);
423                }
424
425                let servers_tx = servers_tx.clone();
426                cx.background()
427                    .spawn(async move {
428                        if fake_server
429                            .try_receive_notification::<lsp::notification::Initialized>()
430                            .await
431                            .is_some()
432                        {
433                            servers_tx.unbounded_send(fake_server).ok();
434                        }
435                    })
436                    .detach();
437                Ok(server)
438            }));
439        }
440
441        let download_dir = self
442            .language_server_download_dir
443            .clone()
444            .ok_or_else(|| anyhow!("language server download directory has not been assigned"))
445            .log_err()?;
446
447        let this = self.clone();
448        let adapter = language.adapter.clone()?;
449        let lsp_binary_statuses = self.lsp_binary_statuses_tx.clone();
450        let login_shell_env_loaded = self.login_shell_env_loaded.clone();
451        Some(cx.spawn(|cx| async move {
452            login_shell_env_loaded.await;
453            let server_binary_path = this
454                .lsp_binary_paths
455                .lock()
456                .entry(adapter.name.clone())
457                .or_insert_with(|| {
458                    get_server_binary_path(
459                        adapter.clone(),
460                        language.clone(),
461                        http_client,
462                        download_dir,
463                        lsp_binary_statuses,
464                    )
465                    .map_err(Arc::new)
466                    .boxed()
467                    .shared()
468                })
469                .clone()
470                .map_err(|e| anyhow!(e));
471
472            let server_binary_path = server_binary_path.await?;
473            let server_args = &adapter.server_args;
474            let server = lsp::LanguageServer::new(
475                server_id,
476                &server_binary_path,
477                &server_args,
478                &root_path,
479                cx,
480            )?;
481            Ok(server)
482        }))
483    }
484
485    pub fn language_server_binary_statuses(
486        &self,
487    ) -> async_broadcast::Receiver<(Arc<Language>, LanguageServerBinaryStatus)> {
488        self.lsp_binary_statuses_rx.clone()
489    }
490}
491
492async fn get_server_binary_path(
493    adapter: Arc<CachedLspAdapter>,
494    language: Arc<Language>,
495    http_client: Arc<dyn HttpClient>,
496    download_dir: Arc<Path>,
497    statuses: async_broadcast::Sender<(Arc<Language>, LanguageServerBinaryStatus)>,
498) -> Result<PathBuf> {
499    let container_dir = download_dir.join(adapter.name.0.as_ref());
500    if !container_dir.exists() {
501        smol::fs::create_dir_all(&container_dir)
502            .await
503            .context("failed to create container directory")?;
504    }
505
506    let path = fetch_latest_server_binary_path(
507        adapter.clone(),
508        language.clone(),
509        http_client,
510        &container_dir,
511        statuses.clone(),
512    )
513    .await;
514    if let Err(error) = path.as_ref() {
515        if let Some(cached_path) = adapter.cached_server_binary(container_dir).await {
516            statuses
517                .broadcast((language.clone(), LanguageServerBinaryStatus::Cached))
518                .await?;
519            return Ok(cached_path);
520        } else {
521            statuses
522                .broadcast((
523                    language.clone(),
524                    LanguageServerBinaryStatus::Failed {
525                        error: format!("{:?}", error),
526                    },
527                ))
528                .await?;
529        }
530    }
531    path
532}
533
534async fn fetch_latest_server_binary_path(
535    adapter: Arc<CachedLspAdapter>,
536    language: Arc<Language>,
537    http_client: Arc<dyn HttpClient>,
538    container_dir: &Path,
539    lsp_binary_statuses_tx: async_broadcast::Sender<(Arc<Language>, LanguageServerBinaryStatus)>,
540) -> Result<PathBuf> {
541    let container_dir: Arc<Path> = container_dir.into();
542    lsp_binary_statuses_tx
543        .broadcast((
544            language.clone(),
545            LanguageServerBinaryStatus::CheckingForUpdate,
546        ))
547        .await?;
548    let version_info = adapter
549        .fetch_latest_server_version(http_client.clone())
550        .await?;
551    lsp_binary_statuses_tx
552        .broadcast((language.clone(), LanguageServerBinaryStatus::Downloading))
553        .await?;
554    let path = adapter
555        .fetch_server_binary(version_info, http_client, container_dir.to_path_buf())
556        .await?;
557    lsp_binary_statuses_tx
558        .broadcast((language.clone(), LanguageServerBinaryStatus::Downloaded))
559        .await?;
560    Ok(path)
561}
562
563impl Language {
564    pub fn new(config: LanguageConfig, ts_language: Option<tree_sitter::Language>) -> Self {
565        Self {
566            config,
567            grammar: ts_language.map(|ts_language| {
568                Arc::new(Grammar {
569                    highlights_query: None,
570                    brackets_query: None,
571                    indents_query: None,
572                    outline_query: None,
573                    ts_language,
574                    highlight_map: Default::default(),
575                })
576            }),
577            adapter: None,
578
579            #[cfg(any(test, feature = "test-support"))]
580            fake_adapter: None,
581        }
582    }
583
584    pub fn lsp_adapter(&self) -> Option<Arc<CachedLspAdapter>> {
585        self.adapter.clone()
586    }
587
588    pub fn with_highlights_query(mut self, source: &str) -> Result<Self> {
589        let grammar = self.grammar_mut();
590        grammar.highlights_query = Some(Query::new(grammar.ts_language, source)?);
591        Ok(self)
592    }
593
594    pub fn with_brackets_query(mut self, source: &str) -> Result<Self> {
595        let grammar = self.grammar_mut();
596        grammar.brackets_query = Some(Query::new(grammar.ts_language, source)?);
597        Ok(self)
598    }
599
600    pub fn with_indents_query(mut self, source: &str) -> Result<Self> {
601        let grammar = self.grammar_mut();
602        grammar.indents_query = Some(Query::new(grammar.ts_language, source)?);
603        Ok(self)
604    }
605
606    pub fn with_outline_query(mut self, source: &str) -> Result<Self> {
607        let grammar = self.grammar_mut();
608        grammar.outline_query = Some(Query::new(grammar.ts_language, source)?);
609        Ok(self)
610    }
611
612    fn grammar_mut(&mut self) -> &mut Grammar {
613        Arc::get_mut(self.grammar.as_mut().unwrap()).unwrap()
614    }
615
616    pub fn with_lsp_adapter(mut self, lsp_adapter: Arc<CachedLspAdapter>) -> Self {
617        self.adapter = Some(lsp_adapter);
618        self
619    }
620
621    #[cfg(any(test, feature = "test-support"))]
622    pub async fn set_fake_lsp_adapter(
623        &mut self,
624        fake_lsp_adapter: Arc<FakeLspAdapter>,
625    ) -> mpsc::UnboundedReceiver<lsp::FakeLanguageServer> {
626        let (servers_tx, servers_rx) = mpsc::unbounded();
627        self.fake_adapter = Some((servers_tx, fake_lsp_adapter.clone()));
628        let adapter = CachedLspAdapter::new(fake_lsp_adapter).await;
629        self.adapter = Some(adapter);
630        servers_rx
631    }
632
633    pub fn name(&self) -> Arc<str> {
634        self.config.name.clone()
635    }
636
637    pub fn line_comment_prefix(&self) -> Option<&str> {
638        self.config.line_comment.as_deref()
639    }
640
641    pub async fn disk_based_diagnostic_sources(&self) -> &[String] {
642        match self.adapter.as_ref() {
643            Some(adapter) => &adapter.disk_based_diagnostic_sources,
644            None => &[],
645        }
646    }
647
648    pub async fn disk_based_diagnostics_progress_token(&self) -> Option<&str> {
649        if let Some(adapter) = self.adapter.as_ref() {
650            adapter.disk_based_diagnostics_progress_token.as_deref()
651        } else {
652            None
653        }
654    }
655
656    pub async fn process_diagnostics(&self, diagnostics: &mut lsp::PublishDiagnosticsParams) {
657        if let Some(processor) = self.adapter.as_ref() {
658            processor.process_diagnostics(diagnostics).await;
659        }
660    }
661
662    pub async fn label_for_completion(
663        &self,
664        completion: &lsp::CompletionItem,
665    ) -> Option<CodeLabel> {
666        self.adapter
667            .as_ref()?
668            .label_for_completion(completion, self)
669            .await
670    }
671
672    pub async fn label_for_symbol(&self, name: &str, kind: lsp::SymbolKind) -> Option<CodeLabel> {
673        self.adapter
674            .as_ref()?
675            .label_for_symbol(name, kind, self)
676            .await
677    }
678
679    pub fn highlight_text<'a>(
680        &'a self,
681        text: &'a Rope,
682        range: Range<usize>,
683    ) -> Vec<(Range<usize>, HighlightId)> {
684        let mut result = Vec::new();
685        if let Some(grammar) = &self.grammar {
686            let tree = grammar.parse_text(text, None);
687            let mut offset = 0;
688            for chunk in BufferChunks::new(text, range, Some(&tree), self.grammar.as_ref(), vec![])
689            {
690                let end_offset = offset + chunk.text.len();
691                if let Some(highlight_id) = chunk.syntax_highlight_id {
692                    if !highlight_id.is_default() {
693                        result.push((offset..end_offset, highlight_id));
694                    }
695                }
696                offset = end_offset;
697            }
698        }
699        result
700    }
701
702    pub fn brackets(&self) -> &[BracketPair] {
703        &self.config.brackets
704    }
705
706    pub fn path_suffixes(&self) -> &[String] {
707        &self.config.path_suffixes
708    }
709
710    pub fn should_autoclose_before(&self, c: char) -> bool {
711        c.is_whitespace() || self.config.autoclose_before.contains(c)
712    }
713
714    pub fn set_theme(&self, theme: &SyntaxTheme) {
715        if let Some(grammar) = self.grammar.as_ref() {
716            if let Some(highlights_query) = &grammar.highlights_query {
717                *grammar.highlight_map.lock() =
718                    HighlightMap::new(highlights_query.capture_names(), theme);
719            }
720        }
721    }
722
723    pub fn grammar(&self) -> Option<&Arc<Grammar>> {
724        self.grammar.as_ref()
725    }
726}
727
728impl Grammar {
729    fn parse_text(&self, text: &Rope, old_tree: Option<Tree>) -> Tree {
730        PARSER.with(|parser| {
731            let mut parser = parser.borrow_mut();
732            parser
733                .set_language(self.ts_language)
734                .expect("incompatible grammar");
735            let mut chunks = text.chunks_in_range(0..text.len());
736            parser
737                .parse_with(
738                    &mut move |offset, _| {
739                        chunks.seek(offset);
740                        chunks.next().unwrap_or("").as_bytes()
741                    },
742                    old_tree.as_ref(),
743                )
744                .unwrap()
745        })
746    }
747
748    pub fn highlight_map(&self) -> HighlightMap {
749        self.highlight_map.lock().clone()
750    }
751
752    pub fn highlight_id_for_name(&self, name: &str) -> Option<HighlightId> {
753        let capture_id = self
754            .highlights_query
755            .as_ref()?
756            .capture_index_for_name(name)?;
757        Some(self.highlight_map.lock().get(capture_id))
758    }
759}
760
761impl CodeLabel {
762    pub fn plain(text: String, filter_text: Option<&str>) -> Self {
763        let mut result = Self {
764            runs: Vec::new(),
765            filter_range: 0..text.len(),
766            text,
767        };
768        if let Some(filter_text) = filter_text {
769            if let Some(ix) = result.text.find(filter_text) {
770                result.filter_range = ix..ix + filter_text.len();
771            }
772        }
773        result
774    }
775}
776
777#[cfg(any(test, feature = "test-support"))]
778impl Default for FakeLspAdapter {
779    fn default() -> Self {
780        Self {
781            name: "the-fake-language-server",
782            capabilities: lsp::LanguageServer::full_capabilities(),
783            initializer: None,
784            disk_based_diagnostics_progress_token: None,
785            disk_based_diagnostics_sources: Vec::new(),
786        }
787    }
788}
789
790#[cfg(any(test, feature = "test-support"))]
791#[async_trait]
792impl LspAdapter for Arc<FakeLspAdapter> {
793    async fn name(&self) -> LanguageServerName {
794        LanguageServerName(self.name.into())
795    }
796
797    async fn fetch_latest_server_version(
798        &self,
799        _: Arc<dyn HttpClient>,
800    ) -> Result<Box<dyn 'static + Send + Any>> {
801        unreachable!();
802    }
803
804    async fn fetch_server_binary(
805        &self,
806        _: Box<dyn 'static + Send + Any>,
807        _: Arc<dyn HttpClient>,
808        _: PathBuf,
809    ) -> Result<PathBuf> {
810        unreachable!();
811    }
812
813    async fn cached_server_binary(&self, _: PathBuf) -> Option<PathBuf> {
814        unreachable!();
815    }
816
817    async fn process_diagnostics(&self, _: &mut lsp::PublishDiagnosticsParams) {}
818
819    async fn disk_based_diagnostic_sources(&self) -> Vec<String> {
820        self.disk_based_diagnostics_sources.clone()
821    }
822
823    async fn disk_based_diagnostics_progress_token(&self) -> Option<String> {
824        self.disk_based_diagnostics_progress_token.clone()
825    }
826}
827
828pub fn point_to_lsp(point: PointUtf16) -> lsp::Position {
829    lsp::Position::new(point.row, point.column)
830}
831
832pub fn point_from_lsp(point: lsp::Position) -> PointUtf16 {
833    PointUtf16::new(point.line, point.character)
834}
835
836pub fn range_to_lsp(range: Range<PointUtf16>) -> lsp::Range {
837    lsp::Range {
838        start: point_to_lsp(range.start),
839        end: point_to_lsp(range.end),
840    }
841}
842
843pub fn range_from_lsp(range: lsp::Range) -> Range<PointUtf16> {
844    let mut start = point_from_lsp(range.start);
845    let mut end = point_from_lsp(range.end);
846    if start > end {
847        mem::swap(&mut start, &mut end);
848    }
849    start..end
850}