Allow formatting of use behind #[cfg*]

Fix #6666

We would like to allow formatting items with attributes that
conceptually allows outer styling as well.

Signed-off-by: Xiangfei Ding <dingxiangfei2009@protonmail.ch>

Author: dingxiangfei2009

src/imports.rs

@@ -1,6 +1,6 @@
-use std::borrow::Cow;
 use std::cmp::Ordering;
 use std::fmt;
+use std::{borrow::Cow, vec};
 
 use core::hash::{Hash, Hasher};
 
@@ -123,7 +123,7 @@ pub(crate) struct UseTree {
 
 impl PartialEq for UseTree {
     fn eq(&self, other: &UseTree) -> bool {
-        self.path == other.path
+        self.path == other.path && !self.has_attrs() && !other.has_attrs()
     }
 }
 impl Eq for UseTree {}
@@ -226,13 +226,19 @@ pub(crate) fn normalize_use_trees_with_granularity(
 
     let mut result = Vec::with_capacity(use_trees.len());
     for use_tree in use_trees {
-        if use_tree.contains_comment() || use_tree.attrs.is_some() {
+        if use_tree.contains_comment() || use_tree.has_attrs_disallow_outer_style() {
             result.push(use_tree);
             continue;
         }
+        let attrs = use_tree.attrs.clone();
+        let result_buf = if attrs.is_some() {
+            &mut vec![]
+        } else {
+            &mut result
+        };
 
         for mut flattened in use_tree.flatten(import_granularity) {
-            if let Some(tree) = result
+            if let Some(tree) = result_buf
                 .iter_mut()
                 .find(|tree| tree.share_prefix(&flattened, merge_by))
             {
@@ -242,9 +248,19 @@ pub(crate) fn normalize_use_trees_with_granularity(
                 if merge_by == SharedPrefix::Module {
                     flattened = flattened.nest_trailing_self();
                 }
-                result.push(flattened);
+                result_buf.push(flattened);
             }
         }
+        if let Some(attrs) = attrs {
+            let result_buf: Vec<_> = result_buf
+                .drain(..)
+                .map(|mut use_tree| {
+                    use_tree.attrs = Some(attrs.clone());
+                    use_tree
+                })
+                .collect();
+            result.extend(result_buf);
+        }
     }
     result
 }
@@ -552,7 +568,7 @@ impl UseTree {
 
         // Remove foo::{} or self without attributes.
         match last.kind {
-            _ if self.attrs.is_some() => (),
+            _ if self.has_attrs() => (),
             UseSegmentKind::List(ref list) if list.is_empty() => {
                 self.path = vec![];
                 return self;
@@ -650,6 +666,17 @@ impl UseTree {
         self.has_comment() || self.path.iter().any(|path| path.contains_comment())
     }
 
+    fn has_attrs(&self) -> bool {
+        self.attrs.is_some()
+    }
+
+    fn has_attrs_disallow_outer_style(&self) -> bool {
+        !self.attrs.iter().flatten().all(|attr| match &attr.kind {
+            ast::AttrKind::Normal(attr) => attr.item.is_valid_for_outer_style(),
+            ast::AttrKind::DocComment(..) => false,
+        })
+    }
+
     fn same_visibility(&self, other: &UseTree) -> bool {
         match (&self.visibility, &other.visibility) {
             (
@@ -675,7 +702,8 @@ impl UseTree {
     fn share_prefix(&self, other: &UseTree, shared_prefix: SharedPrefix) -> bool {
         if self.path.is_empty()
             || other.path.is_empty()
-            || self.attrs.is_some()
+            || self.has_attrs()
+            || other.has_attrs()
             || self.contains_comment()
             || !self.same_visibility(other)
         {
@@ -692,7 +720,8 @@ impl UseTree {
     }
 
     fn flatten(self, import_granularity: ImportGranularity) -> Vec<UseTree> {
-        if self.path.is_empty() || self.contains_comment() {
+        if self.path.is_empty() || self.contains_comment() || self.has_attrs_disallow_outer_style()
+        {
             return vec![self];
         }
         match &self.path.clone().last().unwrap().kind {

tests/source/imports/imports_granularity_crate.rs

@@ -63,3 +63,13 @@ use b::v::{
 };
 use b::t::{/* Before b::t::self */ self};
 use b::c;
+
+// https://github.com/rust-lang/rustfmt/issues/6666
+#[cfg(true)]
+use a::{b::c, d::e, d::f};
+
+#[my_attribute]
+use a::{b::c,d::e,d::f};
+
+/// Some doc comment
+use a::{b::c,d::e,d::f};

tests/source/imports/imports_granularity_default-with-dups.rs

@@ -4,3 +4,7 @@ use crate::lexer::{tokens::TokenData};
 use crate::lexer::self;
 use crate::lexer::{self};
 use crate::lexer::{self, tokens::TokenData};
+
+use a::{b::c, d::e, d::f};
+#[cfg(unix)]
+use a::{b::c, d::e, d::f};

tests/source/imports/imports_granularity_item-with-dups-StdExternalCrate-no-reorder.rs

@@ -11,3 +11,11 @@ use crate::lexer;
 use crate::lexer;
 use crate::lexer::{self};
 use crate::lexer::{self, tokens::TokenData};
+
+use crate::{b::c, d::e, d::f};
+#[cfg(unix)]
+use crate::{b::c, d::e, d::f};
+#[cfg(windows)]
+use crate::{b::c, d::e, d::f};
+// my comment
+use crate::b;

tests/source/imports/imports_granularity_item-with-dups.rs

@@ -9,3 +9,5 @@ use crate::lexer;
 use crate::lexer;
 use crate::lexer::{self};
 use crate::lexer::{self, tokens::TokenData};
+#[cfg(unix)]
+use crate::lexer;

tests/source/imports/imports_granularity_item.rs

@@ -32,3 +32,11 @@ use b::v::{
 };
 use b::t::{/* Before b::t::self */ self};
 use b::c;
+
+use c::{a::b, c::d, c::e};
+#[cfg(unix)]
+use c::{a::b, c::d, c::e};
+#[cfg(windows)]
+use c::{a::b, c::d, c::e};
+// my comment
+use c::a::b;

tests/source/imports/imports_granularity_module.rs

@@ -45,3 +45,9 @@ use b::v::{
 };
 use b::t::{/* Before b::t::self */ self};
 use b::c;
+
+#[my_attribute]
+use a::{b::c,d::e,d::f};
+
+/// Some doc comment
+use a::{b::c,d::e,d::f};

tests/target/imports/imports_granularity_crate.rs

@@ -57,3 +57,16 @@ use b::{
     /* Before b::l group */ l::{self, m, n::o, p::*},
     q,
 };
+
+// https://github.com/rust-lang/rustfmt/issues/6666
+#[cfg(true)]
+use a::{
+    b::c,
+    d::{e, f},
+};
+
+#[my_attribute]
+use a::{b::c, d::e, d::f};
+
+/// Some doc comment
+use a::{b::c, d::e, d::f};

tests/target/imports/imports_granularity_default-with-dups.rs

@@ -4,3 +4,7 @@ use crate::lexer::tokens::TokenData;
 use crate::lexer::tokens::TokenData;
 use crate::lexer::{self};
 use crate::lexer::{self, tokens::TokenData};
+
+use a::{b::c, d::e, d::f};
+#[cfg(unix)]
+use a::{b::c, d::e, d::f};

tests/target/imports/imports_granularity_item-with-dups-StdExternalCrate-no-reorder.rs

@@ -5,3 +5,20 @@
 use crate::lexer;
 use crate::lexer::tokens::TokenData;
 use crate::lexer::{self};
+use crate::b::c;
+use crate::d::e;
+use crate::d::f;
+#[cfg(unix)]
+use crate::b::c;
+#[cfg(unix)]
+use crate::d::e;
+#[cfg(unix)]
+use crate::d::f;
+#[cfg(windows)]
+use crate::b::c;
+#[cfg(windows)]
+use crate::d::e;
+#[cfg(windows)]
+use crate::d::f;
+// my comment
+use crate::b;