From 6a94f203fc5e2dc6f48a592c26b48f3a98638d77 Mon Sep 17 00:00:00 2001 From: Evgenii P Date: Sat, 3 Aug 2019 17:07:20 +0700 Subject: [PATCH] Use std::future::Future trait from stdlib --- crates/ra_hir/src/source_binder.rs | 57 ++++++++++++------- .../ra_ide_api/src/completion/complete_dot.rs | 19 +++---- 2 files changed, 44 insertions(+), 32 deletions(-) diff --git a/crates/ra_hir/src/source_binder.rs b/crates/ra_hir/src/source_binder.rs index 7720329e33a..6a9f228b82d 100644 --- a/crates/ra_hir/src/source_binder.rs +++ b/crates/ra_hir/src/source_binder.rs @@ -7,6 +7,20 @@ /// purely for "IDE needs". use std::sync::Arc; +use crate::{ + expr::{ + self, + scope::{ExprScopes, ScopeId}, + BodySourceMap, + }, + ids::LocationCtx, + name, + path::{PathKind, PathSegment}, + ty::method_resolution::implements_trait, + AsName, AstId, Const, Crate, DefWithBody, Either, Enum, Function, HirDatabase, HirFileId, + MacroDef, Module, ModuleDef, Name, Path, PerNs, Resolution, Resolver, Static, Struct, Trait, + Ty, +}; use ra_db::{FileId, FilePosition}; use ra_syntax::{ algo::find_node_at_offset, @@ -17,19 +31,6 @@ }; use rustc_hash::{FxHashMap, FxHashSet}; -use crate::{ - expr, - expr::{ - scope::{ExprScopes, ScopeId}, - BodySourceMap, - }, - ids::LocationCtx, - lang_item::LangItemTarget, - ty::method_resolution::implements_trait, - AsName, AstId, Const, Crate, DefWithBody, Either, Enum, Function, HirDatabase, HirFileId, - MacroDef, Module, Name, Path, PerNs, Resolver, Static, Struct, Trait, Ty, -}; - /// Locates the module by `FileId`. Picks topmost module in the file. pub fn module_from_file_id(db: &impl HirDatabase, file_id: FileId) -> Option { module_from_source(db, file_id.into(), None) @@ -411,18 +412,32 @@ pub fn autoderef<'a>( crate::ty::autoderef(db, &self.resolver, canonical).map(|canonical| canonical.value) } - /// Checks that particular type `ty` implements `Future` trait (`future_trait` lang item). + /// Checks that particular type `ty` implements `std::future::Future`. /// This function is used in `.await` syntax completion. pub fn impls_future(&self, db: &impl HirDatabase, ty: Ty) -> bool { - let krate = self.resolver.krate(); - if let Some(krate) = krate { - let future_trait = match db.lang_item(krate, "future_trait".into()) { - Some(LangItemTarget::Trait(t)) => t, - _ => return false, + let std_future_path = Path { + kind: PathKind::Abs, + segments: vec![ + PathSegment { name: name::STD, args_and_bindings: None }, + PathSegment { name: name::FUTURE_MOD, args_and_bindings: None }, + PathSegment { name: name::FUTURE_TYPE, args_and_bindings: None }, + ], + }; + + let std_future_trait = + match self.resolver.resolve_path_segments(db, &std_future_path).into_fully_resolved() { + PerNs { types: Some(Resolution::Def(ModuleDef::Trait(trait_))), .. } => { + Some(trait_) + } + _ => None, }; - let canonical_ty = crate::ty::Canonical { value: ty, num_vars: 0 }; - return implements_trait(&canonical_ty, db, &self.resolver, krate, future_trait); + let krate = self.resolver.krate(); + if let Some(krate) = krate { + if let Some(trait_) = std_future_trait { + let canonical_ty = crate::ty::Canonical { value: ty, num_vars: 0 }; + return implements_trait(&canonical_ty, db, &self.resolver, krate, trait_); + } } false diff --git a/crates/ra_ide_api/src/completion/complete_dot.rs b/crates/ra_ide_api/src/completion/complete_dot.rs index e8fd37bca46..9a3b353a926 100644 --- a/crates/ra_ide_api/src/completion/complete_dot.rs +++ b/crates/ra_ide_api/src/completion/complete_dot.rs @@ -425,28 +425,25 @@ fn test_completion_await_impls_future() { assert_debug_snapshot_matches!( do_completion( r###" - // Mock Future trait from stdlib - pub mod std { - pub mod future { - #[lang = "future_trait"] - pub trait Future {} - } - } - + //- /main.rs use std::future::*; struct A {} impl Future for A {} - fn foo(a: A) { a.<|> } + + //- /std/lib.rs + pub mod future { + pub trait Future {} + } "###, CompletionKind::Keyword), @r###" ⋮[ ⋮ CompletionItem { ⋮ label: "await", - ⋮ source_range: [358; 358), - ⋮ delete: [358; 358), + ⋮ source_range: [74; 74), + ⋮ delete: [74; 74), ⋮ insert: "await", ⋮ detail: "expr.await", ⋮ },