lib/docman/deployers/deployer.rb in docman-0.0.4 vs lib/docman/deployers/deployer.rb in docman-0.0.5

@@ -2,19 +2,34 @@
   module Deployers
     class Deployer
 
       attr_reader :deploy_target
 
-      def build(root, info)
-        return if @deployed.include? info['name']
-        build_type = build_type(info['type'])
-        Docman::Builders::Builder.create(build_type['handler'], root, build_type, info).do()
-        @deployed << info['name']
+      @@subclasses = {}
+
+      def self.create(type, deploy_target)
+        c = @@subclasses[type]
+        if c
+          c.new(deploy_target)
+        else
+          raise "Bad deployer type: #{type}"
+        end
       end
 
+      def self.register_deployer(name)
+        @@subclasses[name] = self
+      end
+
       def initialize(deploy_target)
         @deployed = []
         @deploy_target = deploy_target
+      end
+
+      def build(root, info)
+        return if @deployed.include? info['name']
+        build_type = build_type(info['type'])
+        Docman::Builders::Builder.create(build_type['handler'], root, build_type, info).do()
+        @deployed << info['name']
       end
 
       def build_type(type)
         @deploy_target['builders'][type]
       end
\ No newline at end of file