class AWS::SimpleEmailService::IdentityCollection

def [] email_or_domain
  Identity.new(email_or_domain, :config => config)
end

def _each_item next_token, limit, options = {}, &block
  options[:max_items] = limit if limit
  options[:next_token] = next_token if next_token
  options[:identity_type] = @type if @type
  resp = client.list_identities(options)
  resp.data[:identities].each do |identity|
    yield(self[identity])
  end
  resp.data[:next_token]
end

def domains
  self.class.new(:type => 'Domain', :config => config)
end

def email_addresses
  self.class.new(:type => 'EmailAddress', :config => config)
end

def initialize options = {}
  @type = options[:type]
  super
end

def verify email_or_domain
  resp = email_or_domain =~ /@/ ?
    client.verify_email_identity(:email_address => email_or_domain) :
    client.verify_domain_identity(:domain => email_or_domain)
  Identity.new(email_or_domain,
    :verification_token => resp.data[:verification_token],
    :config => config)
end