lib/fulmar/domain/task/versions.rake in fulmar-1.5.2 vs lib/fulmar/domain/task/versions.rake in fulmar-1.6.0

@@ -12,33 +12,42 @@
   end
 
   unless @versioned_servers.empty?
 
     @versioned_servers.each_key do |env|
-      target_count = @versioned_servers.keys.reduce(0) { |sum, target| target.split(':').first == env.split(':').first ? sum + 1 : sum }
-      namespace :list do
+      target_count = @versioned_servers.keys.reduce(0) { |a, e| e.split(':').first == env.split(':').first ? a + 1 : a }
 
+      task_environment = (target_count > 1 ? env : env.split(':').first)
+
+      namespace :list do
         # Count of there are multiple targets within the environment
         # if not, we can omit the target name in the task and shorten it a bit
-        # This should work in most cases.
+        # This should apply for almost all cases.
 
         desc "List available versions for environment/target \"#{env}\""
-        task (target_count > 1 ? env : env.split(':').first) do
+        task task_environment do
           configuration.environment = env.split(':').first
           configuration.target = env.split(':').last
           file_sync.list_releases(false).each { |item| puts item }
         end
       end
 
-
       namespace :clean do
         desc "Delete obsolete versions for target \"#{env}\""
-        task (target_count > 1 ? env : env.split(':').first) do
+        task task_environment do
           configuration.environment = env.split(':').first
           configuration.target = env.split(':').last
           file_sync.cleanup
         end
       end
-    end
 
+      namespace :revert do
+        desc "Revert to the previous version for \"#{env}\""
+        task task_environment do
+          configuration.environment = env.split(':').first
+          configuration.target = env.split(':').last
+          error 'Cannot revert to previous version.' unless file_sync.revert
+        end
+      end
+    end
   end
 end