begin
  north_america = Spree::Zone.find_by!(name: 'North America')
rescue ActiveRecord::RecordNotFound
  puts 'Couldn\'t find \'North America\' zone. Did you run `rake db:seed` first?'
  puts 'That task will set up the countries, states and zones required for Spree.'
  exit
end

europe_vat = Spree::Zone.find_by!(name: 'EU_VAT')
shipping_category = Spree::ShippingCategory.find_or_create_by!(name: 'Default')

shipping_methods = [
  {
    name: 'UPS Ground (USD)',
    zones: [north_america],
    display_on: 'both',
    shipping_categories: [shipping_category]
  },
  {
    name: 'UPS Two Day (USD)',
    zones: [north_america],
    display_on: 'both',
    shipping_categories: [shipping_category]
  },
  {
    name: 'UPS One Day (USD)',
    zones: [north_america],
    display_on: 'both',
    shipping_categories: [shipping_category]
  },
  {
    name: 'UPS Ground (EU)',
    zones: [europe_vat],
    display_on: 'both',
    shipping_categories: [shipping_category]
  },
  {
    name: 'UPS Ground (EUR)',
    zones: [europe_vat],
    display_on: 'both',
    shipping_categories: [shipping_category]
  }
]

shipping_methods.each do |attributes|
  Spree::ShippingMethod.where(name: attributes[:name]).first_or_create! do |shipping_method|
    shipping_method.calculator = Spree::Calculator::Shipping::FlatRate.create!
    shipping_method.zones = attributes[:zones]
    shipping_method.display_on = attributes[:display_on]
    shipping_method.shipping_categories = attributes[:shipping_categories]
  end
end

{
  'UPS Ground (USD)' => [5, 'USD'],
  'UPS Ground (EU)' => [5, 'USD'],
  'UPS One Day (USD)' => [15, 'USD'],
  'UPS Two Day (USD)' => [10, 'USD'],
  'UPS Ground (EUR)' => [8, 'EUR']
}.each do |shipping_method_name, (price, currency)|
  shipping_method = Spree::ShippingMethod.find_by!(name: shipping_method_name)
  shipping_method.calculator.preferences = {
    amount: price,
    currency: currency
  }
  shipping_method.calculator.save!
  shipping_method.save!
end