Close SSH session before exiting

lib/producer/core/cli.rb

@@ -54,6 +54,7 @@ module Producer
 
       def run
         worker.process load_recipe.tasks
+        env.cleanup
       end
 
       def load_recipe

lib/producer/core/env.rb

@@ -44,6 +44,10 @@ module Producer
       def dry_run?
         @dry_run
       end
+
+      def cleanup
+        remote.cleanup
+      end
     end
   end
 end

lib/producer/core/remote.rb

@@ -2,6 +2,7 @@ module Producer
   module Core
     class Remote
       attr_reader :hostname
+      attr_writer :session
 
       def initialize(hostname)
         @hostname = hostname
@@ -43,6 +44,10 @@ module Producer
       def environment
         Environment.new_from_string(execute 'env')
       end
+
+      def cleanup
+        session.close if @session
+      end
     end
   end
 end