module Neo4j
  module NodeMixin
    module ClassMethods

      def load_wrapper(node)
        wrapped_node = self.orig_new
        wrapped_node.init_on_load(node)
        wrapped_node
      end


      # Creates a new node or loads an already existing Neo4j node.
      #
      # You can use two callback method to initialize the node
      # init_on_load:: this method is called when the node is loaded from the database
      # init_on_create:: called when the node is created, will be provided with the same argument as the new method
      #
      #
      # Does
      # * sets the neo4j property '_classname' to self.class.to_s
      # * creates a neo4j node java object (in @_java_node)
      #
      # If you want to provide your own initialize method you should instead implement the
      # method init_on_create method.
      #
      # === Example
      #
      #   class MyNode
      #     include Neo4j::NodeMixin
      #
      #     def init_on_create(name, age)
      #        self[:name] = name
      #        self[:age] = age
      #     end
      #   end
      #
      #   node = MyNode.new('jimmy', 23)
      #
      def new(*args)
        node = Neo4j::Node.create
        wrapped_node = super()
        Neo4j::IdentityMap.add(node, wrapped_node)
        wrapped_node.init_on_load(node)
        wrapped_node.init_on_create(*args)
        wrapped_node
      end

      alias_method :create, :new
    end
  end
end