1 files changed, 56 insertions, 0 deletions
diff --git a/nikola/plugins/shortcode/gist.py b/nikola/plugins/shortcode/gist.py
new file mode 100644
index 0000000..64fd0d9
--- /dev/null
+++ b/nikola/plugins/shortcode/gist.py
@@ -0,0 +1,56 @@
+# -*- coding: utf-8 -*-
+# This file is public domain according to its author, Brian Hsu
+
+"""Gist directive for reStructuredText."""
+
+import requests
+
+from nikola.plugin_categories import ShortcodePlugin
+
+
+class Plugin(ShortcodePlugin):
+    """Plugin for gist directive."""
+
+    name = "gist"
+
+    def set_site(self, site):
+        """Set Nikola site."""
+        self.site = site
+        site.register_shortcode('gist', self.handler)
+        return super(Plugin, self).set_site(site)
+
+    def get_raw_gist_with_filename(self, gistID, filename):
+        """Get raw gist text for a filename."""
+        url = '/'.join(("https://gist.github.com/raw", gistID, filename))
+        return requests.get(url).text
+
+    def get_raw_gist(self, gistID):
+        """Get raw gist text."""
+        url = "https://gist.github.com/raw/{0}".format(gistID)
+        try:
+            return requests.get(url).text
+        except requests.exceptions.RequestException:
+            raise self.error('Cannot get gist for url={0}'.format(url))
+
+    def handler(self, gistID, filename=None, site=None, data=None, lang=None, post=None):
+        """Create HTML for gist."""
+        if 'https://' in gistID:
+            gistID = gistID.split('/')[-1].strip()
+        else:
+            gistID = gistID.strip()
+        embedHTML = ""
+        rawGist = ""
+
+        if filename is not None:
+            rawGist = (self.get_raw_gist_with_filename(gistID, filename))
+            embedHTML = ('<script src="https://gist.github.com/{0}.js'
+                         '?file={1}"></script>').format(gistID, filename)
+        else:
+            rawGist = (self.get_raw_gist(gistID))
+            embedHTML = ('<script src="https://gist.github.com/{0}.js">'
+                         '</script>').format(gistID)
+
+        output = '''{}
+        <noscript><pre>{}</pre></noscript>'''.format(embedHTML, rawGist)
+
+        return output, []