spree/spree

module Spree
  module V2
    module Storefront
      class CountrySerializer < BaseSerializer
        set_type :country

        attributes :iso, :iso3, :iso_name, :name, :states_required,
                   :zipcode_required

        attribute :default do |object, params|
          object.default?(params[:store])
        end

        has_many :states, if: proc { |_record, params| params && params[:include_states] }

        has_many :checkout_zone_applicable_states,
                 serializer: ::Spree::V2::Storefront::StateSerializer,
                 if: proc { |_record, params| params && params[:store].present? } do |object, params|
          params[:store].states_available_for_checkout(object)
        end
      end
    end
  end
end