mod_spox 0.1.0.1 → 0.2.0

data/data/mod_spox/extras/Roulette.rb

@@ -1,7 +1,7 @@
 class Roulette < ModSpox::Plugin
 
     include Models
-    
+
     def initialize(pipeline)
         super(pipeline)
         Signature.find_or_create(:signature => 'roulette', :plugin => name, :method => 'roulette', :requirement => 'public')
@@ -15,7 +15,7 @@ class Roulette < ModSpox::Plugin
         @banner = nil
         @pipeline.hook(self, :get_banner, :Internal_PluginResponse)
     end
-    
+
     # message:: ModSpox::Messages::Incoming::Privmsg
     # Display chamber statistics
     def chambers(m, p)
@@ -32,13 +32,13 @@ class Roulette < ModSpox::Plugin
             reply m.replyto, "\2Error:\2 No games found"
         end
     end
-    
+
     # message:: ModSpox::Messages::Internal::PluginResponse
     # Get the banner plugin
     def get_banner(message)
         @banner = message.plugin if message.found?
     end
-    
+
     # message:: ModSpox::Messages::Incoming::Privmsg
     # params:: empty
     # Play roulette
@@ -46,7 +46,7 @@ class Roulette < ModSpox::Plugin
         return unless message.is_public?
         do_shot(message.source, message.target)
     end
-    
+
     # message:: ModSpox::Messages::Incoming::Privmsg
     # params:: empty
     # Kill self
@@ -54,7 +54,7 @@ class Roulette < ModSpox::Plugin
         return unless message.is_public?
         do_suicide(message.source, message.target)
     end
-    
+
     # message:: ModSpox::Messages::Incoming::Privmsg
     # params:: nick
     # Shoot other person playing current game
@@ -76,7 +76,6 @@ class Roulette < ModSpox::Plugin
         return unless message.is_public?
         ds = Database.db[:infos].left_outer_join(:games, :id => :game_id)
         ds.select!(:nick_id, :COUNT[:win] => :wins).where!(:channel_id => message.target.pk, :win => true).group!(:nick_id).reverse_order!(:wins).limit!(10)
-        Logger.log(ds.sql)
         ids = ds.map(:nick_id)
         top = []
         ids.each do |id|
@@ -106,9 +105,9 @@ class Roulette < ModSpox::Plugin
         end
         reply(message.replyto, "\2Roulette stats #{nick.nick}:\2 #{win_loss_ratio(nick, message.target)}% survival rate. Has won #{games_won(nick, message.target)} games and lost #{games_lost(nick, message.target)} games, taking a total of #{total_shots(nick, message.target)} shots.")
     end
-    
+
     private
-    
+
     # nick:: ModSpox::Models::Nick
     # channel:: ModSpox::Models::Channel
     # Report win/loss ratio for nick
@@ -129,7 +128,7 @@ class Roulette < ModSpox::Plugin
     def games_won(nick, channel)
         Info.left_outer_join(:games, :id => :game_id).filter('nick_id = ?', nick.pk).filter('channel_id = ?', channel.pk).filter('win = ?', true).size
     end
-    
+
     # nick:: ModSpox::Models::Nick
     # channel:: ModSpox::Models::Channel
     # Return number of games nick has lost
@@ -155,7 +154,7 @@ class Roulette < ModSpox::Plugin
 
     # nick:: ModSpox::Models::Nick
     # channel:: ModSpox::Models::Channel
-    # Fire shot    
+    # Fire shot
     def do_shot(nick, channel)
         begin
             shot(nick, channel)
@@ -165,7 +164,7 @@ class Roulette < ModSpox::Plugin
             kill_nick(nick, channel)
         end
     end
-    
+
     # nick:: ModSpox::Models::Nick
     # channel:: ModSpox::Models::Channel
     # Commit suicide
@@ -179,7 +178,7 @@ class Roulette < ModSpox::Plugin
             kill_nick(nick, channel)
         end
     end
-    
+
     def kill_nick(nick, channel)
         unless(@banner.nil?)
             begin
@@ -188,7 +187,7 @@ class Roulette < ModSpox::Plugin
                 reply(channel, "#{nick.nick}: *BANG*")
             rescue Object => boom
                 reply(channel, "#{nick.nick}: *BANG*")
-                Logger.log("Error: Roulette ban generated an unexpected error: #{boom}")
+                Logger.warn("Error: Roulette ban generated an unexpected error: #{boom}")
             end
         else
             reply(channel, "#{nick.nick}: *BANG*")
@@ -207,7 +206,7 @@ class Roulette < ModSpox::Plugin
         end
         return game
     end
-    
+
     # nick:: ModSpox::Models::Nick
     # channel:: ModSpox::Models::Channel
     # Process shot
@@ -218,7 +217,7 @@ class Roulette < ModSpox::Plugin
         cur_game.update_with_params(:shots => cur_game.shots - 1)
         raise Bullet.new(cur_game) if cur_game.shots < 1
     end
-    
+
     # nick:: ModSpox::Models::Nick
     # game:: Game
     # Return number of games nick has won
@@ -227,7 +226,7 @@ class Roulette < ModSpox::Plugin
             info.update_with_params(:win => true) unless info.nick_id == nick.pk
         end
     end
-    
+
     class Game < Sequel::Model
         set_schema do
             primary_key :id
@@ -236,16 +235,16 @@ class Roulette < ModSpox::Plugin
             integer :chamber, :null => false, :default => 1
             foreign_key :channel_id, :null => false, :table => :channels
         end
-        
+
         before_create do
             self.stamp = Time.now
         end
-        
+
         def channel
             Models::Channel[channel_id]
         end
     end
-    
+
     class Info < Sequel::Model
         set_schema do
             primary_key :id
@@ -254,16 +253,16 @@ class Roulette < ModSpox::Plugin
             foreign_key :nick_id, :null => false, :table => :nicks
             foreign_key :game_id, :null => false, :table => :games
         end
-        
+
         def nick
             Models::Nick[nick_id]
         end
-        
+
         def game
             Models::Game[game_id]
         end
     end
-    
+
     class Bullet < Exception
         attr_reader :game
         def initialize(game)

data/data/mod_spox/extras/Search.rb

@@ -8,12 +8,12 @@ class Search < ModSpox::Plugin
         super(pipeline)
         Models::Signature.find_or_create(:signature => 'search (\d+)? ?(.+)', :plugin => name, :method => 'search', :description => 'Search the internet').params = [:number, :terms]
     end
-    
+
     def search(message, params)
         limit = params[:number] ? params[:number].to_i : 1
         limit = limit > 5 || limit < 1 ? 1 : limit
         begin
-            resp = Net::HTTP.new('www.scroogle.org', 80).request_get("/cgi-bin/nbbw.cgi?Gw=#{CGI::escape(params[:terms])}", nil)
+            resp = Net::HTTP.new('www.scroogle.org', 80).request_get("/cgi-bin/nbbw.cgi?Gw=#{CGI::escape(params[:terms])}")
             resp.value
             page = resp.body
             results = []
@@ -29,6 +29,7 @@ class Search < ModSpox::Plugin
             reply message.replyto, output
         rescue Object => boom
             @pipeline << Privmsg.new(message.replyto, "Failed to find any results for: #{params[:terms]} Reason: #{boom}")
+            Logger.warn("Error: #{boom}\n#{boom.backtrace.join("\n")}")
         end
     end
 end

data/data/mod_spox/extras/Slashdot.rb

@@ -0,0 +1,35 @@
+require 'rexml/document'
+require 'open-uri'
+
+class Slashdot < ModSpox::Plugin
+
+    def initialize(pipeline)
+        begin
+            require 'htmlentities'
+        rescue Object => boom
+            Logger.warn('Error: This plugin requires the HTMLEntities gem. Please install and reload plugin.')
+            raise Exceptions::BotException.new("Missing required HTMLEntities library")
+        end
+        super
+        add_sig(:sig => '(slashdot|\/\.) ?(\d+)?', :method => 'show_slashdot', :desc => 'Slashdot headlines', :params => [:trig, :num])
+        @coder = HTMLEntities.new
+    end
+
+    def show_slashdot(message, params)
+        src = open('http://rss.slashdot.org/Slashdot/slashdot')
+        doc = REXML::Document.new(src.read)
+        num = params[:num].nil? || params[:num].to_i < 1 ? 1 : params[:num].to_i
+        num = 5 if num > 5
+        output = []
+        doc.elements.each('rdf:RDF/item') do |item|
+            num -= 1
+            title = CGI::unescapeHTML(item.elements['title'].text)
+            title = title.gsub(/<.+?>/, ' ').gsub(/[\r\n]/, '').gsub(/\s+/, ' ')
+            title = @coder.decode(title)
+            output << "\2/.\2 #{title.strip} \2->\2 #{Helpers.tinyurl(item.elements['link'].text)}"
+            break if num < 1
+        end
+        reply message.replyto, output
+    end
+
+end

data/data/mod_spox/extras/Topten.rb

@@ -11,7 +11,7 @@ class Topten < ModSpox::Plugin
             :description => 'Show topten from given date', :requirement => 'public').params = [:date]
         Signature.find_or_create(:signature => 'stats ?(\S+)?', :plugin => name, :method => 'stats', :description => 'Show stats on nick',
             :requirement => 'public').params = [:nick]
-        Signature.find_or_create(:signature => 'stats lifetime? (\S+)?', :plugin => name, :method => 'life_stats',
+        Signature.find_or_create(:signature => 'stats lifetime (\S+)?', :plugin => name, :method => 'life_stats',
             :description => 'Show stat totals for given nick', :requirement => 'public').params = [:nick]
         @pipeline.hook(self, :log_stats, :Incoming_Privmsg)
     end
@@ -62,9 +62,9 @@ class Topten < ModSpox::Plugin
     def life_stats(m, p)
         nick = p[:nick] ? Nick.locate(p[:nick], false) : m.source
         if(nick)
-            result = ChatStat.group(:nick_id).select(:SUM[:bytes] => :tbytes, :SUM[:words] => :twords, :SUM[:questions] => :tquestions).filter(:nick_id => nick.pk).first
-            bytes = (result.tbytes > 1023) ? "#{sprintf('%.2f', (result.tbytes / 1024.0))}kb" : "#{result.tbytes}b"
-            reply m.replyto, "#{nick.nick} (total): #{bytes} logged, #{result.twords} words spoken and #{result.tquestions} questions asked"
+            result = ChatStat.group(:nick_id).select(:SUM[:bytes].as(:tbytes), :SUM[:words].as(:twords), :SUM[:questions].as(:tquestions)).filter(:nick_id => nick.pk).first
+            bytes = (result[:tbytes] > 1023) ? "#{sprintf('%.2f', (result[:tbytes] / 1024.0))}kb" : "#{result[:tbytes]}b"
+            reply m.replyto, "#{nick.nick} (total): #{bytes} logged, #{result[:twords]} words spoken and #{result[:tquestions]} questions asked"
         else
             reply m.replyto, "\2Error:\2 Failed to find #{p[:nick]}"
         end
@@ -76,7 +76,7 @@ class Topten < ModSpox::Plugin
         stat = ChatStat.find_or_create(:nick_id => m.source.pk, :channel_id => m.target.pk, :daykey => key)
         words = m.message.scan(/([^ ]+ |[^ ]$)/).size
         bytes = m.message.gsub(/[^a-zA-Z0-9`\~\!@#\$%\^&\*\(\)_\+\[\]\}\{;:'"\/\?\.>,<\\|\-=]/, '').length
-        questions = m.message.scan(/.+?(\? |\?$)/).size
+        questions = m.message.scan(/.+?(!?\?!? |!?\?!?$)/).size
         stat.words += words
         stat.bytes += bytes
         stat.questions += questions

data/data/mod_spox/extras/TracTicket.rb

@@ -0,0 +1,68 @@
+require 'net/http'
+
+class TracTicket < ModSpox::Plugin
+    
+    def initialize(pipeline)
+        super
+        admin = Group.find_or_create(:name => 'admin')
+        add_sig(:sig => 'ticket (defect|enhancement|task) (.+?)( # (.+))?', :method => :add_ticket,
+                :desc => 'Add new ticket to trac ticket tracker', :params => [:type, :short, :trash, :long])
+        add_sig(:sig => 'ticket trac( (\S+))?', :method => :trac_location, :group => admin,
+                :desc => 'Show/set trac site', :params => [:trash, :site])
+        add_sig(:sig => 'ticket site', :method => :trac_location, :desc => 'Show trac site')
+        @site = Config[:trac]
+    end
+    
+    def trac_location(message, params)
+        if(params[:site])
+            site = Config.find_or_create(:name => 'trac')
+            site.value = params[:site]
+            site.save
+            @site = params[:site]
+            reply message.replyto, "\2Trac Update:\2 Location of trac system has been updated: #{params[:site]}"
+        else
+            output = "\2Error:\2 Trac site has not yet been set"
+            unless(@site.nil?)
+                output = "\2Trac location:\2 #{@site}"
+            end
+            reply message.replyto, output
+        end
+    end
+    
+    def add_ticket(message, params)
+        if(@site =~ /(http:\/\/)?([^\/:]+)(:\d+)?(.+)$/)
+            addr = $2
+            port = $3
+            path = $4
+            port = port.nil? ? 80 : port.gsub(/:/, '').to_i
+            con = Net::HTTP.new(addr, port)
+            begin
+                cookie, fid = get_form_info(con, path)
+                long = params[:long] ? params[:long] : params[:short]
+                headers = {
+                    'Cookie' => cookie,
+                    'Referer' => @site,
+                    'Content-Type' => 'application/x-www-form-urlencoded'
+                }   
+                data = "reporter=#{URI.escape(message.source.nick)}&summary=#{URI.escape(params[:short])}&description=#{URI.escape(long)}&type=#{URI.escape(params[:type])}&action=create&status=new&priority=minor&__FORM_TOKEN=#{URI.escape(fid)}"
+                resp, data = con.post(path, data, headers)
+                reply message.replyto, 'Ticket was successfully added to tracker'
+            rescue Object => boom
+                reply message.replyto, "\2Error:\2 Encountered unexpected error while submitting ticket: #{boom}"
+            end
+        else
+            reply message.replyto, "\2Error:\2 Trac location has not been set"
+        end
+    end
+    
+    def get_form_info(con, path)
+        res, content = con.get(path, {})
+        cookie = res['set-cookie']
+        if(content =~ /name="__FORM_TOKEN" value="(.+?)"/i)
+            return cookie, $1
+        else
+            raise 'Failed to locate proper identification'
+        end
+    end
+    
+end

data/data/mod_spox/extras/Translate.rb

@@ -4,37 +4,64 @@ class Translate < ModSpox::Plugin
     
     def initialize(pipeline)
         super(pipeline)
+        begin
+            require 'htmlentities'
+        rescue Object => boom
+            Logger.warn('Error: This plugin requires the HTMLEntities gem. Please install and reload plugin.')
+            raise Exceptions::BotException.new("Missing required HTMLEntities library")
+        end
         Signature.find_or_create(:signature => 'translate ([a-z]{2}\|[a-z]{2}) (.+)', :plugin => name, :method => 'translate',
             :description => 'Translate text').params = [:lang, :text]
         Signature.find_or_create(:signature => 'autotranslate add ([a-z]{2}) (\S+)', :plugin => name, :method => 'auto_add',
             :description => 'Add a nick to the autotranslate service').params = [:lang, :nick]
         Signature.find_or_create(:signature => 'autotranslate remove (\S+)', :plugin => name, :method => 'auto_remove',
             :description => 'Remove a nick from the autotranslate service').params = [:nick]
+        add_sig(:sig => 'translate languages', :method => 'langs', :description => 'Show available languages')
+        @pipeline.hook(self, :listener, :Incoming_Privmsg)
         @watchers = {}
         @cache = {}
+        @coder = HTMLEntities.new
+        @allowed = {'zh'=>'Chinese-simplified','zt'=>'Chinese-traditional','en'=>'English','nl'=>'Dutch',
+                    'fr'=>'French','de'=>'German','el'=>'Greek','it'=>'Italian','ja'=>'Japanese',
+                    'ko'=>'Korean','pt'=>'Portuguese','ru'=>'Russian','es'=>'Spanish'}
+        @allowed_trans = ['zh_en','zh_zt', 'zt_en', 'zt_zh', 'en_zh', 'en_zt', 'en_nl', 'en_fr', 'en_de',
+                          'en_el', 'en_it', 'en_ja', 'en_ko', 'en_pt', 'en_ru', 'en_es', 'nl_en', 'nl_fr',
+                          'fr_nl', 'fr_en', 'fr_de', 'fr_el', 'fr_it', 'fr_pt', 'fr_es', 'de_en', 'de_fr',
+                          'el_en', 'el_fr', 'it_en', 'it_fr', 'ja_en', 'ko_en', 'pt_en', 'pt_fr', 'ru_en',
+                          'es_en', 'es_fr']
+    end
+    
+    def langs(m, params)
+        output = ['Available languages for translate:']
+        s = []
+        @allowed.each_pair{|k,v| s << "#{v} (\2#{k}\2)"}
+        output << s.join(', ')
+        reply m.replyto, output
     end
     
     def auto_add(message, params)
         return unless message.is_public?
-        nick = Helpers.find_model(params[:nick], false)
-        if(nick && nick.channels.include?(message.target))
-            @watchers[message.target.pk] = {} unless @watchers.has_key?(message.target.pk)
-            @watchers[message.target.pk][nick.pk] = params[:lang] unless @watchers[message.target.pk].has_key?(nick.pk)
-            hook
-            reply message.replyto, "#{params[:nick]} is now being tracked for auto translation"
+        if(@allowed_trans.include?("en_#{params[:lang]}") && @allowed_trans.include?("#{params[:lang]}_en"))
+            nick = Helpers.find_model(params[:nick], false)
+            if(nick && nick.channels.include?(message.target))
+                @watchers[message.target.pk] = {} unless @watchers.has_key?(message.target.pk)
+                @watchers[message.target.pk][nick.pk] = params[:lang] unless @watchers[message.target.pk].has_key?(nick.pk)
+                reply message.replyto, "#{params[:nick]} is now being tracked for auto translation"
+            else
+                reply message.replyto, "\2Error:\2 Failed to locate #{params[:nick]}"
+            end
         else
-            reply message.replyto, "\2Error:\2 Failed to locate #{params[:nick]}"
+            reply message.replyt, "\2Error:\2 Unsupported bi-directional translation"
         end
     end
     
     def auto_remove(message, params)
         return unless message.is_public?
-        nick = Helpers.find_model(params[:nick], false)
+        nick = Models::Nick.locate(params[:nick], false)
         if(nick)
             if(@watchers.has_key?(message.target.pk))
                 @watchers[message.target.pk].delete(nick.pk) if @watchers[message.target.pk].has_key?(nick.pk)
                 @watchers.delete(message.target.pk) if @watchers[message.target.pk].empty?
-                hook
                 reply message.replyto, "#{params[:nick]} is no longer being tracked for auto translation"
             else
                 reply message.replyto, "No one is currently being tracked"
@@ -45,15 +72,20 @@ class Translate < ModSpox::Plugin
     end
     
     def translate(message, params)
-        reply message.replyto, "\2Translation:\2 #{do_translation(params[:lang], params[:text])}"
+        begin
+            reply message.replyto, "\2Translation:\2 #{do_translation(params[:lang], params[:text])}"
+        rescue Object => boom
+            reply message.replyto, "\2Error:\2 #{boom}"
+        end
     end
     
     def listener(message)
         if(message.is_public? && @watchers.has_key?(message.target.pk))
             if(@watchers[message.target.pk].has_key?(message.source.pk))
-                reply message.replyto, "\2Translation (#{message.source.nick}):\2 #{do_translation("#{@watchers[message.target.pk][message.source.pk]}en", message.message)}"
+                trans_message = do_translation("#{@watchers[message.target.pk][message.source.pk]}en", message.message)
+                reply message.replyto, "\2Translation (#{message.source.nick}):\2 #{trans_message}" unless trans_message == message.message
             elsif(message.message =~ /^(\S+)[:,]/)
-                Logger.log("Matched a nick: #{$1}")
+                Logger.info("Translate matched a followed nick: #{$1}")
                 nick = Helpers.find_model($1, false)
                 return unless nick
                 if(@watchers[message.target.pk].has_key?(nick.pk))
@@ -66,29 +98,36 @@ class Translate < ModSpox::Plugin
     private
     
     def do_translation(langs, text)
+        raise 'Unsupported language combination for translation' unless @allowed_trans.include?(langs.gsub(/\|/, '_'))
         if(@cache.has_key?(langs) && @cache[langs].has_key?(text))
             return @cache[langs][text]
         end
-        connection = Net::HTTP.new('babelfish.altavista.com', 80)
-        response = connection.request_get("/tr?tt=urltext&trtext=#{CGI::escape(text)}&lp=#{langs.gsub(/\|/, '_')}", nil)
-        response.value()
-        if response.body.gsub(/[\r\n]/, '') =~ /<div style=padding:10px;>(.+?)<\/div>/
-            result = $1
-            if(text.length < 15)
-                @cache[langs] = {} unless @cache.has_key?(langs)
-                @cache[langs][text] = result
+        content = Net::HTTP.post_form(URI.parse('http://babelfish.yahoo.com/translate_txt'), {
+            'ei' => 'UTF-8',
+            'doit' => 'done',
+            'fr' => 'bf-home',
+            'intl' => '1',
+            'tt' => 'urltext',
+            'trtext' => text,
+            'lp' => langs.gsub(/\|/, '_'),
+            'btnTrTxt' => 'Translate'
+        }).body
+        if(content)
+            if(content =~ /<div id="result">(.+?)<input/im)
+                tr = $1
+                tr.gsub!(/<.+?>/, '')
+                tr.gsub!(/[\r\n]/, ' ')
+                tr.gsub!(/\s+/, ' ')
+                if(text.length < 15)
+                    @cache[langs] = {} unless @cache.has_key?(langs)
+                    @cache[langs][text] = tr
+                end
+                return @coder.decode(tr).strip
+            else
+                raise 'Failed to locate translation'
             end
-            return result.gsub(/\s+/, ' ')
-        else
-            raise "Failed to extract translation"
-        end
-    end
-    
-    def hook
-        if(@watchers.size > 0)
-            @pipeline.hook(self, :listener, :Incoming_Privmsg)
         else
-            @pipeline.unhook(self, :listener, :Incoming_Privmsg)
+            raise "Failed to receive result from server"
         end
     end