adding local logging of chat

Author: lukeis

.gitignore

@@ -1,4 +1,4 @@
 notes.yml
 selbotseen.yml
 tmp/
-.bundle
+

bin/selbot2.rb

@@ -18,7 +18,12 @@
 #      Selbot2::SeleniumHQ,
       Selbot2::CI,
       Selbot2::Google,
+<<<<<<< HEAD
 #      Selbot2::WhoBrokeIt
+=======
+      Selbot2::Log,
+      Selbot2::WhoBrokeIt
+>>>>>>> adding local logging of chat
     ]
 
     # if File.exist?("twitter.conf")

lib/selbot2.rb

@@ -5,7 +5,11 @@
 
 module Selbot2
   PREFIX = ":"
+<<<<<<< HEAD
   CHANNELS = Array(ENV['SELBOT_CHANNEL'] || ["#selendroid"])
+=======
+  CHANNELS = Array(ENV['SELBOT_CHANNEL'])
+>>>>>>> adding local logging of chat
   HELPS = []
 end
 
@@ -25,3 +29,4 @@ module Selbot2
 # require 'selbot2/twitter'
 require 'selbot2/google'
 require 'selbot2/whobrokeit'
+require 'selbot2/log'

lib/selbot2/log.rb

@@ -0,0 +1,51 @@
+require 'fileutils'
+
+module Selbot2
+  class Log
+    include Cinch::Plugin
+
+    listen_to :channel, :join, :part, :quit, :nick
+
+    def listen(m)
+      return unless m.user
+      nick = m.command == 'NICK' ? m.user.last_nick : m.user.nick
+
+      e = Event.new(nick, m.message, m.command, Time.now)
+      FileUtils.mkpath "log/#{(Time.new).strftime('%Y/%m')}"
+      File.open("log/#{(Time.new).strftime('%Y/%m/%d')}.txt", 'a') {|f| f.puts(e.to_s) }
+    end
+
+    private
+
+    class Event
+      def initialize(nick, message, type, time)
+        @nick    = nick
+        @message = message
+        @type    = type
+        @time    = time
+      end
+
+      def to_s
+        msg = "#{(Time.new).strftime('[%Y-%m-%d %H:%M:%S] ')}"
+
+        case @type
+        when 'JOIN'
+          msg << "#{@nick}, joining. (#{@message})"
+        when 'PART'
+          msg << "#{@nick}, leaving. (#{@message})"
+        when 'QUIT'
+          msg << "#{@nick}, quitting. (#{@message})"
+        when 'NICK'
+          msg << "#{@nick}, changing nick to #{@message}."
+        else
+          if @message =~ /^\001ACTION (.+?)\001/
+            msg << "*#{@nick} #{$1}'"
+          else
+            msg << "#{@nick}: #{@message}"
+          end
+        end
+      end
+    end
+
+  end
+end