implement logger

Signed-off-by: David A. Cuadrado <krawek@gmail.com>

models/channel.rb

@@ -3,6 +3,7 @@ class Channel
 
   key :_id, String
   has_many :users
+  has_many :logs
 
   def self.get_user(name, nick, create = true)
     name = name.downcase
@@ -16,4 +17,17 @@ def self.get_user(name, nick, create = true)
 
     user
   end
+
+  def self.add_log_message(name, message)
+    channel = Channel.find(name)
+
+    log = Log.first(:channel_id => channel.id, :order => "created_at desc", :select => %w[messages_count channel_id])
+    if log.nil? || log.messages_count >= 100
+      log = Log.create(:channel_id => channel.id)
+    end
+
+    Log.push({:_id => log.id}, {:messages => message})
+    Log.increment({:_id => log.id},{:messages_count => 1})
+  end
 end
+

models/log.rb

@@ -0,0 +1,29 @@
+class Log
+  include MongoMapper::Document
+
+  timestamps!
+
+  key :_id, String
+  key :_type, String
+
+  key :channel_id, String
+  belongs_to :channel
+
+  key :messages, Array
+  key :messages_count, Integer, :default => 0
+
+
+  def self.format_message(msg)
+    if msg["type"] == "normal"
+      "[#{msg["date"]}] <#{msg["nick"]}> #{msg["text"]}"
+    elsif msg["type"] == "action"
+      "[#{msg["date"]}] * #{msg["nick"]} #{msg["text"]}"
+    elsif msg["type"] == "join"
+      "[#{msg["date"]}] JOIN #{msg["nick"]}"
+    elsif msg["type"] == "part"
+      "[#{msg["date"]}] PART #{msg["nick"]}"
+    elsif msg["type"] == "ctcp"
+      "[#{msg["date"]}] CTCP #{msg["nick"]} #{msg["text"]}"
+    end
+  end
+end

plugins/logger.rb

@@ -0,0 +1,69 @@
+class LoggerPlugin
+  include NinjaPlugin
+
+  match /grep (.+)/, :method => :grep
+
+  listen_to :channel, :method => :log_channel
+  listen_to :join, :method => :log_channel
+  listen_to :part, :method => :log_channel
+
+  def log_channel(m)
+    return if !m.channel?
+
+    message = m.message
+    nick = m.user.nick
+    type = "normal"
+    if m.events.include?(:notice)
+      type = "notice"
+    elsif m.events.include?(:ctcp)
+      type = "ctcp"
+      message = m.ctcp_message
+
+      if message =~ /^ACTION /
+        type = "action"
+        message.gsub!(/^ACTION /, "")
+      end
+    elsif m.events.include?(:join)
+      type = "join"
+      message = ""
+    elsif m.events.include?(:part)
+      type = "part"
+    end
+
+    data = {
+      :nick => nick,
+      :date => Time.now,
+      :type => type
+    }
+    if !message.empty?
+      data[:text] = message
+    end
+
+    Channel.add_log_message(m.channel.name, data)
+  end
+
+  def grep(m, query)
+    if m.channel?
+      regexp = Regexp.new(query) rescue Regexp.new(Regexp.escape(query))
+      matches = []
+      count = 0
+      Log.find_each(:"messages.text" => regexp, :channel_id => m.channel.name, :limit => 5, :order => "created_at desc") do |log|
+        if count >= 10
+          m.user.send ">>>> TOO MANY RESULTS<<<<<"
+          break
+        end
+
+        log.messages.each do |message|
+          if message["text"] =~ regexp && message["text"] !~ /^\!grep\s/
+            m.user.send Log.format_message(message)
+
+            count += 1
+          end
+        end
+      end
+
+    end
+  end
+end
+
+register_plugin LoggerPlugin