ws-io 1.0.1 → 1.1.0

data/VERSION

@@ -1 +1 @@
-1.0.1
+1.1.0

data/bin/ws-irb

@@ -0,0 +1,8 @@
+#!/usr/bin/env ruby
+$:.unshift File.dirname(__FILE__) + "/../lib"
+require "ws-io"
+require 'irb'
+
+WsIo.start {
+  IRB.start
+}.open.join

data/examples/sh.rb

@@ -1,14 +1,8 @@
 #!/usr/bin/env ruby
 $:.unshift File.dirname(__FILE__) + "/../lib"
+
 require "ws-io"
 
-threads = []
-threads << Thread.start do
-  WsIo.start(['*'], 8080) do
-    system '/bin/sh'
-  end
-end
-threads << Thread.start do
-  system 'open', File.expand_path('../index.html', __FILE__)
-end
-threads.each { |thread| thread.join }
+WsIo.start {
+  system '/bin/sh'
+}.open.join

data/{examples/index.html → lib/index.html.erb}

@@ -4,7 +4,7 @@
     <script type="text/javascript" src="http://ajax.googleapis.com/ajax/libs/jquery/1.5.0/jquery.min.js?v=1"></script>
     <script type="text/javascript" charset="utf-8">
       $(function() {
-        var ws = new WebSocket("ws://127.0.0.1:8080");
+        var ws = new WebSocket("ws://127.0.0.1:<%= port %>");
         ws.onmessage = function(e) {
           $("#result").append($("<div>").html(e.data));
           $("body").scrollTop(document.height);

data/lib/ws-io.rb

@@ -1,9 +1,12 @@
-require 'web_socket'
 require 'erb'
+require 'thread'
+require 'tempfile'
+require 'web_socket'
+require "launchy"
 
 class WsIo
   class << self
-    attr_accessor :ws
+    attr_accessor :ws, :domains, :port
 
     if ENV['WSIO_DEBUG']
       require 'g'
@@ -11,39 +14,40 @@ class WsIo
       def g(*args);end
     end
 
-    def start(domains = ["*"], port = 8080)
+    def start(port = 8080, domains = ["*"])
+      @port = port
+      @domains = domains
+
       fake_io
 
-      threads = []
+      m = Mutex.new
+      c = ConditionVariable.new
 
-      threads << Thread.start do
+      @server_thread = Thread.start do
         @server = WebSocketServer.new(:accepted_domains => domains, :port => port)
-        @server.run() do |ws|
-          if ws.path == "/"
-            ws.handshake()
-            WsIo.ws = ws
-            while data = ws.receive()
-              WsIo.input(data)
-            end
-          else
-            ws.handshake("404 Not Found")
-          end
-          stop_server
-        end
-      end
-
-      threads << Thread.start do
         begin
-          yield
-        ensure
-          unfake_io
-          stop_server
-        end
-      end
-
-      threads << Thread.start do
+          @server.run() do |ws|
+            if ws.path == "/"
+              ws.handshake()
+              WsIo.ws = ws
+              ws.send("connected")
+              c.signal                      # ####
+              while data = ws.receive()          #
+                WsIo.input(data)                 #
+              end                                #
+            else                                 #  ##     ## ##     ## ######## ######## ##     ## ####
+              ws.handshake("404 Not Found")      #  ###   ### ##     ##    ##    ##        ##   ##  ####
+            end                                  #  #### #### ##     ##    ##    ##         ## ##   ####
+            stop_server                          #  ## ### ## ##     ##    ##    ######      ###     ##
+          end                                    #  ##     ## ##     ##    ##    ##         ## ##
+        rescue => e                              #  ##     ## ##     ##    ##    ##        ##   ##  ####
+          g e                                    #  ##     ##  #######     ##    ######## ##     ## ####
+        end                                      #
+      end                                        #
+                                                 #
+      Thread.start do                            #
+        m.synchronize { c.wait(m) }         # <###
         loop do
-          break if @ws && @ws.instance_variable_get(:@socket).closed?
           if @ws
             begin
               @ws.send(escape(output))
@@ -54,13 +58,23 @@ class WsIo
         end
       end
 
-      threads.each do |thread|
-        thread.join
+      Thread.start do
+        begin
+          yield
+        rescue => e
+          g e
+        ensure
+          unfake_io
+          stop_server
+        end
       end
+
+      self
     rescue SignalException, StandardError => e
       g e
       unfake_io
       stop_server
+      raise
     rescue Exception => e
       g e
       unfake_io
@@ -68,6 +82,23 @@ class WsIo
       raise
     end
 
+    def after
+      yield
+      self
+    end
+
+    def join
+      @server_thread.join if @server_thread
+    end
+
+    def open
+      tempfile = Tempfile.open('ws-io')
+      tempfile << ERB.new(File.read(File.expand_path('../index.html.erb', __FILE__))).result(binding)
+      tempfile.flush
+      Launchy::Browser.run(tempfile.path)
+      self
+    end
+
     def fake_io
       @in_read, @in_write = IO.pipe
       @stdin = STDIN.clone
@@ -84,17 +115,20 @@ class WsIo
       STDIN.reopen(@stdin)
       STDOUT.reopen(@stdout)
       STDERR.reopen(@stderr)
-      @in_read.close
-      @in_write.close
-      @out_write.close
-      @out_read.close
+      [@in_read, @in_write, @out_read, @out_write].each do |io|
+        io.close unless io.closed?
+      end
       g 'unfake_io'
+    rescue => e
+      g e
     end
 
     def stop_server
       ws.close
       @server.tcp_server.close
       g 'stop_server'
+    rescue => e
+      g e
     end
 
     def stop

metadata

@@ -2,7 +2,7 @@
 name: ws-io
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 1.0.1
+  version: 1.1.0
 platform: ruby
 authors: 
 - jugyo
@@ -11,7 +11,7 @@ bindir: bin
 cert_chain: []
 
 date: 2011-02-10 00:00:00 +09:00
-default_executable: 
+default_executable: ws-irb
 dependencies: 
 - !ruby/object:Gem::Dependency 
   name: web-socket-ruby
@@ -70,8 +70,8 @@ dependencies:
   version_requirements: *id005
 description: It provides a bridge between WebSocket and IO
 email: jugyo.org@gmail.com
-executables: []
-
+executables: 
+- ws-irb
 extensions: []
 
 extra_rdoc_files: 
@@ -86,8 +86,9 @@ files:
 - README.rdoc
 - Rakefile
 - VERSION
-- examples/index.html
+- bin/ws-irb
 - examples/sh.rb
+- lib/index.html.erb
 - lib/ws-io.rb
 - spec/spec_helper.rb
 - spec/ws-io_spec.rb