require 'aws-sdk'
require 'multi_json'

class Arsenicum::Queue::Sqs < Arsenicum::Queue
  attr_reader :sqs_queue, :via_sns

  def initialize(name, options = {})
    super name, options
    sqs_args = [options[:aws_account]].tap(&:compact!)
    sqs = AWS::SQS.new *sqs_args
    @sqs_queue = sqs.queues.named(name)
    @via_sns = options[:via_sns]
  end

  def pick
    message = sqs_message = sqs_queue.receive_message
    return unless message
    message = message.as_sns_message if via_sns

    [MultiJson.decode(message.body), sqs_message]
  end

  def handle_success(original_message)
    original_message.delete
  end
end