maxdemarzi/neography

module Neography
  module NodeRelationship

    DIRECTIONS = ["incoming", "in", "outgoing", "out", "all", "both"]

    def outgoing(types=nil)
      NodeTraverser.new(self).outgoing(types)
    end

    def incoming(types=nil)
      NodeTraverser.new(self).incoming(types)
    end

    def both(types=nil)
      NodeTraverser.new(self).both(types)
    end

    def rels(*types)
      Neography::RelationshipTraverser.new(self, types, :both)
    end

    def rel(dir, type)
      rel = Neography::RelationshipTraverser.new(self, type, dir)
      rel = rel.first unless rel.empty?
      rel
    end

    def rel?(dir=nil, type=nil)
      if DIRECTIONS.include?(dir.to_s)
        !self.neo_server.get_node_relationships(self, dir, type).empty? 
      else
        !self.neo_server.get_node_relationships(self, type, dir).empty? 
      end
    end

  end
end