From 403deb90a395dea64a8036e606524ccd79bf431c Mon Sep 17 00:00:00 2001
From: Tom French <15848336+TomAFrench@users.noreply.github.com>
Date: Wed, 4 Oct 2023 16:04:16 +0100
Subject: [PATCH] feat(lsp): add "info" codelens (#2982)

---
 tooling/lsp/src/codelens/mod.rs | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/tooling/lsp/src/codelens/mod.rs b/tooling/lsp/src/codelens/mod.rs
index 02a91b70074..db6ab4300f2 100644
--- a/tooling/lsp/src/codelens/mod.rs
+++ b/tooling/lsp/src/codelens/mod.rs
@@ -24,6 +24,8 @@ const TEST_COMMAND: &str = "nargo.test";
 const TEST_CODELENS_TITLE: &str = "Run Test";
 const COMPILE_COMMAND: &str = "nargo.compile";
 const COMPILE_CODELENS_TITLE: &str = "Compile";
+const INFO_COMMAND: &str = "nargo.info";
+const INFO_CODELENS_TITLE: &str = "Info";
 const EXECUTE_COMMAND: &str = "nargo.execute";
 const EXECUTE_CODELENS_TITLE: &str = "Execute";
 
@@ -148,6 +150,16 @@ fn on_code_lens_request_inner(
 
                 lenses.push(compile_lens);
 
+                let info_command = Command {
+                    title: INFO_CODELENS_TITLE.to_string(),
+                    command: INFO_COMMAND.into(),
+                    arguments: Some(package_selection_args(&workspace, package)),
+                };
+
+                let info_lens = CodeLens { range, command: Some(info_command), data: None };
+
+                lenses.push(info_lens);
+
                 let execute_command = Command {
                     title: EXECUTE_CODELENS_TITLE.to_string(),
                     command: EXECUTE_COMMAND.into(),
@@ -186,6 +198,16 @@ fn on_code_lens_request_inner(
                 let compile_lens = CodeLens { range, command: Some(compile_command), data: None };
 
                 lenses.push(compile_lens);
+
+                let info_command = Command {
+                    title: INFO_CODELENS_TITLE.to_string(),
+                    command: INFO_COMMAND.into(),
+                    arguments: Some(package_selection_args(&workspace, package)),
+                };
+
+                let info_lens = CodeLens { range, command: Some(info_command), data: None };
+
+                lenses.push(info_lens);
             }
         }
     }