Merge pull request #30 from neatc0der/feature/encryptcontent

feature/encryptcontent

Author: neatc0der

changelog/v2.2.0.md

@@ -1,4 +1,5 @@
 # v2.2.0
 
-* Fixes distribute workflow
+* Adds [support for mkdocs-encryptcontent-plugin](https://github.com/neatc0der/mkdocs-markmap/issues/26)
 * Adds troubleshooting for [nav tree lists markmap files](https://github.com/neatc0der/mkdocs-markmap/issues/33)
+* Fixes distribute workflow

mkdocs_markmap/plugin.py

@@ -91,20 +91,24 @@ def on_config(self, config: Config) -> Config:
 
         return config
 
-    def on_post_page(self, output_content: str, config: Config, **kwargs) -> str:
-        soup: BeautifulSoup = BeautifulSoup(output_content, 'html.parser')
-        page: Page = kwargs.get('page')
-
-        markmaps: ResultSet = soup.find_all(class_='language-markmap')
-        if not any(markmaps):
+    def on_post_page(self, html: str, page: Page, config: Config, **kwargs) -> str:
+        if not getattr(page, '_found_markmap', False):
             log.info(f"no markmap found: {page.file.name}")
-            return output_content
+            return html
 
+        soup: BeautifulSoup = BeautifulSoup(html, 'html.parser')
         script_base_url: str = re.sub(r'[^/]+?/', '../', re.sub(r'/+?', '/', page.url)) + 'js/'
         js_path: Path = Path(config['site_dir']) / 'js'
         self._load_scripts(soup, script_base_url, js_path)
         self._add_statics(soup)
 
+        return str(soup)
+
+    def on_page_content(self, html: str, page: Page, **kwargs) -> str:
+        soup: BeautifulSoup = BeautifulSoup(html, 'html.parser')
+        markmaps: ResultSet = soup.find_all(class_='language-markmap')
+        setattr(page, '_found_markmap', any(markmaps))
+
         for index, markmap in enumerate(markmaps):
             markmap: Tag
             tag_id: str = f'markmap-{index}'

mkdocs_markmap/static_files/mkdocs-markmap.js

@@ -1,19 +1,33 @@
 (function initializeMarkmap() {
-    const markmap_transformer = new markmap.Transformer();
-    const markmaps = document.getElementsByClassName('mkdocs-markmap');
-    var el, content, svg, root, m;
-    for (var i = 0; i < markmaps.length; i++) {
-        el = markmaps[i];
-        content = el.getAttribute('data-markdown').replaceAll('&#10;', '\n');
-        svg = el.querySelector('svg');
-        root = markmap_transformer.transform(content).root;
-        m = markmap.Markmap.create(svg, null, root);
+    function update_markmaps() {
+        const markmap_transformer = new markmap.Transformer();
+        const markmaps = document.getElementsByClassName('mkdocs-markmap');
+        var el, content, svg, root, m;
+        for (var i = 0; i < markmaps.length; i++) {
+            el = markmaps[i];
+            content = el.getAttribute('data-markdown').replaceAll('&#10;', '\n');
+            svg = el.querySelector('svg');
+            root = markmap_transformer.transform(content).root;
+            m = markmap.Markmap.create(svg, null, root);
 
-        (function(obj, e, r) {
-            obj.rescale(1).then(function() {
-                e.parentElement.style.height = (e.getBBox().height + 10) + "px";
-                requestAnimationFrame(() => { obj.fit(); })
-            });
-        })(m, svg, root);
+            (function(obj, e, r) {
+                obj.rescale(1).then(function() {
+                    e.parentElement.style.height = (e.getBBox().height + 10) + "px";
+                    requestAnimationFrame(() => { obj.fit(); })
+                });
+            })(m, svg, root);
+        }
     }
+
+    var MutationObserver = window.MutationObserver || window.WebKitMutationObserver;
+    var observer = new MutationObserver(function(mutations) {
+        update_markmaps();
+    });
+
+    var target = document.getElementById('mkdocs-decrypted-content');
+    if (undefined != target) {
+        observer.observe(target, { childList: true });
+    }
+
+    update_markmaps();
 })();