amazon_integration.rb

require "sinatra"
require "endpoint_base"
require 'mws-connect'

require_all 'lib'

class AmazonIntegration < EndpointBase::Sinatra::Base
  set :logging, true

  Honeybadger.configure do |config|
    config.api_key = ENV['HONEYBADGER_KEY']
    config.environment_name = ENV['RACK_ENV']
  end

  # NOTE: Can only be used in development this will break production if left in uncommented.
  # configure :development do
  #   enable :logging, :dump_errors, :raise_errors
  #   log = File.new("tmp/sinatra.log", "a")
  #   STDOUT.reopen(log)
  #   STDERR.reopen(log)
  # end

  before do
    if @config
      @mws = Mws.connect(
        merchant: @config['merchant_id'],
        access:   @config['aws_access_key_id'],
        secret:   @config['secret_key']
      )
    end
  end

  post '/add_product' do
    begin
      code, response = submit_product_feed
    rescue => e
      log_exception(e)
      code, response = handle_error(e)
    end
    result code, response
  end

  post '/get_customers' do
    begin
      client = MWS.customer_information(
        aws_access_key_id:     @config['aws_access_key_id'],
        aws_secret_access_key: @config['secret_key'],
        marketplace_id:        @config['marketplace_id'],
        merchant_id:           @config['merchant_id']
      )
      amazon_response = client.list_customers(date_range_start: Time.parse(@config['amazon_customers_last_polling_datetime']).iso8601.to_s, date_range_type: 'LastUpdatedDate').parse

      customers = if amazon_response['CustomerList']
        collection = amazon_response['CustomerList']['Customer'].is_a?(Array) ? amazon_response['CustomerList']['Customer'] : [amazon_response['CustomerList']['Customer']]
        collection.map { |customer| Customer.new(customer) }
      else
        []
      end

      unless customers.empty?
        customers.each { |customer| add_object :customer, customer.to_message }
        # We want to set the time to be right after the last received so convert to unix stamp to increment and convert back to iso8601.
        add_parameter 'amazon_customers_last_polling_datetime', Time.at(Time.parse(customers.last.last_updated_on).to_i + 1).utc.iso8601
      end

      code     = 200
      response = if customers.size > 0
        "Successfully received #{customers.size} customer(s) from Amazon MWS."
      else
        nil
      end
    rescue => e
      code, response = handle_error(e)
    end

    result code, response
  end

  post '/get_orders' do
    begin
      uri   = URI.parse(ENV["REDIS_HOST"] || "redis://localhost:6379")
      redis = Redis.new(host: uri.host, port: uri.port, password: uri.password)

      statuses = %w(PartiallyShipped Unshipped Shipped Canceled Unfulfillable)

      client = MWS.orders(
        aws_access_key_id:     @config['aws_access_key_id'],
        aws_secret_access_key: @config['secret_key'],
        marketplace_id:        @config['marketplace_id'],
        merchant_id:           @config['merchant_id']
      )

      amazon_response = client.list_orders(last_updated_after: Time.parse(@config['amazon_orders_last_polling_datetime']).iso8601.to_s, order_status: statuses, max_results_per_page: 25).parse

      orders = if amazon_response['Orders']
        collection = amazon_response['Orders']['Order'].is_a?(Array) ? amazon_response['Orders']['Order'] : [amazon_response['Orders']['Order']]
        collection.map { |order| Order.new(order, client, redis) }
      else
        []
      end

      unless orders.empty?
        shipment_count = 0
        orders.each do |order|
          add_object :order, order.to_message
          if !order.fulfilled_by_amazon? && @config['create_shipments'] == '1'
            add_object :shipment, order.to_shipment
            shipment_count += 1
          end
        end
        # We want to set the time to be right after the last received so convert to unix stamp to increment and convert back to iso8601.
        add_parameter 'amazon_orders_last_polling_datetime', Time.at(Time.parse(orders.last.last_update_date).to_i + 1).utc.iso8601
      end

      code     = 200
      response = if orders.size > 0
        "Successfully received #{orders.size} order(s) and #{shipment_count} shipment(s) from Amazon MWS."
      else
        nil
      end
    rescue => e
      code, response = handle_error(e)
    end

    redis.quit
    result code, response
  end

  post '/set_inventory' do
    begin
      inventory_feed = @mws.feeds.inventory.update(
        Mws::Inventory(@payload['inventory']['product_id'],
          quantity: @payload['inventory']['quantity'],
          fulfillment_type: :mfn
        )
      )
      response = "Submitted SKU #{@payload['inventory']['product_id']} MWS Inventory Feed ID: #{inventory_feed.id}"
      code = 200
    rescue => e
      code, response = handle_error(e)
    end

    result code, response
  end

  post '/update_product' do
    begin
      code, response = submit_product_feed
    rescue => e
      log_exception(e)
      code, response = handle_error(e)
    end
    result code, response
  end

  post '/update_shipment' do
    begin
      feed = AmazonFeed.new(@config)
      order = Feeds::OrderFulfillment.new(@payload['shipment'], @config)

      id = feed.submit(order.feed_type, order.to_xml)
      response = "Submited Feed: feed id - #{id}"
      code = 200
    rescue => e
      code, response = handle_error(e)
    end

    result code, response
  end

  # post '/feed_status' do
  #   begin
  #     raise 'TODO?'
  #     code = 200
  #   rescue => e
  #     code, response = handle_error(e)
  #   end
  #
  #   result code, response
  # end

  private

  def submit_product_feed
    mws = Mws.connect(
      merchant: @config['merchant_id'],
      access: @config['aws_access_key_id'],
      secret: @config['secret_key']
    )

    # Assign vars outside Mws::Product block as @payload can't be accessed in it.
    brand_name = @payload['product']['properties']['brand']
    sku = @payload['product']['sku']
    title = @payload['product']['name']
    desc = @payload['product']['description']
    upc_code = @payload['product']['properties']['upc_code']

    product = Mws::Product(@payload['product']['sku']) {
      brand brand_name
      description desc
      tax_code 'GEN_TAX_CODE'
      upc upc_code
      name title
    }
    product_feed = mws.feeds.products.add(product)

    if @payload['product']['price']
      price_feed = mws.feeds.prices.update(
        Mws::PriceListing(sku, @payload['product']['price'])
      )
    end

    [200, "Submitted SKU #{sku} with MWS Feed IDs: Product #{product_feed.id}, Price #{price_feed ? price_feed.id : 'not submitted'}."]
  end

  def handle_error(e)
    response = if e.message =~ /403 Forbidden/
      "403 Forbidden.  Please ensure your connection credentials are correct.  If using /get_customers webhook ensure you've enabled the Customer Information API. For further help read: https://support.wombat.co/hc/en-us/articles/203066480"
    elsif e.message =~ /Access denied/
      "Access denied.  Please ensure your connection credentials are correct."
    else
      "Error processing request: #{e.message}.  Please contact support if this error persists."
    end
    [500, response]
  end

end