require 'yaml'
module Database

  # Load the standard vision toolkit database and re-arrage it to be simply exportable 
  # to liquid as assigns. All this is based on Shopify
  def self.tables
    @tables ||= begin
      db = YAML.load_file(File.dirname(__FILE__) + '/vision.database.yml')

      # From vision source      
      db['products'].each do |product|
        collections = db['collections'].find_all do |collection|
          collection['products'].any? { |p| p['id'].to_i == product['id'].to_i }
        end
        product['collections'] = collections
      end

      # key the tables by handles, as this is how liquid expects it.      
      db = db.inject({}) do |assigns, (key, values)|
        assigns[key] = values.inject({}) { |h, v| h[v['handle']] = v; h; }
        assigns
      end

      # Some standard direct accessors so that the specialized templates 
      # render correctly
      db['collection'] = db['collections'].values.first
      db['product'] = db['products'].values.first
      db['blog'] = db['blogs'].values.first
      db['article'] = db['blog']['articles'].first

      db['cart'] = {
          'total_price' => db['line_items'].values.inject(0) { |sum, item| sum += item['line_price'] * item['quantity'] },
          'item_count' => db['line_items'].values.inject(0) { |sum, item| sum += item['quantity'] },
          'items' => db['line_items'].values
      }

      db
    end
  end
end

if __FILE__ == $0
  p Database.tables['collections']['frontpage'].keys
  #p Database.tables['blog']['articles']
end