keep empty lines while parsing comments (#272)

webrpc · Apr 8, 2024 · 3f48bd0 · 3f48bd0
1 parent 0e4e294
commit 3f48bd0
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 5 deletions.
diff --git a/_examples/golang-basics/example.gen.go b/_examples/golang-basics/example.gen.go
diff --git a/_examples/golang-basics/example.ridl b/_examples/golang-basics/example.ridl
@@ -15,7 +15,9 @@ enum Intent: string
 
 struct Empty
 
-
+# User struct
+#
+# More information about user struct
 struct User
   - id: uint64
     + json = id

diff --git a/schema/ridl/ridl.go b/schema/ridl/ridl.go
@@ -276,7 +276,7 @@ func (p *Parser) parse() (*schema.WebRPCSchema, error) {
 				StreamOutput: method.StreamOutput(),
 				Inputs:       inputs,
 				Outputs:      outputs,
-				Comments:     strings.FieldsFunc(method.Comment(), func(r rune) bool { return r == '\n' }),
+				Comments:     parseComment(method.Comment()),
 			}
 
 			methods = append(methods, m)
@@ -358,5 +358,9 @@ func buildArgumentsList(s *schema.WebRPCSchema, args []*ArgumentNode) ([]*schema
 }
 
 func parseComment(comment string) []string {
-	return strings.FieldsFunc(comment, func(r rune) bool { return r == '\n' })
+	if comment == "" {
+		return []string{}
+	}
+
+	return strings.Split(comment, "\n")
 }