New runnables API

This commit is contained in:
Aleksey Kladov 2020-06-02 17:22:23 +02:00
parent f137b3a4e6
commit 0303982119
8 changed files with 108 additions and 118 deletions

View file

@ -92,15 +92,16 @@ impl NavigationTarget {
let name = module.name(db).map(|it| it.to_string().into()).unwrap_or_default(); let name = module.name(db).map(|it| it.to_string().into()).unwrap_or_default();
if let Some(src) = module.declaration_source(db) { if let Some(src) = module.declaration_source(db) {
let frange = original_range(db, src.as_ref().map(|it| it.syntax())); let frange = original_range(db, src.as_ref().map(|it| it.syntax()));
return NavigationTarget::from_syntax( let mut res = NavigationTarget::from_syntax(
frange.file_id, frange.file_id,
name, name,
None, None,
frange.range, frange.range,
src.value.syntax().kind(), src.value.syntax().kind(),
src.value.doc_comment_text(),
src.value.short_label(),
); );
res.docs = src.value.doc_comment_text();
res.description = src.value.short_label();
return res;
} }
module.to_nav(db) module.to_nav(db)
} }
@ -130,11 +131,9 @@ impl NavigationTarget {
} }
/// Allows `NavigationTarget` to be created from a `NameOwner` /// Allows `NavigationTarget` to be created from a `NameOwner`
fn from_named( pub(crate) fn from_named(
db: &RootDatabase, db: &RootDatabase,
node: InFile<&dyn ast::NameOwner>, node: InFile<&dyn ast::NameOwner>,
docs: Option<String>,
description: Option<String>,
) -> NavigationTarget { ) -> NavigationTarget {
//FIXME: use `_` instead of empty string //FIXME: use `_` instead of empty string
let name = node.value.name().map(|it| it.text().clone()).unwrap_or_default(); let name = node.value.name().map(|it| it.text().clone()).unwrap_or_default();
@ -148,8 +147,6 @@ impl NavigationTarget {
focus_range, focus_range,
frange.range, frange.range,
node.value.syntax().kind(), node.value.syntax().kind(),
docs,
description,
) )
} }
@ -159,8 +156,6 @@ impl NavigationTarget {
focus_range: Option<TextRange>, focus_range: Option<TextRange>,
full_range: TextRange, full_range: TextRange,
kind: SyntaxKind, kind: SyntaxKind,
docs: Option<String>,
description: Option<String>,
) -> NavigationTarget { ) -> NavigationTarget {
NavigationTarget { NavigationTarget {
file_id, file_id,
@ -169,8 +164,8 @@ impl NavigationTarget {
full_range, full_range,
focus_range, focus_range,
container_name: None, container_name: None,
description, description: None,
docs, docs: None,
} }
} }
} }
@ -238,12 +233,11 @@ where
{ {
fn to_nav(&self, db: &RootDatabase) -> NavigationTarget { fn to_nav(&self, db: &RootDatabase) -> NavigationTarget {
let src = self.source(db); let src = self.source(db);
NavigationTarget::from_named( let mut res =
db, NavigationTarget::from_named(db, src.as_ref().map(|it| it as &dyn ast::NameOwner));
src.as_ref().map(|it| it as &dyn ast::NameOwner), res.docs = src.value.doc_comment_text();
src.value.doc_comment_text(), res.description = src.value.short_label();
src.value.short_label(), res
)
} }
} }
@ -258,15 +252,7 @@ impl ToNav for hir::Module {
} }
}; };
let frange = original_range(db, src.with_value(syntax)); let frange = original_range(db, src.with_value(syntax));
NavigationTarget::from_syntax( NavigationTarget::from_syntax(frange.file_id, name, focus, frange.range, syntax.kind())
frange.file_id,
name,
focus,
frange.range,
syntax.kind(),
None,
None,
)
} }
} }
@ -285,8 +271,6 @@ impl ToNav for hir::ImplDef {
None, None,
frange.range, frange.range,
src.value.syntax().kind(), src.value.syntax().kind(),
None,
None,
) )
} }
} }
@ -296,12 +280,12 @@ impl ToNav for hir::Field {
let src = self.source(db); let src = self.source(db);
match &src.value { match &src.value {
FieldSource::Named(it) => NavigationTarget::from_named( FieldSource::Named(it) => {
db, let mut res = NavigationTarget::from_named(db, src.with_value(it));
src.with_value(it), res.docs = it.doc_comment_text();
it.doc_comment_text(), res.description = it.short_label();
it.short_label(), res
), }
FieldSource::Pos(it) => { FieldSource::Pos(it) => {
let frange = original_range(db, src.with_value(it.syntax())); let frange = original_range(db, src.with_value(it.syntax()));
NavigationTarget::from_syntax( NavigationTarget::from_syntax(
@ -310,8 +294,6 @@ impl ToNav for hir::Field {
None, None,
frange.range, frange.range,
it.syntax().kind(), it.syntax().kind(),
None,
None,
) )
} }
} }
@ -322,12 +304,10 @@ impl ToNav for hir::MacroDef {
fn to_nav(&self, db: &RootDatabase) -> NavigationTarget { fn to_nav(&self, db: &RootDatabase) -> NavigationTarget {
let src = self.source(db); let src = self.source(db);
log::debug!("nav target {:#?}", src.value.syntax()); log::debug!("nav target {:#?}", src.value.syntax());
NavigationTarget::from_named( let mut res =
db, NavigationTarget::from_named(db, src.as_ref().map(|it| it as &dyn ast::NameOwner));
src.as_ref().map(|it| it as &dyn ast::NameOwner), res.docs = src.value.doc_comment_text();
src.value.doc_comment_text(), res
None,
)
} }
} }

View file

@ -1,19 +1,19 @@
use std::fmt;
use hir::{AsAssocItem, Attrs, HirFileId, InFile, Semantics}; use hir::{AsAssocItem, Attrs, HirFileId, InFile, Semantics};
use itertools::Itertools; use itertools::Itertools;
use ra_cfg::CfgExpr;
use ra_ide_db::RootDatabase; use ra_ide_db::RootDatabase;
use ra_syntax::{ use ra_syntax::{
ast::{self, AstNode, AttrsOwner, ModuleItemOwner, NameOwner}, ast::{self, AstNode, AttrsOwner, DocCommentsOwner, ModuleItemOwner, NameOwner},
match_ast, SyntaxNode, TextRange, match_ast, SyntaxNode,
}; };
use crate::FileId; use crate::{display::ToNav, FileId, NavigationTarget};
use ast::DocCommentsOwner;
use ra_cfg::CfgExpr;
use std::fmt::Display;
#[derive(Debug)] #[derive(Debug)]
pub struct Runnable { pub struct Runnable {
pub range: TextRange, pub nav: NavigationTarget,
pub kind: RunnableKind, pub kind: RunnableKind,
pub cfg_exprs: Vec<CfgExpr>, pub cfg_exprs: Vec<CfgExpr>,
} }
@ -24,8 +24,8 @@ pub enum TestId {
Path(String), Path(String),
} }
impl Display for TestId { impl fmt::Display for TestId {
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match self { match self {
TestId::Name(name) => write!(f, "{}", name), TestId::Name(name) => write!(f, "{}", name),
TestId::Path(path) => write!(f, "{}", path), TestId::Path(path) => write!(f, "{}", path),
@ -131,7 +131,8 @@ fn runnable_fn(
let cfg_exprs = let cfg_exprs =
attrs.by_key("cfg").tt_values().map(|subtree| ra_cfg::parse_cfg(subtree)).collect(); attrs.by_key("cfg").tt_values().map(|subtree| ra_cfg::parse_cfg(subtree)).collect();
Some(Runnable { range: fn_def.syntax().text_range(), kind, cfg_exprs }) let nav = NavigationTarget::from_named(sema.db, InFile::new(file_id.into(), &fn_def));
Some(Runnable { nav, kind, cfg_exprs })
} }
#[derive(Debug)] #[derive(Debug)]
@ -183,7 +184,6 @@ fn runnable_mod(
if !has_test_function { if !has_test_function {
return None; return None;
} }
let range = module.syntax().text_range();
let module_def = sema.to_def(&module)?; let module_def = sema.to_def(&module)?;
let path = module_def let path = module_def
@ -197,7 +197,8 @@ fn runnable_mod(
let cfg_exprs = let cfg_exprs =
attrs.by_key("cfg").tt_values().map(|subtree| ra_cfg::parse_cfg(subtree)).collect(); attrs.by_key("cfg").tt_values().map(|subtree| ra_cfg::parse_cfg(subtree)).collect();
Some(Runnable { range, kind: RunnableKind::TestMod { path }, cfg_exprs }) let nav = module_def.to_nav(sema.db);
Some(Runnable { nav, kind: RunnableKind::TestMod { path }, cfg_exprs })
} }
#[cfg(test)] #[cfg(test)]

View file

@ -4,7 +4,6 @@ use std::{collections::HashMap, path::PathBuf};
use lsp_types::request::Request; use lsp_types::request::Request;
use lsp_types::{Position, Range, TextDocumentIdentifier}; use lsp_types::{Position, Range, TextDocumentIdentifier};
use rustc_hash::FxHashMap;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
pub enum AnalyzerStatus {} pub enum AnalyzerStatus {}
@ -121,25 +120,30 @@ pub struct RunnablesParams {
pub position: Option<Position>, pub position: Option<Position>,
} }
// Must strictly correspond to the executable name #[derive(Deserialize, Serialize, Debug)]
#[serde(rename_all = "camelCase")]
pub struct Runnable {
pub label: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub location: Option<lsp_types::LocationLink>,
pub kind: RunnableKind,
pub args: CargoRunnable,
}
#[derive(Serialize, Deserialize, Debug)] #[derive(Serialize, Deserialize, Debug)]
#[serde(rename_all = "lowercase")] #[serde(rename_all = "lowercase")]
pub enum RunnableKind { pub enum RunnableKind {
Cargo, Cargo,
Rustc,
Rustup,
} }
#[derive(Deserialize, Serialize, Debug)] #[derive(Deserialize, Serialize, Debug)]
#[serde(rename_all = "camelCase")] #[serde(rename_all = "camelCase")]
pub struct Runnable { pub struct CargoRunnable {
pub range: Range, pub workspace_root: Option<PathBuf>,
pub label: String, // command, --package and --lib stuff
pub kind: RunnableKind, pub cargo_args: Vec<String>,
pub args: Vec<String>, // stuff after --
pub extra_args: Vec<String>, pub executable_args: Vec<String>,
pub env: FxHashMap<String, String>,
pub cwd: Option<PathBuf>,
} }
pub enum InlayHints {} pub enum InlayHints {}

View file

@ -23,7 +23,6 @@ use ra_ide::{
use ra_prof::profile; use ra_prof::profile;
use ra_project_model::TargetKind; use ra_project_model::TargetKind;
use ra_syntax::{AstNode, SyntaxKind, TextRange, TextSize}; use ra_syntax::{AstNode, SyntaxKind, TextRange, TextSize};
use rustc_hash::FxHashMap;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use serde_json::to_value; use serde_json::to_value;
use stdx::format_to; use stdx::format_to;
@ -401,7 +400,7 @@ pub fn handle_runnables(
let cargo_spec = CargoTargetSpec::for_file(&world, file_id)?; let cargo_spec = CargoTargetSpec::for_file(&world, file_id)?;
for runnable in world.analysis().runnables(file_id)? { for runnable in world.analysis().runnables(file_id)? {
if let Some(offset) = offset { if let Some(offset) = offset {
if !runnable.range.contains_inclusive(offset) { if !runnable.nav.full_range().contains_inclusive(offset) {
continue; continue;
} }
} }
@ -422,25 +421,31 @@ pub fn handle_runnables(
Some(spec) => { Some(spec) => {
for &cmd in ["check", "test"].iter() { for &cmd in ["check", "test"].iter() {
res.push(lsp_ext::Runnable { res.push(lsp_ext::Runnable {
range: Default::default(),
label: format!("cargo {} -p {}", cmd, spec.package), label: format!("cargo {} -p {}", cmd, spec.package),
location: None,
kind: lsp_ext::RunnableKind::Cargo, kind: lsp_ext::RunnableKind::Cargo,
args: vec![cmd.to_string(), "--package".to_string(), spec.package.clone()], args: lsp_ext::CargoRunnable {
extra_args: Vec::new(), workspace_root: workspace_root.map(|root| root.to_owned()),
env: FxHashMap::default(), cargo_args: vec![
cwd: workspace_root.map(|root| root.to_owned()), cmd.to_string(),
"--package".to_string(),
spec.package.clone(),
],
executable_args: Vec::new(),
},
}) })
} }
} }
None => { None => {
res.push(lsp_ext::Runnable { res.push(lsp_ext::Runnable {
range: Default::default(),
label: "cargo check --workspace".to_string(), label: "cargo check --workspace".to_string(),
location: None,
kind: lsp_ext::RunnableKind::Cargo, kind: lsp_ext::RunnableKind::Cargo,
args: vec!["check".to_string(), "--workspace".to_string()], args: lsp_ext::CargoRunnable {
extra_args: Vec::new(), workspace_root: workspace_root.map(|root| root.to_owned()),
env: FxHashMap::default(), cargo_args: vec!["check".to_string(), "--workspace".to_string()],
cwd: workspace_root.map(|root| root.to_owned()), executable_args: Vec::new(),
},
}); });
} }
} }
@ -782,10 +787,11 @@ pub fn handle_code_lens(
} }
}; };
let mut r = to_proto::runnable(&world, file_id, runnable)?; let range = to_proto::range(&line_index, runnable.nav.range());
let r = to_proto::runnable(&world, file_id, runnable)?;
if world.config.lens.run { if world.config.lens.run {
let lens = CodeLens { let lens = CodeLens {
range: r.range, range,
command: Some(Command { command: Some(Command {
title: run_title.to_string(), title: run_title.to_string(),
command: "rust-analyzer.runSingle".into(), command: "rust-analyzer.runSingle".into(),
@ -797,13 +803,8 @@ pub fn handle_code_lens(
} }
if debugee && world.config.lens.debug { if debugee && world.config.lens.debug {
if r.args[0] == "run" {
r.args[0] = "build".into();
} else {
r.args.push("--no-run".into());
}
let debug_lens = CodeLens { let debug_lens = CodeLens {
range: r.range, range,
command: Some(Command { command: Some(Command {
title: "Debug".into(), title: "Debug".into(),
command: "rust-analyzer.debugSingle".into(), command: "rust-analyzer.debugSingle".into(),

View file

@ -8,7 +8,6 @@ use ra_ide::{
}; };
use ra_syntax::{SyntaxKind, TextRange, TextSize}; use ra_syntax::{SyntaxKind, TextRange, TextSize};
use ra_vfs::LineEndings; use ra_vfs::LineEndings;
use rustc_hash::FxHashMap;
use crate::{ use crate::{
cargo_target_spec::CargoTargetSpec, lsp_ext, semantic_tokens, world::WorldSnapshot, Result, cargo_target_spec::CargoTargetSpec, lsp_ext, semantic_tokens, world::WorldSnapshot, Result,
@ -638,9 +637,8 @@ pub(crate) fn runnable(
) -> Result<lsp_ext::Runnable> { ) -> Result<lsp_ext::Runnable> {
let spec = CargoTargetSpec::for_file(world, file_id)?; let spec = CargoTargetSpec::for_file(world, file_id)?;
let target = spec.as_ref().map(|s| s.target.clone()); let target = spec.as_ref().map(|s| s.target.clone());
let (args, extra_args) = let (cargo_args, executable_args) =
CargoTargetSpec::runnable_args(spec, &runnable.kind, &runnable.cfg_exprs)?; CargoTargetSpec::runnable_args(spec, &runnable.kind, &runnable.cfg_exprs)?;
let line_index = world.analysis().file_line_index(file_id)?;
let label = match &runnable.kind { let label = match &runnable.kind {
RunnableKind::Test { test_id, .. } => format!("test {}", test_id), RunnableKind::Test { test_id, .. } => format!("test {}", test_id),
RunnableKind::TestMod { path } => format!("test-mod {}", path), RunnableKind::TestMod { path } => format!("test-mod {}", path),
@ -650,18 +648,16 @@ pub(crate) fn runnable(
target.map_or_else(|| "run binary".to_string(), |t| format!("run {}", t)) target.map_or_else(|| "run binary".to_string(), |t| format!("run {}", t))
} }
}; };
let location = location_link(world, None, runnable.nav)?;
Ok(lsp_ext::Runnable { Ok(lsp_ext::Runnable {
range: range(&line_index, runnable.range),
label, label,
location: Some(location),
kind: lsp_ext::RunnableKind::Cargo, kind: lsp_ext::RunnableKind::Cargo,
args, args: lsp_ext::CargoRunnable {
extra_args, workspace_root: world.workspace_root_for(file_id).map(|root| root.to_owned()),
env: { cargo_args,
let mut m = FxHashMap::default(); executable_args,
m.insert("RUST_BACKTRACE".to_string(), "short".to_string());
m
}, },
cwd: world.workspace_root_for(file_id).map(|root| root.to_owned()),
}) })
} }

View file

@ -114,8 +114,8 @@ async function getDebugConfiguration(ctx: Ctx, runnable: ra.Runnable): Promise<v
} }
async function getDebugExecutable(runnable: ra.Runnable): Promise<string> { async function getDebugExecutable(runnable: ra.Runnable): Promise<string> {
const cargo = new Cargo(runnable.cwd || '.', debugOutput); const cargo = new Cargo(runnable.args.workspaceRoot || '.', debugOutput);
const executable = await cargo.executableFromArgs(runnable.args); const executable = await cargo.executableFromArgs(runnable.args.cargoArgs);
// if we are here, there were no compilation errors. // if we are here, there were no compilation errors.
return executable; return executable;
@ -127,8 +127,8 @@ function getLldbDebugConfig(runnable: ra.Runnable, executable: string, sourceFil
request: "launch", request: "launch",
name: runnable.label, name: runnable.label,
program: executable, program: executable,
args: runnable.extraArgs, args: runnable.args.executableArgs,
cwd: runnable.cwd, cwd: runnable.args.workspaceRoot,
sourceMap: sourceFileMap, sourceMap: sourceFileMap,
sourceLanguages: ["rust"] sourceLanguages: ["rust"]
}; };
@ -140,8 +140,8 @@ function getCppvsDebugConfig(runnable: ra.Runnable, executable: string, sourceFi
request: "launch", request: "launch",
name: runnable.label, name: runnable.label,
program: executable, program: executable,
args: runnable.extraArgs, args: runnable.args.executableArgs,
cwd: runnable.cwd, cwd: runnable.args.workspaceRoot,
sourceFileMap: sourceFileMap, sourceFileMap: sourceFileMap,
}; };
} }

View file

@ -46,16 +46,15 @@ export interface RunnablesParams {
position: lc.Position | null; position: lc.Position | null;
} }
export type RunnableKind = "cargo" | "rustc" | "rustup";
export interface Runnable { export interface Runnable {
range: lc.Range;
label: string; label: string;
kind: RunnableKind; location?: lc.LocationLink;
args: string[]; kind: "cargo";
extraArgs: string[]; args: {
env: { [key: string]: string }; workspaceRoot?: string;
cwd: string | null; cargoArgs: string[];
executableArgs: string[];
};
} }
export const runnables = new lc.RequestType<RunnablesParams, Runnable[], void>("rust-analyzer/runnables"); export const runnables = new lc.RequestType<RunnablesParams, Runnable[], void>("rust-analyzer/runnables");

View file

@ -103,18 +103,27 @@ interface CargoTaskDefinition extends vscode.TaskDefinition {
env?: { [key: string]: string }; env?: { [key: string]: string };
} }
export function createTask(spec: ra.Runnable): vscode.Task { export function createTask(runnable: ra.Runnable): vscode.Task {
const TASK_SOURCE = 'Rust'; const TASK_SOURCE = 'Rust';
let command;
switch (runnable.kind) {
case "cargo": command = toolchain.getPathForExecutable("cargo");
}
const args = runnable.args.cargoArgs;
if (runnable.args.executableArgs.length > 0) {
args.push('--', ...runnable.args.executableArgs);
}
const definition: CargoTaskDefinition = { const definition: CargoTaskDefinition = {
type: 'cargo', type: 'cargo',
label: spec.label, label: runnable.label,
command: toolchain.getPathForExecutable(spec.kind), command,
args: spec.extraArgs ? [...spec.args, '--', ...spec.extraArgs] : spec.args, args,
env: Object.assign({}, process.env, spec.env), env: Object.assign({}, process.env as { [key: string]: string }, { "RUST_BACKTRACE": "short" }),
}; };
const execOption: vscode.ShellExecutionOptions = { const execOption: vscode.ShellExecutionOptions = {
cwd: spec.cwd || '.', cwd: runnable.args.workspaceRoot || '.',
env: definition.env, env: definition.env,
}; };
const exec = new vscode.ShellExecution( const exec = new vscode.ShellExecution(