datamapper/dm-core

module DataMapper
  module Resource
    class PersistenceState

      # a not-persisted/modifiable resource
      class Transient < PersistenceState
        def get(subject, *args)
          set_default_value(subject)
          super
        end

        def set(subject, value)
          track(subject)
          super
        end

        def delete
          self
        end

        def commit
          set_child_keys
          set_default_values
          assert_valid_attributes
          create_resource
          set_repository
          add_to_identity_map
          Clean.new(resource)
        end

        def rollback
          self
        end

        def original_attributes
          @original_attributes ||= {}
        end

      private

        def repository
          @repository ||= model.repository
        end

        def set_default_values
          (properties | relationships).each do |subject|
            set_default_value(subject)
          end
        end

        def set_default_value(subject)
          return if subject.loaded?(resource) || !subject.default?
          default = typecast_default(subject, subject.default_for(resource))
          set(subject, default)
        end

        def typecast_default(subject, default)
          return default unless subject.respond_to?(:typecast)

          typecasted_default = subject.send(:typecast, default)
          unless typecasted_default.eql?(default)
            warn "Automatic typecasting of default property values is deprecated " +
                 "(#{default.inspect} was casted to #{typecasted_default.inspect}). " +
                 "Specify the correct type for #{resource.class}."
          end
          typecasted_default
        end

        def track(subject)
          original_attributes[subject] = nil
        end

        def create_resource
          repository.create([ resource ])
        end

        def set_repository
          resource.instance_variable_set(:@_repository, repository)
        end

        def assert_valid_attributes
          properties.each do |property|
            value = get(property)
            unless property.serial? && value.nil?
              property.assert_valid_value(value)
            end
          end
        end

      end # class Transient
    end # class PersistenceState
  end # module Resource
end # module DataMapper