nyoibo 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,6 @@
+source "http://rubygems.org"
+
+gem "shoulda-matchers", :git => "https://github.com/thoughtbot/shoulda-matchers.git"
+gem "shoulda-context",  :git => "https://github.com/thoughtbot/shoulda-context.git"
+gem "turn",             :git => "https://github.com/yalab/turn.git"
+gemspec

data/MIT-LICENSE

@@ -0,0 +1,20 @@
+Copyright 2011 YOURNAME
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/Rakefile

@@ -0,0 +1,11 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks
+require 'rake/testtask'
+desc "Default Task"
+Rake::TestTask.new do |t|
+  files = FileList['test/test_helper.rb', 'test/**/*test.rb']
+  t.test_files = files
+  t.libs << "test"
+#  t.warning = true
+end
+task :default => :test

data/app/helpers/nyoibo_helper.rb

@@ -0,0 +1,15 @@
+module NyoiboHelper
+  def ws_form_for(model, options={}, &block)
+    options[:html] ||= {}
+    options[:html][:id] ||= 'ws-form'
+    output = ''
+    output << javascript_tag(<<EOS)
+  WEB_SOCKET_DEBUG = true;
+  WEB_SOCKET_SWF_LOCATION = '/WebSocketMain.swf';
+  var nyoibo =  #{I18n.t('nyoibo').to_json};
+EOS
+    output << form_for(model, options, &block)
+    output << content_tag(:div, '', :id => 'ws-progress')
+    output.html_safe
+  end
+end

data/lib/generators/nyoibo/install_generator.rb

@@ -0,0 +1,20 @@
+require 'rails'
+require 'rails/generators'
+module Nyoibo
+  module Generators
+    class InstallGenerator < Rails::Generators::Base
+      source_root File.expand_path("../templates", __FILE__)
+
+      desc "Creates a Nyoibo initializer and copy javascript and locale files to your application."
+      def copy_initializer
+        template "nyoibo.rb.erb", "config/initializers/nyoibo.rb"
+        template "nyoibo_en.yml", "config/locales/nyoibo_en.yml"
+        template "nyoibo.js.coffee", "app/assets/javascripts/nyoibo.js.coffee"
+        template "../../../../vendor/html5jp/progress.js", "app/assets/javascripts/progress.js"
+        template "../../../../vendor/web-socket-js/web_socket.js", "app/assets/javascripts/web_socket.js"
+        template "../../../../vendor/web-socket-js/swfobject.js", "app/assets/javascripts/swfobject.js"
+        template "../../../../vendor/web-socket-js/WebSocketMain.swf", "public/WebSocketMain.swf"
+      end
+    end
+  end
+end

data/lib/generators/nyoibo/templates/nyoibo.js.coffee

@@ -0,0 +1,65 @@
+jQuery ($)->
+  progress = new html5jp.progress("ws-progress")
+  progress.draw()
+
+  form = $('#ws-form')
+  progress_bar = $("#ws-progress")
+
+  fire = (obj, name, data) ->
+    event = $.Event(name)
+    obj.trigger(event, data)
+    return event.result != false
+
+  form.bind 'ws:upload', (event) ->
+    file = form.find('input[type=file]').get(0).files[0]
+    try
+      max_length = file.size
+    catch e
+      alert nyoibo.file_not_found
+      return
+    if max_length == 0
+      alert nyoibo.file_not_found
+      return
+    chunk = 102400
+    start = 0
+    ws = new WebSocket("ws://localhost:3030/")
+
+    fire(form, 'ws:before_upload')
+    ws.onclose = ->
+      progress.set_val(100)
+      fire(form, 'ws:after_upload')
+      progress.reset()
+      ws = null
+
+    ws.onmessage = (evt) ->
+      switch evt.data
+        when 'OK Ready'
+          params = {filename: file.name, comment: $('#post_comment').val(), size: max_length, session_string:  $('#session_string').val()}
+          ws.send("JSON: " + JSON.stringify(params))
+        when 'OK Bye'
+          ws.close()
+        when 'EMPTY'
+          ws.send("QUIT")
+        when 'NEXT'
+          val = Math.floor(start / max_length * 100)
+          progress.set_val(val)
+          stop = start + chunk - 1
+          if stop >= max_length
+            stop = max_length
+
+          blob = if typeof(file.mozSlice) == "function"
+                   file.mozSlice(start, stop)
+                 else if typeof(file.webkitSlice) == "function"
+                   file.webkitSlice(start, stop)
+          start = stop
+          reader = new FileReader()
+          reader.onloadend = (e) ->
+            ws.send(e.target.result)
+
+          setTimeout ->
+            reader.readAsBinaryString(blob)
+          , 300
+
+  $('#ws-form input[type=submit]').click ->
+    fire(form, 'ws:upload') if fire(form, 'ws:prepare_upload')
+    return false

data/lib/generators/nyoibo/templates/nyoibo.rb.erb

@@ -0,0 +1,4 @@
+Nyoibo.configure do
+  host "localhost"
+  port 3030
+end

data/lib/generators/nyoibo/templates/nyoibo_en.yml

@@ -0,0 +1,3 @@
+en:
+  nyoibo:
+    file_not_found: `File is not exists`

data/lib/nyoibo.rb

@@ -0,0 +1,17 @@
+require 'em-websocket'
+require 'json'
+require 'nyoibo/configure'
+require 'nyoibo/daemon'
+require 'nyoibo/callback'
+
+module Nyoibo
+  extend Configure
+  extend Daemon
+  extend Callback::Runner
+end
+
+if defined?(Rails)
+  require 'rails'
+  require 'nyoibo/rails'
+  require 'nyoibo/railtie'
+end

data/lib/nyoibo/callback.rb

@@ -0,0 +1,28 @@
+module Nyoibo
+  module Callback
+    def self.callbacks
+      @callbacks ||= {}
+    end
+    def self.included(base)
+      base.extend ClassMethod
+    end
+
+    module ClassMethod
+      def uploaded(path, &block)
+        if ENV["NYOIBO_ENV"] == "production" && Nyoibo::Callback.callbacks[path]
+          raise "Already defined '#{path}' updated callback."
+        end
+        Nyoibo::Callback.callbacks[path] = block
+
+        Process.kill(:TERM, Nyoibo.pid) && Nyoibo.run if ENV["NYOIBO_ENV"] == "development"
+      end
+    end
+
+    module Runner
+      def run_callback(path="/", *args)
+        block = Nyoibo::Callback.callbacks[path]
+        block.call(*args)
+      end
+    end
+  end
+end

data/lib/nyoibo/configure.rb

@@ -0,0 +1,19 @@
+module Nyoibo
+  module Configure
+    attr_accessor :config
+    def configure(&block)
+      @config = Config.new
+      @config.instance_exec(&block)
+    end
+  end
+
+  class Config
+    [:host, :port].each do |method|
+      module_eval <<-EOS, __FILE__, __LINE__
+        def #{method}(value = nil)
+          (value) ? (@#{method} = value) : @#{method}
+        end
+      EOS
+    end
+  end
+end

data/lib/nyoibo/daemon.rb

@@ -0,0 +1,58 @@
+module Nyoibo
+  module Daemon
+    attr_reader :pid
+    CMD_QUIT = /^QUIT$/
+    CMD_JSON = /^JSON: /
+    TYPE_BASE64 = %r|^data:application/octet-stream;base64,|
+    def run
+      return if defined?(IRB)
+      daemon = lambda{
+        EventMachine::WebSocket.start(:host => config.host, :port => config.port) do |ws|
+          ws.onopen{
+            ws.send "OK Ready"
+          }
+          ws.onclose{
+            ws.send "OK Bye"
+          }
+          ws.onmessage{|msg|
+            @binary ||= ""
+            @binary_type ||= 'binary'
+            case msg
+            when CMD_QUIT
+              if @binary_type == 'base64'
+                @binary = Base64.decode64(@binary)
+              elsif @binary.encoding == Encoding::UTF_8
+                @binary = @binary.unpack('U*').pack('c*')
+              end
+              Nyoibo.run_callback(ws.request["path"], @json, @binary)
+              ws.close_connection
+            when CMD_JSON
+              msg.gsub!(CMD_JSON, '')
+              @json = JSON.parse(msg)
+              @json['size'] = @json['size'].to_i
+              ws.send("NEXT")
+            else
+
+              if msg =~ TYPE_BASE64
+                msg.gsub!(TYPE_BASE64, '')
+                @binary_type = 'base64'
+              end
+
+              if msg.length > 0
+                @binary << msg
+                ws.send("NEXT")
+              else
+                ws.send("EMPTY")
+              end
+            end
+          }
+        end
+      }
+      if ENV["NYOIBO_ENV"] == "test"
+        daemon.call
+      else
+        @pid = Process.fork &daemon
+      end
+    end
+  end
+end

data/lib/nyoibo/rails.rb

@@ -0,0 +1,4 @@
+module Nyoibo
+  class Rails < Rails::Engine
+  end
+end

data/lib/nyoibo/railtie.rb

@@ -0,0 +1,8 @@
+require "nyoibo"
+
+class Nyoibo::Railtie < Rails::Railtie
+  config.after_initialize do
+    ENV["NYOIBO_ENV"] = Rails.env
+    Nyoibo.run
+  end
+end

data/lib/nyoibo/version.rb

@@ -0,0 +1,3 @@
+module Nyoibo
+  VERSION = "0.0.1"
+end

data/nyoibo.gemspec

@@ -0,0 +1,26 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "nyoibo/version"
+
+Gem::Specification.new do |s|
+  s.name        = "nyoibo"
+  s.version     = Nyoibo::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["yalab"]
+  s.email       = ["rudeboyjet@gmail.com"]
+  s.homepage    = ""
+  s.summary     = %q{Websocket uploader with progressbar.}
+  s.description = %q{}
+  s.license     = 'MIT'
+
+  s.rubyforge_project = "nyoibo"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+  s.add_dependency("em-websocket", "~>0.3.0")
+  s.add_dependency("json", "~>1.5.3")
+  s.add_development_dependency('em-http-request', '~> 0.2.6')
+  s.add_development_dependency('railties')
+end

data/test/test_helper.rb

@@ -0,0 +1,19 @@
+require 'test/unit'
+require 'socket'
+require 'bundler'
+ENV['BUNDLE_GEMFILE'] = File.expand_path("../../Gemfile", __FILE__)
+require 'bundler/setup'
+require 'shoulda/context'
+require 'em-http'
+require 'turn'
+
+Dir.glob(File.expand_path("../../lib/**/*.rb", __FILE__)).each {|f| require f }
+
+ENV["NYOIBO_ENV"] = "test"
+
+Test::Unit::TestCase.module_eval do
+  def failed
+    EventMachine.stop
+    fail
+  end
+end

data/test/unit/callback_test.rb

@@ -0,0 +1,17 @@
+require 'test_helper'
+class Nyoibo::CallbackTest < Test::Unit::TestCase
+  class TestApp
+    include Nyoibo::Callback
+    uploaded "/" do
+      "uploaded!"
+    end
+  end
+  context "Run callback" do
+    setup do
+      @response = Nyoibo.run_callback("/")
+    end
+    should "fetch response" do
+      assert_equal "uploaded!", @response
+    end
+  end
+end

data/test/unit/configure_test.rb

@@ -0,0 +1,15 @@
+require 'test_helper'
+class Nyoibo::ConfigureTest < Test::Unit::TestCase
+  context ".configure" do
+    setup do
+      Nyoibo.configure do
+        host 'localhost'
+        port 3030
+      end
+    end
+    should "reflect config" do
+      assert_equal 'localhost', Nyoibo.config.host
+      assert_equal 3030, Nyoibo.config.port
+    end
+  end
+end

data/test/unit/daemon_test.rb

@@ -0,0 +1,60 @@
+require 'test_helper'
+class Nyoibo::DaemonTest < Test::Unit::TestCase
+  class TestApp < Test::Unit::TestCase
+    include Nyoibo::Callback
+    uploaded "/" do |json, binary|
+      File.open("/tmp/test.jpg", "w:binary"){|f|
+        f.write(binary)
+      }
+      raise "finame is not 'test.jpg'" if json["filename"] != "test.jpg"
+      raise "Binary size is not '3137' byte" if binary.length != 3137
+    end
+  end
+
+  context "run" do
+    setup do
+      @file = File.new(File.expand_path("../../fixtures/test.jpg", __FILE__), "rb")
+      Nyoibo.configure do
+        host "localhost"
+        port 3030
+      end
+    end
+
+    should "listen port" do
+      EM.run do
+        EventMachine.add_timer(0.1) do
+          http = EventMachine::HttpRequest.new('ws://localhost:3030/').get(:timeout => 0)
+          http.errback { failed }
+          @start = 0
+          @sendsize = 1024
+          http.stream{|msg|
+            case  msg
+            when "OK Ready"
+              http.send("JSON: " + {:filename => File.basename(@file.path), :size => @file.size}.to_json)
+            when "NEXT"
+              @end = @start + @sendsize
+              @encoded ||= Base64.encode64(@file.read)
+
+              @end = @encoded.size - 1 if @end >= @encoded.size
+              if @start == @encoded.size
+                http.send("QUIT")
+              else
+                chunk = @encoded.slice(@start..@end)
+                if @start == 0
+                  chunk = "data:application/octet-stream;base64," + chunk
+                end
+
+                http.send(chunk)
+              end
+              @start = @end + 1
+            end
+          }
+          http.disconnect{
+            EventMachine.stop
+          }
+        end
+        Nyoibo.run
+      end
+    end
+  end
+end