Fix attachment of inner doc comments

crates/syntax/src/ast.rs

@@ -115,7 +115,22 @@ fn test_doc_comment_none() {
 }
 
 #[test]
-fn test_doc_comment_of_items() {
+fn test_outer_doc_comment_of_items() {
+    let file = SourceFile::parse(
+        r#"
+        /// doc
+        // non-doc
+        mod foo {}
+        "#,
+    )
+    .ok()
+    .unwrap();
+    let module = file.syntax().descendants().find_map(Module::cast).unwrap();
+    assert_eq!("doc", module.doc_comment_text().unwrap());
+}
+
+#[test]
+fn test_inner_doc_comment_of_items() {
     let file = SourceFile::parse(
         r#"
         //! doc
@@ -126,7 +141,7 @@ fn test_doc_comment_of_items() {
     .ok()
     .unwrap();
     let module = file.syntax().descendants().find_map(Module::cast).unwrap();
-    assert_eq!("doc", module.doc_comment_text().unwrap());
+    assert!(module.doc_comment_text().is_none());
 }
 
 #[test]

crates/syntax/src/parsing/text_tree_sink.rs

@@ -5,6 +5,7 @@ use std::mem;
 use parser::{ParseError, TreeSink};
 
 use crate::{
+    ast,
     parsing::Token,
     syntax_node::GreenNode,
     SmolStr, SyntaxError,
@@ -153,24 +154,22 @@ fn n_attached_trivias<'a>(
 
             while let Some((i, (kind, text))) = trivias.next() {
                 match kind {
-                    WHITESPACE => {
+                    WHITESPACE if text.contains("\n\n") => {
-                        if text.contains("\n\n") {
                         // we check whether the next token is a doc-comment
                         // and skip the whitespace in this case
-                            if let Some((peek_kind, peek_text)) =
+                        if let Some((COMMENT, peek_text)) = trivias.peek().map(|(_, pair)| pair) {
-                                trivias.peek().map(|(_, pair)| pair)
+                            let comment_kind = ast::CommentKind::from_text(peek_text);
-                            {
+                            if comment_kind.doc == Some(ast::CommentPlacement::Outer) {
-                                if *peek_kind == COMMENT
-                                    && peek_text.starts_with("///")
-                                    && !peek_text.starts_with("////")
-                                {
                                 continue;
                             }
                         }
                         break;
                     }
-                    }
                     COMMENT => {
+                        let comment_kind = ast::CommentKind::from_text(text);
+                        if comment_kind.doc == Some(ast::CommentPlacement::Inner) {
+                            break;
+                        }
                         res = i + 1;
                     }
                     _ => (),

crates/syntax/test_data/parser/ok/0037_mod.rast

@@ -1,9 +1,9 @@
 SOURCE_FILE@0..93
   COMMENT@0..60 "// https://github.com ..."
   WHITESPACE@60..62 "\n\n"
-  MODULE@62..93
   COMMENT@62..70 "//! docs"
   WHITESPACE@70..71 "\n"
+  MODULE@71..93
     COMMENT@71..82 "// non-docs"
     WHITESPACE@82..83 "\n"
     MOD_KW@83..86 "mod"