lib/pec2/pssh.rb in pec2-0.7.0 vs lib/pec2/pssh.rb in pec2-0.7.1

@@ -20,10 +20,11 @@
         else
           color_index = color_index + 1
         end
         result
       }
+      @color = options[:color]
       @user = options[:user]
       @print = options[:print]
       @sudo_password = options[:sudo_password]
       @ssh_options = {
         verify_host_key: :never,
@@ -57,10 +58,11 @@
               if data =~ /^\[sudo\] password for /
                 channel.send_data "#{@sudo_password}\n"
               else
                 data.to_s.lines.each do |line|
                   if @print
-                    print %Q{#{server[:host]}:#{line}}.colorize(server[:color])
+                    output = %Q{#{server[:host]}:#{line}}
+                    puts (@color ? output.colorize(server[:color]) : output)
                   end
                 end
               end
             end
             channel.request_pty