require "active_support/concern"

module Protobuf
  module ActiveRecord
    module MassAssignmentSecurity
      module Persistence
        extend ::ActiveSupport::Concern

        module ClassMethods
          # :nodoc:
          def create(attributes, options = {})
            attributes = attributes_from_proto(attributes) if attributes.is_a?(::Protobuf::Message)

            super(attributes, options)
          end

          # :nodoc:
          def create!(attributes, options = {})
            attributes = attributes_from_proto(attributes) if attributes.is_a?(::Protobuf::Message)

            super(attributes, options)
          end
        end

        # :nodoc:
        def assign_attributes(attributes, options = {})
          attributes = attributes_from_proto(attributes) if attributes.is_a?(::Protobuf::Message)

          super(attributes, options)
        end

        # :nodoc:
        def update_attributes(attributes, options = {})
          attributes = attributes_from_proto(attributes) if attributes.is_a?(::Protobuf::Message)

          super(attributes, options)
        end

        # :nodoc:
        def update_attributes!(attributes, options = {})
          attributes = attributes_from_proto(attributes) if attributes.is_a?(::Protobuf::Message)

          super(attributes, options)
        end
      end

      # Override Active Record's initialize method so it can accept a protobuf
      # message as it's attributes.
      # :noapi:
      def initialize(*args, &block)
        args[0] = attributes_from_proto(args.first) if args.first.is_a?(::Protobuf::Message)

        super(*args, &block)
      end
    end
  end
end