"Dig" through hash using dot notation

hash_dig.rb

class Hash
  def dig(dotted_path)
    parts = dotted_path.split '.', 2
    first_part = parts[0]
    match = self[first_part] || self[first_part.to_sym]
    if !parts[1] || match.nil?
      return match
    else
      return match.dig(parts[1])
    end
  end
end