Integrate atx inline feature

1 files changed, 8 insertions, 1 deletions

diff --git a/src/index.js b/src/index.js
index 0c3e47d..30d216d 100644
--- a/src/index.js
+++ b/src/index.js
@@ -299,6 +299,8 @@ function remarkAttr(userConfig) {
     extend: {},
     scope: 'extended',
     mdAttrConfig: undefined,
+    enableAtxHeaderInline: true,
+    disableBlockElements: false,
   };
   const config = {...defaultConfig, ...userConfig};
 
@@ -312,7 +314,7 @@ function remarkAttr(userConfig) {
   // For each elements, replace the old tokenizer by the new one
   config.elements.forEach(elem => {
     if (supportedElements.indexOf(elem) >= 0) {
-      if (blockElements.indexOf(elem) >= 0) {
+      if (!config.disableBlockElements && blockElements.indexOf(elem) >= 0) {
         const oldElem = tokenizersBlock[elem];
         tokenizersBlock[elem] = tokenizeGenerator('\n', oldElem, config);
       } else if (particularElements.indexOf(elem) >= 0) {
@@ -324,6 +326,11 @@ function remarkAttr(userConfig) {
         elemTokenize.locator = tokenizers[elem].locator;
         tokenizers[elem] = elemTokenize;
       }
+
+      if (config.enableAtxHeaderInline && elem === 'atxHeading') {
+        const oldElem = tokenizersBlock[elem];
+        tokenizersBlock[elem] = tokenizeModifierGenerator(oldElem, config);
+      }
     }
   });
 }