cogbot 0.0.2

data/plugins/manager.rb

@@ -0,0 +1,102 @@
+# thx dominikh for this code :) (then slightly modified)
+module Cinch
+  module Plugins
+    class Manager
+      include Cinch::Plugin
+
+      match(/m list/, method: :list_plugins)
+      match(/m load (\S+)/, method: :load_plugin)
+      match(/m unload (\S+)/, method: :unload_plugin)
+      match(/m reload (\S+)/, method: :reload_plugin)
+      match(/m set (\S+) (\S+) (.+)$/, method: :set_option)
+
+      def list_plugins(m)
+        back = ''
+        @bot.plugins.each { |p| back += "#{p.class.name.split('::').last.downcase} " }
+        m.reply back
+      end
+
+      def load_plugin(m, mapping)
+        plugin = mapping.downcase.camelize
+        p plugin
+        p mapping
+
+        file_name = "#{ROOT_DIR}/plugins/#{mapping.downcase}.rb"
+        unless File.exist?(file_name)
+          m.reply "Could not load #{plugin} because #{file_name} does not exist."
+          return
+        end
+
+        begin
+          load(file_name)
+        rescue Exception
+          m.reply "Could not load #{plugin}."
+          raise
+        end
+
+        begin
+          const = Cinch::Plugins.const_get(plugin)
+        rescue NameError
+          m.reply "Could not load #{plugin} because no matching class was found."
+          return
+        end
+
+        @bot.plugins.register_plugin(const)
+        m.reply "Successfully loaded #{plugin}"
+      end
+
+      def unload_plugin(m, plugin)
+        begin
+          plugin_class = Cinch::Plugins.const_get(plugin.downcase.camelize)
+        rescue NameError
+          m.reply "Could not unload #{plugin} because no matching class was found."
+          return
+        end
+
+        @bot.plugins.select {|p| p.class == plugin_class}.each do |p|
+          @bot.plugins.unregister_plugin(p)
+        end
+
+        ## FIXME not doing this at the moment because it'll break
+        ## plugin options. This means, however, that reloading a
+        ## plugin is relatively dirty: old methods will not be removed
+        ## but only overwritten by new ones. You will also not be able
+        ## to change a classes superclass this way.
+        # Cinch::Plugins.__send__(:remove_const, plugin)
+
+        # Because we're not completely removing the plugin class,
+        # reset everything to the starting values.
+        plugin_class.hooks.clear
+        plugin_class.matchers.clear
+        plugin_class.listeners.clear
+        plugin_class.timers.clear
+        plugin_class.ctcps.clear
+        plugin_class.react_on = :message
+        plugin_class.plugin_name = nil
+        plugin_class.help = nil
+        plugin_class.prefix = nil
+        plugin_class.suffix = nil
+        plugin_class.required_options.clear
+
+        m.reply "Successfully unloaded #{plugin}"
+      end
+
+      def reload_plugin(m, plugin)
+        unload_plugin(m, plugin)
+        load_plugin(m, plugin)
+      end
+
+      def set_option(m, plugin, option, value)
+        begin
+          const = Cinch::Plugins.const_get(plugin.downcase.camelize)
+        rescue NameError
+          m.reply "Could not set plugin option for #{plugin} because no matching class was found."
+          return
+        end
+        @bot.config.plugins.options[const][option.to_sym] = eval(value)
+
+        m.reply "Successfuly set option."
+      end
+    end
+  end
+end

data/plugins/redmine.rb

@@ -0,0 +1,61 @@
+require 'open-uri'
+require 'yajl'
+require 'cgi'
+
+module Cinch
+  module Plugins
+    class Redmine
+      include Cinch::Plugin
+
+      timer 60, method: :check_redmine
+
+      set :plugin_name, 'redmine'
+      set :help, <<EOT
+Redmine plugins checks the new issues in the specified redmine instance
+EOT
+
+      def check_redmine
+        @issues ||= Array.new
+        api_key = @bot.config.options['cogconf']['redmine']['api_key']
+        redmine_url = "%s/issues.json?project_id=%s&limit=20" % [
+          @bot.config.options['cogconf']['redmine']['url'],
+          @bot.config.options['cogconf']['redmine']['project']
+        ]
+        res = Yajl::Parser.parse(open(redmine_url, "X-Redmine-API-Key" => api_key))
+        subjs = Hash[ res['issues'].map { |i| [ i['id'], i['subject'] ] } ]
+        issues = subjs.keys
+        newones = Array.new
+        if @issues.count == 0
+          @issues = issues
+        else
+          issues.each { |i| newones << i unless @issues.include? i  }
+          if newones.count > 0
+            newones.each do |i|
+              @issues.push i
+              @bot.config.options['cogconf']['main']['channels'].each do |channel|
+                Channel(channel).msg "[%s] >>> %s - %s/issues/%s" % [
+                  @bot.config.options['cogconf']['redmine']['project'],
+                  subjs[i],
+                  @bot.config.options['cogconf']['redmine']['url'],
+                  i
+                ]
+              end
+            end
+            @issues = @issues[0..50]
+          end
+        end
+        # puts "redmine: done #{newones.count} new issues, #{@issues.count} cached"
+      rescue Exception => e
+        puts "*** #{e.class}\n"
+        puts e.to_s
+        puts e.backtrace
+      end
+
+      def new(bot)
+        @bot = bot
+        @issues = Array.new
+      end
+
+    end
+  end
+end

data/plugins/rubygems.rb

@@ -0,0 +1,51 @@
+require 'open-uri'
+require 'yajl'
+require 'cgi'
+
+module Cinch
+  module Plugins
+    class Rubygems
+      include Cinch::Plugin
+
+      match /r (.*)$/
+
+      set :plugin_name, 'rubygems'
+      set :help, <<EOT
+Rubygems queries http://rubygems.org for finding gems
+EOT
+
+      def query(query)
+        url = "https://rubygems.org/api/v1/search.json?query=#{CGI.escape(query)}"
+        back = "Requesting ... \n"
+        begin
+          file = open(url)
+          if file.class == StringIO
+            json = Yajl::Parser.parse(file)
+            if json.empty?
+              return "Nothing matches '#{query}'"
+            end
+          else
+            json = Yajl::Parser.parse(File.read(file))
+          end
+          4.times do |it|
+            back += "#{json[it]['name']} : #{json[it]['homepage_uri']}\n"
+          end
+        rescue Exception => e
+          p file.class
+          back = "*** #{e.class}\n"
+          back += e.to_s
+        end
+        return back
+      end
+
+      def new(bot)
+        @bot = bot
+      end
+
+      def execute(m,words)
+        m.reply(query(words.strip))
+      end
+
+    end
+  end
+end

data/plugins/shorturl.rb

@@ -0,0 +1,32 @@
+require 'open-uri'
+
+module Cinch
+  module Plugins
+    class Shorturl
+      include Cinch::Plugin
+
+      match /(http:\/\/[^ ]*)/, :use_prefix => false
+
+      set :plugin_name, 'shorturl'
+      set :help, <<EOT
+Shorturl catches urls said on channel and shorten then
+EOT
+
+      def shorten(url)
+        url = open("http://tinyurl.com/api-create.php?url=#{URI.escape(url)}").read
+        url == "Error" ? nil : url
+      rescue OpenURI::HTTPError
+        nil
+      end
+
+      def new(bot)
+        @bot = bot
+      end
+
+      def execute(m,url)
+        m.reply("Shortening: " + shorten(url))
+      end
+
+    end
+  end
+end

data/plugins/stackoverflow.rb

@@ -0,0 +1,37 @@
+require 'open-uri'
+require 'json'
+
+class Stackoverflow
+  include Cinch::Plugin
+
+  match /s (.+)/
+
+  set :plugin_name, 'stackoverflow'
+  set :help, <<EOT
+Stackoverflow returns the first match on a search
+.s  <keywords> : searches on those keywords
+EOT
+
+  def new(bot)
+    @bot = bot
+  end
+
+  def search(query)
+    url = "http://api.stackoverflow.com/docs/search#site=stackoverflow&sort=activity&intitle=#{CGI.escape(query)}"
+p url
+    res = JSON.parse(open(url))
+p res
+    res.items.each do |i|
+      title = i.title
+      link = i.link
+      desc = i.tags.join(', ')
+    end
+    "#{title} - #{desc} (#{link})"
+  rescue
+    "No results found"
+  end
+
+  def execute(m, query)
+    m.reply(search(query))
+  end
+end

data/plugins/tests.rb

@@ -0,0 +1,18 @@
+class Tests
+  include Cinch::Plugin
+
+  set :plugin_name, 'tests'
+  set :help, 'just a plugin for dev tests'
+
+  match /hey you/
+
+  def new(bot)
+    @bot = bot
+  end
+
+
+  def execute(m)
+    m.reply "Who ? me ?"
+  end
+
+end

data/plugins/tweet.rb

@@ -0,0 +1,54 @@
+require 'twitter'
+
+module Cinch
+  module Plugins
+    class Tweet
+      include Cinch::Plugin
+
+      match /t ?([^ ]*)?( ?.*)/
+
+      set :plugin_name, 'tweet'
+      set :help, <<EOT
+Tweet makes the bot can query twittter API
+.t search <term> : searches the public timelines
+EOT
+
+=begin
+  Twitter.configure do |c|
+    c.consumer_key = @bot.config.options.cogconf['tweet']['consumer_key']
+    c.consumer_secret = @bot.config.options.cogconf['tweet']['consumer_secret']
+    c.oauth_token = @bot.config.options.cogconf['tweet']['oauth_token']
+    c.oauth_token_secret = @bot.config.options.cogconf['tweet']['oauth_token_secret']
+  end
+=end
+
+      def new(bot)
+        @bot = bot
+      end
+
+      def exec(command,args)
+        back = ''
+        #back += "command: #{command} / args: #{args}\n"
+        begin
+          case command
+          when 'search'
+            Twitter.search(args, { :lang => 'en', :rpp => '3' }).each do|status|
+              back += status.full_text + "\n"
+            end
+          else
+            back += Twitter.send(command,args.split(','))
+          end
+        rescue Exception => e
+          back += "Bad request\n"
+          back += e.inspect
+        end
+        return back
+      end
+
+      def execute(m,command,args)
+        m.reply(exec(command,args.strip))
+      end
+
+    end
+  end
+end

data/plugins/urban.rb

@@ -0,0 +1,36 @@
+require 'open-uri'
+require 'nokogiri'
+require 'cgi'
+
+module Cinch
+  module Plugins
+    class Urban
+      include Cinch::Plugin
+
+      match /u (.*)$/
+
+      set :plugin_name, 'urban'
+      set :help, <<EOT
+Urban connects to urban dictionary and returns the first result for a given query, replying with the result directly to the sender
+EOT
+
+      def query(query)
+        url = "http://www.urbandictionary.com/define.php?term=#{CGI.escape(query)}"
+        begin
+          CGI.unescape_html Nokogiri::HTML(open(url)).at("div.definition").text.gsub(/\s+/, ' ')
+        rescue
+          "no result found"
+        end
+      end
+
+      def new(bot)
+        @bot = bot
+      end
+
+      def execute(m,words)
+        m.reply(query(words.strip))
+      end
+
+    end
+  end
+end

data/plugins/users.rb

@@ -0,0 +1,39 @@
+
+module Cinch
+  module Plugins
+    class Users
+
+      class UserStruct < Struct.new(:user, :channel, :text, :time)
+      end
+
+      include Cinch::Plugin
+
+      def initialize(*args)
+        super
+        storage[:users] ||= {}
+      end
+
+      match /u (.+ )?([^\s]+)( .+)/
+
+      def exec(query,nick)
+        back ''
+        query.strip! rescue ''
+        nick.strip! rescue ''
+        if query.blank?
+        else
+          case query
+          when 'add'
+          else
+            back = "Unknown command #{query}."
+          end
+        end
+        return back
+      end
+
+      def execute(m, query, nick, msg)
+        m.reply(exec(query, nick, msg))
+      end
+
+    end
+  end
+end

data/test/helper.rb

@@ -0,0 +1,13 @@
+require 'minitest/ci'
+require 'simplecov'
+require 'simplecov-rcov'
+
+SimpleCov.formatter = SimpleCov::Formatter::RcovFormatter
+SimpleCov.start
+
+if $0 == __FILE__
+  TestHelpers::load_env
+  puts ENV.inspect
+end
+
+MiniTest::Ci.auto_clean = false