diff options
Diffstat (limited to 'crates/assists/src/utils')
-rw-r--r-- | crates/assists/src/utils/insert_use.rs | 35 |
1 files changed, 32 insertions, 3 deletions
diff --git a/crates/assists/src/utils/insert_use.rs b/crates/assists/src/utils/insert_use.rs index f6025c99a..bfd457d18 100644 --- a/crates/assists/src/utils/insert_use.rs +++ b/crates/assists/src/utils/insert_use.rs | |||
@@ -200,7 +200,18 @@ fn recursive_merge( | |||
200 | return None; | 200 | return None; |
201 | } | 201 | } |
202 | let rhs_path = rhs_t.path(); | 202 | let rhs_path = rhs_t.path(); |
203 | match use_trees.binary_search_by(|p| path_cmp_bin_search(p.path(), rhs_path.clone())) { | 203 | match use_trees.binary_search_by(|lhs_t| { |
204 | let (lhs_t, rhs_t) = match lhs_t | ||
205 | .path() | ||
206 | .zip(rhs_path.clone()) | ||
207 | .and_then(|(lhs, rhs)| common_prefix(&lhs, &rhs)) | ||
208 | { | ||
209 | Some((lhs_p, rhs_p)) => (lhs_t.split_prefix(&lhs_p), rhs_t.split_prefix(&rhs_p)), | ||
210 | None => (lhs_t.clone(), rhs_t.clone()), | ||
211 | }; | ||
212 | |||
213 | path_cmp_bin_search(lhs_t.path(), rhs_t.path()) | ||
214 | }) { | ||
204 | Ok(idx) => { | 215 | Ok(idx) => { |
205 | let lhs_t = &mut use_trees[idx]; | 216 | let lhs_t = &mut use_trees[idx]; |
206 | let lhs_path = lhs_t.path()?; | 217 | let lhs_path = lhs_t.path()?; |
@@ -327,11 +338,11 @@ fn path_cmp_for_sort(a: Option<ast::Path>, b: Option<ast::Path>) -> Ordering { | |||
327 | 338 | ||
328 | /// Path comparison func for binary searching for merging. | 339 | /// Path comparison func for binary searching for merging. |
329 | fn path_cmp_bin_search(lhs: Option<ast::Path>, rhs: Option<ast::Path>) -> Ordering { | 340 | fn path_cmp_bin_search(lhs: Option<ast::Path>, rhs: Option<ast::Path>) -> Ordering { |
330 | match (lhs, rhs) { | 341 | match (lhs.and_then(|path| path.segment()), rhs.and_then(|path| path.segment())) { |
331 | (None, None) => Ordering::Equal, | 342 | (None, None) => Ordering::Equal, |
332 | (None, Some(_)) => Ordering::Less, | 343 | (None, Some(_)) => Ordering::Less, |
333 | (Some(_), None) => Ordering::Greater, | 344 | (Some(_), None) => Ordering::Greater, |
334 | (Some(ref a), Some(ref b)) => path_cmp_short(a, b), | 345 | (Some(ref a), Some(ref b)) => path_segment_cmp(a, b), |
335 | } | 346 | } |
336 | } | 347 | } |
337 | 348 | ||
@@ -802,6 +813,24 @@ use std::foo::bar::{Qux, quux::{Fez, Fizz}};", | |||
802 | } | 813 | } |
803 | 814 | ||
804 | #[test] | 815 | #[test] |
816 | fn merge_groups_full_nested_long() { | ||
817 | check_full( | ||
818 | "std::foo::bar::Baz", | ||
819 | r"use std::{foo::bar::Qux};", | ||
820 | r"use std::{foo::bar::{Baz, Qux}};", | ||
821 | ); | ||
822 | } | ||
823 | |||
824 | #[test] | ||
825 | fn merge_groups_last_nested_long() { | ||
826 | check_full( | ||
827 | "std::foo::bar::Baz", | ||
828 | r"use std::{foo::bar::Qux};", | ||
829 | r"use std::{foo::bar::{Baz, Qux}};", | ||
830 | ); | ||
831 | } | ||
832 | |||
833 | #[test] | ||
805 | fn merge_groups_skip_pub() { | 834 | fn merge_groups_skip_pub() { |
806 | check_full( | 835 | check_full( |
807 | "std::io", | 836 | "std::io", |