// Copyright 2018-2024 the Deno authors. All rights reserved. MIT license. use super::client::Client; use super::config::Config; use super::config::WorkspaceSettings; use super::documents::Documents; use super::documents::DocumentsFilter; use super::jsr::CliJsrSearchApi; use super::lsp_custom; use super::npm::CliNpmSearchApi; use super::registries::ModuleRegistry; use super::resolver::LspResolver; use super::search::PackageSearchApi; use super::tsc; use crate::jsr::JsrFetchResolver; use crate::util::path::is_importable_ext; use crate::util::path::relative_specifier; use deno_graph::source::ResolutionMode; use deno_graph::Range; use deno_runtime_tauri::deno_node::SUPPORTED_BUILTIN_NODE_MODULES; use deno_runtime_tauri::fs_util::specifier_to_file_path; use deno_ast::LineAndColumnIndex; use deno_ast::SourceTextInfo; use deno_core::resolve_path; use deno_core::resolve_url; use deno_core::serde::Deserialize; use deno_core::serde::Serialize; use deno_core::serde_json::json; use deno_core::url::Position; use deno_core::ModuleSpecifier; use deno_semver::jsr::JsrPackageReqReference; use deno_semver::package::PackageNv; use import_map::ImportMap; use indexmap::IndexSet; use lsp_types::CompletionList; use once_cell::sync::Lazy; use regex::Regex; use tower_lsp::lsp_types as lsp; static FILE_PROTO_RE: Lazy = lazy_regex::lazy_regex!(r#"^file:/{2}(?:/[A-Za-z]:)?"#); const CURRENT_PATH: &str = "."; const PARENT_PATH: &str = ".."; const LOCAL_PATHS: &[&str] = &[CURRENT_PATH, PARENT_PATH]; pub(crate) const IMPORT_COMMIT_CHARS: &[&str] = &["\"", "'"]; #[derive(Debug, Serialize, Deserialize)] #[serde(rename_all = "camelCase")] pub struct CompletionItemData { #[serde(skip_serializing_if = "Option::is_none")] pub documentation: Option, #[serde(skip_serializing_if = "Option::is_none")] pub tsc: Option, } /// Check if the origin can be auto-configured for completions, and if so, send /// a notification to the client. async fn check_auto_config_registry( url_str: &str, workspace_settings: &WorkspaceSettings, client: &Client, module_registries: &ModuleRegistry, ) { // check to see if auto discovery is enabled if workspace_settings.suggest.imports.auto_discover { if let Ok(specifier) = resolve_url(url_str) { let scheme = specifier.scheme(); let path = &specifier[Position::BeforePath..]; if scheme.starts_with("http") && !path.is_empty() && url_str.ends_with(path) { // check to see if this origin is already explicitly set let in_config = workspace_settings .suggest .imports .hosts .iter() .any(|(h, _)| { resolve_url(h).map(|u| u.origin()) == Ok(specifier.origin()) }); // if it isn't in the configuration, we will check to see if it supports // suggestions and send a notification to the client. if !in_config { let origin = specifier.origin().ascii_serialization(); let suggestions = module_registries.check_origin(&origin).await.is_ok(); // we are only sending registry state when enabled now, but changing // the custom notification would make older versions of the plugin // incompatible. // TODO(@kitsonk) clean up protocol when doing v2 of suggestions if suggestions { client.send_registry_state_notification( lsp_custom::RegistryStateNotificationParams { origin, suggestions, }, ); } } } } } } /// Ranges from the graph for specifiers include the leading and maybe trailing quote, /// which we want to ignore when replacing text. fn to_narrow_lsp_range( text_info: &SourceTextInfo, range: &deno_graph::Range, ) -> lsp::Range { let end_byte_index = text_info .loc_to_source_pos(LineAndColumnIndex { line_index: range.end.line, column_index: range.end.character, }) .as_byte_index(text_info.range().start); let start_byte_index = text_info .loc_to_source_pos(LineAndColumnIndex { line_index: range.start.line, column_index: range.start.character, }) .as_byte_index(text_info.range().start); let text_bytes = text_info.text_str().as_bytes(); let is_empty = end_byte_index - 1 == start_byte_index; let has_trailing_quote = !is_empty && matches!(text_bytes[end_byte_index - 1], b'"' | b'\''); lsp::Range { start: lsp::Position { line: range.start.line as u32, // skip the leading quote character: (range.start.character + 1) as u32, }, end: lsp::Position { line: range.end.line as u32, character: if has_trailing_quote { range.end.character - 1 // do not include it } else { range.end.character } as u32, }, } } /// Given a specifier, a position, and a snapshot, optionally return a /// completion response, which will be valid import completions for the specific /// context. #[allow(clippy::too_many_arguments)] pub async fn get_import_completions( specifier: &ModuleSpecifier, position: &lsp::Position, config: &Config, client: &Client, module_registries: &ModuleRegistry, jsr_search_api: &CliJsrSearchApi, npm_search_api: &CliNpmSearchApi, documents: &Documents, resolver: &LspResolver, maybe_import_map: Option<&ImportMap>, ) -> Option { let document = documents.get(specifier)?; let file_referrer = document.file_referrer(); let (text, _, range) = document.get_maybe_dependency(position)?; let range = to_narrow_lsp_range(document.text_info(), &range); let resolved = resolver .as_graph_resolver(file_referrer) .resolve( &text, &Range { specifier: specifier.clone(), start: deno_graph::Position::zeroed(), end: deno_graph::Position::zeroed(), }, ResolutionMode::Execution, ) .ok(); if let Some(completion_list) = get_jsr_completions( specifier, &text, &range, resolved.as_ref(), jsr_search_api, Some(jsr_search_api.get_resolver()), ) .await { Some(lsp::CompletionResponse::List(completion_list)) } else if let Some(completion_list) = get_npm_completions(specifier, &text, &range, npm_search_api).await { Some(lsp::CompletionResponse::List(completion_list)) } else if let Some(completion_list) = get_node_completions(&text, &range) { Some(lsp::CompletionResponse::List(completion_list)) } else if let Some(completion_list) = get_import_map_completions(specifier, &text, &range, maybe_import_map) { // completions for import map specifiers Some(lsp::CompletionResponse::List(completion_list)) } else if text.starts_with("./") || text.starts_with("../") || text.starts_with('/') { // completions for local relative modules Some(lsp::CompletionResponse::List(CompletionList { is_incomplete: false, items: get_local_completions(specifier, &text, &range, resolver)?, })) } else if !text.is_empty() { // completion of modules from a module registry or cache check_auto_config_registry( &text, config.workspace_settings_for_specifier(specifier), client, module_registries, ) .await; let maybe_list = module_registries .get_completions(&text, &range, resolved.as_ref(), |s| { documents.exists(s, file_referrer) }) .await; let maybe_list = maybe_list .or_else(|| module_registries.get_origin_completions(&text, &range)); let list = maybe_list.unwrap_or_else(|| CompletionList { items: get_workspace_completions(specifier, &text, &range, documents), is_incomplete: false, }); Some(lsp::CompletionResponse::List(list)) } else { // the import specifier is empty, so provide all possible specifiers we are // aware of let mut items: Vec = LOCAL_PATHS .iter() .map(|s| lsp::CompletionItem { label: s.to_string(), kind: Some(lsp::CompletionItemKind::FOLDER), detail: Some("(local)".to_string()), sort_text: Some("1".to_string()), insert_text: Some(s.to_string()), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect(), ), ..Default::default() }) .collect(); let mut is_incomplete = false; if let Some(import_map) = maybe_import_map { items.extend(get_base_import_map_completions(import_map, specifier)); } if let Some(origin_items) = module_registries.get_origin_completions(&text, &range) { is_incomplete = origin_items.is_incomplete; items.extend(origin_items.items); } Some(lsp::CompletionResponse::List(CompletionList { is_incomplete, items, })) } } /// When the specifier is an empty string, return all the keys from the import /// map as completion items. fn get_base_import_map_completions( import_map: &ImportMap, referrer: &ModuleSpecifier, ) -> Vec { import_map .entries_for_referrer(referrer) .map(|entry| { // for some strange reason, keys that start with `/` get stored in the // import map as `file:///`, and so when we pull the keys out, we need to // change the behavior let mut label = if entry.key.starts_with("file://") { FILE_PROTO_RE.replace(entry.key, "").to_string() } else { entry.key.to_string() }; let kind = if entry.key.ends_with('/') { label.pop(); Some(lsp::CompletionItemKind::FOLDER) } else { Some(lsp::CompletionItemKind::FILE) }; lsp::CompletionItem { label: label.clone(), kind, detail: Some("(import map)".to_string()), sort_text: Some(label.clone()), insert_text: Some(label), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect(), ), ..Default::default() } }) .collect() } /// Given an existing specifier, return any completions that could apply derived /// from the import map. There are two main type of import map keys, those that /// a literal, which don't end in `/`, which expects a one for one replacement /// of specifier to specifier, and then those that end in `/` which indicates /// that the path post the `/` should be appended to resolved specifier. This /// handles both cases, pulling any completions from the workspace completions. fn get_import_map_completions( _specifier: &ModuleSpecifier, text: &str, range: &lsp::Range, maybe_import_map: Option<&ImportMap>, ) -> Option { if !text.is_empty() { if let Some(import_map) = maybe_import_map { let mut specifiers = IndexSet::new(); for key in import_map.imports().keys() { // for some reason, the import_map stores keys that begin with `/` as // `file:///` in its index, so we have to reverse that here let key = if key.starts_with("file://") { FILE_PROTO_RE.replace(key, "").to_string() } else { key.to_string() }; if key.starts_with(text) && key != text { specifiers.insert(key.trim_end_matches('/').to_string()); } } if !specifiers.is_empty() { let items = specifiers .into_iter() .map(|specifier| lsp::CompletionItem { label: specifier.clone(), kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(import map)".to_string()), sort_text: Some("1".to_string()), text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range: *range, new_text: specifier, })), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect(), ), ..Default::default() }) .collect(); return Some(CompletionList { items, is_incomplete: false, }); } } } None } /// Return local completions that are relative to the base specifier. fn get_local_completions( base: &ModuleSpecifier, text: &str, range: &lsp::Range, resolver: &LspResolver, ) -> Option> { if base.scheme() != "file" { return None; } let parent = base.join(text).ok()?.join(".").ok()?; let resolved_parent = resolver .as_graph_resolver(Some(base)) .resolve( parent.as_str(), &Range { specifier: base.clone(), start: deno_graph::Position::zeroed(), end: deno_graph::Position::zeroed(), }, ResolutionMode::Execution, ) .ok()?; let resolved_parent_path = specifier_to_file_path(&resolved_parent).ok()?; let raw_parent = &text[..text.char_indices().rfind(|(_, c)| *c == '/')?.0 + 1]; if resolved_parent_path.is_dir() { let cwd = std::env::current_dir().ok()?; let items = std::fs::read_dir(resolved_parent_path).ok()?; Some( items .filter_map(|de| { let de = de.ok()?; let label = de.path().file_name()?.to_string_lossy().to_string(); let entry_specifier = resolve_path(de.path().to_str()?, &cwd).ok()?; if entry_specifier == *base { return None; } let full_text = format!("{raw_parent}{label}"); let text_edit = Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range: *range, new_text: full_text.clone(), })); let filter_text = Some(full_text); match de.file_type() { Ok(file_type) if file_type.is_dir() => Some(lsp::CompletionItem { label, kind: Some(lsp::CompletionItemKind::FOLDER), detail: Some("(local)".to_string()), filter_text, sort_text: Some("1".to_string()), text_edit, commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect(), ), ..Default::default() }), Ok(file_type) if file_type.is_file() => { if is_importable_ext(&de.path()) { Some(lsp::CompletionItem { label, kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(local)".to_string()), filter_text, sort_text: Some("1".to_string()), text_edit, commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect(), ), ..Default::default() }) } else { None } } _ => None, } }) .collect(), ) } else { None } } fn get_relative_specifiers( base: &ModuleSpecifier, specifiers: Vec, ) -> Vec { specifiers .iter() .filter_map(|s| { if s != base { Some(relative_specifier(base, s).unwrap_or_else(|| s.to_string())) } else { None } }) .collect() } /// Find the index of the '@' delimiting the package name and version, if any. fn parse_bare_specifier_version_index(bare_specifier: &str) -> Option { if bare_specifier.starts_with('@') { bare_specifier .find('/') .filter(|idx| !bare_specifier[1..*idx].is_empty()) .and_then(|idx| { bare_specifier[idx..] .find('@') .filter(|idx2| !bare_specifier[idx..][1..*idx2].is_empty()) .filter(|idx2| !bare_specifier[idx..][1..*idx2].contains('/')) .map(|idx2| idx + idx2) }) } else { bare_specifier .find('@') .filter(|idx| !bare_specifier[1..*idx].is_empty()) .filter(|idx| !bare_specifier[1..*idx].contains('/')) } } async fn get_jsr_completions( referrer: &ModuleSpecifier, specifier: &str, range: &lsp::Range, resolved: Option<&ModuleSpecifier>, jsr_search_api: &impl PackageSearchApi, jsr_resolver: Option<&JsrFetchResolver>, ) -> Option { // First try to match `jsr:some-package@some-version/`. let req_ref = resolved .and_then(|s| JsrPackageReqReference::from_specifier(s).ok()) .or_else(|| JsrPackageReqReference::from_str(specifier).ok()); if let Some(req_ref) = req_ref { let sub_path = req_ref.sub_path(); if sub_path.is_some() || specifier.ends_with('/') { let export_prefix = sub_path.unwrap_or(""); let req = req_ref.req(); let nv = match jsr_resolver { Some(jsr_resolver) => jsr_resolver.req_to_nv(req).await, None => None, }; let nv = nv.or_else(|| PackageNv::from_str(&req.to_string()).ok())?; let exports = jsr_search_api.exports(&nv).await.ok()?; let items = exports .iter() .enumerate() .filter_map(|(idx, export)| { if export == "." { return None; } let export = export.strip_prefix("./").unwrap_or(export.as_str()); if !export.starts_with(export_prefix) { return None; } let specifier = format!( "{}/{export}", specifier.strip_suffix(export_prefix)?.trim_end_matches('/') ); let command = Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!([&specifier]), json!(referrer), json!({ "forceGlobalCache": true }), ]), }); let text_edit = Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range: *range, new_text: specifier.clone(), })); Some(lsp::CompletionItem { label: specifier, kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(jsr)".to_string()), sort_text: Some(format!("{:0>10}", idx + 1)), text_edit, command, commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect(), ), ..Default::default() }) }) .collect(); return Some(CompletionList { is_incomplete: false, items, }); } } // Then try to match `jsr:some-package@`. let bare_specifier = specifier.strip_prefix("jsr:")?; if let Some(v_index) = parse_bare_specifier_version_index(bare_specifier) { let package_name = &bare_specifier[..v_index]; let v_prefix = &bare_specifier[(v_index + 1)..]; let versions = jsr_search_api.versions(package_name).await.ok()?; let items = versions .iter() .enumerate() .filter_map(|(idx, version)| { let version = version.to_string(); if !version.starts_with(v_prefix) { return None; } let specifier = format!("jsr:{}@{}", package_name, version); let command = Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!([&specifier]), json!(referrer), json!({ "forceGlobalCache": true }), ]), }); let text_edit = Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range: *range, new_text: specifier.clone(), })); Some(lsp::CompletionItem { label: specifier, kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(jsr)".to_string()), sort_text: Some(format!("{:0>10}", idx + 1)), text_edit, command, commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect(), ), ..Default::default() }) }) .collect(); return Some(CompletionList { is_incomplete: false, items, }); } // Otherwise match `jsr:`. let names = jsr_search_api.search(bare_specifier).await.ok()?; let items = names .iter() .enumerate() .map(|(idx, name)| { let specifier = format!("jsr:{}", name); let command = Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!([&specifier]), json!(referrer), json!({ "forceGlobalCache": true }), ]), }); let text_edit = Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range: *range, new_text: specifier.clone(), })); lsp::CompletionItem { label: specifier, kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(jsr)".to_string()), sort_text: Some(format!("{:0>10}", idx + 1)), text_edit, command, commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect(), ), ..Default::default() } }) .collect(); Some(CompletionList { is_incomplete: true, items, }) } /// Get completions for `npm:` specifiers. async fn get_npm_completions( referrer: &ModuleSpecifier, specifier: &str, range: &lsp::Range, npm_search_api: &impl PackageSearchApi, ) -> Option { // First try to match `npm:some-package@`. let bare_specifier = specifier.strip_prefix("npm:")?; if let Some(v_index) = parse_bare_specifier_version_index(bare_specifier) { let package_name = &bare_specifier[..v_index]; let v_prefix = &bare_specifier[(v_index + 1)..]; let versions = npm_search_api.versions(package_name).await.ok()?; let items = versions .iter() .enumerate() .filter_map(|(idx, version)| { let version = version.to_string(); if !version.starts_with(v_prefix) { return None; } let specifier = format!("npm:{}@{}", package_name, version); let command = Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!([&specifier]), json!(referrer), json!({ "forceGlobalCache": true }), ]), }); let text_edit = Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range: *range, new_text: specifier.clone(), })); Some(lsp::CompletionItem { label: specifier, kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(npm)".to_string()), sort_text: Some(format!("{:0>10}", idx + 1)), text_edit, command, commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect(), ), ..Default::default() }) }) .collect(); return Some(CompletionList { is_incomplete: false, items, }); } // Otherwise match `npm:`. let names = npm_search_api.search(bare_specifier).await.ok()?; let items = names .iter() .enumerate() .map(|(idx, name)| { let specifier = format!("npm:{}", name); let command = Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!([&specifier]), json!(referrer), json!({ "forceGlobalCache": true }), ]), }); let text_edit = Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range: *range, new_text: specifier.clone(), })); lsp::CompletionItem { label: specifier, kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(npm)".to_string()), sort_text: Some(format!("{:0>10}", idx + 1)), text_edit, command, commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect(), ), ..Default::default() } }) .collect(); Some(CompletionList { is_incomplete: true, items, }) } /// Get completions for `node:` specifiers. fn get_node_completions( specifier: &str, range: &lsp::Range, ) -> Option { if !specifier.starts_with("node:") { return None; } let items = SUPPORTED_BUILTIN_NODE_MODULES .iter() .map(|name| { let specifier = format!("node:{}", name); let text_edit = Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range: *range, new_text: specifier.clone(), })); lsp::CompletionItem { label: specifier, kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(node)".to_string()), text_edit, commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect(), ), ..Default::default() } }) .collect(); Some(CompletionList { is_incomplete: false, items, }) } /// Get workspace completions that include modules in the Deno cache which match /// the current specifier string. fn get_workspace_completions( specifier: &ModuleSpecifier, current: &str, range: &lsp::Range, documents: &Documents, ) -> Vec { let workspace_specifiers = documents .documents(DocumentsFilter::AllDiagnosable) .into_iter() .map(|d| d.specifier().clone()) .collect(); let specifier_strings = get_relative_specifiers(specifier, workspace_specifiers); specifier_strings .into_iter() .filter_map(|label| { if label.starts_with(current) { let detail = Some( if label.starts_with("http:") || label.starts_with("https:") { "(remote)".to_string() } else if label.starts_with("data:") { "(data)".to_string() } else { "(local)".to_string() }, ); let text_edit = Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range: *range, new_text: label.clone(), })); Some(lsp::CompletionItem { label, kind: Some(lsp::CompletionItemKind::FILE), detail, sort_text: Some("1".to_string()), text_edit, commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect(), ), ..Default::default() }) } else { None } }) .collect() } #[cfg(test)] mod tests { use super::*; use crate::cache::HttpCache; use crate::lsp::cache::LspCache; use crate::lsp::documents::Documents; use crate::lsp::documents::LanguageId; use crate::lsp::search::tests::TestPackageSearchApi; use deno_core::resolve_url; use deno_graph::Range; use pretty_assertions::assert_eq; use std::collections::HashMap; use test_util::TempDir; fn setup( open_sources: &[(&str, &str, i32, LanguageId)], fs_sources: &[(&str, &str)], ) -> Documents { let temp_dir = TempDir::new(); let cache = LspCache::new(Some(temp_dir.uri().join(".deno_dir").unwrap())); let mut documents = Documents::default(); documents.update_config( &Default::default(), &Default::default(), &cache, &Default::default(), ); for (specifier, source, version, language_id) in open_sources { let specifier = resolve_url(specifier).expect("failed to create specifier"); documents.open(specifier, *version, *language_id, (*source).into(), None); } for (specifier, source) in fs_sources { let specifier = resolve_url(specifier).expect("failed to create specifier"); cache .global() .set(&specifier, HashMap::default(), source.as_bytes()) .expect("could not cache file"); let document = documents .get_or_load(&specifier, Some(&temp_dir.uri().join("$").unwrap())); assert!(document.is_some(), "source could not be setup"); } documents } #[test] fn test_get_relative_specifiers() { let base = resolve_url("file:///a/b/c.ts").unwrap(); let specifiers = vec![ resolve_url("file:///a/b/c.ts").unwrap(), resolve_url("file:///a/b/d.ts").unwrap(), resolve_url("file:///a/c/c.ts").unwrap(), resolve_url("file:///a/b/d/d.ts").unwrap(), resolve_url("https://deno.land/x/a/b/c.ts").unwrap(), ]; assert_eq!( get_relative_specifiers(&base, specifiers), vec![ "./d.ts".to_string(), "../c/c.ts".to_string(), "./d/d.ts".to_string(), "https://deno.land/x/a/b/c.ts".to_string(), ] ); } #[test] fn test_get_local_completions() { let temp_dir = TempDir::new(); let fixtures = temp_dir.path().join("fixtures"); std::fs::create_dir(&fixtures).expect("could not create"); let dir_a = fixtures.join("a"); std::fs::create_dir(&dir_a).expect("could not create"); let dir_b = dir_a.join("b"); std::fs::create_dir(&dir_b).expect("could not create"); let file_c = dir_a.join("c.ts"); std::fs::write(&file_c, b"").expect("could not create"); let file_d = dir_b.join("d.ts"); std::fs::write(file_d, b"").expect("could not create"); let file_e = dir_a.join("e.txt"); std::fs::write(file_e, b"").expect("could not create"); let file_f = dir_a.join("f.mjs"); std::fs::write(file_f, b"").expect("could not create"); let file_g = dir_a.join("g.json"); std::fs::write(file_g, b"").expect("could not create"); let specifier = ModuleSpecifier::from_file_path(file_c).expect("could not create"); let actual = get_local_completions( &specifier, "./", &lsp::Range { start: lsp::Position { line: 0, character: 20, }, end: lsp::Position { line: 0, character: 22, }, }, &Default::default(), ); assert!(actual.is_some()); let actual = actual.unwrap(); assert_eq!(actual.len(), 3); for item in actual { match item.text_edit { Some(lsp::CompletionTextEdit::Edit(text_edit)) => { assert!(["./b", "./f.mjs", "./g.json"] .contains(&text_edit.new_text.as_str())); } _ => unreachable!(), } } } #[tokio::test] async fn test_get_workspace_completions() { let specifier = resolve_url("file:///a/b/c.ts").unwrap(); let range = lsp::Range { start: lsp::Position { line: 0, character: 20, }, end: lsp::Position { line: 0, character: 21, }, }; let documents = setup( &[ ( "file:///a/b/c.ts", "import * as d from \"h\"", 1, LanguageId::TypeScript, ), ("file:///a/c.ts", r#""#, 1, LanguageId::TypeScript), ], &[("https://deno.land/x/a/b/c.ts", "console.log(1);\n")], ); let actual = get_workspace_completions(&specifier, "h", &range, &documents); assert_eq!( actual, vec![lsp::CompletionItem { label: "https://deno.land/x/a/b/c.ts".to_string(), kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(remote)".to_string()), sort_text: Some("1".to_string()), text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range: lsp::Range { start: lsp::Position { line: 0, character: 20 }, end: lsp::Position { line: 0, character: 21, } }, new_text: "https://deno.land/x/a/b/c.ts".to_string(), })), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect() ), ..Default::default() }] ); } #[test] fn test_parse_bare_specifier_version_index() { assert_eq!(parse_bare_specifier_version_index(""), None); assert_eq!(parse_bare_specifier_version_index("/"), None); assert_eq!(parse_bare_specifier_version_index("/@"), None); assert_eq!(parse_bare_specifier_version_index("@"), None); assert_eq!(parse_bare_specifier_version_index("@/"), None); assert_eq!(parse_bare_specifier_version_index("@/@"), None); assert_eq!(parse_bare_specifier_version_index("foo"), None); assert_eq!(parse_bare_specifier_version_index("foo/bar"), None); assert_eq!(parse_bare_specifier_version_index("foo/bar@"), None); assert_eq!(parse_bare_specifier_version_index("@org/foo/bar"), None); assert_eq!(parse_bare_specifier_version_index("@org/foo/bar@"), None); assert_eq!(parse_bare_specifier_version_index("foo@"), Some(3)); assert_eq!(parse_bare_specifier_version_index("foo@1."), Some(3)); assert_eq!(parse_bare_specifier_version_index("@org/foo@"), Some(8)); assert_eq!(parse_bare_specifier_version_index("@org/foo@1."), Some(8)); // Regression test for https://github.com/denoland/deno/issues/22325. assert_eq!( parse_bare_specifier_version_index( "@longer_than_right_one/arbitrary_string@" ), Some(39) ); } #[tokio::test] async fn test_get_jsr_completions() { let jsr_search_api = TestPackageSearchApi::default() .with_package_version("@std/archive", "1.0.0", &[]) .with_package_version("@std/assert", "1.0.0", &[]) .with_package_version("@std/async", "1.0.0", &[]) .with_package_version("@std/bytes", "1.0.0", &[]); let range = lsp::Range { start: lsp::Position { line: 0, character: 23, }, end: lsp::Position { line: 0, character: 29, }, }; let referrer = ModuleSpecifier::parse("file:///referrer.ts").unwrap(); let actual = get_jsr_completions( &referrer, "jsr:as", &range, None, &jsr_search_api, None, ) .await .unwrap(); assert_eq!( actual, CompletionList { is_incomplete: true, items: vec![ lsp::CompletionItem { label: "jsr:@std/assert".to_string(), kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(jsr)".to_string()), sort_text: Some("0000000001".to_string()), text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range, new_text: "jsr:@std/assert".to_string(), })), command: Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!(["jsr:@std/assert"]), json!(&referrer), json!({ "forceGlobalCache": true }) ]) }), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect() ), ..Default::default() }, lsp::CompletionItem { label: "jsr:@std/async".to_string(), kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(jsr)".to_string()), sort_text: Some("0000000002".to_string()), text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range, new_text: "jsr:@std/async".to_string(), })), command: Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!(["jsr:@std/async"]), json!(&referrer), json!({ "forceGlobalCache": true }) ]) }), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect() ), ..Default::default() }, ], } ); } #[tokio::test] async fn test_get_jsr_completions_for_versions() { let jsr_search_api = TestPackageSearchApi::default() .with_package_version("@std/assert", "0.3.0", &[]) .with_package_version("@std/assert", "0.4.0", &[]) .with_package_version("@std/assert", "0.5.0", &[]); let range = lsp::Range { start: lsp::Position { line: 0, character: 23, }, end: lsp::Position { line: 0, character: 39, }, }; let referrer = ModuleSpecifier::parse("file:///referrer.ts").unwrap(); let actual = get_jsr_completions( &referrer, "jsr:@std/assert@", &range, None, &jsr_search_api, None, ) .await .unwrap(); assert_eq!( actual, CompletionList { is_incomplete: false, items: vec![ lsp::CompletionItem { label: "jsr:@std/assert@0.5.0".to_string(), kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(jsr)".to_string()), sort_text: Some("0000000001".to_string()), text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range, new_text: "jsr:@std/assert@0.5.0".to_string(), })), command: Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!(["jsr:@std/assert@0.5.0"]), json!(&referrer), json!({ "forceGlobalCache": true }), ]) }), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect() ), ..Default::default() }, lsp::CompletionItem { label: "jsr:@std/assert@0.4.0".to_string(), kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(jsr)".to_string()), sort_text: Some("0000000002".to_string()), text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range, new_text: "jsr:@std/assert@0.4.0".to_string(), })), command: Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!(["jsr:@std/assert@0.4.0"]), json!(&referrer), json!({ "forceGlobalCache": true }), ]) }), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect() ), ..Default::default() }, lsp::CompletionItem { label: "jsr:@std/assert@0.3.0".to_string(), kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(jsr)".to_string()), sort_text: Some("0000000003".to_string()), text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range, new_text: "jsr:@std/assert@0.3.0".to_string(), })), command: Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!(["jsr:@std/assert@0.3.0"]), json!(&referrer), json!({ "forceGlobalCache": true }), ]) }), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect() ), ..Default::default() }, ], } ); } #[tokio::test] async fn test_get_jsr_completions_for_exports() { let jsr_search_api = TestPackageSearchApi::default().with_package_version( "@std/path", "0.1.0", &[".", "./basename", "./common", "./constants", "./dirname"], ); let range = lsp::Range { start: lsp::Position { line: 0, character: 23, }, end: lsp::Position { line: 0, character: 45, }, }; let referrer = ModuleSpecifier::parse("file:///referrer.ts").unwrap(); let actual = get_jsr_completions( &referrer, "jsr:@std/path@0.1.0/co", &range, None, &jsr_search_api, None, ) .await .unwrap(); assert_eq!( actual, CompletionList { is_incomplete: false, items: vec![ lsp::CompletionItem { label: "jsr:@std/path@0.1.0/common".to_string(), kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(jsr)".to_string()), sort_text: Some("0000000003".to_string()), text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range, new_text: "jsr:@std/path@0.1.0/common".to_string(), })), command: Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!(["jsr:@std/path@0.1.0/common"]), json!(&referrer), json!({ "forceGlobalCache": true }), ]) }), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect() ), ..Default::default() }, lsp::CompletionItem { label: "jsr:@std/path@0.1.0/constants".to_string(), kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(jsr)".to_string()), sort_text: Some("0000000004".to_string()), text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range, new_text: "jsr:@std/path@0.1.0/constants".to_string(), })), command: Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!(["jsr:@std/path@0.1.0/constants"]), json!(&referrer), json!({ "forceGlobalCache": true }), ]) }), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect() ), ..Default::default() }, ], } ); } #[tokio::test] async fn test_get_jsr_completions_for_exports_import_mapped() { let jsr_search_api = TestPackageSearchApi::default().with_package_version( "@std/path", "0.1.0", &[".", "./common"], ); let range = lsp::Range { start: lsp::Position { line: 0, character: 23, }, end: lsp::Position { line: 0, character: 45, }, }; let referrer = ModuleSpecifier::parse("file:///referrer.ts").unwrap(); let actual = get_jsr_completions( &referrer, "@std/path/co", &range, Some(&ModuleSpecifier::parse("jsr:@std/path@0.1.0/co").unwrap()), &jsr_search_api, None, ) .await .unwrap(); assert_eq!( actual, CompletionList { is_incomplete: false, items: vec![lsp::CompletionItem { label: "@std/path/common".to_string(), kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(jsr)".to_string()), sort_text: Some("0000000002".to_string()), text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range, new_text: "@std/path/common".to_string(), })), command: Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!(["@std/path/common"]), json!(&referrer), json!({ "forceGlobalCache": true }), ]) }), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect() ), ..Default::default() }, ], } ); } #[tokio::test] async fn test_get_npm_completions() { let npm_search_api = TestPackageSearchApi::default() .with_package_version("puppeteer", "1.0.0", &[]) .with_package_version("puppeteer-core", "1.0.0", &[]) .with_package_version("puppeteer-extra-plugin", "1.0.0", &[]) .with_package_version("puppeteer-extra-plugin-stealth", "1.0.0", &[]); let range = lsp::Range { start: lsp::Position { line: 0, character: 23, }, end: lsp::Position { line: 0, character: 32, }, }; let referrer = ModuleSpecifier::parse("file:///referrer.ts").unwrap(); let actual = get_npm_completions(&referrer, "npm:puppe", &range, &npm_search_api) .await .unwrap(); assert_eq!( actual, CompletionList { is_incomplete: true, items: vec![ lsp::CompletionItem { label: "npm:puppeteer".to_string(), kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(npm)".to_string()), sort_text: Some("0000000001".to_string()), text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range, new_text: "npm:puppeteer".to_string(), })), command: Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!(["npm:puppeteer"]), json!(&referrer), json!({ "forceGlobalCache": true }), ]) }), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect() ), ..Default::default() }, lsp::CompletionItem { label: "npm:puppeteer-core".to_string(), kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(npm)".to_string()), sort_text: Some("0000000002".to_string()), text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range, new_text: "npm:puppeteer-core".to_string(), })), command: Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!(["npm:puppeteer-core"]), json!(&referrer), json!({ "forceGlobalCache": true }), ]) }), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect() ), ..Default::default() }, lsp::CompletionItem { label: "npm:puppeteer-extra-plugin".to_string(), kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(npm)".to_string()), sort_text: Some("0000000003".to_string()), text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range, new_text: "npm:puppeteer-extra-plugin".to_string(), })), command: Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!(["npm:puppeteer-extra-plugin"]), json!(&referrer), json!({ "forceGlobalCache": true }), ]) }), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect() ), ..Default::default() }, lsp::CompletionItem { label: "npm:puppeteer-extra-plugin-stealth".to_string(), kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(npm)".to_string()), sort_text: Some("0000000004".to_string()), text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range, new_text: "npm:puppeteer-extra-plugin-stealth".to_string(), })), command: Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!(["npm:puppeteer-extra-plugin-stealth"]), json!(&referrer), json!({ "forceGlobalCache": true }), ]) }), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect() ), ..Default::default() }, ], } ); } #[tokio::test] async fn test_get_npm_completions_for_versions() { let npm_search_api = TestPackageSearchApi::default() .with_package_version("puppeteer", "20.9.0", &[]) .with_package_version("puppeteer", "21.0.0", &[]) .with_package_version("puppeteer", "21.0.1", &[]) .with_package_version("puppeteer", "21.0.2", &[]); let range = lsp::Range { start: lsp::Position { line: 0, character: 23, }, end: lsp::Position { line: 0, character: 37, }, }; let referrer = ModuleSpecifier::parse("file:///referrer.ts").unwrap(); let actual = get_npm_completions(&referrer, "npm:puppeteer@", &range, &npm_search_api) .await .unwrap(); assert_eq!( actual, CompletionList { is_incomplete: false, items: vec![ lsp::CompletionItem { label: "npm:puppeteer@21.0.2".to_string(), kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(npm)".to_string()), sort_text: Some("0000000001".to_string()), text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range, new_text: "npm:puppeteer@21.0.2".to_string(), })), command: Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!(["npm:puppeteer@21.0.2"]), json!(&referrer), json!({ "forceGlobalCache": true }), ]) }), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect() ), ..Default::default() }, lsp::CompletionItem { label: "npm:puppeteer@21.0.1".to_string(), kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(npm)".to_string()), sort_text: Some("0000000002".to_string()), text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range, new_text: "npm:puppeteer@21.0.1".to_string(), })), command: Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!(["npm:puppeteer@21.0.1"]), json!(&referrer), json!({ "forceGlobalCache": true }), ]) }), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect() ), ..Default::default() }, lsp::CompletionItem { label: "npm:puppeteer@21.0.0".to_string(), kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(npm)".to_string()), sort_text: Some("0000000003".to_string()), text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range, new_text: "npm:puppeteer@21.0.0".to_string(), })), command: Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!(["npm:puppeteer@21.0.0"]), json!(&referrer), json!({ "forceGlobalCache": true }), ]) }), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect() ), ..Default::default() }, lsp::CompletionItem { label: "npm:puppeteer@20.9.0".to_string(), kind: Some(lsp::CompletionItemKind::FILE), detail: Some("(npm)".to_string()), sort_text: Some("0000000004".to_string()), text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit { range, new_text: "npm:puppeteer@20.9.0".to_string(), })), command: Some(lsp::Command { title: "".to_string(), command: "deno.cache".to_string(), arguments: Some(vec![ json!(["npm:puppeteer@20.9.0"]), json!(&referrer), json!({ "forceGlobalCache": true }), ]) }), commit_characters: Some( IMPORT_COMMIT_CHARS.iter().map(|&c| c.into()).collect() ), ..Default::default() }, ], } ); } #[test] fn test_to_narrow_lsp_range() { let text_info = SourceTextInfo::from_string(r#""te""#.to_string()); let range = to_narrow_lsp_range( &text_info, &Range { specifier: ModuleSpecifier::parse("https://deno.land").unwrap(), start: deno_graph::Position { line: 0, character: 0, }, end: deno_graph::Position { line: 0, character: text_info.text_str().chars().count(), }, }, ); assert_eq!(range.start.character, 1); assert_eq!( range.end.character, (text_info.text_str().chars().count() - 1) as u32 ); } #[test] fn test_to_narrow_lsp_range_no_trailing_quote() { let text_info = SourceTextInfo::from_string(r#""te"#.to_string()); let range = to_narrow_lsp_range( &text_info, &Range { specifier: ModuleSpecifier::parse("https://deno.land").unwrap(), start: deno_graph::Position { line: 0, character: 0, }, end: deno_graph::Position { line: 0, character: text_info.text_str().chars().count(), }, }, ); assert_eq!(range.start.character, 1); assert_eq!( range.end.character, text_info.text_str().chars().count() as u32 ); } }