Class: Participle::LogQueue

Inherits:

Object

• Object
• Participle::LogQueue

show all

Includes:

Colors

Defined in:

lib/logger.rb

Overview

Constant Summary

Constant Summary

Constants included from Colors

Colors::ANSI_COLORS_HEX, Colors::ANSI_COLORS_NAMED, Colors::ANSI_STYLES

Instance Attribute Summary (collapse)

• - (Boolean) pretty_display

  Whether we are using colors or not.

• - (Boolean) shutdown writeonly

  Set to true to terminate the logging loop once all logs have been flushed.

Instance Method Summary (collapse)

• - (void) add_logger(room)

  Add a logger to the loop.

• - (void) error(line)

  Push an error message onto the queue.

• - (void) flush

  Log waiting messages and shut down.

• - (LogQueue) initialize(bot) constructor

  Start up the logger loop.

• - (void) log(type, line, room, username = nil)

  Push a log message onto the queue.

• - (void) puts(line)

  Replacement for Kernel.puts to replace the prompt after each line.

• - (void) remove_logger(room)

  Remove a logger from the loop.

Methods included from Colors

#ansi_colorize, #ansi_stylize, #chromaticied, #closest_to, #color_difference, #colorize, #embolden, #format_html, #format_user, #get_term_size, #italicize, #loggerify, #room_name, #timestamp, #to_hex, #unchrome, #underscore

33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

# File 'lib/logger.rb', line 33

def initialize(bot)
  @bot = bot
  @loggers  = {}
  @queue    = Queue.new
  @shutdown = false
  @th       = Thread.new {
    loop {
      break if @shutdown && @queue.size.zero?
      msg = @queue.pop
      if String === msg
        puts(colorize("!!", :red) + msg)
      else
        loggerify(msg.type, msg.content, msg.room, msg.user).word_wrap(get_term_size).split(/\n/).each{|l|puts l}
        begin
          @loggers[msg.room] << loggerify(msg.type, msg.content, msg.room, msg.user).remove_ansi_colors
        rescue
          # no logger is set up, do something here i suppose
        end
      end
    }
  }
end

16
17
18

# File 'lib/logger.rb', line 16

def pretty_display
  @pretty_display
end

14
15
16

# File 'lib/logger.rb', line 14

def shutdown=(value)
  @shutdown = value
end

59
60
61
62
63

# File 'lib/logger.rb', line 59

def add_logger room
  Dir.mkdir("log") unless File.directory?("log")
  open("log/#{room}.txt", "a").close
  @loggers[room] = Logger.new("log/#{room}.txt")
end

83
84
85

# File 'lib/logger.rb', line 83

def error line
  @queue << line
end

89
90
91
92

# File 'lib/logger.rb', line 89

def flush
  @shutdown = true
  @th.join
end

76
77
78

# File 'lib/logger.rb', line 76

def log type, line, room, username = nil
  @queue << LogMessage.new(type, line, room, username)
end

22
23
24
25
26
27
28
29
30

# File 'lib/logger.rb', line 22

def puts line
  if $opts[:prompt] && RUBY_PLATFORM =~ /linux|darwin/
    print "\e[1K\e[200D"
    Kernel.puts line
    print "\e[1000D\e[K##{@bot.current_room}> #{@bot.__cli_buffer}"
  else
    Kernel.puts line
  end
end

68
69
70

# File 'lib/logger.rb', line 68

def remove_logger room
  @loggers.delete(room)
end