lib/caravan/deploy.rb in caravan-0.4.0 vs lib/caravan/deploy.rb in caravan-0.5.0.alpha1

@@ -1,45 +1,135 @@
 require "colorize"
 
 module Caravan
   module Deploy
     class << self
-      def create_deployer(method = "shell")
+      def create_deployer(src, dst, method = "shell")
         case method
         when "shell"
-          return Caravan::DeployMethods::Shell.new
+          Caravan::DeployMethods::Shell.new(src, dst)
         when "scp"
-          return Caravan::DeployMethods::Scp.new
+          Caravan::DeployMethods::Scp.new(src, dst)
         when "rsync"
-          return Caravan::DeployMethods::Rsync.new
+          Caravan::DeployMethods::Rsync.new(src, dst)
         when "rsync_local"
-          return Caravan::DeployMethods::RsyncLocal.new
+          Caravan::DeployMethods::RsyncLocal.new(src, dst)
         else
           Message.error("Unknown deploy method \"#{method}\"")
-          return nil
+          nil
         end
       end
     end
 
     class Base
       attr_accessor :debug
+      attr_reader :src, :dst
 
-      def initialize
+      def initialize(src, dst)
         Message.info("Creating #{self.class.name}...")
+        @src = src
+        @dst = dst
         @debug = false
       end
 
-      def run(src, dst)
+      def handle_change(modified, added, removed)
+        @modified = modified
+        @added = added
+        @removed = removed
+
+        after_change
+      end
+
+      def after_create
+        Message.info("#{self.class.name} is created.")
+
+        if block_given?
+          ret_val = yield
+          debug_msg("Block `after_create` returned #{ret_val}")
+          false if ret_val == false
+        end
+
+        true
+      end
+
+      def after_change
+        @modified.each do |change|
+          Caravan::Message.info("#{change} was changed.")
+        end
+
+        @added.each do |change|
+          Caravan::Message.info("#{change} was created.")
+        end
+
+        @removed.each do |change|
+          Caravan::Message.info("#{change} was removed.")
+        end
+
+        if block_given?
+          ret_val = yield @modified, @added, @removed
+          debug_msg("Block `after_change` returned #{ret_val}")
+          false if ret_val == false
+        end
+
+        true
+      end
+
+      def before_deploy
+        if block_given?
+          ret_val = yield
+          debug_msg("Block `before_deploy` returned #{ret_val}")
+          false if ret_val == false
+        end
+
+        true
+      end
+
+      def run
         time_str = Time.now.strftime("%H:%M:%S").green
         Message.info("#{time_str} Deploying #{src} to #{dst}...")
 
         status = 0
         if block_given?
           status, output = yield src, dst
+          debug_msg("Block `run` returned #{status}")
         end
 
-        Message.error("deploying block returned false") unless status == 0
-        return status
+        Message.error("Deploying block returned false") unless status == 0
+        status
+      end
+
+      def after_deploy
+        if block_given?
+          ret_val = yield
+          debug_msg("Block `after_deploy` returned #{ret_val}")
+          false if ret_val == false
+        end
+
+        true
+      end
+
+      def before_destroy
+        if block_given?
+          ret_val = yield
+          debug_msg("Block `before_destroy` returned #{ret_val}")
+          false if ret_val == false
+        end
+
+        true
+      end
+
+      def relative_path(path)
+        working_dir = Dir.pwd
+        path_routes = path.split(working_dir)
+        return nil if path_routes.nil? || path_routes.empty? || path_routes.size < 2
+
+        path_routes[-1]
+      end
+
+      private
+
+      def debug_msg(msg)
+        Caravan::Message.debug(msg) if @debug
       end
     end
   end
 end