Add support for collection

GitHub: fix GH-8

Reported by Jeremy Lloyd Conlin. Thanks!!!

Author: kou

example/Gemfile

@@ -16,3 +16,4 @@ source "https://rubygems.org/"
 
 gem "jekyll"
 gem "jekyll-jupyter-notebook", path: ".."
+gem "webrick"

example/_config.yml

@@ -16,3 +16,7 @@ exclude:
   - "Gemfile.lock"
 plugins:
   - jekyll-jupyter-notebook
+collections:
+  members:
+    output: true
+    permalink: /members/:path

example/_members/alice.md

@@ -0,0 +1,9 @@
+---
+layout: default
+name: alice
+title: Alice
+---
+
+## Title
+
+{% jupyter_notebook "python-alice.ipynb" %}

example/_members/python-alice.ipynb

@@ -0,0 +1,48 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "metadata": {
+    "collapsed": false
+   },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "sys.version_info(major=3, minor=5, micro=2, releaselevel='final', serial=0)"
+      ]
+     },
+     "execution_count": 1,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "import sys\n",
+    "sys.version_info"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.5.2"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

example/index.md

@@ -22,3 +22,7 @@ Here is a Jupyter Notebook:
 {% for post in site.posts %}
   * [{{ post.title }} ({{ post.date | date: "%Y-%m-%d" }})]({{ post.url }})
 {% endfor %}
+
+### Collection: members
+
+  * [Alice]({% link _members/alice.md %})

lib/jekyll-jupyter-notebook/generator.rb

@@ -11,26 +11,48 @@
 # limitations under the License.
 
 module JekyllJupyterNotebook
-  module IFramePage
+  module IFramable
     def output_ext
-      ext + super
+      extname + super
     end
   end
 
   class Generator < Jekyll::Generator
     def generate(site)
+      generate_site_static_files(site)
+      site.collections.each_value do |collection|
+        generate_collection_filtered_entries(collection)
+      end
+    end
+
+    private
+    def generate_site_static_files(site)
       site.static_files.reject! do |static_file|
-        if static_file.extname == ".ipynb"
-          base = static_file.instance_variable_get(:@base)
-          dir = static_file.instance_variable_get(:@dir)
-          name = static_file.name
-          page = Jekyll::Page.new(site, base, dir, name)
-          page.extend(IFramePage)
-          site.pages << page
-          true
-        else
-          false
-        end
+        next false unless static_file.extname == ".ipynb"
+
+        base = static_file.instance_variable_get(:@base)
+        dir = static_file.instance_variable_get(:@dir)
+        name = static_file.name
+        page = Jekyll::Page.new(site, base, dir, name)
+        page.extend(IFramable)
+        site.pages << page
+        true
+      end
+    end
+
+    def generate_collection_filtered_entries(collection)
+      collection.filtered_entries.reject! do |file_path|
+        full_path = collection.collection_dir(file_path)
+        next false unless File.extname(file_path) == ".ipynb"
+        next false if Jekyll::Utils.has_yaml_header?(full_path)
+
+        document = Jekyll::Document.new(full_path,
+                                        site: collection.site,
+                                        collection: collection)
+        document.extend(IFramable)
+        document.read
+        collection.docs << document
+        true
       end
     end
   end