From 4573594c91ca40e1af6dfe2adc1453c7c4560779 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ramiro=20G=C3=B3mez?= <code@ramiro.org>
Date: Thu, 15 Oct 2020 23:08:14 +0200
Subject: [PATCH] Add filesource function.

---
 logya/sites/base/content/using-tags.md | 16 +++++-----------
 logya/template.py                      | 25 ++++++++++++++++++++++++-
 2 files changed, 29 insertions(+), 12 deletions(-)

diff --git a/logya/sites/base/content/using-tags.md b/logya/sites/base/content/using-tags.md
index 27188f7..69b5b0d 100644
--- a/logya/sites/base/content/using-tags.md
+++ b/logya/sites/base/content/using-tags.md
@@ -3,16 +3,10 @@ title: Using Tags
 description: An example post showing how to use tags.
 created: 2020-10-07 22:08:20
 template: post.html
-tags: [Example, Example, Example, Documentation]
+tags: [Example, Example, Documentation]
 ---
-Donec id elit non mi porta gravida at eget metus. Fusce dapibus, tellus ac
-cursus commodo, tortor mauris condimentum nibh, ut fermentum massa justo sit
-amet risus. Etiam porta sem malesuada magna mollis euismod. Donec sed odio dui.
+This page shows how to use collections using tags as an example. Values are unique and duplicates will be ignored.
 
-Donec id elit non mi porta gravida at eget metus. Fusce dapibus, tellus ac
-cursus commodo, tortor mauris condimentum nibh, ut fermentum massa justo sit
-amet risus. Etiam porta sem malesuada magna mollis euismod. Donec sed odio dui.
-
-Donec id elit non mi porta gravida at eget metus. Fusce dapibus, tellus ac
-cursus commodo, tortor mauris condimentum nibh, ut fermentum massa justo sit
-amet risus. Etiam porta sem malesuada magna mollis euismod. Donec sed odio dui.
\ No newline at end of file
+<pre>
+{{ filesource('content/using-tags.md', lines=8) }}
+</pre>
\ No newline at end of file
diff --git a/logya/template.py b/logya/template.py
index 668efad..0ba4738 100644
--- a/logya/template.py
+++ b/logya/template.py
@@ -1,8 +1,9 @@
 # -*- coding: utf-8 -*-
 from operator import itemgetter
+from pathlib import Path
 from string import ascii_lowercase
 
-from jinja2 import Environment, FileSystemLoader
+from jinja2 import Environment, FileSystemLoader, escape
 
 from logya.util import deduplicate
 
@@ -44,6 +45,24 @@ def _content_list(index: dict, url: str = '') -> list:
     return []
 
 
+def _filesource(root: Path, name: str, lines: int = None, raw: bool = False) -> str:
+    """Read and return source of text files.
+
+    A template function that reads the source of the given file and returns it. Content is escaped by default so it can
+    be rendered safely on a Web page. The lines keyword argument is used to limit the number of lines returned. To not
+    escape the content you can set the raw keyword argument to False. A use case is for documentation projects to show
+    the source code used to render the current example.
+    """
+
+    # Call lstrip to prevent loading files outside the site directory.
+    text = root.joinpath(name.lstrip('/')).read_text()
+    if lines:
+        text = '\n'.join(text.split('\n')[:lines])
+    if raw:
+        return text
+    return escape(text)
+
+
 def _get_docs(index: dict, url: str = '', sort_attr: str = 'created', sort_order: str = 'descending') -> list:
     docs = []
     if url:
@@ -65,8 +84,12 @@ def init_env(L):
     # Enable expression-statement extension that adds the do tag.
     env.add_extension('jinja2.ext.do')
 
+    # Create an alphabetical index for a list of objects.
     env.filters['alpha_index'] = _alpha_index
 
+    # Include the source of a file.
+    env.globals['filesource'] = lambda name, **kwargs: _filesource(L.paths.root, name, **kwargs)
+
     # Get a document from its URL.
     env.globals['get_doc'] = lambda url: L.index.get(url)['doc']