Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'crates/ide-assists/src/utils.rs')
| -rw-r--r-- | crates/ide-assists/src/utils.rs | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/crates/ide-assists/src/utils.rs b/crates/ide-assists/src/utils.rs index 0806d5feb1..31924773b4 100644 --- a/crates/ide-assists/src/utils.rs +++ b/crates/ide-assists/src/utils.rs @@ -24,7 +24,7 @@ use syntax::{ make, syntax_factory::SyntaxFactory, }, - ted, + ted::{self, Position}, }; use crate::assist_context::{AssistContext, SourceChangeBuilder}; @@ -212,6 +212,10 @@ pub fn add_trait_assoc_items_to_impl( }); let assoc_item_list = impl_.get_or_create_assoc_item_list(); + if trait_.is_unsafe(sema.db) { + ted::insert(Position::first_child_of(impl_.syntax()), make::token(T![unsafe])); + } + let mut first_item = None; for item in items { first_item.get_or_insert_with(|| item.clone()); |