fog--fog/lib/fog/local/models/directories.rb

require 'fog/collection'
require 'fog/local/models/directory'

module Fog
  module Local

    class Real
      def directories
        Fog::Local::Directories.new(:connection => self)
      end
    end

    class Mock
      def directories
        Fog::Local::Directories.new(:connection => self)
      end
    end

    class Directories < Fog::Collection

      model Fog::Local::Directory

      def all
        data = Dir.entries(connection.local_root).select do |entry|
          entry[0...1] != '.' && ::File.directory?(connection.path_to(entry))
        end.map do |entry| 
          {:key => entry}
        end
        load(data)
      end

      def get(key)
        if ::File.directory?(connection.path_to(key))
          new(:key => key)
        else
          nil
        end
      end

    end

  end
end