Contents

module Ahnnotate
  module Refinement
    module Dig
      if Gem::Version.new(RUBY_VERSION) < Gem::Version.new("2.3.0")
        refine Hash do
          def dig(head, *tail)
            value = self[head]

            if tail.empty? || value.nil?
              return value
            end

            if value.respond_to?(:dig) || value.is_a?(Hash) || value.is_a?(Array)
              value.dig(*tail)
            else
              raise TypeError, "#{value.class} does not have #dig method"
            end
          end
        end

        refine Array do
          def dig(head, *tail)
            value = self.at(head)

            if tail.empty? || value.nil?
              return value
            end

            if value.respond_to?(:dig) || value.is_a?(Hash) || value.is_a?(Array)
              value.dig(*tail)
            else
              raise TypeError, "#{value.class} does not have #dig method"
            end
          end
        end
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
ahnnotate-0.5.1	lib/ahnnotate/refinement/dig.rb
ahnnotate-0.5.0	lib/ahnnotate/refinement/dig.rb
ahnnotate-0.4.0	lib/ahnnotate/refinement/dig.rb
ahnnotate-0.3.0	lib/ahnnotate/refinement/dig.rb