data_page_generator.rb

# Generate pages from individual records in yml files
# (c) 2014 Adolfo Villafiorita
# Distributed under the conditions of the MIT License

module Jekyll

  class DataPage < Page
    def initialize(site, base, dir, data, name, template, extension)
      @site = site
      @base = base
      @dir = dir
      @name = sanitize_filename(data[name]).to_s + "." + extension.to_s

      self.process(@name)
      self.read_yaml(File.join(base, '_layouts'), template + ".html")
      self.data.merge!(data)
      self.data['name'] = data[name] #change from self.data['title'] to self.data['name']
    end

    private

    # strip characters and whitespace to create valid filenames, also lowercase
    def sanitize_filename(name)
      if(name.is_a? Integer)
        return name.to_s
      end
      return name.downcase.strip.gsub(' ', '-').gsub(/[^\w-]/, '')
    end
  end

  class DataPagesGenerator < Generator
    safe true

    def generate(site)
      data = site.config['page_gen']
      if data
        data.each do |data_spec|
          # todo: check input data correctness
          template = data_spec['template'] || data_spec['data']
          name = data_spec['name']
          dir = data_spec['dir'] || data_spec['data']
          extension = data_spec['extension'] || "html"

          if site.layouts.key? template
            records =  site.data[data_spec['data']]
            records.each do |record|
              site.pages << DataPage.new(site, site.source, dir, record, name, template, extension)
            end
          else
            puts "error. could not find #{data_file}" if not File.exists?(data_file)
            puts "error. could not find template #{template}" if not site.layouts.key? template
          end
        end
      end
    end
  end

  module DataPageLinkGenerator
    # use it like this: {{input | datapage_url: dir}}
    # output: dir / input .html
    def datapage_url(input, dir)
      dir + "/" + sanitize_filename(input) + ".html"
    end

    private

    # strip characters and whitespace to create valid filenames, also lowercase
    def sanitize_filename(name)
      return name.downcase.strip.gsub(' ', '-').gsub(/[^\w-]/, '')
    end
  end

end

Liquid::Template.register_filter(Jekyll::DataPageLinkGenerator)