From 4ac39f0c98648b7095332d8b821cf93c935fdc3f Mon Sep 17 00:00:00 2001 From: Ddystopia Date: Thu, 27 Apr 2023 18:05:04 +0200 Subject: [PATCH] Propagating `sysroot` down + Refactoring --- crates/ide/src/doc_links.rs | 45 ++++++++++---------- crates/ide/src/doc_links/tests.rs | 26 +++++++++-- crates/ide/src/lib.rs | 5 ++- crates/rust-analyzer/src/handlers/request.rs | 25 +++++++---- 4 files changed, 66 insertions(+), 35 deletions(-) diff --git a/crates/ide/src/doc_links.rs b/crates/ide/src/doc_links.rs index 39c290264bb..92021da4c09 100644 --- a/crates/ide/src/doc_links.rs +++ b/crates/ide/src/doc_links.rs @@ -133,6 +133,7 @@ pub(crate) fn external_docs( db: &RootDatabase, position: &FilePosition, target_dir: Option<&OsStr>, + sysroot: Option<&OsStr>, ) -> Option { let sema = &Semantics::new(db); let file = sema.parse(position.file_id).syntax().clone(); @@ -163,7 +164,7 @@ pub(crate) fn external_docs( } }; - Some(get_doc_links(db, definition, target_dir)) + Some(get_doc_links(db, definition, target_dir, sysroot)) } /// Extracts all links from a given markdown text returning the definition text range, link-text @@ -325,6 +326,7 @@ fn get_doc_links( db: &RootDatabase, def: Definition, target_dir: Option<&OsStr>, + sysroot: Option<&OsStr>, ) -> DocumentationLinks { let join_url = |base_url: Option, path: &str| -> Option { base_url.and_then(|url| url.join(path).ok()) @@ -332,7 +334,7 @@ fn get_doc_links( let Some((target, file, frag)) = filename_and_frag_for_def(db, def) else { return Default::default(); }; - let (mut web_url, mut local_url) = get_doc_base_urls(db, target, target_dir); + let (mut web_url, mut local_url) = get_doc_base_urls(db, target, target_dir, sysroot); if let Some(path) = mod_path_of_def(db, target) { web_url = join_url(web_url, &path); @@ -360,7 +362,7 @@ fn rewrite_intra_doc_link( let (link, ns) = parse_intra_doc_link(target); let resolved = resolve_doc_path_for_def(db, def, link, ns)?; - let mut url = get_doc_base_urls(db, resolved, None).0?; + let mut url = get_doc_base_urls(db, resolved, None, None).0?; let (_, file, frag) = filename_and_frag_for_def(db, resolved)?; if let Some(path) = mod_path_of_def(db, resolved) { @@ -379,7 +381,7 @@ fn rewrite_url_link(db: &RootDatabase, def: Definition, target: &str) -> Option< return None; } - let mut url = get_doc_base_urls(db, def, None).0?; + let mut url = get_doc_base_urls(db, def, None, None).0?; let (def, file, frag) = filename_and_frag_for_def(db, def)?; if let Some(path) = mod_path_of_def(db, def) { @@ -461,27 +463,29 @@ fn get_doc_base_urls( db: &RootDatabase, def: Definition, target_dir: Option<&OsStr>, + sysroot: Option<&OsStr>, ) -> (Option, Option) { - let local_doc_path = target_dir - .and_then(|path: &OsStr| -> Option { - let mut with_prefix = OsStr::new("file:///").to_os_string(); - with_prefix.push(path); - with_prefix.push("/"); - with_prefix.to_str().and_then(|s| Url::parse(s).ok()) - }) + let local_doc = target_dir + .and_then(|path| path.to_str()) + .and_then(|path| Url::parse(&format!("file:///{path}/")).ok()) .and_then(|it| it.join("doc/").ok()); + let system_doc = sysroot + .and_then(|it| it.to_str()) + .map(|sysroot| format!("file:///{sysroot}/share/doc/rust/html/")) + .and_then(|it| Url::parse(&it).ok()); + // special case base url of `BuiltinType` to core // https://github.com/rust-lang/rust-analyzer/issues/12250 if let Definition::BuiltinType(..) = def { - let weblink = Url::parse("https://doc.rust-lang.org/nightly/core/").ok(); - return (weblink, None); + let web_link = Url::parse("https://doc.rust-lang.org/nightly/core/").ok(); + let system_link = system_doc.and_then(|it| it.join("core/").ok()); + return (web_link, system_link); }; let Some(krate) = def.krate(db) else { return Default::default() }; let Some(display_name) = krate.display_name(db) else { return Default::default() }; let crate_data = &db.crate_graph()[krate.into()]; let channel = crate_data.channel.map_or("nightly", ReleaseChannel::as_str); - let sysroot = "/home/ddystopia/.rustup/toolchains/stable-x86_64-unknown-linux-gnu"; let (web_base, local_base) = match &crate_data.origin { // std and co do not specify `html_root_url` any longer so we gotta handwrite this ourself. @@ -493,13 +497,10 @@ fn get_doc_base_urls( | LangCrateOrigin::Std | LangCrateOrigin::Test), ) => { - let local_url = format!("file:///{sysroot}/share/doc/rust/html/{origin}/index.html"); - let local_url = Url::parse(&local_url).ok(); + let system_url = system_doc.and_then(|it| it.join(&format!("{origin}")).ok()); let web_url = format!("https://doc.rust-lang.org/{channel}/{origin}"); - println!("local_url: {:?}", local_url.unwrap().to_string()); - panic!(); - (Some(web_url), local_url) - }, + (Some(web_url), system_url) + } CrateOrigin::Lang(_) => return (None, None), CrateOrigin::Rustc { name: _ } => { (Some(format!("https://doc.rust-lang.org/{channel}/nightly-rustc/")), None) @@ -519,7 +520,7 @@ fn get_doc_base_urls( version = version.as_deref().unwrap_or("*") )) }); - (weblink, local_doc_path) + (weblink, local_doc) } CrateOrigin::Library { repo: _, name } => { let weblink = krate.get_html_root_url(db).or_else(|| { @@ -535,7 +536,7 @@ fn get_doc_base_urls( version = version.as_deref().unwrap_or("*") )) }); - (weblink, local_doc_path) + (weblink, local_doc) } }; let web_base = web_base diff --git a/crates/ide/src/doc_links/tests.rs b/crates/ide/src/doc_links/tests.rs index 4159d3ed816..05a64b33bfd 100644 --- a/crates/ide/src/doc_links/tests.rs +++ b/crates/ide/src/doc_links/tests.rs @@ -20,9 +20,10 @@ fn check_external_docs( target_dir: Option<&OsStr>, expect_web_url: Option, expect_local_url: Option, + sysroot: Option<&OsStr>, ) { let (analysis, position) = fixture::position(ra_fixture); - let links = analysis.external_docs(position, target_dir).unwrap(); + let links = analysis.external_docs(position, target_dir, sysroot).unwrap(); let web_url = links.web_url; let local_url = links.local_url; @@ -128,7 +129,8 @@ let x: u3$02 = 0; "#, Some(&OsStr::new("/home/user/project")), Some(expect![[r#"https://doc.rust-lang.org/nightly/core/primitive.u32.html"#]]), - None, + Some(expect![[r#"file:///sysroot/share/doc/rust/html/core/primitive.u32.html"#]]), + Some(&OsStr::new("/sysroot")), ); } @@ -144,6 +146,7 @@ pub struct Foo; Some(&OsStr::new("/home/user/project")), Some(expect![[r#"https://docs.rs/foo/*/foo/index.html"#]]), Some(expect![[r#"file:///home/user/project/doc/foo/index.html"#]]), + Some(&OsStr::new("/sysroot")), ); } @@ -156,7 +159,8 @@ use self$0; "#, Some(&OsStr::new("/home/user/project")), Some(expect!["https://doc.rust-lang.org/stable/std/index.html"]), - None, + Some(expect!["file:///sysroot/share/doc/rust/html/std/index.html"]), + Some(&OsStr::new("/sysroot")), ); } @@ -170,6 +174,7 @@ pub struct Fo$0o; Some(&OsStr::new("/home/user/project")), Some(expect![[r#"https://docs.rs/foo/*/foo/struct.Foo.html"#]]), Some(expect![[r#"file:///home/user/project/doc/foo/struct.Foo.html"#]]), + Some(&OsStr::new("/sysroot")), ); } @@ -183,6 +188,7 @@ pub struct Fo$0o; Some(&OsStr::new(r"C:\Users\user\project")), Some(expect![[r#"https://docs.rs/foo/*/foo/struct.Foo.html"#]]), Some(expect![[r#"file:///C:/Users/user/project/doc/foo/struct.Foo.html"#]]), + Some(&OsStr::new("/sysroot")), ); } @@ -196,6 +202,7 @@ pub struct Fo$0o; Some(&OsStr::new(r"C:/Users/user/project")), Some(expect![[r#"https://docs.rs/foo/*/foo/struct.Foo.html"#]]), Some(expect![[r#"file:///C:/Users/user/project/doc/foo/struct.Foo.html"#]]), + Some(&OsStr::new("/sysroot")), ); } @@ -211,6 +218,7 @@ pub struct Foo { None, Some(expect![[r##"https://docs.rs/foo/*/foo/struct.Foo.html#structfield.field"##]]), None, + None, ); } @@ -224,6 +232,7 @@ pub fn fo$0o() {} None, Some(expect![[r#"https://docs.rs/foo/*/foo/fn.foo.html"#]]), None, + None, ); } @@ -240,6 +249,7 @@ impl Foo { None, Some(expect![[r##"https://docs.rs/foo/*/foo/struct.Foo.html#method.method"##]]), None, + None, ); check_external_docs( r#" @@ -252,6 +262,7 @@ impl Foo { None, Some(expect![[r##"https://docs.rs/foo/*/foo/struct.Foo.html#associatedconstant.CONST"##]]), None, + None, ); } @@ -271,6 +282,7 @@ impl Trait for Foo { None, Some(expect![[r##"https://docs.rs/foo/*/foo/struct.Foo.html#method.method"##]]), None, + None, ); check_external_docs( r#" @@ -286,6 +298,7 @@ impl Trait for Foo { None, Some(expect![[r##"https://docs.rs/foo/*/foo/struct.Foo.html#associatedconstant.CONST"##]]), None, + None, ); check_external_docs( r#" @@ -301,6 +314,7 @@ impl Trait for Foo { None, Some(expect![[r##"https://docs.rs/foo/*/foo/struct.Foo.html#associatedtype.Type"##]]), None, + None, ); } @@ -316,6 +330,7 @@ pub trait Foo { None, Some(expect![[r##"https://docs.rs/foo/*/foo/trait.Foo.html#tymethod.method"##]]), None, + None, ); check_external_docs( r#" @@ -327,6 +342,7 @@ pub trait Foo { None, Some(expect![[r##"https://docs.rs/foo/*/foo/trait.Foo.html#associatedconstant.CONST"##]]), None, + None, ); check_external_docs( r#" @@ -338,6 +354,7 @@ pub trait Foo { None, Some(expect![[r##"https://docs.rs/foo/*/foo/trait.Foo.html#associatedtype.Type"##]]), None, + None, ); } @@ -351,6 +368,7 @@ trait Trait$0 {} None, Some(expect![[r#"https://docs.rs/foo/*/foo/trait.Trait.html"#]]), None, + None, ) } @@ -366,6 +384,7 @@ pub mod foo { None, Some(expect![[r#"https://docs.rs/foo/*/foo/foo/bar/index.html"#]]), None, + None, ) } @@ -389,6 +408,7 @@ fn foo() { None, Some(expect![[r#"https://docs.rs/foo/*/foo/wrapper/module/struct.Item.html"#]]), None, + None, ) } diff --git a/crates/ide/src/lib.rs b/crates/ide/src/lib.rs index 7678ef5250c..1af4d39265f 100644 --- a/crates/ide/src/lib.rs +++ b/crates/ide/src/lib.rs @@ -475,8 +475,11 @@ impl Analysis { &self, position: FilePosition, target_dir: Option<&OsStr>, + sysroot: Option<&OsStr>, ) -> Cancellable { - self.with_db(|db| doc_links::external_docs(db, &position, target_dir).unwrap_or_default()) + self.with_db(|db| { + doc_links::external_docs(db, &position, target_dir, sysroot).unwrap_or_default() + }) } /// Computes parameter information at the given position. diff --git a/crates/rust-analyzer/src/handlers/request.rs b/crates/rust-analyzer/src/handlers/request.rs index 3d939b98471..65541bfe141 100644 --- a/crates/rust-analyzer/src/handlers/request.rs +++ b/crates/rust-analyzer/src/handlers/request.rs @@ -1534,20 +1534,26 @@ pub(crate) fn handle_semantic_tokens_range( pub(crate) fn handle_open_docs( snap: GlobalStateSnapshot, - params: lsp_types::TextDocumentPositionParams, -) -> Result<(Option, Option)> { + params: lsp_types::TextDocumentPositionParams, + ) -> Result<(Option, Option)> { let _p = profile::span("handle_open_docs"); - let file_uri = ¶ms.text_document.uri; - let file_id = from_proto::file_id(&snap, file_uri)?; let position = from_proto::file_position(&snap, params)?; - let cargo = match &*snap.analysis.crates_for(file_id)? { - &[crate_id, ..] => snap.cargo_target_for_crate_root(crate_id).map(|(it, _)| it), - _ => None, - }; + let ws_and_sysroot = snap.workspaces.iter().find_map(|ws| match ws { + ProjectWorkspace::Cargo { cargo, sysroot, .. } => Some((cargo, sysroot.as_ref().ok())), + ProjectWorkspace::Json { .. } => None, + ProjectWorkspace::DetachedFiles { .. } => None, + }); + let (cargo, sysroot) = match ws_and_sysroot { + Some((ws, Some(sysroot))) => (Some(ws), Some(sysroot)), + _ => (None, None), + }; + + let sysroot = sysroot.map(|p| p.root().as_os_str()); let target_dir = cargo.map(|cargo| cargo.target_directory()).map(|p| p.as_os_str()); - let Ok(remote_urls) = snap.analysis.external_docs(position, target_dir) else { return Ok((None, None)); }; + + let Ok(remote_urls) = snap.analysis.external_docs(position, target_dir, sysroot) else { return Ok((None, None)); }; let web_url = remote_urls.web_url.and_then(|it| Url::parse(&it).ok()); let local_url = remote_urls.local_url.and_then(|it| Url::parse(&it).ok()); @@ -1555,6 +1561,7 @@ pub(crate) fn handle_open_docs( Ok((web_url, local_url)) } + pub(crate) fn handle_open_cargo_toml( snap: GlobalStateSnapshot, params: lsp_ext::OpenCargoTomlParams,