Pulse 1.2.0 → 1.2.1

data/library/pulse.rb

@@ -2,6 +2,7 @@
 
 require 'yaml'
 require 'socket'
+require 'pulse/dcc'
 require 'pulse/user'
 require 'pulse/queue'
 require 'pulse/client'

data/library/pulse/client.rb

@@ -22,6 +22,8 @@ module Pulse
     def connect
       trap 2 do
         unload_scripts
+        @connection.close
+        Thread.list.each &:kill
         transmit :QUIT, "I was interrupted" 
       end
       @connection.establish
@@ -69,6 +71,15 @@ module Pulse
       end
     end
 
+    def send_file path, recipient
+      Thread.new path, recipient do
+        DCC.new(path).listen do |addr|
+          # FIXME: find the right local ip address and convert it to an integer
+          transmit :PRIVMSG, recipient, "\x01DCC SEND #{File.basename path} 1489377357 #{addr[1]} #{File.size path}\x01"
+        end
+      end
+    end
+
   private
 
     def emit name, *args

data/library/pulse/connection.rb

@@ -10,6 +10,8 @@ module Pulse
 
     def establish
       TCPSocket.open @settings.hostname, @settings.port do |socket|
+        @socket = socket
+
         Thread.start socket, &@queue.method(:process)
         @delegate.connection_established self
 
@@ -21,6 +23,10 @@ module Pulse
       @delegate.connection_terminated self
     end
 
+    def close
+      @socket.close
+    end
+
     def transmit name, *arguments
       Command.new(name, arguments).tap { |this| @queue << this }
     end

data/library/pulse/dcc.rb

@@ -0,0 +1,37 @@
+# encoding: utf-8
+
+require 'timeout'
+
+module Pulse
+  class DCC
+    Duration = 150
+
+    def initialize path
+      @path = path
+    end
+
+    def listen
+      client = nil
+
+      TCPServer.open '0.0.0.0', 0 do |server|
+        yield server.addr if block_given?
+
+        begin
+          timeout(Duration) { client = server.accept }
+        rescue Timeout::Error
+          server.close
+        end
+
+        each_chunk { |chunk| client.write chunk } if client
+        client.close if client
+      end
+    end
+
+    def each_chunk
+      File.open @path, ?r do |file|
+        yield file.readpartial 1024 until file.eof?
+      end
+    end
+
+  end
+end

data/library/pulse/handling.rb

@@ -54,7 +54,12 @@ module Pulse
 
           (@conversations[command.sender.nickname] ||= Conversation.new user, self).tap do |conversation|
             user.channel = conversation
-            emit :conversation, user, conversation, message
+
+            if message.starts_with? "\x01DCC"
+              emit :DCC, user, conversation, message.split[1..-1]
+            else
+              emit :conversation, user, conversation, message
+            end
           end
         end
       end

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 1
   - 2
-  - 0
-  version: 1.2.0
+  - 1
+  version: 1.2.1
 platform: ruby
 authors: 
 - Mikkel Kroman
@@ -14,7 +14,7 @@ autorequire:
 bindir: executables
 cert_chain: []
 
-date: 2010-11-25 00:00:00 +01:00
+date: 2010-11-26 00:00:00 +01:00
 default_executable: 
 dependencies: []
 
@@ -38,6 +38,7 @@ files:
 - library/pulse/user.rb
 - library/pulse/conversation.rb
 - library/pulse/channel.rb
+- library/pulse/dcc.rb
 - library/pulse/enhancements.rb
 - library/pulse.rb
 has_rdoc: true