require 'simnos/utils'

module Simnos
  class DSL
    class Subscription
      include Simnos::TemplateHelper

      def create
        Simnos.logger.info("Create Topic(#{@aws_topic[:topic].topic_arn.split(':').last}) Subscription. protocol: #{protocol.inspect}, endpoint: #{masked_endpoint.inspect}#{@options[:dry_run] ? ' [dry-run]' : ''}")
        return if @options[:dry_run]

        client.subscribe(
          topic_arn: @aws_topic[:topic].topic_arn,
          protocol: protocol,
          endpoint: endpoint,
        )
      end

      def aws_topic(aws_topic)
        @aws_topic = aws_topic
        self
      end

      def initialize(context, topic: , protocol: , endpoint: )
        @context = context
        @options = @context.options
        @topic = topic
        @protocol = protocol
        @endpoint = endpoint
      end

      attr_reader :topic, :protocol

      # We have to mask endpoint because SNS returns masked endpoint from API
      def masked_endpoint
        if URI.extract(@endpoint, ['http', 'https']).empty?
          return endpoint
        end
        uri = URI.parse(endpoint)
        if md = uri.userinfo&.match(/(.*):(.*)/)
          uri.userinfo = "#{md[1]}:****"
        end
        uri.to_s
      end

      def endpoint
        secret_expander = @options[:secret_expander]
        if secret_expander
          secret_expander.expand(@endpoint)
        else
          @endpoint
        end
      end

      private

      def client
        @client ||= Simnos::ClientWrapper.new(@context)
      end
    end
  end
end