module Neo4j::ActiveRel
  module Persistence
    extend ActiveSupport::Concern
    include Neo4j::Shared::Persistence

    class RelInvalidError < RuntimeError
      attr_reader :record

      def initialize(record)
        @record = record
        super(@record.errors.full_messages.join(", "))
      end
    end

    class ModelClassInvalidError < RuntimeError; end

    def save(*)
      update_magic_properties
      create_or_update
    end

    def save!(*args)
      unless save(*args)
        raise RelInvalidError.new(self)
      end
    end

    def create_model(*)
      confirm_node_classes
      create_magic_properties
      set_timestamps
      properties = convert_properties_to :db, props
      rel = _create_rel(properties)
      init_on_load(rel, to_node, from_node, @rel_type)
      true
    end

    module ClassMethods

      # Creates a new relationship between objects
      # @param [Hash] props the properties the new relationship should have
      def create(props = {})
        relationship_props = extract_association_attributes!(props) || {}
        new(props).tap do |obj|
          relationship_props.each do |prop, value|
            obj.send("#{prop}=", value)
          end
          obj.save
        end
      end

      # Same as #create, but raises an error if there is a problem during save.
      def create!(*args)
        unless create(*args)
          raise RelInvalidError.new(self)
        end
      end
    end

    private 

    def confirm_node_classes
      [from_node, to_node].each do |node|
        check = from_node == node ? :_from_class : :_to_class
        next if self.class.send(check) == :any
        unless self.class.send(check) == node.class        
          raise ModelClassInvalidError, "Node class was #{node.class}, expected #{self.class.send(check)}" 
        end
      end
    end

    def _create_rel(*args)
      session = self.class.neo4j_session
      props = self.class.default_property_values(self)
      props.merge!(args[0]) if args[0].is_a?(Hash)
      set_classname(props)
      from_node.create_rel(type, to_node, props)
    end
  end
end