apply to ruby

highlightjs · Mar 3, 2020 · 52beaf2 · 52beaf2
1 parent 2978c8b
commit 52beaf2
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 8 deletions.
diff --git a/src/languages/c-like.js b/src/languages/c-like.js
@@ -32,10 +32,6 @@ export default function(hljs) {
   // https://en.cppreference.com/w/cpp/language/escape
   // \\ \x \xFF \u2837 \u00323747 \374
   var CHARACTER_ESCAPES = '\\\\(x[0-9A-Fa-f]{2}|u[0-9A-Fa-f]{4,8}|[0-7]{3}|\\S)'
-  var END_GROUP_SAME_AS_BEGIN = {
-    'after:begin': (m, resp) => { resp.data.heredoc = m[1]; },
-    'before:end': (m, resp) => { if (resp.data.heredoc !== m[1]) resp.ignoreMatch(); }
-  };
   var STRINGS = {
     className: 'string',
     variants: [
@@ -51,7 +47,7 @@ export default function(hljs) {
       Object.assign({
         begin: /(?:u8?|U|L)?R"([^()\\ ]{0,16})\(/,
         end: /\)([^()\\ ]{0,16})"/,
-      }, END_GROUP_SAME_AS_BEGIN)
+      }, hljs.END_FIRST_MATCH_SAME_AS_BEGIN)
     ]
   };
 

diff --git a/src/languages/ruby.js b/src/languages/ruby.js
@@ -72,10 +72,10 @@ export default function(hljs) {
         returnBegin: true,
         contains: [
           { begin: /<<[-~]?'?/ },
-          { begin: /\w+/,
-            endSameAsBegin: true,
+          Object.assign({
+            begin: /(\w+)/, end: /(\w+)/,
             contains: [hljs.BACKSLASH_ESCAPE, SUBST],
-          }
+          }, hljs.END_FIRST_MATCH_SAME_AS_BEGIN)
         ]
       }
     ]

diff --git a/src/lib/modes.js b/src/lib/modes.js
@@ -112,3 +112,8 @@ export const METHOD_GUARD = {
   begin: '\\.\\s*' + UNDERSCORE_IDENT_RE,
   relevance: 0
 };
+
+export const END_FIRST_MATCH_SAME_AS_BEGIN = {
+  'after:begin': (m, resp) => { resp.data.heredoc = m[1]; },
+  'before:end': (m, resp) => { if (resp.data.heredoc !== m[1]) resp.ignoreMatch(); }
+};