fix: walkTokens uses marked as this (#2251)

Author: UziTech

docs/USING_PRO.md

@@ -406,19 +406,21 @@ const description = {
     return `\n<dt>${this.parser.parseInline(token.dt)}</dt><dd>${this.parser.parseInline(token.dd)}</dd>`;
   },
   childTokens: ['dt', 'dd'],                 // Any child tokens to be visited by walkTokens
-  walkTokens(token) {                        // Post-processing on the completed token tree
-    if (token.type === 'strong') {
-      token.text += ' walked';
-    }
-  }
 };
 
-marked.use({ extensions: [descriptionlist, description] });
+function walkTokens(token) {                        // Post-processing on the completed token tree
+  if (token.type === 'strong') {
+    token.text += ' walked';
+    token.tokens = this.Lexer.lexInline(token.text)
+  }
+}
+marked.use({ extensions: [descriptionlist, description], walkTokens });
 
-\\ EQUIVALENT TO:
+// EQUIVALENT TO:
 
-marked.use({extensions: [descriptionList] });
-marked.use({extensions: [description]     });
+marked.use({ extensions: [descriptionList] });
+marked.use({ extensions: [description]     });
+marked.use({ walkTokens })
 
 console.log(marked('A Description List:\n'
                  + ':   Topic 1   :  Description 1\n'

src/marked.js

@@ -235,10 +235,10 @@ marked.use = function(...args) {
     // ==-- Parse WalkTokens extensions --== //
     if (pack.walkTokens) {
       const walkTokens = marked.defaults.walkTokens;
-      opts.walkTokens = (token) => {
+      opts.walkTokens = function(token) {
         pack.walkTokens.call(this, token);
         if (walkTokens) {
-          walkTokens(token);
+          walkTokens.call(this, token);
         }
       };
     }
@@ -257,7 +257,7 @@ marked.use = function(...args) {
 
 marked.walkTokens = function(tokens, callback) {
   for (const token of tokens) {
-    callback(token);
+    callback.call(marked, token);
     switch (token.type) {
       case 'table': {
         for (const cell of token.header) {

test/unit/marked-spec.js

@@ -1046,4 +1046,16 @@ br
       ['text', 'br']
     ]);
   });
+
+  it('should asign marked to `this`', () => {
+    marked.use({
+      walkTokens(token) {
+        if (token.type === 'em') {
+          token.text += ' walked';
+          token.tokens = this.Lexer.lexInline(token.text);
+        }
+      }
+    });
+    expect(marked('*text*').trim()).toBe('<p><em>text walked</em></p>');
+  });
 });