cogbot 0.0.2

data/.gitignore

@@ -0,0 +1,3 @@
+*.gem
+.bundle
+config/cogbot.yml

data/.rvmrc

@@ -0,0 +1,2 @@
+rvm_gemset_create_on_use_flag=1
+rvm gemset use cogbot

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+gemspec
+

data/Gemfile.lock

@@ -0,0 +1,74 @@
+PATH
+  remote: .
+  specs:
+    cogbot (0.0.2)
+      cinch
+      daemons
+      eventmachine
+      eventmachine_httpserver
+      fortune_gem
+      json
+      nokogiri
+      thor
+      twitter
+      yajl-ruby
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    awesome_print (1.1.0)
+    cinch (2.0.4)
+    coderay (1.0.9)
+    daemons (1.1.9)
+    eventmachine (1.0.3)
+    eventmachine_httpserver (0.2.1)
+    faraday (0.8.7)
+      multipart-post (~> 1.1)
+    flog (4.0.0)
+      ruby_parser (~> 3.1, > 3.1.0)
+      sexp_processor (~> 4.0)
+    fortune_gem (0.0.6)
+    json (1.7.7)
+    method_source (0.8.1)
+    minitest (4.7.4)
+    minitest-ci (2.4.0)
+      minitest (>= 3.0, < 5.0)
+    multi_json (1.7.2)
+    multipart-post (1.2.0)
+    nokogiri (1.5.9)
+    pry (0.9.12.1)
+      coderay (~> 1.0.5)
+      method_source (~> 0.8)
+      slop (~> 3.4)
+    rake (10.0.4)
+    ruby_parser (3.1.3)
+      sexp_processor (~> 4.1)
+    sexp_processor (4.2.1)
+    simple_oauth (0.2.0)
+    simplecov (0.7.1)
+      multi_json (~> 1.0)
+      simplecov-html (~> 0.7.1)
+    simplecov-html (0.7.1)
+    simplecov-rcov (0.2.3)
+      simplecov (>= 0.4.1)
+    slop (3.4.4)
+    thor (0.18.1)
+    twitter (4.6.2)
+      faraday (~> 0.8, < 0.10)
+      multi_json (~> 1.0)
+      simple_oauth (~> 0.2)
+    yajl-ruby (1.1.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  awesome_print
+  cogbot!
+  flog
+  minitest
+  minitest-ci
+  pry
+  rake
+  simplecov
+  simplecov-rcov

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 mose
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,47 @@
+# Cogbot
+
+
+
+Cogbot is an irc bot written in ruby based on [Cinch bot framework](https://github.com/cinchrb/cinch).
+
+It has been in servvice at [Code Green](http://codegreenit.com) since summer 2012 and his stability
+is unquestionned. It is used with a collection of custom plugins that are focused on helping a coding
+team that uses irc as a main shared communication space:
+
+* git notifications pushed on the channel
+* redmine issues polled from our redmine and announced
+* commands to ask rubygems or stack overflow
+* the urban dictionary to make us laugh
+* and some other more or less used features
+
+## Installation
+
+    gem install cogbot
+
+## Usage
+
+At first launch:
+
+    cogbot start
+
+you will be prompted to create a configuration file in ~/.cogbot/cogbot.yml
+When this is done you can launch again and it will just run according to your configuration.
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request
+
+## Todo
+
+* write some documentation of the commands of plugins
+* write tests
+
+## Licence
+
+MIT license
+
+Copyright (c) 2012-13 mose at Code Green

data/Rakefile

@@ -0,0 +1,15 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"
+require "rake/testtask"
+
+Rake::TestTask.new do |t|
+  t.libs = ["lib","plugins"]
+  t.test_files = FileList['test/**/*_test.rb']
+end
+
+task :ci_cleanup do
+  require 'minitest/ci'
+  MiniTest::Ci.clean
+end
+
+task :default => :test

data/bin/cogbot

@@ -0,0 +1,12 @@
+#!/usr/bin/env ruby
+
+ROOT_DIR = File.expand_path('../..', __FILE__)
+$LOAD_PATH.unshift(ROOT_DIR)
+CONFIG_DIR = File.join(File.expand_path('~'),'.cogbot')
+CONFIG_FILE = File.join(CONFIG_DIR,'cogbot.yml')
+LOG_DIR = File.join(CONFIG_DIR,'logs')
+
+require 'yaml'
+require 'lib/cogbot'
+
+Cogbot::Bot.start

data/cogbot.gemspec

@@ -0,0 +1,39 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/cogbot/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["mose"]
+  gem.email         = ["mose@mose.com"]
+  gem.description   = 'Irc bot based on Cinch'
+  gem.summary       = 'Yet another irc bot, focused on helping development teams.'
+  gem.homepage      = "https://github.com/mose/cogbot"
+  gem.license       = "MIT"
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = ['cogbot']
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "cogbot"
+  gem.require_paths = ["lib"]
+  gem.version       = Cogbot::VERSION
+
+  gem.add_dependency 'cinch'
+  gem.add_dependency "thor"
+  gem.add_dependency "eventmachine"
+  gem.add_dependency "eventmachine_httpserver"
+  gem.add_dependency 'nokogiri'
+  gem.add_dependency "daemons"
+
+  gem.add_dependency 'twitter'  # twitter plugin
+  gem.add_dependency 'json' # stackoverflow plugin
+  gem.add_dependency 'yajl-ruby' # rubygems plugin
+  gem.add_dependency 'fortune_gem' # fortune plugin
+
+  gem.add_development_dependency 'awesome_print'
+  gem.add_development_dependency 'rake'
+  gem.add_development_dependency 'minitest'
+  gem.add_development_dependency 'minitest-ci'
+  gem.add_development_dependency 'pry'
+  gem.add_development_dependency 'simplecov'
+  gem.add_development_dependency 'simplecov-rcov'
+  gem.add_development_dependency 'flog'
+end

data/config/cogbot.yml.defaults

@@ -0,0 +1,26 @@
+main:
+  server: 'irc.freenode.net'
+  port: '6667'
+  ssl: false
+  nick: 'cogbot'
+  channels:
+  - '#cinch-bots'
+  prefix: '.'
+  plugins:
+  - 'manager'
+  - 'google'
+  - 'dice'
+  - 'urban'
+  - 'tweet'
+server:
+  ip: 127.0.0.1
+  port: 9090
+tweet:
+  consumer_key: ''
+  consumer_secret: ''
+  oauth_token: ''
+  oauth_token_secret: ''
+redmine:
+  api_key: ''
+  url: ''
+  project: ''

data/lib/cogbot/server.rb

@@ -0,0 +1,22 @@
+class Server < EM::Connection
+  include EM::HttpServer
+
+  def initialize(bot)
+    @bot = bot
+  end
+
+  def post_init
+    super
+    no_environment_strings
+  end
+
+  def process_http_request
+    if @http_request_method == "POST"
+      @bot.handlers.dispatch(:api_callback, nil, @http_post_content)
+    end
+
+    response = EM::DelegatedHttpResponse.new(self)
+    response.status = 200
+    response.send_response
+  end
+end

data/lib/cogbot/setup.rb

@@ -0,0 +1,70 @@
+module Cogbot
+  module Setup
+
+    def self.setvalue default
+      input = $stdin.gets.chomp
+      if input == ''
+        default
+      else
+        input
+      end
+    end
+
+    def self.setbinary default
+      input = $stdin.gets.chomp
+      if input == ''
+        default
+      else
+        input.downcase == 'y'
+      end
+    end
+
+    def self.setlist default
+      input = $stdin.gets.chomp
+      if input == ''
+        default
+      else
+        input.split(',')
+      end
+    end
+
+    def self.init
+      st = "\033[0;33m"
+      en = "\033[m"
+
+      default = YAML::load_file(File.join(ROOT_DIR,'config','cogbot.yml.defaults'))
+      main = {}
+      puts "You don't have a configuration file yet,"
+      puts "let's make one in ~/.cogbot/config.yml"
+      puts
+      puts "Please answer those few questions"
+
+      print "#{st}What name will cogbot use ?#{en} [#{default['main']['nick']}] "
+      main['nick'] = setvalue default['main']['nick']
+
+      print "#{st}What irc server #{main['nick']} will connect to ?#{en} [#{default['main']['server']}] "
+      main['server'] = setvalue default['main']['server']
+
+      print "#{st}What port #{main['nick']} will use to connect to #{main['server']} ?#{en} [#{default['main']['port']}] "
+      main['port'] = setvalue default['main']['port']
+
+      print "#{st}Will #{main['nick']} use SSL to connect to #{main['server']}:#{main['port']} ?#{en} [#{default['main']['ssl'] ? 'Yn' : 'yN'}] "
+      main['ssl'] = setbinary default['main']['ssl']
+
+      print "#{st}What channels #{main['nick']} should join ?#{en} [#{default['main']['channels'].join(',')}] "
+      main['channels'] = setlist default['main']['channels']
+
+      print "#{st}What prefix #{main['nick']} will use to understand he's talked to ?#{en} [#{default['main']['prefix']}] "
+      main['prefix'] = setvalue default['main']['prefix']
+
+      print "#{st}What plugins will be enabled for #{main['nick']} ?#{en} [#{default['main']['plugins'].join(',')}] "
+      main['plugins'] = setlist default['main']['plugins']
+
+      FileUtils.mkdir_p(CONFIG_DIR) unless File.directory?(CONFIG_DIR)
+      FileUtils.mkdir_p(LOG_DIR) unless File.directory?(LOG_DIR)
+      File.open(CONFIG_FILE,'w') { |f| YAML::dump({ 'main' => main }, f) }
+      return main
+    end
+  end
+end
+

data/lib/cogbot/utils.rb

@@ -0,0 +1,9 @@
+class String
+
+  def camelize
+    self.downcase!
+    self.gsub!(/(?:_| )?(.)([^_ ]*)/) { "#{$1.upcase}#{$2}" }
+  end
+
+end
+

data/lib/cogbot/version.rb

@@ -0,0 +1,3 @@
+module Cogbot
+  VERSION = "0.0.2"
+end

data/lib/cogbot.rb

@@ -0,0 +1,106 @@
+require 'cinch'
+require 'eventmachine'
+require 'evma_httpserver'
+require 'nokogiri'
+require 'net/http'
+require 'daemons'
+require 'thor'
+require 'yajl'
+#require 'cinch/storage/yaml'
+require "lib/cogbot/version"
+require "lib/cogbot/utils"
+require "lib/cogbot/server"
+
+module Cogbot
+
+  class Bot < Thor
+
+    desc "start", "start cogbot"
+    def start
+
+      # prepare config
+      config = {}
+      begin
+        config = YAML::load_file(CONFIG_FILE)
+      rescue Exception => e
+        load "lib/cogbot/setup.rb"
+        config['main'] = Cogbot::Setup.init
+      end
+
+      # prepare daemon
+      Daemons.daemonize(
+        :app_name => 'cogbot',
+        :dir_mode => :normal,
+        :log_dir => LOG_DIR,
+        :log_output => true,
+        :dir => File.join('/', 'tmp')
+      )
+
+      # checkout plugins
+      plugins = []
+      config['main']['plugins'].each do |p|
+        puts p
+        puts File.join(ROOT_DIR,'plugins',"#{p}.rb")
+        if File.exists?(File.join(ROOT_DIR,'plugins',"#{p}.rb"))
+          puts p
+          require File.join(ROOT_DIR,'plugins',p)
+          plugins.push Cinch::Plugins.const_get(p.camelize)
+        end
+      end
+
+      # create bot
+      bot = Cinch::Bot.new do
+        configure do |c|
+          c.server = config['main']['server']
+          c.ssl.use = ( config['main']['ssl'] == 'true' )
+          c.nick = config['main']['nick']
+          c.user = config['main']['nick']
+          c.realname = config['main']['nick']
+          c.channels = config['main']['channels']
+          c.options = { 'cogconf' => config }
+          c.plugins.prefix = config['main']['prefix']
+          c.plugins.plugins = plugins
+          #c.storage.backend = Cinch::Storage::YAML
+          #c.storage.basedir = File.join(ROOT_DIR,"yaml")
+          #c.storage.autosave = true
+        end
+        on :message, "hi" do |m|
+          m.reply "Hello, #{m.user.nick}"
+        end
+      end
+      bot.loggers.debug(plugins.inspect)
+
+      Signal.trap('TERM') { EM.stop }
+
+      EM.run do
+        EM.defer { bot.start }
+        if config['server']
+          EM.add_timer(3) do
+            EM.start_server(
+              config['server']['ip'],
+              config['server']['port'],
+              Server,
+              bot
+            )
+          end
+        end
+      end
+
+      bot.quit
+    end
+
+    desc "stop", "stop cogbot"
+    def stop
+      pid_file = File.join('/', 'tmp', 'cogbot.pid')
+      pid = File.read(pid_file).to_i if File.exist?(pid_file)
+      Process.kill('TERM', pid)
+    end
+
+    desc "restart", "restart cogbot"
+    def restart
+      stop
+      start
+    end
+
+  end
+end

data/plugins/dice.rb

@@ -0,0 +1,24 @@
+module Cinch
+  module Plugins
+    class Dice
+      include Cinch::Plugin
+
+      match /^roll$/
+
+      set :plugin_name, 'dice'
+      set :help, <<EOT
+.roll Rolls a dice.
+EOT
+
+      def new(bot)
+        @bot = bot
+      end
+
+      def execute(m)
+        score = rand(100) + 1
+        m.reply("#{m.user.nick} rolls #{score}")
+      end
+
+    end
+  end
+end

data/plugins/exec.rb

@@ -0,0 +1,56 @@
+module Cinch
+  module Plugins
+    class Exec
+      include Cinch::Plugin
+
+      match /do ([^ ]*)( ?.*)/
+
+      set :plugin_name, 'exec'
+      set :help, <<EOT
+Exec is a commandline execution proxy: do <command> <pipe-grep> (only the 5 first lines will be displayed)
+.do  psaux    : rnning processes
+.do  df       : Disk free
+.do  last     : last logged
+.do  free     : free mem
+.do  who      : who is logged
+.do  uptime   : uptime and load average
+EOT
+
+      def new(bot)
+        @bot = bot
+      end
+
+      def exec(command,args)
+        commands = {
+          'psaux' => 'ps -aux',
+          'df' => 'df -h',
+          'last' => 'last',
+          'free' => 'free -mol',
+          'who' => 'w -shuo',
+          'f' => 'fortune',
+          'uptime' => 'uptime'
+        }
+        if command == 'help'
+          return commands.inspect
+        elsif !commands[command].nil?
+          args.strip!
+          line = commands[command]
+          unless args.nil? || args == ''
+            line += ' | grep ' + args.gsub(/;|&/,'')
+          end
+          line += '| head -5'
+          #back = line + "\n"
+          back = `#{line}`
+          return back
+        else
+          return "Command #{command} not found"
+        end
+      end
+
+      def execute(m,command,args)
+        m.reply(exec(command,args))
+      end
+
+    end
+  end
+end

data/plugins/fortune.rb

@@ -0,0 +1,32 @@
+require "fortune_gem"
+
+module Cinch
+  module Plugins
+    class Fortune
+      include Cinch::Plugin
+
+      match /f$/
+
+      set :plugin_name, 'fortune'
+      set :help, <<EOT
+.f gives you a bit of random wisdom.
+EOT
+
+      def new(bot)
+        @bot = bot
+      end
+
+      def fortune
+        max_length = 256
+        fortune = FortuneGem.give_fortune(:max_length => max_length)
+        fortune.gsub(/[\n]/, "").gsub(/[\t]/, " ").gsub("?A:", "? A:")
+      end
+
+      def execute(m)
+        m.reply(fortune)
+      end
+
+    end
+    end
+end
+

data/plugins/gitlistener.rb

@@ -0,0 +1,38 @@
+require "cgi"
+
+module Cinch
+  module Plugins
+    class Gitlistener
+      include Cinch::Plugin
+
+      set :plugin_name, 'gitlistener'
+
+      listen_to :api_callback
+
+      def listen(m, json)
+        hash = Yajl::Parser.parse(URI.unescape(json[8..-1]))
+        bot.loggers.debug(hash.inspect)
+        @bot.config.options['cogconf']['main']['channels'].each do |channel|
+        #['#dev'].each do |channel|
+          hash['commits'].each do |c|
+            files = ""
+            files += '[= ' + short(c['modified']).join(' ') + ' ] ' unless c['modified'].empty?
+            files += '[- ' + short(c['removed']).join(' ') + ' ] ' unless c['removed'].empty?
+            files += '[+ ' + short(c['added']).join(' ') + ' ] ' unless c['added'].empty?
+            Channel(channel).msg "%s@%s: %s %s" % [
+              c['committer']['name'],
+              hash['repository']['name'],
+              CGI::unescape(c['message']),
+              files
+            ]
+          end
+        end
+      end
+
+      def short(x)
+        x.map { |l| l.gsub(/([^\/]*)\//) { |s| s[0,1] + '/'} }
+      end
+
+    end
+  end
+end

data/plugins/google.rb

@@ -0,0 +1,40 @@
+require 'open-uri'
+require 'nokogiri'
+require 'cgi'
+
+module Cinch
+  module Plugins
+    class Google
+      include Cinch::Plugin
+
+      match /g (.+)/
+
+      set :plugin_name, 'google'
+      set :help, <<EOT
+Google returns the first google match on a search
+.g  <keywords> : searches on those keywords
+EOT
+
+      def new(bot)
+        @bot = bot
+      end
+
+      def search(query)
+        url = "http://www.google.com/search?q=#{CGI.escape(query)}"
+        res = Nokogiri::HTML(open(url)).at("h3.r")
+
+        title = res.text
+        link = res.at('a')[:href]
+        link.gsub!(/^.*q=(.*)&sa.*$/) { $1 }
+        desc = res.at("./following::div").children.first.text
+        CGI.unescape_html "#{title} - #{desc} (#{link})"
+      rescue
+        "No results found"
+      end
+
+      def execute(m, query)
+        m.reply(search(query))
+      end
+    end
+  end
+end