maxdemarzi/neography

module Neography
  class Rest
    module RelationshipIndexes
      include Neography::Rest::Helpers

      def list_relationship_indexes
        @connection.get("/index/relationship")
      end

      def create_relationship_index(name, type = "exact", provider = "lucene", extra_config = nil)
        config = {
          :type => type,
          :provider => provider
        }
        config.merge!(extra_config) unless extra_config.nil?
        options = {
          :body => (
            { :name => name,
              :config => config
            }
          ).to_json,
          :headers => json_content_type
        }
        @connection.post("/index/relationship", options)
      end

      def create_relationship_auto_index(type = "exact", provider = "lucene")
        create_relationship_index("relationship_auto_index", type, provider)
      end

      def add_relationship_to_index(index, key, value, id, unique = false)
        options = {
          :body => (
            { :uri   => @connection.configuration + "/relationship/#{get_id(id)}",
              :key   => key,
              :value => value
            }
          ).to_json,
          :headers => json_content_type
        }
        path = unique ? "/index/relationship/%{index}?unique" % {:index => index} : "/index/relationship/%{index}" % {:index => index}
        @connection.post(path, options)
      end

      def get_relationship_index(index, key, value)
        index = @connection.get("/index/relationship/%{index}/%{key}/%{value}" % {:index => index, :key => key, :value => encode(value)}) || []
        return nil if index.empty?
        index
      end

      def find_relationship_index(index, key_or_query, value = nil)
        if value
          index = find_relationship_index_by_key_value(index, key_or_query, value)
        else
          index = find_relationship_index_by_query(index, key_or_query)
        end
        return nil if index.empty?
        index
      end

      def find_relationship_index_by_key_value(index, key, value)
        @connection.get("/index/relationship/%{index}/%{key}/%{value}" % {:index => index, :key => key, :value => encode(value)}) || []
      end

      def find_relationship_index_by_query(index, query)
        @connection.get("/index/relationship/%{index}?query=%{query}" % {:index => index, :query => encode(query)}) || []
      end

      # Mimick original neography API in Rest class.
      def remove_relationship_from_index(index, id_or_key, id_or_value = nil, id = nil)
        if id
          remove_relationship_index_by_value(index, id, id_or_key, id_or_value)
        elsif id_or_value
          remove_relationship_index_by_key(index, id_or_value, id_or_key)
        else
          remove_relationship_index_by_id(index, id_or_key)
        end
      end

      def remove_relationship_index_by_id(index, id)
        @connection.delete("/index/relationship/%{index}/%{id}" % {:index => index, :id => get_id(id)})
      end

      def remove_relationship_index_by_key(index, id, key)
        @connection.delete("/index/relationship/%{index}/%{key}/%{id}" % {:index => index, :id => get_id(id), :key => key})
      end

      def remove_relationship_index_by_value(index, id, key, value)
        @connection.delete("/index/relationship/%{index}/%{key}/%{value}/%{id}" % {:index => index, :id => get_id(id), :key => key, :value => encode(value)})
      end

      def drop_relationship_index(index)
        @connection.delete("/index/relationship/%{index}" % {:index => index})
      end

      def create_unique_relationship(index, key, value, type, from, to, props = nil)
        body = {
          :key   => key,
          :value => value,
          :type  => type,
          :start => @connection.configuration + "/node/#{get_id(from)}",
          :end   => @connection.configuration + "/node/#{get_id(to)}",
          :properties => props
        }
        options = { :body => body.to_json, :headers => json_content_type }

        @connection.post("/index/relationship/%{index}?unique" % {:index => index}, options)
      end

      def get_or_create_unique_relationship(index, key, value, properties = {})
        options = {
          :body => (
            { :properties => properties,
              :key => key,
              :value => value
            }
          ).to_json,
          :headers => json_content_type
        }
        @connection.post("/index/relationship/%{index}?uniqueness=%{function}" %  {:index => index, :function => 'get_or_create'}, options)

      end

      def create_or_fail_unique_relationship(index, key, value, properties = {})
        options = {
          :body => (
            { :properties => properties,
              :key => key,
              :value => value
            }
          ).to_json,
          :headers => json_content_type
        }
        @connection.post("/index/relationship/%{index}?uniqueness=%{function}" %  {:index => index, :function => 'create_or_fail'}, options)

      end

    end
  end
end