Directory tree printer

tree

#!/usr/bin/env ruby
# encoding: utf-8

@bar   = '─'
@elbow = '└'
@pipe  = '│'
@space = ' '
@tee   = '├'
@slash = '/'

def print_tree(dir = '.', depth = 32, indent = '')
  return if depth < 0

  (entries = Dir.glob(File.join(dir, '*'))).each do |e|
    f = File.basename(e)
    d = File.directory?(e)

    branch = (e == entries.last) ? @elbow : @tee
    line = "#{indent}#{branch}#{@bar}#{f}"
    line << @slash if d

    puts line

    if d
      column = (e == entries.last) ? @space : @pipe
      print_tree(e, depth - 1, "#{indent}#{column}#{@space}")
    end
  end
end

ARGV << '.' << '32' if ARGV.empty?
print_tree(ARGV[0], ARGV[1].to_i)