require 'pathname'
module Exegesis
  class Design
    
    include Exegesis::Document
    
    def self.design_directory= dir
      @design_directory = Pathname.new(dir)
    end
    
    def self.design_directory
      @design_directory ||= Pathname.new("designs/#{design_name}")
    end
    
    def self.design_name= n
      @design_name = n.to_s
    end
    
    def self.design_name
      @design_name ||= name.scan(/^(?:[A-Za-z0-9\:]+::)([A-Za-z0-9]+)Design$/).first.first.downcase
    end
    
    def self.compose_canonical
      Dir[(design_directory + 'views' + '**/*.js').to_s].each do |view_func|
        path = view_func.split('/')
        func = path.pop.sub(/\.js$/,'')
        name = path.pop
        canonical_design['views'][name] ||= {}
        canonical_design['views'][name][func] = File.read(view_func)
      end
    end
    
    def self.canonical_design
      @canonical_design ||= {
        '_id' => "_design/#{design_name}",
        'views' => {}
      }
    end
    
    def self.views
      @views ||= canonical_design['views'].keys
    end
    
    def self.reduceable? view_name
      view_name = view_name.to_s
      views.include?(view_name) && canonical_design['views'][view_name].has_key?('reduce')
    end
    
    def self.view name, default_options={}
      define_method name do |*opts|
        options = parse_opts opts.shift, opts.first, default_options
        Exegesis::DocumentCollection.new(call_view(name, options), database)
      end
    end
    
    def self.docs name, default_options={}
      view_name = default_options.delete(:view) || name
      raise ArgumentError, "missing view #{view_name}" unless views.include?(view_name.to_s)
      if [:reduce, :group, :group_level].any? {|key| default_options.has_key?(key)}
        raise ArgumentError, "cannot reduce (:group, :group_level, :reduce) on a docs view"
      end
      
      default_options = {:include_docs => true}.merge(default_options)
      default_options.update({:reduce => false}) if reduceable?(view_name)

      define_method name do |*opts|
        key = opts.shift
        options = parse_opts key, opts.first, default_options
        Exegesis::DocumentCollection.new(call_view(view_name, options), database)
      end
    end
    
    def self.hash name, default_options={}
      view_name = default_options.delete(:view) || name
      raise ArgumentError, "missing view #{view_name}" unless views.include?(view_name.to_s)
      raise NameError, "Cannot return a hash for views without a reduce function" unless reduceable?(view_name)
      if default_options.has_key?(:group) && default_options[:group] == false
        raise ArgumentError, "cannot turn off grouping for a hash view" 
      end
      
      default_options = {:group => true}.merge(default_options)
      
      define_method name do |*opts|
        options = parse_opts opts.shift, opts.first, default_options

        if options.has_key?(:group) && options[:group] == false
          raise ArgumentError, "cannot turn off grouping for a hash view"
        end

        if options[:key]
          options.delete(:group)
          options.delete(:group_level)
        end
        
        response = call_view view_name, options
        if response.size == 1 && response.first['key'].nil?
          response.first['value']
        else
          response.inject({}) do |memo, row|
            memo.update(row['key'] => row['value'])
          end
        end
      end
    end
    
    
    def initialize db
      begin
        super db.raw_get("_design/#{design_name}")
        self.database = db
      rescue RestClient::ResourceNotFound
        db.put("_design/#{design_name}", self.class.canonical_design)
        retry
      end
      unless self['views'] == self.class.canonical_design['views']
        self['views'].update(self.class.canonical_design['views'])
        save
      end
    end
    
    def view name, key=nil, opts={}
      call_view name, parse_opts(key, opts)
    end
    
    def call_view name, opts={}
      url = "_design/#{design_name}/_view/#{name}"
      database.raw_get(url, opts)['rows']
    end
    
    def design_name
      self.class.design_name
    end
    
    def parse_opts key, opts={}, defaults={}
      opts = straighten_args key, opts, defaults
      parse_key opts
      parse_keys opts
      parse_range opts
      parse_reduce opts
      opts
    end
    
    private
    
    def straighten_args key, opts, defaults
      opts ||= {}
      if key.is_a?(Hash)
        opts = key
      elsif ! key.nil?
        opts[:key] = key
      end
      defaults.merge(opts)
    end
    
    def parse_key opts
      if opts[:key]
        if opts[:key].is_a?(Range)
          range = opts.delete(:key)
          opts.update({:startkey => range.first, :endkey => range.last})
        elsif opts[:key].is_a?(Array) && 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
    end
    
    def parse_keys opts
      opts.delete(:keys) if opts[:keys] && opts[:keys].empty?
    end
    
    def parse_range opts
      if opts[:startkey] || opts[:endkey]
        raise ArgumentError, "both a startkey and endkey must be specified if either is" unless opts[:startkey] && opts[:endkey]
      end
    end
    
    def parse_reduce opts
      if opts.has_key?(:group)
        opts[:group_level] = opts.delete(:group) if opts[:group].is_a?(Numeric)
      end
      if opts.keys.any? {|key| [:group, :group_level].include?(key) }
        raise ArgumentError, "cannot include_docs when reducing" if opts[:include_docs]
        if opts.has_key?(:reduce) && opts[:reduce] == false
          raise ArgumentError, "cannot reduce=false when either group or group_level is present"
        end
      end
    end
    
  end
end