mruby-mruby/mrbgems/mruby-dir/mrblib/dir.rb

class Dir
  include Enumerable

  #
  # call-seq:
  #   dir.each { |filename| block } -> dir
  #   dir.each                      -> enumerator
  #
  # Calls the block once for each entry in this directory, passing the filename
  # of each entry as a parameter to the block.
  #
  #   d = Dir.new("testdir")
  #   d.each { |x| puts "Got #{x}" }
  #
  def each(&block)
    return to_enum(:each) unless block
    while s = self.read
      block.call(s)
    end
    self
  end

  #
  # call-seq:
  #   dir.each_child { |filename| block } -> dir
  #   dir.each_child                      -> enumerator
  #
  # Calls the block once for each entry in this directory except for "." and "..",
  # passing the filename of each entry as a parameter to the block.
  #
  #   d = Dir.new("testdir")
  #   d.each_child { |x| puts "Got #{x}" }
  #
  def each_child(&block)
    return to_enum(:each_child) unless block
    while s = self.read
      block.call(s) unless s == "." || s == ".."
    end
    self
  end

  alias pos tell
  alias pos= seek

  class << self
    #
    # call-seq:
    #   Dir.entries(dirname) -> array
    #
    # Returns an array containing all of the filenames in the given directory.
    # Will raise a SystemCallError if the named directory doesn't exist.
    #
    #   Dir.entries("testdir")   #=> [".", "..", "config.h", "main.rb"]
    #
    def entries(path)
      a = []
      self.open(path) do |d|
        while s = d.read
          a << s
        end
      end
      a
    end

    #
    # call-seq:
    #   Dir.children(dirname) -> array
    #
    # Returns an array containing all of the filenames except for "." and ".."
    # in the given directory. Will raise a SystemCallError if the named
    # directory doesn't exist.
    #
    #   Dir.children("testdir")   #=> ["config.h", "main.rb"]
    #
    def children(path)
      a = []
      self.open(path) do |d|
        while s = d.read
          a << s unless s == "." || s == ".."
        end
      end
      a
    end

    #
    # call-seq:
    #   Dir.foreach(dirname) { |filename| block } -> nil
    #   Dir.foreach(dirname)                      -> enumerator
    #
    # Calls the block once for each entry in the named directory, passing
    # the filename of each entry as a parameter to the block.
    #
    #   Dir.foreach("testdir") { |x| puts "Got #{x}" }
    #
    def foreach(path, &block)
      return to_enum(:foreach, path) unless block
      self.open(path) do |d|
        d.each(&block)
      end
    end

    #
    # call-seq:
    #   Dir.open(string) -> aDir
    #   Dir.open(string) { |aDir| block } -> obj
    #
    # With no block, open is a synonym for Dir.new. If a block is present, it is
    # passed aDir as a parameter. The directory is closed at the end of the block,
    # and Dir.open returns the value of the block.
    #
    #   Dir.open("testdir") { |d| d.each { |x| puts "Got #{x}" } }
    #
    def open(path, &block)
      if block
        d = self.new(path)
        begin
          block.call(d)
        ensure
          begin
            d.close
          rescue IOError
          end
        end
      else
        self.new(path)
      end
    end

    #
    # call-seq:
    #   Dir.chdir(string) -> 0
    #   Dir.chdir(string) { |path| block } -> obj
    #
    # Changes the current working directory of the calling process to the given
    # string. When called with a block, changes to the directory, executes the
    # block, then changes back to the original directory. Returns the value of
    # the block.
    #
    #   Dir.chdir("/var/spool/mail")
    #   Dir.chdir("/tmp") { Dir.pwd }   #=> "/tmp"
    #
    def chdir(path, &block)
      if block
        wd = self.getwd
        begin
          self._chdir(path)
          block.call(path)
        ensure
          self._chdir(wd)
        end
      else
        self._chdir(path)
      end
    end

    alias exists? exist?
    alias pwd getwd
    alias rmdir delete
    alias unlink delete
  end
end