#
# Copyright 2011, edelight GmbH
#
# Authors:
#       Markus Korn <markus.korn@edelight.de>
#       Seth Chisamore <schisamo@opscode.com>
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

module Search
  module Overrides
    # Overwrite the search method of recipes to operate locally by using
    # data found in data_bags.
    # Only very basic lucene syntax is supported and also sorting the result
    # is not implemented, if this search method does not support a given query
    # an exception is raised.
    # This search() method returns a block iterator or an Array, depending
    # on how this method is called.
    def search(obj, query=nil, sort=nil, start=0, rows=1000, &block)
      if !sort.nil?
        raise "Sorting search results is not supported"
      end
      _query = Query.parse(query)
      if _query.nil?
        raise "Query #{query} is not supported"
      end
      _result = []

      case obj
      when :node
        nodes = search_nodes(_query, start, rows, &block)
        _result += nodes
      when :role
        roles = search_roles(_query, start, rows, &block)
         _result += roles
      else
        bags = search_data_bag(_query, obj, start, rows, &block)
        _result += bags
      end


      if block_given?
        pos = 0
        while (pos >= start and pos < (start + rows) and pos < _result.size)
          yield _result[pos]
          pos += 1
        end
      else
        return _result.slice(start, rows)
      end
    end

    def search_nodes(_query, start, rows, &block)
      _result = []
      node_path = Chef::Config[:nodes_path] || File.join(Chef::Config[:data_bag_path], "node")
      Dir.glob(File.join(node_path, "*.json")).map do |f|
        # parse and hashify the node
        node = Chef::JSONCompat.from_json(IO.read(f))
        if _query.match(node.to_hash)
          _result << node
        end
      end
      return _result
    end

    def search_roles(_query, start, rows, &block)
      _result = []
      Dir.glob(File.join(Chef::Config[:role_path], "*.json")).map do |f|
        # parse and hashify the role
        role = Chef::JSONCompat.from_json(IO.read(f))
        if _query.match(role.to_hash)
          _result << role
        end
      end
      return _result
    end

    def search_data_bag(_query, bag_name, start, rows, &block)
      _result = []
      data_bag(bag_name.to_s).each do |bag_item_id|
        bag_item = data_bag_item(bag_name.to_s, bag_item_id)
        if _query.match(bag_item)
          _result << bag_item
        end
      end
      return _result
    end
  end
end