lib/prj/finder.rb in prj-0.0.1 vs lib/prj/finder.rb in prj-0.0.2

@@ -1,20 +1,25 @@
 require 'find'
 
 module Prj
+
   class Finder
-    def initialize(root)
+    def initialize(root, vcs_directories)
       @root = File.expand_path(root)
+      @vcs_directories = Array(vcs_directories)
     end
 
     ##
     # Returns directories containing .git/ directory, relative to @root
     #
     def find_project_directories
       subdirectories = []
       Find.find(@root) do |d|
-        subdirectories << d && Find.prune if File.exists?(File.join(d, ".git/"))
+        if @vcs_directories.any? { |vcs_dir| Dir.exists?(File.join(d, vcs_dir)) }
+          subdirectories << d && Find.prune
+        end
       end
       subdirectories.map { |r| r.gsub(@root, "") }
     end
   end
+
 end