inspec/lib/resources/aws/aws_sns_topics.rb

require "resource_support/aws/aws_plural_resource_mixin"
require "resource_support/aws/aws_backend_base"
require "aws-sdk-sns"

class AwsSnsTopics < Inspec.resource(1)
  name "aws_sns_topics"
  desc "Verifies settings for SNS Topics in bulk"
  example <<~EXAMPLE
    describe aws_sns_topics do
      its('topic_arns') { should include '' }
    end
  EXAMPLE
  supports platform: "aws"

  include AwsPluralResourceMixin

  def validate_params(resource_params)
    unless resource_params.empty?
      raise ArgumentError, "aws_sns_topics does not accept resource parameters."
    end

    resource_params
  end

  def fetch_from_api
    backend = BackendFactory.create(inspec_runner)
    @table = []
    pagination_opts = nil
    catch_aws_errors do
      loop do
        api_result = backend.list_topics(pagination_opts)
        @table += api_result.topics.map(&:to_h)
        break if api_result.next_token.nil?

        pagination_opts = { next_token: api_result.next_token }
      end
    end
  end

  # Underlying FilterTable implementation.
  filter = FilterTable.create
  filter.register_custom_matcher(:exists?) { |x| !x.entries.empty? }
  filter.register_column(:topic_arns, field: :topic_arn)
  filter.install_filter_methods_on_resource(self, :table)

  def to_s
    "EC2 SNS Topics"
  end

  class Backend
    class AwsClientApi < AwsBackendBase
      BackendFactory.set_default_backend self
      self.aws_client_class = Aws::SNS::Client

      def list_topics(pagination_opts)
        aws_service_client.list_topics(pagination_opts)
      end
    end
  end
end