Add BETA Arithmatex support

Author: CoinK0in

README.md

@@ -34,7 +34,7 @@ Install the package from source with pip:
 ```bash
 cd mkdocs-encryptcontent-plugin/
 python3 setup.py sdist bdist_wheel
-pip3 install dist/mkdocs_encryptcontent_plugin-0.0.12-py3-none-any.whl
+pip3 install dist/mkdocs_encryptcontent_plugin-0.0.13-py3-none-any.whl
 ```
 
 Enable the plugin in your `mkdocs.yml`:
@@ -188,7 +188,10 @@ plugins:
         password_button_text: 'custome_text_button'
 ```
 
-### [:wrench: BETA] Encrypt Something
+
+## [:wrench: BETA] Features
+
+### Encrypt Something
 
 Related to [issue #9](https://github.com/CoinK0in/mkdocs-encryptcontent-plugin/issues/9)
 
@@ -269,6 +272,24 @@ Your configuration like this :
 ```
 
 :gear: This feature is still in beta, all feedback, improvement, fixes, are welcome.
+
+### Arithmatex support
+
+Related to [issue #12](https://github.com/CoinK0in/mkdocs-encryptcontent-plugin/issues/12)
+
+If Arithmatex markdown extension is set in your markdown extensions to improve math equations rendering, reload rendering after decryption process.
+ 
+When enable the following part of the template is add to force math equations rendering on decrypted content.
+
+```jinja
+{% if arithmatex %}MathJax.typesetPromise(){% endif %}
+```
+
+> **NOTE** It has been tested in Arithmatex `generic` mode only. 
+
+:gear: This feature is still in beta, all feedback, improvement, fixes, are welcome.
+
+
 ## Contributing
 
 From reporting a bug to submitting a pull request: every contribution is appreciated and welcome.

encryptcontent/decrypt-form.tpl.html

@@ -138,6 +138,7 @@ <h1>{{ summary }}</h1>
                     decrypted_content.innerHTML = content;
                     encrypted_content.parentNode.removeChild(encrypted_content);
                     // any post processing on the decrypted content should be done here
+                    {% if arithmatex %}MathJax.typesetPromise(){% endif %}
                     {% if hljs %}
                     document.getElementById("mkdocs-decrypted-content").querySelectorAll('pre code').forEach((block) => {
                         hljs.highlightBlock(block);

encryptcontent/plugin.py

@@ -51,6 +51,7 @@ class encryptContentPlugin(BasePlugin):
         ('encryption_info_message', mkdocs.config.config_options.Type(string_types, default=str(settings['encryption_info_message']))),
         ('global_password', mkdocs.config.config_options.Type(string_types, default=None)),
         ('password', mkdocs.config.config_options.Type(string_types, default=None)),
+        ('arithmatex', mkdocs.config.config_options.Type(bool, default=False)),
         ('hljs', mkdocs.config.config_options.Type(bool, default=False)),
         ('remember_password', mkdocs.config.config_options.Type(bool, default=False)),
         ('disable_cookie_protection', mkdocs.config.config_options.Type(bool, default=False)),
@@ -99,6 +100,7 @@ def __encrypt_content__(self, content):
             'ciphertext_bundle': b';'.join(ciphertext_bundle).decode('ascii'),
             'js_libraries': JS_LIBRARIES,
             # enable / disable features
+            'arithmatex': self.arithmatex,
             'hljs': self.hljs,
             'remember_password': self.remember_password,
             'disable_cookie_protection': self.disable_cookie_protection,
@@ -134,7 +136,7 @@ def on_pre_build(self, config):
         # Check if decryption_failure_message description is set on plugin configuration to overwrite
         decryption_failure_message = plugin_config.get('decryption_failure_message')
         setattr(self, 'decryption_failure_message', decryption_failure_message)
-        # Check if encryption_info_message descruption is set on plugin configuration to overwrite
+        # Check if encryption_info_message description is set on plugin configuration to overwrite
         encryption_info_message = plugin_config.get('encryption_info_message')
         setattr(self, 'encryption_info_message', encryption_info_message)
         # Check if hljs feature need to be enabled, based on theme configuration
@@ -143,6 +145,10 @@ def on_pre_build(self, config):
             highlightjs = config['theme']._vars['highlightjs']       
             if highlightjs:
                 setattr(self, 'hljs', highlightjs)
+        # Check if pymdownx.arithmatex feature need to be enabled, based on markdown_extensions configuration
+        setattr(self, 'arithmatex', None)
+        if 'pymdownx.arithmatex' in config['markdown_extensions']:
+            setattr(self, 'arithmatex', True)
         # Check if tag_encrypted_page feature is enable: add an extra attribute `encrypted` is add on page object
         setattr(self, 'tag_encrypted_page', False)
         if 'tag_encrypted_page' in plugin_config.keys():
@@ -173,7 +179,6 @@ def on_pre_build(self, config):
             encrypted_something = self.config.get('encrypted_something')
             setattr(self, 'encrypted_something', encrypted_something)
 
-
     def on_page_markdown(self, markdown, page, config, **kwargs):
         """
         The page_markdown event is called after the page's markdown is loaded from file and 
@@ -189,7 +194,7 @@ def on_page_markdown(self, markdown, page, config, **kwargs):
         """
         if 'password' in page.meta.keys():
             page_password = page.meta.get('password')
-            # If global_password is set, but you dont want to encrypt content
+            # If global_password is set, but you don't want to encrypt content
             if page_password == '':
                 setattr(self, 'password', None)
             else:
@@ -244,7 +249,7 @@ def on_post_page(self, output_content, page, config, **kwargs):
                 #print({'name': name, 'html tag': tag[0], 'type': tag[1]})
                 something_search = soup.findAll(tag[0], { tag[1]: name })
                 if something_search is not None and len(something_search) > 0:
-                    # Loop for multi childs tags on target element
+                    # Loop for multi child tags on target element
                     for item in something_search:
                         # Remove '\n', ' ' useless content generated by bs4 parsing...
                         item.contents = [content for content in item.contents if not content in ['\n', ' ']]
@@ -255,7 +260,7 @@ def on_post_page(self, output_content, page, config, **kwargs):
                             merge_item = item.contents[0]
                         else:
                             merge_item = ""
-                        # Encrypt childs items on target tags with page password
+                        # Encrypt child items on target tags with page password
                         cipher_bundle = self.__encrypt_text_aes__(merge_item, page.password)
                         encrypted_content = b';'.join(cipher_bundle).decode('ascii')
                         # Replace initial content with encrypted one
@@ -269,4 +274,4 @@ def on_post_page(self, output_content, page, config, **kwargs):
                         else:
                             item['style'] = "display:none"
             output_content = str(soup)
-        return output_content
+        return output_content

setup.py

@@ -11,7 +11,7 @@ def read(fname):
 
 setup(
     name='mkdocs-encryptcontent-plugin',
-    version='0.0.12',
+    version='0.0.13',
     author='CoinK0in',
     author_email='[email protected]',
     description='A MkDocs plugin that encrypt/decrypt markdown content with AES',