hkroger-websocket-rails 0.7.1

data/lib/websocket_rails/user_manager.rb

@@ -0,0 +1,276 @@
+module WebsocketRails
+
+  # Contains a Hash of all connected users. This
+  # can be used to trigger an event on a specific
+  # user from outside of a WebsocketRails controller.
+  #
+  # The key for a particular user is defined in the
+  # configuration as `config.user_identifier`.
+  #
+  # If there is a `current_user` method defined
+  # in ApplicationController and a user is signed
+  # in to your application when the connection is
+  # opened, WebsocketRails will call the method
+  # defined in `config.user_identifier` on the
+  # `current_user` object and use that value as
+  # the key.
+  #
+  #   # In your events.rb file
+  #   WebsocketRails.setup do |config|
+  #     # Defaults to :id
+  #     config.user_identifier = :name
+  #   end
+  #
+  #   # In a standard controller or background job
+  #   name = current_user.name
+  #   WebsocketRails.users[name].send_message :event_name, data
+  #
+  # If no `current_user` method is defined or the
+  # user is not signed in when the WebsocketRails
+  # connection is opened, the connection will not be
+  # stored in the UserManager.
+  def self.users
+    @user_manager ||= UserManager.new
+  end
+
+  class UserManager
+
+    attr_reader :users
+
+    def initialize
+      @users = {}
+    end
+
+    def [](identifier)
+      unless user = (@users[identifier.to_s] || find_remote_user(identifier.to_s))
+        user = MissingConnection.new(identifier.to_s)
+      end
+      user
+    end
+
+    def []=(identifier, connection)
+      @users[identifier.to_s] ||= LocalConnection.new
+      @users[identifier.to_s] << connection
+      Synchronization.register_user(connection) if WebsocketRails.synchronize?
+    end
+
+    def delete(connection)
+      identifier = connection.user_identifier.to_s
+
+      if (@users.has_key?(identifier) && @users[identifier].connections.count > 1)
+        @users[identifier].delete(connection)
+      else
+        @users.delete(identifier)
+        Synchronization.destroy_user(identifier) if WebsocketRails.synchronize?
+      end
+    end
+
+    # Behaves similarly to Ruby's Array#each, yielding each connection
+    # object stored in the {UserManager}. If synchronization is enabled,
+    # each connection from every active worker will be yielded.
+    #
+    # You can access the `current_user` object through the #user method.
+    #
+    # You can trigger an event on this user using the #send_message method
+    # which behaves identically to BaseController#send_message.
+    #
+    # If Synchronization is enabled, the state of the `current_user` object
+    # will be equivalent to it's state at the time the connection was opened.
+    # It will not reflect changes made after the connection has been opened.
+    def each(&block)
+      if WebsocketRails.synchronize?
+        users_hash = Synchronization.all_users || return
+        users_hash.each do |identifier, user_json|
+          connection = remote_connection_from_json(identifier, user_json)
+          block.call(connection) if block
+        end
+      else
+        users.each do |_, connection|
+          block.call(connection) if block
+        end
+      end
+    end
+
+    # Behaves similarly to Ruby's Array#map, invoking the given block with
+    # each active connection object and returning a new array with the results.
+    #
+    # See UserManager#each for details on the current usage and limitations.
+    def map(&block)
+      collection = []
+
+      each do |connection|
+        collection << block.call(connection) if block
+      end
+
+      collection
+    end
+
+    private
+
+    def find_remote_user(identifier)
+      return unless WebsocketRails.synchronize?
+      user_hash = Synchronization.find_user(identifier) || return
+
+      remote_connection identifier, user_hash
+    end
+
+    def remote_connection_from_json(identifier, user_json)
+      user_hash = JSON.parse(user_json)
+      remote_connection identifier, user_hash
+    end
+
+    def remote_connection(identifier, user_hash)
+      RemoteConnection.new identifier, user_hash
+    end
+
+    # The UserManager::LocalConnection Class serves as a proxy object
+    # for storing multiple connections that belong to the same
+    # user. It implements the same basic interface as a Connection.
+    # This allows you to work with the object as though it is a
+    # single connection, but still trigger the events on all
+    # active connections belonging to the user.
+    class LocalConnection
+
+      attr_reader :connections
+
+      def initialize
+        @connections = []
+      end
+
+      def <<(connection)
+        @connections << connection
+      end
+
+      def delete(connection)
+        @connections.delete(connection)
+      end
+
+      def connected?
+        true
+      end
+
+      def user_identifier
+        latest_connection.user_identifier
+      end
+
+      def user
+        latest_connection.user
+      end
+
+      def trigger(event)
+        connections.each do |connection|
+          connection.trigger event
+        end
+      end
+
+      def send_message(event_name, data = {}, options = {})
+        options.merge! :user_id => user_identifier
+        options[:data] = data
+
+        event = Event.new(event_name, options)
+
+        # Trigger the event on all active connections for this user.
+        connections.each do |connection|
+          connection.trigger event
+        end
+
+        # Still publish the event in case the user is connected to
+        # other workers as well.
+        Synchronization.publish event if WebsocketRails.synchronize?
+        true
+      end
+
+      private
+
+      def latest_connection
+        @connections.last
+      end
+
+    end
+
+    class RemoteConnection
+
+      attr_reader :user_identifier, :user
+
+      def initialize(identifier, user_hash)
+        @user_identifier = identifier.to_s
+        @user_hash = user_hash
+      end
+
+      def connected?
+        true
+      end
+
+      def user
+        @user ||= load_user
+      end
+
+      def send_message(event_name, data = {}, options = {})
+        options.merge! :user_id => @user_identifier
+        options[:data] = data
+
+        event = Event.new(event_name, options)
+
+        # If the user is connected to this worker, trigger the event
+        # immediately as the event will be ignored by the Synchronization
+        ## dispatcher since the server_token will match.
+        if connection = WebsocketRails.users.users[@user_identifier]
+          connection.trigger event
+        end
+
+        # Still publish the event in case the user is connected to
+        # other workers as well.
+        #
+        # No need to check for Synchronization being enabled here.
+        # If a RemoteConnection has been fetched, Synchronization
+        # must be enabled.
+        Synchronization.publish event
+        true
+      end
+
+      private
+
+      def load_user
+        user = WebsocketRails.config.user_class.new
+        set_user_attributes user, @user_hash
+        user
+      end
+
+      def set_user_attributes(user, attr)
+        attr.each do |k, v|
+          user.send "#{k}=", v
+        end
+        user.instance_variable_set(:@new_record, false)
+        user.instance_variable_set(:@destroyed, false)
+      end
+
+    end
+
+    class MissingConnection
+
+      attr_reader :identifier
+
+      def initialize(identifier)
+        @user_identifier = identifier.to_s
+      end
+
+      def connected?
+        false
+      end
+
+      def user
+        nil
+      end
+
+      def send_message(*args)
+        false
+      end
+
+      def nil?
+        true
+      end
+
+    end
+
+  end
+end

data/lib/websocket_rails/version.rb

@@ -0,0 +1,3 @@
+module WebsocketRails
+  VERSION = "0.7.1"
+end

data/spec/dummy/Rakefile

@@ -0,0 +1,7 @@
+# Add your own tasks in files placed in lib/tasks ending in .rake,
+# for example lib/tasks/capistrano.rake, and they will automatically be available to Rake.
+
+require File.expand_path('../config/application', __FILE__)
+require 'rake'
+
+Dummy::Application.load_tasks

data/spec/dummy/app/controllers/application_controller.rb

@@ -0,0 +1,3 @@
+class ApplicationController < ActionController::Base
+  protect_from_forgery
+end

data/spec/dummy/app/controllers/chat_controller.rb

@@ -0,0 +1,53 @@
+class ChatController < WebsocketRails::BaseController
+
+  before_action do
+    if message_counter > 10
+      self.message_counter = 0
+    end
+  end
+
+  before_action :only => :new_message do
+    true
+  end
+
+  attr_accessor :message_counter
+
+  def initialize
+    # perform application setup here
+    @message_counter = 0
+  end
+
+  def client_connected
+    # do something when a client connects
+  end
+
+  def error_occurred
+    # do something when an error occurs
+  end
+
+  def new_message
+    @message_counter += 1
+    broadcast_message :new_message, message
+  end
+
+  def new_user
+    controller_store[:user] = message
+    broadcast_user_list
+  end
+
+  def change_username
+    controller_store[:user] = message
+    broadcast_user_list
+  end
+
+  def delete_user
+    controller_store[:user] = nil
+    broadcast_user_list
+  end
+
+  def broadcast_user_list
+    users = ['user']
+    broadcast_message :user_list, users
+  end
+
+end

data/spec/dummy/app/helpers/application_helper.rb

@@ -0,0 +1,2 @@
+module ApplicationHelper
+end

data/spec/dummy/app/models/user.rb

@@ -0,0 +1,2 @@
+class User < ActiveRecord::Base
+end

data/spec/dummy/app/views/layouts/application.html.erb

@@ -0,0 +1,14 @@
+<!DOCTYPE html>
+<html>
+<head>
+  <title>Dummy</title>
+  <%= stylesheet_link_tag :all %>
+  <%= javascript_include_tag :defaults %>
+  <%= csrf_meta_tag %>
+</head>
+<body>
+
+<%= yield %>
+
+</body>
+</html>

data/spec/dummy/config.ru

@@ -0,0 +1,4 @@
+# This file is used by Rack-based servers to start the application.
+
+require ::File.expand_path('../config/environment',  __FILE__)
+run Dummy::Application

data/spec/dummy/config/application.rb

@@ -0,0 +1,45 @@
+require File.expand_path('../boot', __FILE__)
+
+require "active_model/railtie"
+require "active_record/railtie"
+require "action_controller/railtie"
+require "action_view/railtie"
+require "action_mailer/railtie"
+
+Bundler.require
+require "websocket-rails"
+
+module Dummy
+  class Application < Rails::Application
+    # Settings in config/environments/* take precedence over those specified here.
+    # Application configuration should go into files in config/initializers
+    # -- all .rb files in that directory are automatically loaded.
+
+    # Custom directories with classes and modules you want to be autoloadable.
+    # config.autoload_paths += %W(#{config.root}/extras)
+
+    # Only load the plugins named here, in the order given (default is alphabetical).
+    # :all can be used as a placeholder for all plugins not explicitly named.
+    # config.plugins = [ :exception_notification, :ssl_requirement, :all ]
+
+    # Activate observers that should always be running.
+    # config.active_record.observers = :cacher, :garbage_collector, :forum_observer
+
+    # Set Time.zone default to the specified zone and make Active Record auto-convert to this zone.
+    # Run "rake -D time" for a list of tasks for finding time zone names. Default is UTC.
+    # config.time_zone = 'Central Time (US & Canada)'
+
+    # The default locale is :en and all translations from config/locales/*.rb,yml are auto loaded.
+    # config.i18n.load_path += Dir[Rails.root.join('my', 'locales', '*.{rb,yml}').to_s]
+    # config.i18n.default_locale = :de
+
+    # JavaScript files you want as :defaults (application.js is always included).
+    # config.action_view.javascript_expansions[:defaults] = %w(jquery rails)
+
+    # Configure the default encoding used in templates for Ruby 1.9.
+    config.encoding = "utf-8"
+
+    # Configure sensitive parameters which will be filtered from the log file.
+    config.filter_parameters += [:password]
+  end
+end

data/spec/dummy/config/boot.rb

@@ -0,0 +1,10 @@
+require 'rubygems'
+gemfile = File.expand_path('../../../../Gemfile', __FILE__)
+
+if File.exist?(gemfile)
+  ENV['BUNDLE_GEMFILE'] = gemfile
+  require 'bundler'
+  Bundler.setup
+end
+
+$:.unshift File.expand_path('../../../../lib', __FILE__)

data/spec/dummy/config/database.yml

@@ -0,0 +1,22 @@
+# SQLite version 3.x
+#   gem install sqlite3
+development:
+  adapter: sqlite3
+  database: db/development.sqlite3
+  pool: 5
+  timeout: 5000
+
+# Warning: The database defined as "test" will be erased and
+# re-generated from your development database when you run "rake".
+# Do not set this db to the same as development or production.
+test:
+  adapter: sqlite3
+  database: db/test.sqlite3
+  pool: 5
+  timeout: 5000
+
+production:
+  adapter: sqlite3
+  database: db/production.sqlite3
+  pool: 5
+  timeout: 5000