Move protocol functionality to separate file.

1 files changed, 85 insertions, 0 deletions

diff --git a/irc.rb b/irc.rb
new file mode 100644
index 0000000..835694e
--- /dev/null
+++ b/irc.rb
@@ -0,0 +1,85 @@
+require 'socket'
+
+module Dinobot
+  class IRC
+    def initialize(server, port, nick, pass=nil)
+      @server = server
+      @port = port
+      @nick = nick
+      @pass = pass
+
+      @socket = nil
+    end
+
+    def connect
+      log :info, "Connecting to #{@server}:#{@port}."
+
+      @socket = TCPSocket.new(@server, @port)
+
+      puts "PASS #{@pass}" if @pass
+      puts "NICK #{@nick}"
+      puts "USER #{@nick} 0 * :#{@nick}"
+    end
+
+    def disconnect
+      @socket.close
+    end
+
+    def connected?
+      !(@socket.nil? || @socket.closed?)
+    end
+
+    def gets
+      str = @socket.gets
+
+      log :in, str.inspect
+
+      str
+    end
+
+    def puts(str)
+      log :out, str.inspect
+
+      @socket.puts str
+    end
+
+    def join(channel)
+      puts "JOIN #{channel}"
+    end
+
+    def part(channel)
+      puts "PART #{channel}"
+    end
+
+    def privmsg(channel, message)
+      puts "PRIVMSG #{channel} :#{message}"
+    end
+
+    def pong(message)
+      puts "PONG #{message}"
+    end
+
+    private
+
+    def log(type, *lines)
+      str = lines.join("\n")
+
+      case type
+      when :in
+        prefix = "\e[32m<<\e[0m "
+      when :out
+        prefix = "\e[36m>>\e[0m "
+      when :error
+        prefix = "\e[31m!!\e[0m "
+      when :info
+        prefix = "\e[33m==\e[0m "
+      when :indent
+        prefix = '   '
+      else
+        raise "unknown type specified -- #{type}"
+      end
+
+      Kernel.puts str.gsub(/^/, prefix)
+    end
+  end
+end