cloudfoundry/cloud_controller_ng

View on GitHub
app/controllers/runtime/helpers/blob_dispatcher.rb

Summary

Maintainability
A
55 mins
Test Coverage
module VCAP::CloudController
  class BlobDispatcher
    def initialize(blobstore:, controller:)
      @blobstore = blobstore
      @controller = controller
    end

    def send_or_redirect(guid:)
      unless guid
        logger.error('No guid for BlobNotFound')
        raise CloudController::Errors::BlobNotFound
      end

      blob = @blobstore.blob(guid)

      unless blob
        logger.error('No blob object for BlobNotFound')
        raise CloudController::Errors::BlobNotFound
      end

      send_or_redirect_blob(blob)
    end

    def send_or_redirect_blob(blob)
      raise CloudController::Errors::BlobNotFound unless blob

      if @blobstore.local?
        blob_sender.send_blob(blob, @controller)
      else
        begin
          if @controller.is_a?(ActionController::Base)
            @controller.redirect_to blob.public_download_url
          else
            @controller.redirect blob.public_download_url
          end
        rescue CloudController::Blobstore::SigningRequestError => e
          logger.error("failed to get download url: #{e.message}")
          raise CloudController::Errors::ApiError.new_from_details('BlobstoreUnavailable')
        end
      end
    end

    private

    def logger
      @logger ||= Steno.logger('cc.blob_dispatcher')
    end

    def blob_sender
      @blob_sender ||= CloudController::DependencyLocator.instance.blob_sender
    end
  end
end