diff --git a/crates/lsp/src/lib.rs b/crates/lsp/src/lib.rs index 7af00fa15a8..f5a7ae6ff29 100644 --- a/crates/lsp/src/lib.rs +++ b/crates/lsp/src/lib.rs @@ -29,6 +29,8 @@ use tower::Service; const TEST_COMMAND: &str = "nargo.test"; const TEST_CODELENS_TITLE: &str = "▶\u{fe0e} Run Test"; +const COMPILE_COMMAND: &str = "nargo.compile"; +const COMPILE_CODELENS_TITLE: &str = "▶\u{fe0e} Compile"; // State for the LSP gets implemented on this struct and is internal to the implementation pub struct LspState { @@ -185,7 +187,7 @@ fn on_code_lens_request( for package in &workspace { let (mut context, crate_id) = prepare_package(package); - // We ignore the warnings and errors produced by compilation for producing codelenses + // We ignore the warnings and errors produced by compilation for producing code lenses // because we can still get the test functions even if compilation fails let _ = check_crate(&mut context, crate_id, false); @@ -226,6 +228,73 @@ fn on_code_lens_request( lenses.push(lens); } + + if package.is_binary() { + if let Some(main_func_id) = context.get_main_function(&crate_id) { + let location = context.function_meta(&main_func_id).name.location; + let file_id = location.file; + + // Ignore diagnostics for any file that wasn't the file we saved + // TODO: In the future, we could create "related" diagnostics for these files + // TODO: This currently just appends the `.nr` file extension that we store as a constant, + // but that won't work if we accept other extensions + if fm.path(file_id).with_extension(FILE_EXTENSION) != file_path { + continue; + } + + let range = byte_span_to_range(files, file_id.as_usize(), location.span.into()) + .unwrap_or_default(); + + let command = Command { + title: COMPILE_CODELENS_TITLE.into(), + command: COMPILE_COMMAND.into(), + arguments: Some(vec![ + "--program-dir".into(), + format!("{}", workspace.root_dir.display()).into(), + "--package".into(), + format!("{}", package.name).into(), + ]), + }; + + let lens = CodeLens { range, command: command.into(), data: None }; + + lenses.push(lens); + } + } + + if package.is_contract() { + // Currently not looking to dedupe this since we don't have a clear decision on if the Contract stuff is staying + for contract in context.get_all_contracts(&crate_id) { + let location = contract.location; + let file_id = location.file; + + // Ignore diagnostics for any file that wasn't the file we saved + // TODO: In the future, we could create "related" diagnostics for these files + // TODO: This currently just appends the `.nr` file extension that we store as a constant, + // but that won't work if we accept other extensions + if fm.path(file_id).with_extension(FILE_EXTENSION) != file_path { + continue; + } + + let range = byte_span_to_range(files, file_id.as_usize(), location.span.into()) + .unwrap_or_default(); + + let command = Command { + title: COMPILE_CODELENS_TITLE.into(), + command: COMPILE_COMMAND.into(), + arguments: Some(vec![ + "--program-dir".into(), + format!("{}", workspace.root_dir.display()).into(), + "--package".into(), + format!("{}", package.name).into(), + ]), + }; + + let lens = CodeLens { range, command: command.into(), data: None }; + + lenses.push(lens); + } + } } let res = if lenses.is_empty() { Ok(None) } else { Ok(Some(lenses)) }; @@ -365,6 +434,9 @@ fn on_did_save_text_document( } } + // We need to refresh lenses when we compile since that's the only time they can be accurately reflected + let _ = state.client.code_lens_refresh(()); + let _ = state.client.publish_diagnostics(PublishDiagnosticsParams { uri: params.text_document.uri, version: None, diff --git a/crates/noirc_frontend/src/hir/def_map/mod.rs b/crates/noirc_frontend/src/hir/def_map/mod.rs index f264d3f40b8..2dc8c5ec96f 100644 --- a/crates/noirc_frontend/src/hir/def_map/mod.rs +++ b/crates/noirc_frontend/src/hir/def_map/mod.rs @@ -148,7 +148,7 @@ impl CrateDefMap { let functions = module.value_definitions().filter_map(|id| id.as_function()).collect(); let name = self.get_module_path(id, module.parent); - Some(Contract { name, functions }) + Some(Contract { name, location: module.location, functions }) } else { None } @@ -194,6 +194,7 @@ impl CrateDefMap { pub struct Contract { /// To keep `name` semi-unique, it is prefixed with the names of parent modules via CrateDefMap::get_module_path pub name: String, + pub location: Location, pub functions: Vec, }