spunk 0.0.1

data/lib/spunk.rb

@@ -0,0 +1,10 @@
+$:.unshift File.dirname(__FILE__)
+
+require 'spunk/base_parser'
+require 'spunk/ping_parser'
+require 'spunk/helpers'
+require 'spunk/origin'
+require 'spunk/bot'
+
+SPUNK_VERSION = '0.0.1'
+

data/lib/spunk/base_parser.rb

@@ -0,0 +1,14 @@
+module Spunk
+  module Processor
+    class Base
+      def call(bot, origin, command, parameters)
+        if command =~ /^INVITE #{bot.nickname}$/i
+          room = Spunk::Helpers.parse_room(parameters)
+          if room
+            bot.join_room(room)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/spunk/bot.rb

@@ -0,0 +1,157 @@
+require 'socket'
+require 'openssl'
+
+module Spunk
+  class Bot
+    attr_accessor :nickname, :server, :joined_rooms, :ssl, :server, :rooms, :token
+    attr_reader :processors, :request_processors, :response_processors
+
+    def initialize(options = {})
+      options.each do |option, value|
+        instance_variable_set("@#{option}", value)
+      end
+      @token = options[:token] || nil
+      @ssl = options[:ssl] || false
+      @processors =  []
+      @request_processors = []
+      @response_processors = []
+      @joined_rooms = []
+      @hostname = Socket.gethostname
+      @server ||= { :hostname => options[:hostname] || "localhost", :port => options[:port] || 6667 }
+      add_request_processor(Spunk::Processor::Ping.new)
+      add_request_processor(Spunk::Processor::Base.new)
+      @rooms = options[:rooms] ||= []
+    end
+
+    def start
+      loop do
+        @buffer ||= ""
+        if @ssl
+          @buffer += @socket.gets
+        else
+          @buffer += @socket.recv(1024)
+        end
+        messages = @buffer.split(/\r|\n/).collect { |s| s != "" && !s.nil? ? s : nil }.compact
+        if messages.any?
+          last_character = @buffer[-1..-1]
+          @buffer = if ["\n", "\r"].include?(last_character)
+            ""
+          else
+            messages.pop.to_s
+          end
+
+          messages.each do |message|
+            message.strip!
+            process_request(*parse_message(message))
+          end
+        end
+        sleep 0.25
+      end
+    end
+    
+    def setup_connection
+      connect
+      authenticate
+      @rooms.each { |room| join_room room }
+    end
+    
+    def run
+      setup_connection
+      start
+    end
+
+    def parse_message(message)
+      prefix, message = if message =~ /^\:([^\ ]*) (.*)/
+        message.scan(/^\:([^\ ]*) (.*)/)[0]
+      else
+        [ nil, message ]
+      end
+
+      command, parameters = message.split(/\:/, 2)
+      [ prefix, command, parameters ].map! { |s| s && s.strip }
+    end
+
+    def process_request(prefix, command, parameters)
+      origin = if !prefix.nil? && prefix != ""
+        Origin.new(prefix)
+      end
+
+      (@request_processors + @processors).each do |processor|
+        begin
+          processor.call(self, origin, command, parameters)
+        rescue => e
+          puts e.class.name + ": " + e.message
+          puts e.backtrace.join("\n")
+        end
+      end
+    end
+
+    def process_response(origin, command, parameters)
+      (@response_processors + @processors).each do |processor|
+        begin
+          processor.call(self, origin, command, parameters)
+        rescue => e
+          puts e.class.name + ": " + e.message
+          puts e.backtrace.join("\n")
+        end
+      end
+      @socket.print("#{command} :#{parameters}\r\n")
+    end
+
+    def origin
+      @origin ||= Origin.new("#{@nickname}!#{@username}@#{@hostname}")
+    end
+
+    def add_request_processor(processor = nil, &block)
+      @request_processors << (processor || block)
+    end
+
+    def add_response_processor(processor = nil, &block)
+      @response_processors << (processor || block)
+    end
+
+    def add_processor(processor = nil, &block)
+      @processors << (processor || block)
+    end
+
+    def connect
+      @socket = TCPSocket.new(@server[:hostname], @server[:port])
+      if @ssl == true
+        @ssl_context = OpenSSL::SSL::SSLContext.new()
+        @ssl_context.verify_mode = OpenSSL::SSL::VERIFY_NONE
+        @socket = OpenSSL::SSL::SSLSocket.new(@socket, @ssl_context)
+        @socket.sync_close = true
+        @socket.connect
+      end
+      trap("INT") do
+        @socket.close
+      end
+      trap("KILL") do
+        @socket.close
+      end
+    end
+
+    def authenticate
+      unless @token.nil?
+        send_message "PASS #{@token}"
+      end
+      send_message "NICK #{@nickname}"
+      send_message "USER #{@username} #{@hostname} bla :#{@fullname}"
+    end
+
+    def join_room(room, password = nil)
+      @joined_rooms << room
+      @joined_rooms.uniq!
+      send_message("JOIN #{room}" + (password ? " #{password}" : ""))
+    end
+
+    def say(to, message)
+      send_message "PRIVMSG #{to} :#{message}"
+    end
+
+    def send_message(message)
+      command, parameters = message.strip.split(/\:/, 2)
+      process_response(origin, command.to_s.strip, parameters)
+    end
+  end
+end

data/lib/spunk/helpers.rb

@@ -0,0 +1,15 @@
+module Spunk
+  module Helpers
+   def Helpers.parse_room(command)
+     command.match(/(\#\S+\b)/i)
+     return $1
+   end
+   
+   def Helpers.hashify(bot, origin, command, msg)
+     hash = {:bot=>bot, :origin=>origin, :command => command, :msg => msg, :room => nil}
+     command.match(/(\#\S+\b)/i)
+     hash[:room] = $1
+     return hash
+   end
+  end
+end

data/lib/spunk/origin.rb

@@ -0,0 +1,24 @@
+module Spunk
+  class Origin
+    attr_accessor :prefix
+
+    def initialize(prefix)
+      @prefix = prefix
+    end
+
+    def nickname
+      @prefix.to_s.split(/!/, 2)[0]
+    end
+
+    def to_s
+      case
+      when nickname != ""
+        nickname
+      when @prefix
+        @prefix
+      else
+        "-unknown-"
+      end
+    end
+  end
+end

data/lib/spunk/ping_parser.rb

@@ -0,0 +1,11 @@
+module Spunk
+  module Processor
+    class Ping
+      def call(bot, origin, command, parameters)
+        if command =~ /^PING$/
+          bot.send_message("PONG :#{parameters}")
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,50 @@
+--- !ruby/object:Gem::Specification
+name: spunk
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Michael Heijmans
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-03-21 00:00:00.000000000 Z
+dependencies: []
+description: A simple little IRCbot framework based on Sprinkles
+email: parabuzzle@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/spunk.rb
+- lib/spunk/bot.rb
+- lib/spunk/origin.rb
+- lib/spunk/base_parser.rb
+- lib/spunk/ping_parser.rb
+- lib/spunk/helpers.rb
+homepage: http://github.com/parabuzzle/spunk
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.25
+signing_key: 
+specification_version: 3
+summary: A simple little IRCbot framework based on Sprinkles
+test_files: []