Added method deprecation reminders and updated comments.

lib/wrack/connection.rb

@@ -62,6 +62,7 @@ def poll
 
     # Takes either an object that responds to a callback or a block that
     # receives a single argument
+    # XXX Remove array callback_type to simplify this method?
     def register_callback(callback_type, options = {}, &block)
       # Validate type
       case callback_type
@@ -103,6 +104,10 @@ def fire_callbacks(type, raw)
       end
     end
 
+    # Convenience method to set signal handlers to terminate a connection
+    # This is implemented outside of the connect method so that classes
+    # reimplementing the connect method can just call this instead of calling
+    # the super method
     def set_signals
       %w{INT TERM QUIT}.each do |signal|
         Signal.trap(signal) do

lib/wrack/session.rb

@@ -1,5 +1,5 @@
-# Assembles the higher level components of an IRC connection on top of a raw
-# socket. Manages callbacks and maintains the connection.
+# Abstraction on top of a raw Wrack::Connection to manage IRC level
+# callbacks
 require 'wrack'
 require 'wrack/irc'
 module Wrack
@@ -17,11 +17,13 @@ def initialize(options = {})
       # Initialize IRC level callback mechanism
       register_message_handler
 
+      # XXX REMOVE ME
       on :ping do |msg|
         pong(msg.params)
       end
     end
 
+    # XXX REMOVE ME
     def connect
       @connection.connect
     end
@@ -35,6 +37,7 @@ def receive(context, &block)
       @receivers << receiver
     end
 
+    # XXX REMOVE ME
     def on(command, options = {}, &block)
       callback = Wrack::IRC::Callback.new(command, options, &block)
       @callbacks[command] ||= []