middleman--middleman/lib/middleman/core_extensions/data.rb

require "yaml"
require "httparty"
require "thor"

module Middleman::CoreExtensions::Data
  class << self
    def registered(app)
      app.set :data_dir, "data"
      app.extend ClassMethods
      app.helpers Helpers
    end
    alias :included :registered
  end
  
  module Helpers
    def data
      @@data ||= DataObject.new(self)
    end
  end
  
  class DataObject
    def initialize(app)
      @app = app
    end
    
    def method_missing(path)
      response = nil
      
      @@local_sources ||= {}
      @@remote_sources ||= {}
      
      if @@local_sources.has_key?(path.to_s)
        response = @@local_sources[path.to_s]
      elsif @@remote_sources.has_key?(path.to_s)
        response = HTTParty.get(@@remote_sources[path.to_s]).parsed_response
      else
        file_path = File.join(@app.class.root, @app.class.data_dir, "#{path}.yml")
        if File.exists? file_path
          response = YAML.load_file(file_path)
        end
      end
      
      if response
        recursively_enhance(response)
      end
    end
    
    def self.add_source(name, json_url)
      @@remote_sources ||= {}
      @@remote_sources[name.to_s] = json_url
    end

    def self.data_content(name, content)
      @@local_sources ||= {}
      @@local_sources[name.to_s] = content
    end
  
  private
    def recursively_enhance(data)
      if data.is_a? Hash
        data = Thor::CoreExt::HashWithIndifferentAccess.new(data)
        data.each do |key, val|
          data[key] = recursively_enhance(val)
        end
        data
      elsif data.is_a? Array
        data.each_with_index do |val, i|
          data[i] = recursively_enhance(val)
        end
        data
      else
        data
      end
    end
  end
  
  module ClassMethods
    # Makes HTTP json data available in the data object
    #
    #     data_source :my_json, "http://my/file.json"
    #
    # Available in templates as:
    #
    #     data.my_json
    def data_source(name, url)
      DataObject.add_source(name, url)
    end
    
    def data_content(name, content)
      DataObject.data_content(name, content)
    end
  end
end