Split grammar into header lines and music lines and handle them separately.

src/languages/abc.js

@@ -5,85 +5,220 @@
   Website: https://abcnotation.com
   Category: common
 */
+
 export default function(hljs) {
-  const accidentalDirectivePrefix = `(?<=%%(propagate|writeout)-accidentals.*)`;
-  const fontDirectivePrefix = `(?<=%%(title|subtitle|composer|parts|tempo|gchord|annotation|info|text|vocal|words|set)font.*)`;
-  const textDirectivePrefix = `(?<=%%(text|center)\\s)`;
+  const untilCommentOrEndOfLine = /(?=[%\n])/
+
+  const inlineHeader = {
+    scope: "inline-header",
+    begin: /\[[MLQPRIV]:/,
+    end: /]/,
+    contains: [
+      {
+        scope: "header-content",
+        match: /[^\]]*/,
+        endsWithParent: true
+      }
+    ]
+  }
+
+  const note = {
+    scope: "note",
+    match: /[_^]?[_^=]?[A-Ga-gZz][,']*\d*\/*\d*-?/
+  }
+
+  const barLine = {
+    scope: "bar",
+    match: /(\|\]|\|\||\[\||\|:|:\||::|\|)\d*([,\-]\d)*/,
+  }
+
+  const tuple = {
+    scope: "tuple",
+    match: /\([2-9](:[1-9])?(:[1-9])?/
+  }
+
+  const graceNotes = {
+    scope: "grace",
+    begin: /\{/, end: /\}/,
+    contains: [
+      {
+        scope: "note",
+        match: /[_^]?[_^=]?[A-Ga-gZz][,']*\d*\/*\d*-?/
+      },
+    ],
+  }
+
+  const chord = {
+    scope: "chord",	// That is: [Ace]
+    begin: /\[/, end: /\]\d*\/*\d*/,
+    contains: [
+      {
+        scope: "note",
+        match: /[_^]?[_^=]?[A-Ga-gZz][,']*\d*-?/
+      },
+    ],
+  }
+
+  const chordSymbol = {
+    scope: "chord-symbol", // That is: "Am"
+    begin: '"', end: '"'
+  }
+
+  const decoration = {
+    scope: "decoration",
+    begin: "!", end: "!"
+  }
+  const decorationShortcut = {
+    scope: "decoration",
+    match: /[\.\~HLMOPSTuv]/
+  }
+  const overlay =	{
+    scope: "overlay",
+    match: /&/
+  }
+
+  const comment =	hljs.COMMENT("%", "$")
+
+  const space = {
+    scope: "space",
+    match: / /
+  }
+  const error = {
+    scope: "error",
+    match: /./
+  }
 
   return {
     name: "Abc Notation",
     aliases: ["abc"],
     contains: [
-      // fields
+      //
+      // header lines and directives (that is, everything except music lines
+      //
       {
-        scope: "meta",
-        match: /^[XTCOAMLQPZNGHKRBDFSImrsUVWw+]:/
+        scope: "header",
+        match: /^[XTCOAMLQPZNGHKRBDFSImrsUVW+]:/
       },
-      // text content in fields
+      // content of header lines
       {
-        scope: "meta string",
-        match: /(?<=[COAZHBDFS]:).*/
+        scope: "title",
+        match: /(?<=\nT:)[^%\n]*/
       },
-      // title content, lyrics content
       {
-        scope: "title",
-        match: /(?<=[TWw]:).*/
+        scope: "part",
+        match: /(?<=\nP:)[^%\n]*/
+      },
+      {
+        scope: "tempo",
+        match: /(?<=\nQ:)[^%\n]*/
       },
-      // duplets, triplets, quadruplets, etc.
       {
-        scope: "operator",
-        match: /\([2-9]/
+        scope: "key",
+        match: /(?<=\nK:)[^%\n]*/
       },
-      // chord symbols
-      hljs.QUOTE_STRING_MODE,
-      // numbers
-      hljs.NUMBER_MODE,
-      // decorations
       {
-        scope: "string",
-        begin: "!", end: "!"
+        scope: "meter",
+        match: /(?<=\nM:)[^%\n]*/
       },
-      // repeat symbols, bar symbols
       {
-        scope: "operator",
-        match: /(\|\||::|\[?:*\|(:|\[?[1-9])*\]?)/
+        scope: "voice",
+        match: /(?<=\nV:)[^%\n]*/
       },
       {
-        scope: "operator",
-        match: /\[".+?"/
+        scope: "history",
+        match: /(?<=\nH:)[^%\n]*/
       },
-      // ties, slurs, grace notes, chords, unisons
       {
-        scope: "operator",
-        match: /(?<!(\w:|%%).*)(-|\(|\)|{|}|\[|\])/
+        scope: "notes",
+        match: /(?<=\nN:)[^%\n]*/
       },
-      // macro expressions
       {
-        scope: "operator",
-        match: "="
+        scope: "words",
+        match: /(?<=\nW:)[^%\n]*/
+      },
+      // Everything not enumerated above - don't need separate styles for these.
+      {
+        scope: "header-content",
+        match: /(?<=[XCOALZGRBDFSImrsU]:)[^%\n]*/
       },
       // directives
       {
-        scope: "keyword",
+        // For staves, we want to style the staff information uniquely
+        scope: "staves",
+        begin: /^%%staves/,
+        end: untilCommentOrEndOfLine,
+        contains: [
+          {
+            scope: "staves-content",
+            match: /[^%\n]*/,
+            endsWithParent: true
+          }
+        ]
+      },
+      {
+        scope: "directive",
         match: /^%%\S+/
       },
-      // accidental directives
       {
-        scope: "keyword",
-        match: new RegExp(accidentalDirectivePrefix +"(not|octave|pitch|none|added|all)")
+        // All other directives just get a simple style
+        scope: "directive-body",
+        match: /(?<=%%\S+).*/
       },
-      // font directives
+      // lyrics content
       {
-        scope: "literal",
-        match: new RegExp(fontDirectivePrefix +"\\*")
+        scope: "lyrics",
+        begin: /^w:/,
+        end: untilCommentOrEndOfLine,
+        contains: [
+          {
+            scope: "lyrics-content",
+            match: /[^%\n]*/,
+            endsWithParent: true
+          }
+        ]
       },
-      // text directives
+
+      //
+      // Music lines
+      //
+
+      inlineHeader,
+      note,
+      barLine,
+      tuple,
+      graceNotes,
+      chord,
+      chordSymbol,
+      decoration,
+      decorationShortcut,
+      overlay,
+      comment,
+      space,
+
+      // slur
       {
-        scope: "string",
-        match: new RegExp(textDirectivePrefix +".*")
+        scope: "slur",
+        begin: /\(/, end: /\)/,
+        contains: [
+          inlineHeader,
+          note,
+          barLine,
+          tuple,
+          graceNotes,
+          chord,
+          chordSymbol,
+          decoration,
+          decorationShortcut,
+          overlay,
+          comment,
+          space,
+          {
+            scope: "error",
+            match: /[^)]/
+          }
+        ]
       },
-      // comments
-      hljs.COMMENT("%", "$"),
+      error,
     ]
   };
 }