Sha256: a813ffa959e3cfd3794ad4ef6db8467e91869cf4ab6628f050c33c6f42f0faf3

Contents?: true

Size: 883 Bytes

Versions: 7

Compression:

Stored size: 883 Bytes

Contents

# -*- coding: binary -*-
require 'rex/ui'

module Rex
module Ui
module Text

###
#
# This class implements output against a file and stdout
#
###
class Output::Tee < Rex::Ui::Text::Output

  attr_accessor :fd

  def initialize(path)
    self.fd = ::File.open(path, "ab")
    super()
  end

  def supports_color?
    case config[:color]
    when true
      return true
    when false
      return false
    else # auto
      term = Rex::Compat.getenv('TERM')
      return (term and term.match(/(?:vt10[03]|xterm(?:-color)?|linux|screen|rxvt)/i) != nil)
    end
  end

  #
  # Prints the supplied message to file output.
  #
  def print_raw(msg = '')
    $stdout.print(msg)
    $stdout.flush

    return if not self.fd
    self.fd.write(msg)
    self.fd.flush
    msg
  end

  alias :write :print_raw

  def close
    self.fd.close if self.fd
    self.fd = nil
  end
end

end
end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version Path
rex-2.0.13 lib/rex/ui/text/output/tee.rb
rex-2.0.12 lib/rex/ui/text/output/tee.rb
rex-2.0.11 lib/rex/ui/text/output/tee.rb
rex-2.0.10 lib/rex/ui/text/output/tee.rb
rex-2.0.9 lib/rex/ui/text/output/tee.rb
rex-2.0.8 lib/rex/ui/text/output/tee.rb
rex-2.0.7 lib/rex/ui/text/output/tee.rb