# frozen_string_literal: true

require 'date'

module Aws
  module Record
    module Marshalers
      class DateMarshaler
        def initialize(opts = {})
          @formatter = opts[:formatter] || Iso8601Formatter
        end

        def type_cast(raw_value)
          case raw_value
          when nil
            nil
          when ''
            nil
          when Date
            raw_value
          when Integer
            Date.parse(Time.at(raw_value).to_s) # assumed timestamp
          else
            Date.parse(raw_value.to_s) # Time, DateTime or String
          end
        end

        def serialize(raw_value)
          date = type_cast(raw_value)
          if date.nil?
            nil
          elsif date.is_a?(Date)
            @formatter.format(date)
          else
            raise ArgumentError, "expected a Date value or nil, got #{date.class}"
          end
        end
      end

      module Iso8601Formatter
        def self.format(date)
          date.iso8601
        end
      end
    end
  end
end