1 files changed, 89 insertions, 0 deletions
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..e9fd447
--- /dev/null
+++ b/main.py
@@ -0,0 +1,89 @@
+import re, jinja2, mistune, shutil, os, pathlib, tarfile
+from pygments.lexers import get_lexer_by_name
+from pygments.formatters import HtmlFormatter
+from pygments import highlight
+
+
+class PygmentsHTMLRenderer(mistune.HTMLRenderer):
+    def block_code(self, code: str, info = None):
+        if not info:
+            return '\n<pre><code>%s</code></pre>\n' % mistune.escape(code)
+        lexer = get_lexer_by_name(info, stripall=True)
+        formatter = HtmlFormatter(lineseparator='<br>')
+        return highlight(code, lexer, formatter)
+
+
+def convert_markdown(page: str):
+    plugins = ['footnotes', 'table', 'strikethrough', 'url']
+    renderer = PygmentsHTMLRenderer(escape=False)
+    return mistune.create_markdown(plugins=plugins, renderer=renderer)(page)
+
+
+def extract_title(page: str):
+    matches = re.match('<h1>(.*?)</h1>', page)
+    if matches:
+        return matches.group(1)
+    return 'BlankHex'
+
+
+def handle_file(path: str, input_dir: str, output_dir: str, template_name: str):
+    # Calculate input and output paths
+    relpath = os.path.relpath(path, input_dir)
+    input_path = path
+    output_path = os.path.join(output_dir, relpath)
+    if input_path.endswith('.md'):
+        output_path = output_path.replace('.md', '.html')
+
+    # Don't convert if output path exists
+    if os.path.exists(output_path):
+        return
+
+    # Run conversion
+    pathlib.Path(os.path.dirname(output_path)).mkdir(parents=True, exist_ok=True)
+    if input_path.endswith('.md'):
+        # Read Markdown document
+        with open(input_path, 'r') as handle:
+            markdown_page = handle.read()
+
+        # Get Pygments styles for light and dark themes
+        light_style = HtmlFormatter(style='default').get_style_defs()
+        dark_style = HtmlFormatter(style='monokai').get_style_defs()
+
+        # Convert Markdown document to HTML document
+        html_page = convert_markdown(markdown_page)
+        html_header = extract_title(html_page)
+        environment = jinja2.Environment(loader=jinja2.FileSystemLoader('template/'))
+        template = environment.get_template(template_name)
+        output_page = template.render(title=html_header,
+                                      body=html_page,
+                                      light_style=light_style,
+                                      dark_style=dark_style)
+
+        # Write HTML document
+        with open(output_path, 'w') as handle:
+            handle.write(output_page)
+    else:
+        # Copy file as is
+        shutil.copy(path, output_path)
+
+
+def convert_dir(input_dir: str, output_dir: str, template_name: str):
+    # Convert or copy every file from the input directory to the output directory
+    for subdir, dirs, files in os.walk(input_dir):
+        for file in files:
+            handle_file(os.path.join(subdir, file), input_dir, output_dir, template_name)
+
+
+# Remove output from previous run
+if os.path.isdir('public'):
+    shutil.rmtree('public')
+if os.path.isfile('public.tgz'):
+    os.remove('public.tgz')
+
+# Run conversion for english, russian and 'common' folder
+convert_dir('content/en', 'public/en', 'en.html')
+convert_dir('content/ru', 'public/ru', 'ru.html')
+convert_dir('content/common', 'public', 'none.html')
+with tarfile.open('public.tgz', 'w:gz') as tar:
+    for file in os.listdir('public'):
+        tar.add(os.path.join('public', file), file)