module Neo4j

  org.neo4j.kernel.impl.core.RelationshipProxy.class_eval do
    include Neo4j::Property
    include Neo4j::Equal

    alias_method :_end_node, :getEndNode
    alias_method :_start_node, :getStartNode
    alias_method :_other_node, :getOtherNode


    def del
      delete
    end

    def end_node # :nodoc:
      getEndNode.wrapper
    end

    def start_node # :nodoc:
      getStartNode.wrapper
    end

    def other_node(node) # :nodoc:
      getOtherNode(node._java_node).wrapper
    end


    # same as _java_rel
    # Used so that we have same method for both relationship and nodes
    def wrapped_entity
      self
    end

    def _java_rel
      self
    end

    # Loads the Ruby wrapper for this node
    # If there is no _classname property for this node then it will simply return itself.
    # Same as Neo4j::Node.load_wrapper(node)
    def wrapper
      self.class.wrapper(self)
    end


    # Returns the relationship name
    #
    # ====Example
    #   a = Neo4j::Node.new
    #   a.outgoing(:friends) << Neo4j::Node.new
    #   a.rels.first.rel_type # => 'friends'
    #
    def rel_type
      getType().name()
    end

    def class
      Neo4j::Relationship
    end

  end

  #
  # A relationship between two nodes in the graph. A relationship has a start node, an end node and a type.
  # You can attach properties to relationships with the API specified in Neo4j::JavaPropertyMixin.
  #
  # Relationship are created by invoking the << operator on the rels method on the node as follow:
  #  node.outgoing(:friends) << other_node << yet_another_node
  #
  # or using the Neo4j::Relationship#new method (which does the same thing):
  #  rel = Neo4j::Relationship.new(:friends, node, other_node)
  #
  # The fact that the relationship API gives meaning to start and end nodes implicitly means that all relationships have a direction.
  # In the example above, rel would be directed from node to otherNode.
  # A relationship's start node and end node and their relation to outgoing and incoming are defined so that the assertions in the following code are true:
  #
  #   a = Neo4j::Node.new
  #   b = Neo4j::Node.new
  #   rel = Neo4j::Relationship.new(:some_type, a, b)
  #   # Now we have: (a) --- REL_TYPE ---> (b)
  #
  #    rel.start_node # => a
  #    rel.end_node   # => b
  #
  # Furthermore, Neo4j guarantees that a relationship is never "hanging freely,"
  # i.e. start_node, end_node and other_node are guaranteed to always return valid, non-null nodes.
  #
  # See also the Neo4j::RelationshipMixin if you want to wrap a relationship with your own Ruby class.
  #
  # === Included Mixins
  # * Neo4j::Property
  # * Neo4j::Equal
  #
  # (Those mixin are actually not included in the Neo4j::Relationship but instead directly included in the java class org.neo4j.kernel.impl.core.RelationshipProxy)
  #
  class Relationship
    extend Neo4j::Index::ClassMethods

    self.rel_indexer self

    class << self
      include Neo4j::Load
      include Neo4j::ToJava


      # Returns a org.neo4j.graphdb.Relationship java object (!)
      # Will trigger a event that the relationship was created.
      #
      # === Parameters
      # type :: the type of relationship
      # from_node :: the start node of this relationship
      # end_node  :: the end node of this relationship
      # props :: optional properties for the created relationship
      #
      # === Returns
      # org.neo4j.graphdb.Relationship java object
      #
      # === Examples
      #
      #  Neo4j::Relationship.new :friend, node1, node2, :since => '2001-01-02', :status => 'okey'
      #
      def new(type, from_node, to_node, props=nil)
        java_type = type_to_java(type)
        rel = from_node._java_node.create_relationship_to(to_node._java_node, java_type)
        props.each_pair {|k,v| rel[k] = v} if props
        rel
      end

      # create is the same as new
      alias_method :create, :new

      def load(rel_id, db = Neo4j.started_db)
        wrapper(db.graph.get_relationship_by_id(rel_id.to_i))
      rescue java.lang.IllegalStateException
        nil # the node has been deleted
      rescue org.neo4j.graphdb.NotFoundException
        nil
      end

    end

  end

end