brew/Library/Homebrew/utils/backtrace.rb

# typed: true
# frozen_string_literal: true

module Utils
  module Backtrace
    # Cleans `sorbet-runtime` gem paths from the backtrace unless...
    # 1. `verbose` is set
    # 2. first backtrace line starts with `sorbet-runtime`
    #   - This implies that the error is related to Sorbet.
    sig { params(error: Exception).returns(T.nilable(T::Array[String])) }
    def self.clean(error)
      backtrace = error.backtrace

      return backtrace if Context.current.verbose?
      return backtrace if backtrace.blank?
      return backtrace if backtrace.fetch(0).start_with?(sorbet_runtime_path)

      old_backtrace_length = backtrace.length
      backtrace.reject { |line| line.start_with?(sorbet_runtime_path) }
               .tap { |new_backtrace| print_backtrace_message if old_backtrace_length > new_backtrace.length }
    end

    def self.sorbet_runtime_path
      @sorbet_runtime_path ||= "#{Gem.paths.home}/gems/sorbet-runtime"
    end

    def self.print_backtrace_message
      return if @print_backtrace_message

      opoo "Removed Sorbet lines from backtrace!"
      puts "Rerun with `--verbose` to see the original backtrace" unless Homebrew::EnvConfig.no_env_hints?

      @print_backtrace_message = true
    end
  end
end