fmt: fix enum/struct_decl/struct_init fields with empty newlines (vla…

…ng#22051)

cmd/tools/vast/vast.v

@@ -647,7 +647,9 @@ fn (t Tree) struct_field(node ast.StructField) &Node {
 	obj.add_terse('is_volatile', t.bool_node(node.is_volatile))
 	obj.add_terse('is_deprecated', t.bool_node(node.is_deprecated))
 	obj.add_terse('attrs', t.array_node_attr(node.attrs))
+	obj.add('pre_comments', t.array_node_comment(node.pre_comments))
 	obj.add('comments', t.array_node_comment(node.comments))
+	obj.add('next_comments', t.array_node_comment(node.next_comments))
 	obj.add('pos', t.pos(node.pos))
 	obj.add_terse('i', t.number_node(node.i))
 	return obj
@@ -683,6 +685,7 @@ fn (t Tree) enum_field(node ast.EnumField) &Node {
 	obj.add_terse('has_expr', t.bool_node(node.has_expr))
 	obj.add_terse('expr', t.expr(node.expr))
 	obj.add('pos', t.pos(node.pos))
+	obj.add('pre_comments', t.array_node_comment(node.pre_comments))
 	obj.add('comments', t.array_node_comment(node.comments))
 	obj.add('next_comments', t.array_node_comment(node.next_comments))
 	return obj
@@ -1611,7 +1614,8 @@ fn (t Tree) struct_init_field(node ast.StructInitField) &Node {
 	obj.add_terse('typ', t.type_node(node.typ))
 	obj.add_terse('expected_type', t.type_node(node.expected_type))
 	obj.add_terse('parent_type', t.type_node(node.parent_type))
-	obj.add('comments', t.array_node_comment(node.comments))
+	obj.add('pre_comments', t.array_node_comment(node.pre_comments))
+	obj.add('end_comments', t.array_node_comment(node.end_comments))
 	obj.add('next_comments', t.array_node_comment(node.next_comments))
 	obj.add('pos', t.pos(node.pos))
 	obj.add('name_pos', t.pos(node.name_pos))

vlib/sync/channels.c.v

@@ -33,7 +33,8 @@ pub struct Channel {
 	ringbuf   &u8 = unsafe { nil } // queue for buffered channels
 	statusbuf &u8 = unsafe { nil } // flags to synchronize write/read in ringbuf
 	objsize   u32
-mut: // atomic
+mut:
+	// atomic
 	writesem           Semaphore // to wake thread that wanted to write, but buffer was full
 	readsem            Semaphore // to wake thread that wanted to read, but buffer was empty
 	writesem_im        Semaphore

vlib/v/ast/ast.v

@@ -330,7 +330,9 @@ pub:
 	pos              token.Pos
 	type_pos         token.Pos
 	option_pos       token.Pos
+	pre_comments     []Comment
 	comments         []Comment
+	next_comments    []Comment
 	i                int
 	has_default_expr bool
 	has_prev_newline bool
@@ -457,7 +459,8 @@ pub struct StructInitField {
 pub:
 	pos              token.Pos
 	name_pos         token.Pos
-	comments         []Comment
+	pre_comments     []Comment
+	end_comments     []Comment
 	next_comments    []Comment
 	has_prev_newline bool
 pub mut:
@@ -1366,6 +1369,7 @@ pub:
 	name             string // just `lock`, or `abc`, etc, no matter if the name is a keyword or not.
 	source_name      string // The name in the source, for example `@lock`, and `abc`. Note that `lock` is a keyword in V.
 	pos              token.Pos
+	pre_comments     []Comment // comment before Enumfield
 	comments         []Comment // comment after Enumfield in the same line
 	next_comments    []Comment // comments between current EnumField and next EnumField
 	has_expr         bool      // true, when .expr has a value

vlib/v/fmt/fmt.v

@@ -1069,6 +1069,9 @@ pub fn (mut f Fmt) enum_decl(node ast.EnumDecl) {
 		if i > 0 && field.has_prev_newline {
 			f.writeln('')
 		}
+		if field.pre_comments.len > 0 {
+			f.comments(field.pre_comments, has_nl: true, level: .indent)
+		}
 		f.write('\t${field.name}')
 		if field.has_expr {
 			f.write(strings.repeat(` `, value_align.max_len(field.pos.line_nr) - field.name.len))

vlib/v/fmt/struct.v

@@ -78,16 +78,10 @@ pub fn (mut f Fmt) struct_decl(node ast.StructDecl, is_anon bool) {
 			}
 			else {}
 		}
-		mut pre_cmts, mut end_cmts, mut next_line_cmts := []ast.Comment{}, []ast.Comment{}, []ast.Comment{}
-		for cmt in field.comments {
-			match true {
-				cmt.pos.pos < field.pos.pos { pre_cmts << cmt }
-				cmt.pos.line_nr > field.pos.last_line { next_line_cmts << cmt }
-				else { end_cmts << cmt }
-			}
-		}
 		// Handle comments before the field
-		f.comments_before_field(pre_cmts)
+		if field.pre_comments.len > 0 {
+			f.comments(field.pre_comments, level: .indent)
+		}
 		volatile_prefix := if field.is_volatile { 'volatile ' } else { '' }
 		f.write('\t${volatile_prefix}${field.name} ')
 		f.write(strings.repeat(` `, type_align.max_len(field.pos.line_nr) - field.name.len))
@@ -115,7 +109,7 @@ pub fn (mut f Fmt) struct_decl(node ast.StructDecl, is_anon bool) {
 			f.single_line_attrs(field.attrs, same_line: true)
 		}
 		// Handle comments at the end of the line
-		if end_cmts.len > 0 {
+		if field.comments.len > 0 {
 			if field.has_default_expr {
 				f.write(strings.repeat(` `, comment_align.max_len(field.pos.line_nr) - field.default_expr.str().len - 2))
 			} else if field.attrs.len > 0 {
@@ -124,13 +118,13 @@ pub fn (mut f Fmt) struct_decl(node ast.StructDecl, is_anon bool) {
 				f.write(strings.repeat(` `, comment_align.max_len(field.pos.line_nr) - field_types[i].len))
 			}
 			f.write(' ')
-			f.comments(end_cmts, level: .indent)
+			f.comments(field.comments, level: .indent)
 		} else {
 			f.writeln('')
 		}
 		// Handle comments on the next lines
-		if next_line_cmts.len > 0 {
-			f.comments(next_line_cmts, level: .indent)
+		if field.next_comments.len > 0 {
+			f.comments(field.next_comments, level: .indent)
 		}
 	}
 	if is_anon || node.end_comments.len > 0 {
@@ -279,23 +273,26 @@ pub fn (mut f Fmt) struct_init(node ast.StructInit) {
 			mut comment_align := new_field_align(use_threshold: true)
 			for init_field in node.init_fields {
 				value_align.add_info(init_field.name.len, init_field.pos.line_nr)
-				if init_field.comments.len > 0 {
+				if init_field.end_comments.len > 0 {
 					comment_align.add_info(init_field.expr.str().len, init_field.pos.line_nr)
 				}
 			}
 			for i, init_field in node.init_fields {
 				if i > 0 && init_field.has_prev_newline {
 					f.writeln('')
 				}
+				if init_field.pre_comments.len > 0 {
+					f.comments(init_field.pre_comments, has_nl: true, level: .keep)
+				}
 				f.write('${init_field.name}: ')
 				if !single_line_fields {
 					f.write(strings.repeat(` `, value_align.max_len(init_field.pos.line_nr) - init_field.name.len))
 				}
 				f.expr(init_field.expr)
-				if init_field.comments.len > 0 {
+				if init_field.end_comments.len > 0 {
 					f.write(strings.repeat(` `, comment_align.max_len(init_field.pos.line_nr) - init_field.expr.str().len))
 					f.write(' ')
-					f.comments(init_field.comments, has_nl: false, level: .indent)
+					f.comments(init_field.end_comments, has_nl: false, level: .indent)
 				}
 				if single_line_fields {
 					if i < node.init_fields.len - 1 {
@@ -305,7 +302,7 @@ pub fn (mut f Fmt) struct_init(node ast.StructInit) {
 					f.writeln('')
 				}
 				f.comments(init_field.next_comments, has_nl: true, level: .keep)
-				if single_line_fields && (init_field.comments.len > 0
+				if single_line_fields && (init_field.end_comments.len > 0
 					|| init_field.next_comments.len > 0
 					|| !expr_is_single_line(init_field.expr) || f.line_len > max_len) {
 					single_line_fields = false

vlib/v/fmt/tests/comments_input.vv

@@ -1,4 +1,5 @@
 import time // foo
+
 /*
 block
 comment

vlib/v/fmt/tests/enum_fields_with_empty_line_2_keep.vv

@@ -0,0 +1,19 @@
+enum Info {
+	aa = 1 // aa
+	bbb // bbb
+
+	cccc = 5 /* cccc
+	--- cccc
+	*/
+	ddddd = 10 // ddddd
+
+	// before
+	ee  = 20
+	fff = 30
+
+	// before comment1
+	// before comment2
+	gg
+}
+
+fn main() {}

vlib/v/fmt/tests/struct_decl_with_newline_and_comments_keep.vv

@@ -0,0 +1,17 @@
+module main
+
+fn main() {
+}
+
+pub struct OperateInfo { // implements IperateInfo
+pub mut:
+	title string // title
+
+	// msg, id
+	msg string // msg
+	id  string // id
+
+	// other1
+	// other2
+	other string
+} // operate info

vlib/v/fmt/tests/struct_init_with_newline_and_comments_2_keep.vv

@@ -0,0 +1,25 @@
+module abcde
+
+pub struct Builder {
+pub mut:
+	// inline before field
+	buf          []u8
+	str_calls    int
+	len          int
+	initial_size int = 1
+}
+
+pub fn new_builder(initial_size int) Builder {
+	return Builder{
+		// buf: make(0, initial_size)
+		buf: []u8{cap: initial_size}
+
+		// before comment
+		str_calls: 0 // after str_calls
+		len:       0 // after len
+
+		// before comment1
+		// before comment2
+		initial_size: initial_size // final
+	}
+}

vlib/v/parser/parser.v

@@ -904,16 +904,16 @@ pub:
 }
 
 fn (mut p Parser) eat_comments(cfg EatCommentsConfig) []ast.Comment {
-	mut line := p.prev_tok.line_nr
+	mut line := p.prev_tok.line_nr + p.prev_tok.lit.count('\n')
 	mut comments := []ast.Comment{}
 	for {
 		if p.tok.kind != .comment || (cfg.same_line && p.tok.line_nr > line)
-			|| (cfg.follow_up && (p.tok.line_nr > line + 1 || p.tok.lit.contains('\n'))) {
+			|| (cfg.follow_up && p.tok.line_nr > line + 1) {
 			break
 		}
 		comments << p.comment()
 		if cfg.follow_up {
-			line = p.prev_tok.line_nr
+			line = p.prev_tok.line_nr + p.prev_tok.lit.count('\n')
 		}
 	}
 	return comments
@@ -4122,6 +4122,7 @@ fn (mut p Parser) enum_decl() ast.EnumDecl {
 	mut uses_exprs := false
 	mut enum_attrs := map[string][]ast.Attr{}
 	for p.tok.kind != .eof && p.tok.kind != .rcbr {
+		pre_comments := p.eat_comments()
 		pos := p.tok.pos()
 		has_prev_newline := p.has_prev_newline()
 		val := p.check_name()
@@ -4143,14 +4144,15 @@ fn (mut p Parser) enum_decl() ast.EnumDecl {
 			p.attrs = []
 		}
 		comments := p.eat_comments(same_line: true)
-		next_comments := p.eat_comments()
+		next_comments := p.eat_comments(follow_up: true)
 		fields << ast.EnumField{
 			name:             val
 			source_name:      source_name(val)
 			pos:              pos
 			expr:             expr
 			has_expr:         has_expr
 			has_prev_newline: has_prev_newline
+			pre_comments:     pre_comments
 			comments:         comments
 			next_comments:    next_comments
 			attrs:            attrs