#!/usr/bin/env python # -*- coding: utf-8 -*- from __future__ import unicode_literals import os import markdown.inlinepatterns import markdown.extensions import markdown.util import macros.plugin import slugify as slugify_impl class ClickHouseLinkMixin(object): def handleMatch(self, m, data): single_page = (os.environ.get('SINGLE_PAGE') == '1') try: el, start, end = super(ClickHouseLinkMixin, self).handleMatch(m, data) except IndexError: return if el is not None: href = el.get('href') or '' is_external = href.startswith('http:') or href.startswith('https:') if is_external: if not href.startswith('https://clickhouse.tech'): el.set('rel', 'external nofollow noreferrer') elif single_page: if '#' in href: el.set('href', '#' + href.split('#', 1)[1]) else: el.set('href', '#' + href.replace('/index.md', '/').replace('.md', '/')) return el, start, end class ClickHouseAutolinkPattern(ClickHouseLinkMixin, markdown.inlinepatterns.AutolinkInlineProcessor): pass class ClickHouseLinkPattern(ClickHouseLinkMixin, markdown.inlinepatterns.LinkInlineProcessor): pass class ClickHousePreprocessor(markdown.util.Processor): def run(self, lines): if os.getenv('QLOUD_TOKEN'): for line in lines: if '<!--hide-->' not in line: yield line else: for line in lines: yield line class ClickHouseMarkdown(markdown.extensions.Extension): def extendMarkdown(self, md, md_globals): md.preprocessors['clickhouse'] = ClickHousePreprocessor() md.inlinePatterns['link'] = ClickHouseLinkPattern(markdown.inlinepatterns.LINK_RE, md) md.inlinePatterns['autolink'] = ClickHouseAutolinkPattern(markdown.inlinepatterns.AUTOLINK_RE, md) def makeExtension(**kwargs): return ClickHouseMarkdown(**kwargs) def slugify(value, separator): return slugify_impl.slugify(value, separator=separator, word_boundary=True, save_order=True) def get_translations(dirname, lang): import babel.support return babel.support.Translations.load( dirname=dirname, locales=[lang, 'en'] ) class PatchedMacrosPlugin(macros.plugin.MacrosPlugin): disabled = False def on_config(self, config): super(PatchedMacrosPlugin, self).on_config(config) self.env.comment_start_string = '{##' self.env.comment_end_string = '##}' def on_env(self, env, config, files): env.add_extension('jinja2.ext.i18n') dirname = os.path.join(config.data['theme'].dirs[0], 'locale') lang = config.data['theme']['language'] env.install_gettext_translations( get_translations(dirname, lang), newstyle=True ) chunk_size = 10240 env.filters['chunks'] = lambda line: [line[i:i+chunk_size] for i in range(0, len(line), chunk_size)] return env def render(self, markdown): if not self.disabled: return self.render_impl(markdown) else: return markdown def render_impl(self, markdown): md_template = self.env.from_string(markdown) return md_template.render(**self.variables) macros.plugin.MacrosPlugin = PatchedMacrosPlugin