Fixed a regression with classDiagram supporting comments

src/diagrams/class/classDiagram.spec.js

@@ -242,69 +242,74 @@ describe('class diagram, ', function () {
 
     it('should handle comments at the start', function () {
       const str =
-        '%% Comment\n' +
+        `%% Comment
-        'classDiagram\n' +
+        classDiagram
-        'class Class1 {\n' +
+        class Class1 {
-        'int : test\n' +
+          int : test
-        'string : foo\n' +
+          string : foo
-        'test()\n' +
+          test()
-        'foo()\n' +
+          foo()
-        '}';
+        }`;
       parser.parse(str);
     });
 
     it('should handle comments at the end', function () {
       const str =
-        'classDiagram\n' +
+        `classDiagram
-        'class Class1 {\n' +
+class Class1 {
-        'int : test\n' +
+int : test
-        'string : foo\n' +
+string : foo
-        'test()\n' +
+test()
-        'foo()\n' +
+foo()
-        '\n}' +
+
-        '%% Comment\n';
+}
+%% Comment
+`;
 
       parser.parse(str);
     });
 
     it('should handle comments at the end no trailing newline', function () {
       const str =
-      'classDiagram\n' +
+      `classDiagram
-      'class Class1 {\n' +
+class Class1 {
-        'int : test\n' +
+int : test
-        'string : foo\n' +
+string : foo
-        'test()\n' +
+test()
-        'foo()\n' +
+foo()
-        '}\n' +
+}
-        '%% Comment';
+%% Comment`;
 
       parser.parse(str);
     });
 
     it('should handle a comment with multiple line feeds', function () {
       const str =
-        'classDiagram\n\n\n' +
+        `classDiagram
-        '%% Comment\n\n' +
+
-        'class Class1 {\n' +
+
-        'int : test\n' +
+%% Comment
-        'string : foo\n' +
+
-        'test()\n' +
+class Class1 {
-        'foo()\n' +
+int : test
-        '}';
+string : foo
+test()
+foo()
+}`;
 
       parser.parse(str);
     });
 
     it('should handle a comment with mermaid class diagram code in them', function () {
       const str =
-        'classDiagram\n' +
+        `classDiagram
-        '%% Comment Class01 <|-- Class02\n' +
+%% Comment Class01 <|-- Class02
-        'class Class1 {\n' +
+class Class1 {
-        'int : test\n' +
+int : test
-        'string : foo\n' +
+string : foo
-        'test()\n' +
+test()
-        'foo()\n' +
+foo()
-        '}';
+}`;
 
       parser.parse(str);
     });

src/diagrams/class/parser/classDiagram.jison

@@ -17,11 +17,11 @@
 <type_directive>":"                                             { this.popState(); this.begin('arg_directive'); return ':'; }
 <type_directive,arg_directive>\}\%\%                            { this.popState(); this.popState(); return 'close_directive'; }
 <arg_directive>((?:(?!\}\%\%).|\n)*)                            return 'arg_directive';
-\%%(?!\{)[^\n]*                                                 /* skip comments */
+\%\%(?!\{)*[^\n]*                                                 /* skip comments */
-[^\}]\%\%[^\n]*                                                 /* skip comments */
+[^\}]\%\%*[^\n]*                                                 /* skip comments */
-\%\%[^\n]*\n*           /* do nothing */
+\%\%*[^\n]*[\n]*           /* do nothing */
-\n+                     return 'NEWLINE';
+[\n]+                     return 'NEWLINE';
-\s+                     /* skip whitespace */
+[\s]+                     /* skip whitespace */
 "classDiagram-v2"       return 'CLASS_DIAGRAM';
 "classDiagram"          return 'CLASS_DIAGRAM';
 [\{]                    { this.begin("struct"); /*console.log('Starting struct');*/return 'STRUCT_START';}
@@ -146,11 +146,15 @@ mermaidDoc
     ;
 
 graphConfig
-    : CLASS_DIAGRAM NEWLINE statements EOF
+    : NEWLINE
+    | NEWLINE graphConfig
+    | graphConfig NEWLINE
+    | CLASS_DIAGRAM NEWLINE statements EOF
     ;
 
 statements
     : statement
+    | NEWLINE statement
     | statement NEWLINE
     | statement NEWLINE statements
     ;