gitlab-foss/lib/gitlab/git/wraps_gitaly_errors.rb

# frozen_string_literal: true

module Gitlab
  module Git
    module WrapsGitalyErrors
      def wrapped_gitaly_errors(&block)
        yield block
      rescue GRPC::BadStatus => e
        # The GRPC::BadStatus is the fundamental error that serves as the basis for all other gRPC error categories,
        # including GRPC::InvalidArgument. It is essential to note that rescuing the specific exception class does not
        # account for all possible cases. In this regard, a status exception can be directly generated from
        # GRPC::BadStatus. Therefore, it is recommended that we capture and rescue the GRPC::BadStatus and assert the
        # status code to ensure adequate coverage of error cases.
        handle_error(e)
      end

      private

      # handle_error first tries to handle the error as a detailed error, if no mapping found,
      # it falls back to handling it as a default error according to the error code.
      def handle_error(exception)
        error = Gitlab::GitalyClient.unwrap_detailed_error(exception)
        handle_detailed_error(error, exception) || handle_default_error(exception)
      end

      def handle_detailed_error(error, exception)
        case error
        when Gitaly::ReferenceNotFoundError
          raise Gitlab::Git::ReferenceNotFoundError.new(exception, error.reference_name)
        when Gitaly::FindCommitsError
          raise Gitlab::Git::Repository::CommitNotFound, exception
        when Gitaly::AmbiguousReferenceError
          raise Gitlab::Git::AmbiguousRef, exception
        when Gitaly::LimitError
          raise ResourceExhaustedError.new(
            "Upstream Gitaly has been exhausted: #{error.error_message}. Try again later", error.retry_after&.seconds
          )
        end
      end

      def handle_default_error(exception)
        case exception.code
        when GRPC::Core::StatusCodes::NOT_FOUND
          raise Gitlab::Git::Repository::NoRepository, exception
        when GRPC::Core::StatusCodes::INVALID_ARGUMENT
          raise ArgumentError, exception
        when GRPC::Core::StatusCodes::DEADLINE_EXCEEDED
          raise Gitlab::Git::CommandTimedOut, exception
        when GRPC::Core::StatusCodes::RESOURCE_EXHAUSTED
          raise ResourceExhaustedError, _("Upstream Gitaly has been exhausted. Try again later")
        else
          raise Gitlab::Git::CommandError, exception
        end
      end
    end
  end
end