0
@@ -2,6 +2,8 @@ require "piston_core/revision"
0
 
0
 module PistonCore
0
   class Repository
0
+    class UnhandledUrl < RuntimeError; end
0
+
0
     class << self
0
       def logger
0
         @@logger
0
@@ -14,7 +16,18 @@ module PistonCore
0
 
0
       def guess(url)
0
         logger.debug {"Guessing the repository type of #{url.inspect}"}
0
-        self.new(url)
0
+
0
+        handler = self.handlers.detect do |handler|
0
+          handler.understands_url?(url)
0
+        end
0
+
0
+        raise UnhandledUrl, "No internal handlers found for #{url.inspect}.  Do you want to help ?" if handler.nil?
0
+        handler.new(url)
0
+      end
0
+
0
+      @@handlers = Array.new
0
+      def handlers
0
+        @@handlers
0
       end
0
     end
0
 

0
@@ -1,5 +1,7 @@
0
 module PistonCore
0
   class WorkingCopy
0
+    class UnhandledWorkingCopy < RuntimeError; end
0
+
0
     class << self
0
       def logger=(logger)
0
         @@logger = logger
0
@@ -11,7 +13,17 @@ module PistonCore
0
 
0
       def guess(path)
0
         logger.debug {"Guessing the working copy type of #{path.inspect}"}
0
-        self.new(path)
0
+        handler = self.handlers.detect do |handler|
0
+          handler.understands_dir?(path)
0
+        end
0
+
0
+        raise UnhandledWorkingCopy, "Don't know what working copy type #{path} is." if handler.nil?
0
+        handler.new(path)
0
+      end
0
+
0
+      @@handlers = Array.new
0
+      def handlers
0
+        @@handlers
0
       end
0
     end
0