1 files changed, 65 insertions, 0 deletions
diff --git a/src/mistune/directives/include.py b/src/mistune/directives/include.py
new file mode 100644
index 0000000..d2180ba
--- /dev/null
+++ b/src/mistune/directives/include.py
@@ -0,0 +1,65 @@
+import os
+from ._base import DirectivePlugin
+
+
+class Include(DirectivePlugin):
+    def parse(self, block, m, state):
+        source_file = state.env.get('__file__')
+        if not source_file:
+            return {'type': 'block_error', 'raw': 'Missing source file'}
+
+        encoding = 'utf-8'
+        options = self.parse_options(m)
+        if options:
+            attrs = dict(options)
+            if 'encoding' in attrs:
+                encoding = attrs['encoding']
+        else:
+            attrs = {}
+
+        relpath = self.parse_title(m)
+        dest = os.path.join(os.path.dirname(source_file), relpath)
+        dest = os.path.normpath(dest)
+
+        if dest == source_file:
+            return {
+                'type': 'block_error',
+                'raw': 'Could not include self: ' + relpath,
+            }
+
+        if not os.path.isfile(dest):
+            return {
+                'type': 'block_error',
+                'raw': 'Could not find file: ' + relpath,
+            }
+
+        with open(dest, 'rb') as f:
+            content = f.read()
+            content = content.decode(encoding)
+
+        ext = os.path.splitext(relpath)[1]
+        if ext in {'.md', '.markdown', '.mkd'}:
+            new_state = block.state_cls()
+            new_state.env['__file__'] = dest
+            new_state.process(content)
+            block.parse(new_state)
+            return new_state.tokens
+
+        elif ext in {'.html', '.xhtml', '.htm'}:
+            return {'type': 'block_html', 'raw': content}
+
+        attrs['filepath'] = dest
+        return {
+            'type': 'include',
+            'raw': content,
+            'attrs': attrs,
+        }
+
+    def __call__(self, directive, md):
+        directive.register('include', self.parse)
+        if md.renderer and md.renderer.NAME == 'html':
+            md.renderer.register('include', render_html_include)
+
+
+def render_html_include(renderer, text, **attrs):
+    return '<pre class="directive-include">\n' + text + '</pre>\n'