setup_intents.rb

module StripeMock
  module RequestHandlers
    module SetupIntents
      ALLOWED_PARAMS = [
        :confirm,
        :customer,
        :description,
        :metadata,
        :on_behalf_of,
        :payment_method,
        :payment_method_options,
        :payment_method_types,
        :return_url,
        :usage
      ]

      def SetupIntents.included(klass)
        klass.add_handler 'post /v1/setup_intents', :new_setup_intent
        klass.add_handler 'get /v1/setup_intents', :get_setup_intents
        klass.add_handler 'get /v1/setup_intents/(.*)', :get_setup_intent
        klass.add_handler 'post /v1/setup_intents/(.*)/confirm', :confirm_setup_intent
        klass.add_handler 'post /v1/setup_intents/(.*)/cancel', :cancel_setup_intent
        klass.add_handler 'post /v1/setup_intents/(.*)', :update_setup_intent
      end

      def new_setup_intent(route, method_url, params, headers)
        id = new_id('si')

        setup_intents[id] = Data.mock_setup_intent(
          params.merge(
            id: id
          )
        )

        setup_intents[id].clone
      end

      def update_setup_intent(route, method_url, params, headers)
        route =~ method_url
        id = $1

        setup_intent = assert_existence :setup_intent, id, setup_intents[id]
        setup_intents[id] = Util.rmerge(setup_intent, params.select { |k, v| ALLOWED_PARAMS.include?(k) })
      end

      def get_setup_intents(route, method_url, params, headers)
        params[:offset] ||= 0
        params[:limit] ||= 10

        clone = setup_intents.clone

        if params[:customer]
          clone.delete_if { |k, v| v[:customer] != params[:customer] }
        end

        Data.mock_list_object(clone.values, params)
      end

      def get_setup_intent(route, method_url, params, headers)
        route =~ method_url
        setup_intent_id = $1 || params[:setup_intent]
        setup_intent = assert_existence :setup_intent, setup_intent_id, setup_intents[setup_intent_id]

        setup_intent = setup_intent.clone

        if params[:expand]&.include?("payment_method")
          setup_intent[:payment_method] = assert_existence :payment_method, setup_intent[:payment_method], payment_methods[setup_intent[:payment_method]]
        end

        setup_intent
      end

      def capture_setup_intent(route, method_url, params, headers)
        route =~ method_url
        setup_intent = assert_existence :setup_intent, $1, setup_intents[$1]

        setup_intent[:status] = 'succeeded'
        setup_intent
      end

      def confirm_setup_intent(route, method_url, params, headers)
        route =~ method_url
        setup_intent = assert_existence :setup_intent, $1, setup_intents[$1]

        setup_intent[:status] = 'succeeded'
        setup_intent
      end

      def cancel_setup_intent(route, method_url, params, headers)
        route =~ method_url
        setup_intent = assert_existence :setup_intent, $1, setup_intents[$1]

        setup_intent[:status] = 'canceled'
        setup_intent
      end
    end
  end
end