Skip to content

Commit 6082b59

Browse files
committed
Fix documentation parsing
1 parent f0028b8 commit 6082b59

1 file changed

Lines changed: 8 additions & 12 deletions

File tree

packages/freezed/lib/src/ast.dart

Lines changed: 8 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,22 +1,18 @@
11
import 'package:analyzer/dart/ast/ast.dart';
2-
import 'package:analyzer/dart/ast/token.dart';
32
import 'package:analyzer/dart/element/element.dart';
43

54
extension AstX on AstNode {
65
String? get documentation {
7-
final builder = StringBuffer();
8-
9-
for (
10-
Token? token = beginToken.precedingComments;
11-
token != null;
12-
token = token.next
13-
) {
14-
builder.writeln(token);
15-
}
6+
final node = switch (this) {
7+
DefaultFormalParameter(:final parameter) => parameter,
8+
_ => this,
9+
};
1610

17-
if (builder.isEmpty) return null;
11+
if (node case AnnotatedNode(:final documentationComment?)) {
12+
return '${documentationComment.tokens.map((token) => token.lexeme).join('\n')}\n';
13+
}
1814

19-
return builder.toString();
15+
return null;
2016
}
2117
}
2218

0 commit comments

Comments
 (0)