fog--fog-aws/lib/fog/aws/models/iam/roles.rb

require 'fog/aws/models/iam/role'

module Fog
  module AWS
    class IAM
      class Roles < Fog::AWS::IAM::PagedCollection

        model Fog::AWS::IAM::Role

        def all
          load(service.list_roles.body['Roles'])
        end

        def get(identity)
          new(service.get_role(identity).body["Role"])
        rescue Excon::Errors::NotFound, Fog::AWS::IAM::NotFound # ignore not found error
          nil
        end

        def new(attributes = {})
          unless attributes.key?(:assume_role_policy_document)
            attributes[:assume_role_policy_document] = Fog::AWS::IAM::EC2_ASSUME_ROLE_POLICY.to_s
          end

          super
        end
      end
    end
  end
end