spox-mod_spox 0.3.1

data/lib/mod_spox/handlers/Kick.rb

@@ -0,0 +1,36 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/Kick'
+module ModSpox
+    module Handlers
+        class Kick < Handler
+            def initialize(handlers)
+                handlers[:KICK] = self
+            end
+            # :spax!~spox@host KICK #m spox :foo
+            def process(string)
+                orig = string.dup
+                string = string.dup
+                begin
+                    string.slice!(0)
+                    source = string.slice!(0..string.index(' ')-1)
+                    2.times{string.slice!(0..string.index(' '))}
+                    channel = string.slice!(0..string.index(' ')-1)
+                    string.slice!(0)
+                    kickee = string.slice!(0..string.index(' ')-1)
+                    string.slice!(0..string.index(':'))
+                    kicker = find_model(source[0..source.index('!')-1])
+                    channel = find_model(channel)
+                    kickee = find_model(kickee)
+                    channel.remove_nick(kickee)
+                    kickee.visible = false if kickee.channels.empty?
+                    kickee.save_changes
+                    channel.save_changes
+                    return Messages::Incoming::Kick.new(orig, channel, kicker, kickee, string)
+                rescue Object => boom
+                    Logger.warn("Failed to parse KICK message: #{orig}")
+                    raise Exceptions::GeneralException.new(boom)
+                end
+            end
+        end
+    end
+end

data/lib/mod_spox/handlers/LuserChannels.rb

@@ -0,0 +1,19 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/LuserChannels'
+module ModSpox
+    module Handlers
+        class LuserChannels < Handler
+            def initialize(handlers)
+                handlers[RFC[:RPL_LUSERCHANNELS][:value]] = self
+            end
+            def process(string)
+                if(string =~ /(\d+)\s:channels/)
+                    return Messages::Incoming::LuserChannels.new(string, $1.to_i)
+                else
+                    Logger.warn('Failed to match RPL_LUSERCHANNELS message')
+                    return nil
+                end
+            end
+        end
+    end
+end

data/lib/mod_spox/handlers/LuserClient.rb

@@ -0,0 +1,18 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/LuserClient'
+module ModSpox
+    module Handlers
+        class LuserClient < Handler
+            def initialize(handlers)
+                handlers[RFC[:RPL_LUSERCLIENT][:value]] = self
+            end
+            def process(string)
+                users = string =~ /(\d+) users/ ? $1 : 0
+                invis = string =~ /(\d+) invisible/ ? $1 : 0
+                servs = string =~ /(\d+) servers/ ? $1 : 0
+                services = string =~ /(\d+) services/ ? $1 : 0
+                return Messages::Incoming::LuserClient.new(string, users, invis, servs, services)
+            end
+        end
+    end
+end

data/lib/mod_spox/handlers/LuserMe.rb

@@ -0,0 +1,16 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/LuserMe'
+module ModSpox
+    module Handlers
+        class LuserMe < Handler
+            def initialize(handlers)
+                handlers[RFC[:RPL_LUSERME][:value]] = self
+            end
+            def process(string)
+                clients = string =~ /(\d+) clients/ ? $1.to_i : 0
+                servers = string =~ /(\d+) server/ ? $1.to_i : 0
+                return Messages::Incoming::LuserMe.new(string, clients, servers)
+            end
+        end
+    end
+end

data/lib/mod_spox/handlers/LuserOp.rb

@@ -0,0 +1,19 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/LuserOp'
+module ModSpox
+    module Handlers
+        class LuserOp < Handler
+            def initialize(handlers)
+                handlers[RFC[:RPL_LUSEROP][:value]] = self
+            end
+            def process(string)
+                if(string =~ /(\d+) :.*?\s*[oO]perators/)
+                    return Messages::Incoming::LuserOp.new(string, $1.to_i)
+                else
+                    Logger.warn('Failed to match RPL_LUSEROP message')
+                    return nil
+                end
+            end
+        end
+    end
+end

data/lib/mod_spox/handlers/LuserUnknown.rb

@@ -0,0 +1,19 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/LuserUnknown'
+module ModSpox
+    module Handlers
+        class LuserUnknown < Handler
+            def initialize(handlers)
+                handlers[RFC[:RPL_LUSERUNKNOWN][:value]] = self
+            end
+            def process(string)
+                if(string =~ /(\d+) :.*[Uu]nknown/)
+                    return Messages::Incoming::LuserUnknown.new(string, $1.to_i)
+                else
+                    Logger.warn('Failed to match RPL_LUSERUNKNOWN message')
+                    return nil
+                end
+            end
+        end
+    end
+end

data/lib/mod_spox/handlers/Mode.rb

@@ -0,0 +1,62 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/Mode'
+module ModSpox
+    module Handlers
+        class Mode < Handler
+            def initialize(handlers)
+                handlers[:MODE] = self
+            end
+
+            # :spax!~spox@host MODE #m +o spax 
+            # :spax MODE spax :+iw
+            def process(string)
+                orig = string.dup
+                string = string.dup
+                begin
+                    string.slice!(0)
+                    if(string.index('!').nil?) # looks like self mode
+                        source = find_model(string.slice!(0..string.index(' ')-1))
+                        2.times{string.slice!(0..string.index(' '))}
+                        target = find_model(string.slice!(0..string.index(' ')-1))
+                        string.slice!(0..string.index(':'))
+                        action = string.slice!(0)
+                        action == '+' ? target.set_mode(string) : target.unset_mode(string)
+                        return Messages::Incoming::Mode.new(orig, "#{action}#{string}", source, target, nil)
+                    else
+                        source = find_model(string.slice!(0..string.index('!')-1))
+                        2.times{string.slice!(0..string.index(' '))}
+                        channel = find_model(string.slice!(0..string.index(' ')-1))
+                        string.slice!(0)
+                        modes = string.index(' ').nil? ? string.dup : string.slice!(0..string.index(' ')-1)
+                        string[0] == ' ' ? string.slice!(0) : string = ''
+                        action = modes.slice!(0)
+                        if(string.size > 0) #nick modes
+                            nicks = []
+                            string.split.each do |n|
+                                ni = find_model(n)
+                                ni.add_channel(channel)
+                                nicks << ni
+                            end
+                            i = 0
+                            modes.each_char do |m|
+                                nm = Models::NickMode.find_or_create(:channel_id => channel.pk, :nick_id => nicks[i].pk)
+                                action == '+' ? nm.set_mode(m) : nm.unset_mode(m)
+                                i += 1
+                            end
+                            nicks = nicks[0] if nicks.size == 1
+                            return Messages::Incoming::Mode.new(orig, "#{action}#{modes}", source, nicks, channel)
+                        else #channel modes
+                            modes.each_char do |m|
+                                action == '+' ? channel.set_mode(m) : channel.unset_mode(c)
+                            end
+                            return Messages::Incoming::Mode.new(orig, "#{action}#{modes}", source, nil, channel)
+                        end
+                    end
+                rescue Object => boom
+                    Logger.warn("Failed to parse MODE message: #{orig}")
+                    raise Exceptions::GeneralException.new(boom)
+                end
+            end
+        end
+    end
+end

data/lib/mod_spox/handlers/Motd.rb

@@ -0,0 +1,38 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/Motd'
+module ModSpox
+    module Handlers
+        class Motd < Handler
+            def initialize(handlers)
+                handlers[RFC[:RPL_MOTDSTART][:value]] = self
+                handlers[RFC[:RPL_MOTD][:value]] = self
+                handlers[RFC[:RPL_ENDOFMOTD][:value]] = self
+                @motds = Hash.new
+                @raw = Hash.new
+            end
+            def process(string)
+                if(string =~ /^:(\S+) #{RFC[:RPL_MOTDSTART][:value]}.*?:-\s?(\S+)/)
+                    @motds[$1] = Array.new
+                    @raw[$1] = [string]
+                    return nil
+                elsif(string =~ /^:(\S+) #{RFC[:RPL_MOTD][:value]}.*?:-\s?(.+)$/)
+                    @motds[$1] ||= []
+                    @raw[$1] ||= []
+                    @motds[$1] << $2
+                    @raw[$1] << string
+                    return nil
+                elsif(string =~ /^:(\S+) #{RFC[:RPL_ENDOFMOTD][:value]}/)
+                    @raw[$1] ||= []
+                    @motds[$1] ||= []
+                    @raw[$1] << string
+                    message = Messages::Incoming::Motd.new(@raw[$1].join("\n"), @motds[$1].join("\n"), $1)
+                    @motds.delete($1)
+                    @raw.delete($1)
+                    return message
+                else
+                    return nil
+                end
+            end
+        end
+    end
+end

data/lib/mod_spox/handlers/MyInfo.rb

@@ -0,0 +1,24 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/MyInfo'
+module ModSpox
+    module Handlers
+        class MyInfo < Handler
+            def initialize(handlers)
+                handlers[RFC[:RPL_MYINFO][:value]] = self
+            end
+
+            def process(string)
+                if(string =~ /^:\S+ \S+ \S+ (\S+) (\S+) (\S+) (\S+)/)
+                    servername = $1
+                    version = $2
+                    usermodes = $3
+                    channelmodes = $4
+                    return Messages::Incoming::MyInfo.new(string, servername, version, usermodes, channelmodes)
+                else
+                    Logger.warn('Failed to match My Info message')
+                    return nil
+                end
+            end
+        end
+    end
+end

data/lib/mod_spox/handlers/Names.rb

@@ -0,0 +1,86 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/Names'
+module ModSpox
+    module Handlers
+        class Names < Handler
+            def initialize(handlers)
+                handlers[RFC[:RPL_NAMREPLY][:value]] = self
+                handlers[RFC[:RPL_ENDOFNAMES][:value]] = self
+                @names = Hash.new
+                @raw = Hash.new
+            end
+     
+            def process(string)
+                orig = string.dup
+                string = string.dup
+                begin
+                    string.slice!(0..string.index(' '))
+                    type = string.slice!(0..string.index(' ')-1)
+                    if(type == RFC[:RPL_NAMREPLY][:value])
+                        3.times{ string.slice!(0..string.index(' ')) }
+                        chan = string.slice!(0..string.index(' ')-1)
+                        string.slice!(0..string.index(':'))
+                        @names[chan] = Array.new unless @names[chan]
+                        @raw[chan] = Array.new unless @raw[chan]
+                        @raw[chan] << orig
+                        @names[chan] += string.split
+                        return nil
+                    else
+                        2.times{ string.slice!(0..string.index(' ')) }
+                        chan = string.slice!(0..string.index(' ')-1)
+                        channel = find_model(chan)
+                        @raw[chan] << orig if @raw[chan]
+                        nicks = Array.new
+                        ops = Array.new
+                        voice = Array.new
+                        raw = @raw[chan].dup if @raw[chan]
+                        @names[chan] = [] unless @names[chan].is_a?(Array)
+                        @names[chan].each do |n|
+                            nick = Models::Nick.find_or_create(:nick => n.gsub(/^[@+]/, ''))
+                            nick.visible = true
+                            nicks << nick
+                            if(n[0] == '@')
+                                ops << nick
+                                m = Models::NickMode.find_or_create(:nick_id => nick.pk, :channel_id => channel.pk)
+                                m.set_mode('o')
+                            elsif(n[0] == '+')
+                                voice << nick
+                                m = Models::NickMode.find_or_create(:nick_id => nick.pk, :channel_id => channel.pk)
+                                m.set_mode('v')
+                            else
+                                m = Models::NickMode.find_or_create(:nick_id => nick.pk, :channel_id => channel.pk)
+                                m.clear_modes
+                            end
+                            nick.add_channel(channel)
+                            nick.refresh
+                            channel.refresh
+                        end
+                        check_visibility(nicks, channel)
+                        @names.delete(chan)
+                        @raw.delete(chan)
+                        return Messages::Incoming::Names.new(raw, channel, nicks, ops, voice)
+                    end
+                rescue Object => boom
+                    Logger.error("Failed to parse NAMES message: #{orig}")
+                    raise Exceptions::GeneralException.new(boom)
+                end
+            end
+
+            # nicks:: list of nicks in channel
+            # channel:: channel nicks are in
+            # Remove visibility from any nicks that aren't really
+            # in the channel
+            def check_visibility(nicks, channel)
+                channel.nicks.each do |nick|
+                    unless(nicks.include?(nick))
+                        channel.remove_nick(nick)
+                        unless(nick.botnick)
+                            nick.refresh
+                            nick.update(:visible => false) if (Models::Nick.filter(:botnick => true).first.channels & nick.channels).empty?
+                        end
+                    end
+                end
+            end
+        end
+    end
+end

data/lib/mod_spox/handlers/Nick.rb

@@ -0,0 +1,50 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/Nick'
+module ModSpox
+    module Handlers
+        class Nick < Handler
+            def initialize(handlers)
+                handlers[:NICK] = self
+            end
+            # :spox!~spox@some.random.host NICK :flock_of_deer
+            def process(string)
+                orig = string.dup
+                string = string.dup
+                begin
+                    string.slice!(0)
+                    old_nick = find_model(string.slice!(0..string.index('!')-1))
+                    string.slice!(0..string.index(':'))
+                    new_nick = find_model(string)
+                    old_nick.channels.each do |channel|
+                        channel.remove_nick(old_nick)
+                        channel.add_nick(new_nick)
+                        m = Models::NickMode.filter(:nick_id => old_nick.pk, :channel_id => channel.pk).first
+                        if(m)
+                            m.nick_id = new_nick.pk
+                            m.save
+                        end
+                    end
+                    new_nick.username = old_nick.username
+                    new_nick.address = old_nick.address
+                    new_nick.real_name = old_nick.real_name
+                    new_nick.connected_to = old_nick.connected_to
+                    new_nick.away = old_nick.away
+                    new_nick.visible = true
+                    new_nick.save_changes
+                    old_nick.visible = false
+                    old_nick.remove_all_channels
+                    if(old_nick.botnick == true)
+                        old_nick.botnick = false
+                        new_nick.botnick = true
+                    end
+                    new_nick.save
+                    old_nick.save
+                    return Messages::Incoming::Nick.new(orig, old_nick, new_nick)
+                rescue Object => boom
+                    Logger.error("Failed to parse NICK message: #{orig}")
+                    raise Exceptions::GeneralException.new(boom)
+                end
+            end
+        end
+    end
+end

data/lib/mod_spox/handlers/NickInUse.rb

@@ -0,0 +1,19 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/NickInUse'
+module ModSpox
+    module Handlers
+        class NickInUse < Handler
+            def initialize(handlers)
+                handlers[RFC[:ERR_NICKNAMEINUSE][:value]] = self
+            end
+            def process(string)
+                if(string =~ /#{RFC[:ERR_NICKNAMEINUSE][:value]}\s\S+\s(\S+)\s:/)
+                    return Messages::Incoming::NickInUse.new(string, $1)
+                else
+                    Logger.warn('Failed to parse ERR_NICKNAMEINUSE message')
+                    return nil
+                end
+            end
+        end
+    end
+end

data/lib/mod_spox/handlers/Notice.rb

@@ -0,0 +1,35 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/Notice'
+module ModSpox
+    module Handlers
+        class Notice < Handler
+            def initialize(handlers)
+                handlers[:NOTICE] = self
+            end
+            
+            def process(string)
+                if(string =~ /:(\S+)\sNOTICE\s(\S+)\s:(.+)$/)
+                    base_source = $1
+                    target = $2
+                    message = $3
+                    if(base_source =~ /\!/)
+                        source = find_model(base_source.gsub(/!.+$/, ''))
+                        if(base_source =~ /!(.+)@(.+)$/)
+                            source.username == $1
+                            source.address = $2
+                            source.source = base_source
+                            source.save_changes
+                        end
+                        target = find_model(target)
+                    else
+                        source = base_source
+                    end
+                    return Messages::Incoming::Notice.new(string, source, target, message)
+                else
+                    Logger.warn('Failed to match NOTICE message')
+                    return nil
+                end
+            end
+        end
+    end
+end

data/lib/mod_spox/handlers/Part.rb

@@ -0,0 +1,39 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/Part'
+module ModSpox
+    module Handlers
+        class Part < Handler
+            def initialize(handlers)
+                handlers[:PART] = self
+            end
+
+            # :mod_spox!~mod_spox@host PART #m :
+            # :guy2net!~guy2net@92-239-14-145.cable.ubr15.haye.blueyonder.co.uk PART #php
+            
+            def process(string)
+                string = string.dup
+                orig = string.dup
+                begin
+                    string.slice!(0)
+                    nick = find_model(string.slice!(0..string.index('!')-1))
+                    2.times{ string.slice!(0..string.index(' ')) }
+                    unless(string.index(' ').nil?)
+                        channel = find_model(string.slice!(0..string.index(' ')-1))
+                        string.slice!(0..string.index(':'))
+                    else
+                        channel = find_model(string)
+                        string = ''
+                    end
+                    channel.remove_nick(nick)
+                    nick.visible = false if nick.channels.empty?
+                    nick.save_changes
+                    channel.save_changes
+                    return Messages::Incoming::Part.new(orig, channel, nick, string)
+                rescue Object => boom
+                    Logger.error("Failed to parse PART message: #{orig}")
+                    raise Exceptions::GeneralException.new(boom)
+                end
+            end
+        end
+    end
+end

data/lib/mod_spox/handlers/Ping.rb

@@ -0,0 +1,25 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/Ping'
+module ModSpox
+    module Handlers
+        class Ping < Handler
+            def initialize(handlers)
+                handlers[:PING] = self
+            end
+            def process(string)
+                orig = string.dup
+                string = string.dup
+                begin
+                    string.slice!(0) if string[0] == ':'
+                    server = string[0..string.index(' ')-1]
+                    message = string[string.index(':')+1..string.size]
+                    server = message.dup if server == 'PING'
+                    return Messages::Incoming::Ping.new(orig, server, message)
+                rescue Object => boom
+                    Logger.error("Failed to parse PING message: #{string}")
+                    raise Exceptions::GeneralException.new(boom)
+                end
+            end
+        end
+    end
+end

data/lib/mod_spox/handlers/Pong.rb

@@ -0,0 +1,23 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/Pong'
+module ModSpox
+    module Handlers
+        class Pong < Handler
+            def initialize(handlers)
+                handlers[:PONG] = self
+            end
+            def process(string)
+                orig = string.dup
+                string = string.dup
+                begin
+                    a = string.slice!(string.rindex(':')+1, string.size)
+                    string.slice!(-2..string.size)
+                    return Messages::Incoming::Pong.new(orig, string.slice(string.rindex(' ')+1..string.size), a)
+                rescue Object => boom
+                    Logger.error("Failed to parse PONG message: #{orig}")
+                    raise Exceptions::GeneralException.new(boom)
+                end
+            end
+        end
+    end
+end

data/lib/mod_spox/handlers/Privmsg.rb

@@ -0,0 +1,39 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/Privmsg'
+module ModSpox
+    module Handlers
+        class Privmsg < Handler
+            def initialize(handlers)
+                handlers[:PRIVMSG] = self
+            end
+
+            # :spox!~spox@host PRIVMSG #m :foobar
+            def process(string)
+                orig = string.dup
+                string = string.dup
+                begin
+                    string.slice!(0)
+                    base_source = string.slice!(0..string.index(' ')-1)
+                    orig_source = base_source.dup
+                    string.slice!(0..string.index(' ',3))
+                    target = find_model(string.slice!(0..string.index(' ')-1))
+                    string.slice!(0..string.index(':'))
+                    string.strip!
+                    source = find_model(base_source.slice!(0..base_source.index('!')-1))
+                    base_source.slice!(0)
+                    source.username = base_source.slice!(0..base_source.index('@')-1)
+                    base_source.slice!(0)
+                    source.address = base_source.strip
+                    source.source = orig_source
+                    source.save_changes
+                    source.add_channel(target) if target.is_a?(Models::Channel)
+                    return Messages::Incoming::Privmsg.new(orig, source, target, string)
+                rescue Object => boom
+                    Logger.warn("Failed to parse PRIVMSG message: #{orig}")
+                    Logger.warn("Reason for PRIVMSG failure: #{boom}")
+                    raise Exceptions::GeneralException.new(boom)
+                end
+            end
+        end
+    end
+end

data/lib/mod_spox/handlers/Quit.rb

@@ -0,0 +1,29 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/Quit'
+module ModSpox
+    module Handlers
+        class Quit < Handler
+            def initialize(handlers)
+                handlers[:QUIT] = self
+            end
+
+            #:spox!~spox@host QUIT :Ping timeout
+            def process(string)
+                orig = string.dup
+                string = string.dup
+                begin
+                    string.slice!(0)
+                    nick = find_model(string.slice!(0..string.index('!')-1))
+                    string.slice!(0..string.index(':'))
+                    nick.remove_all_channels
+                    nick.visible = false
+                    nick.save
+                    return Messages::Incoming::Quit.new(orig, nick, string)
+                rescue Object => boom
+                    Logger.error("Failed to parse QUIT message: #{orig}")
+                    raise Exceptions::GeneralException.new(boom)
+                end
+            end
+        end
+    end
+end

data/lib/mod_spox/handlers/Topic.rb

@@ -0,0 +1,38 @@
+require 'mod_spox/handlers/Handler'
+require 'mod_spox/messages/incoming/Topic'
+require 'mod_spox/messages/incoming/TopicInfo'
+module ModSpox
+    module Handlers
+        class Topic < Handler
+            def initialize(handlers)
+                handlers[RFC[:RPL_TOPIC][:value]] = self
+                handlers[RFC[:RPL_NOTOPIC][:value]] = self
+                handlers[RFC[:RPL_TOPICINFO][:value]] = self
+                @topics = Hash.new
+            end
+            
+#:irc.host 332 spox #mod_spox : the topic is here
+#:irc.host 333 spox #mod_spox spox 1232126516
+            
+            def process(string)
+                if(string =~ /#{RFC[:RPL_TOPIC][:value]}.+?(\S+)\s:(.+)$/)
+                    channel = find_model($1)
+                    channel.update(:topic => $2)
+                    return Messages::Incoming::Topic.new(string, channel, $2)
+                elsif(string =~ /#{RFC[:RPL_NOTOPIC][:value]}.+?(\S+)\s:/)
+                    channel = find_model($1)
+                    return Messages::Incoming::Topic.new(string, channel, nil)
+                elsif(string =~ /#{RFC[:RPL_TOPICINFO][:value]}\s\S+\s(\S+)\s(\S+)\s(.+)$/)
+                    channel = find_model($1)
+                    nick = find_model($1)
+                    time = Time.at($3.to_i)
+                    return Messages::Incoming::TopicInfo.new(string, channel, nick, time)
+                else
+                    Logger.warn('Failed to parse TOPIC type string')
+                    return nil
+                end                    
+            end
+        end
+    end
+end
+