Encrypt multi ToC

Author: CoinK0in

encryptcontent/decrypt-form.tpl.html

@@ -86,30 +86,34 @@ <h1>{{ summary }}</h1>
                 decrypt_button = document.getElementById("mkdocs-decrypt-button"),
                 {% endif %}
                 {% if encrypted_toc %}
-                encrypted_toc = document.getElementById('mkdocs-encrypted-toc'),
+                //encrypted_toc = document.getElementById('mkdocs-encrypted-toc'),
+                // HTMLCollection, cause we can have multiples ToC items ...
+                encrypted_toc = document.getElementsByClassName('mkdocs-encrypted-toc'),
                 {% endif %}
                 decrypt_form = document.getElementById('mkdocs-decrypt-form');
             // Adjust password field width to placeholder length
             let input = document.getElementById("mkdocs-content-password");
             input.setAttribute('size', input.getAttribute('placeholder').length);
             {% if encrypted_toc %}
-            // Decrypt Table Of Content
+            // Decrypt all Table Of Content on the page
             var decrypt_toc = function() {
-                // grab the cipher ToC bundle
-                var parts = encrypted_toc.innerHTML.split(';');
-                // decrypt it
-                var content = decrypt_content(
-                    password_input.value,
-                    parts[0],
-                    parts[1],
-                    parts[2]
-                );
-                if (content) {
-                    // success; display the decrypted ToC
-                    encrypted_toc.innerHTML = content;
-                    encrypted_toc.style.display = null;
-                    // any post processing on the decrypted toc should be done here
-                 }
+                for (i = 0; i < encrypted_toc.length; i++) {
+                    // grab the cipher ToC bundle
+                    var parts = encrypted_toc[i].innerHTML.split(';');
+                    // decrypt it
+                    var content = decrypt_content(
+                        password_input.value,
+                        parts[0],
+                        parts[1],
+                        parts[2]
+                    );
+                    if (content) {
+                        // success; display the decrypted ToC
+                        encrypted_toc[i].innerHTML = content;
+                        encrypted_toc[i].style.display = null;
+                        // any post processing on the decrypted toc should be done here
+                    }
+                }
             }
             {% endif %}
             // Decrypt content
@@ -194,4 +198,4 @@ <h1>{{ summary }}</h1>
         };
         document.addEventListener('DOMContentLoaded', init_decryptor);
     })();
-</script>
+</script>

encryptcontent/plugin.py

@@ -57,7 +57,7 @@ class encryptContentPlugin(BasePlugin):
         ('tag_encrypted_page', mkdocs.config.config_options.Type(bool, default=False)),
         ('password_button', mkdocs.config.config_options.Type(bool, default=False)),
         ('password_button_text', mkdocs.config.config_options.Type(string_types, default=str(settings['password_button_text']))),
-        ('encrypted_toc', mkdocs.config.config_options.Type(bool, default=False)),
+        ('encrypted_something', mkdocs.config.config_options.Type(list, default=[])),
     )
 
     def __hash_md5__(self, text):
@@ -102,7 +102,7 @@ def __encrypt_content__(self, content):
             'hljs': self.hljs,
             'remember_password': self.remember_password,
             'disable_cookie_protection': self.disable_cookie_protection,
-            'encrypted_toc': self.encrypted_toc,
+            'encrypted_something': self.encrypted_something,
         })
         return decrypt_form
 
@@ -167,7 +167,7 @@ def on_pre_build(self, config):
             if 'password_button_text' in plugin_config.keys():
                 password_button_text = plugin_config.get('password_button_text')
                 setattr(self, 'password_button_text', password_button_text)
-        # Check if encrypted_toc feature is enable: encrypt table of content (PoC)
+        # Check if encrypted_something feature is enable: encrypt each elke
         setattr(self, 'encrypted_toc', False)
         if 'encrypted_toc' in plugin_config.keys():
             encrypted_toc = self.config.get('encrypted_toc')
@@ -237,23 +237,32 @@ def on_post_page(self, output_content, page, config, **kwargs):
         :param config:  global configuration object
         :return: output of rendered template as string
         """
-        # limit this process only if encrypted_toc feature is enable *(speedup x4)*
+        # Limit this process only if encrypted_toc feature is enable *(speedup x4)*
         if self.encrypted_toc and hasattr(page, 'encrypted'):
             soup = BeautifulSoup(output_content, 'html.parser')
-            toc_search = soup.find("div", { "id" : "mkdocs-encrypted-toc" })
-            if toc_search is not None and len(toc_search.contents) > 0:
-                # Remove '\n', ' ' useless content generated by bs4 parsing...
-                toc_search.contents = [content for content in toc_search.contents if not content in ['\n', ' ']]
-                # Select childs items on tags div and encrypt all content with page password
-                ciphertoc_bundle = self.__encrypt_text_aes__(toc_search.contents[0], page.password)
-                encrypted_toc = b';'.join(ciphertoc_bundle).decode('ascii')
-                toc_search.contents[0].replaceWith(encrypted_toc)
-                if toc_search.has_attr('style'):
-                    if isinstance(toc_search['style'], list):
-                        toc_search['style'].append("display:none")
+            toc_search = soup.findAll("nav", { "class" : "mkdocs-encrypted-toc" })
+            if toc_search is not None and len(toc_search) > 0:
+                # Loop for multi tags (a.k.a: Mobile and Desktop)
+                for item in toc_search:
+                    # Remove '\n', ' ' useless content generated by bs4 parsing...
+                    item.contents = [content for content in item.contents if not content in ['\n', ' ']]
+                    # Merge the content in case there are several elements
+                    if len(item.contents) > 1:
+                        merge_item = ''.join([str(s) for s in item.contents])
                     else:
-                        toc_search['style'] = toc_search['class'] + "display:none"
-                else:
-                    toc_search['style'] = "display:none"
+                        merge_item = item.contents[0]
+                    # Select childs items on target tags and encrypt all content with page password
+                    ciphertoc_bundle = self.__encrypt_text_aes__(merge_item, page.password)
+                    encrypted_toc = b';'.join(ciphertoc_bundle).decode('ascii')
+                    # Replace initial content with encrypted one
+                    bs4_encrypted_content = BeautifulSoup(encrypted_toc, 'html.parser')
+                    item.contents = [bs4_encrypted_content]
+                    if item.has_attr('style'):
+                        if isinstance(item['style'], list):
+                            item['style'].append("display:none")
+                        else:
+                            item['style'] = item['class'] + "display:none"
+                    else:
+                        item['style'] = "display:none"
                 output_content = str(soup)
         return output_content