$:.unshift File.dirname(__FILE__)
require 'design/design_docs'

module Exegesis
  class Design
    include Exegesis::Design::DesignDocs
    
    attr_accessor :database
    
    def initialize(db)
      @database = db
    end
    
    def self.use_design_doc_name name
      @design_doc_name = name.to_s
    end
    
    def self.design_doc_name
      @design_doc_name ||= name.to_s.sub(/(Design)$/,'').downcase
    end
    
    def design_doc_name
      self.class.design_doc_name
    end
    
    def get(id)
      doc = Exegesis::Document.instantiate database.get(id)
      doc.database = self.database
      doc
    end
    
    def parse_opts(opts={})
      if opts[:key]
        case opts[:key]
        when Range
          range = opts.delete(:key)
          opts.update({:startkey => range.first, :endkey => range.last})
        when Array
          if opts[:key].any?{|v| v.kind_of?(Range) }
            key = opts.delete(:key)
            opts[:startkey] = key.map {|v| v.kind_of?(Range) ? v.first : v }
            opts[:endkey]   = key.map {|v| v.kind_of?(Range) ? v.last : v }
          end
        end
      elsif opts[:keys] && opts[:keys].empty?
        opts.delete(:keys)
      end

      opts
    end
    
    def view view_name, opts={}
      opts = parse_opts opts
      return [] unless opts[:key] || opts[:startkey] || opts[:keys] || opts[:all]
      opts.delete(:all)
      database.view("#{design_doc_name}/#{view_name}", opts)['rows']
    end
    
    def docs_for view_name, opts={}
      response = view view_name, opts.update({:include_docs => true})
      response.map do |doc| 
        model = Exegesis::Document.instantiate doc['doc']
        model.database = database
        model
      end
    end
    
    def values_for view_name, opts={}
      response = view view_name, opts
      response.map {|row| row['value'] }
    end
    
    def keys_for view_name, opts={}
      response = view view_name, opts
      response.map {|row| row['key'] }
    end
    
    def ids_for view_name, opts={}
      response = view view_name, opts
      response.map {|row| row['id'] }
    end
    
  end
end