require 'pubnub/error'
require 'pubnub/uuid'
require 'pubnub/formatter'
require 'pubnub/crypto'
require 'pubnub/configuration'
require 'pubnub/message'
require 'pubnub/event'
require 'pubnub/single_event'
require 'pubnub/subscribe_event/callbacks'
require 'pubnub/subscribe_event/formatter'
require 'pubnub/subscribe_event/heartbeat'
require 'pubnub/subscribe_event/adding'
require 'pubnub/subscribe_event/removing'
require 'pubnub/subscribe_event'
require 'pubnub/pam'
require 'pubnub/heart'
require 'pubnub/subscriber'
require 'pubnub/envelope'
require 'pubnub/error_envelope'
require 'pubnub/client/connections'
require 'pubnub/client/events'
require 'pubnub/client/paged_history'
require 'pubnub/client/helpers'
require 'pubnub/validators/common_validator'
require 'pubnub/validators/client'
require 'pubnub/validators/audit'
require 'pubnub/validators/channel_registration'
require 'pubnub/validators/grant'
require 'pubnub/validators/heartbeat'
require 'pubnub/validators/here_now'
require 'pubnub/validators/history'
require 'pubnub/validators/leave'
require 'pubnub/validators/presence'
require 'pubnub/validators/publish'
require 'pubnub/validators/revoke'
require 'pubnub/validators/set_state'
require 'pubnub/validators/state'
require 'pubnub/validators/subscribe'
require 'pubnub/validators/time'
require 'pubnub/validators/where_now'
require 'pubnub/events/audit'
require 'pubnub/events/channel_registration'
require 'pubnub/events/grant'
require 'pubnub/events/heartbeat'
require 'pubnub/events/here_now'
require 'pubnub/events/history'
require 'pubnub/events/leave'
require 'pubnub/events/presence'
require 'pubnub/events/publish'
require 'pubnub/events/revoke'
require 'pubnub/events/set_state'
require 'pubnub/events/state'
require 'pubnub/events/subscribe'
require 'pubnub/events/time'
require 'pubnub/events/where_now'
# Toplevel Pubnub module.
module Pubnub
# Pubnub client Class.
class Client
include Configuration
include Connections
include Events
include PagedHistory
include Helpers
attr_reader :env, :subscriber
VERSION = Pubnub::VERSION
# Parameters:
# ===========
#
#
# - subscribe_key
# - required. Your subscribe key.
#
# - publish_key
# - optional. Your publish key, without it you can't push messages.
#
# - secret_key
# - optional. Your secret key, required for PAM operations.
#
# - auth_key
# - optional. This client auth key.
#
# - cipher_key
# - optional. Required to encrypt messages.
#
# - uuid
# - optional. Sets given uuid as client uuid, does not generates random uuid on init as usually.
#
# - origin
# - optional. Specifies the fully qualified domain name of the PubNub origin.
# By default this value is set to
pubsub.pubnub.com
but it should be set to the appropriate origin
# specified in the PubNub Admin Portal.
#
# - callback
# - optional. Default callback function for all events if not overwrote while firing event.
#
# - error_callback
# - optional. Default error callback.
#
# - connect_callback
# - optional. Callback that is called when connection with origin is established.
# It should take string as parameter.
#
# - disconnect_callback
# - optional. Callback that is called when connection with origin is teared down
# (it's called each time when reestablishing connection didn't work).
# It should take string as parameter.
#
# - reconnect_callback
# - optional. Callback that is called when connection with origin is reestablished.
# It should take string as parameter.
#
# - ssl
# - optional. Your connection will use ssl if set to true.
#
# - heartbeat
# - optional. Heartbeat interval, if not set heartbeat will not be running.
#
# - subscribe_timeout
# - optional, be careful when modifying this. Timeout for subscribe connection in seconds.
#
# - non_subscribe_timeout
# - optional, be careful when modifying this. Timeout for non-subscribe connection in seconds.
#
# - max_retries
# - optional. How many times client should try to reestablish connection before fail.
#
# - ttl
# - optional. Default ttl for grant and revoke events.
#
# _examples:_
# ```ruby
# # Minimal initialize
# pubnub = Pubnub.new(subscribe_key: :my_sub_key)
# ````
#
# ```ruby
# # More complex initialize
# pubnub = Pubnub.new(
# subscribe_key: :demo,
# publish_key: :demo,
# secret_key: :secret,
# cipher_key: :other_secret,
# uuid: :mad_max,
# origin: 'custom.pubnub.com',
# callback: -> (envelope) { puts envelope.message },
# error_callback: -> (envelope) { puts envelope.message },
# connect_callback: -> (message) { puts message },
# heartbeat: 60,
# subscribe_timeout: 310,
# non_subscribe_timeout: 10,
# max_retries: 10,
# ttl: 0
# )
# ```
# Returns:
# ========
#
# Initialized Pubnub::Client ready to use.
#
def initialize(options)
env_hash = symbolize_options_keys(options)
setup_app env_hash
clean_env
prepare_env
validate! @env
Pubnub.logger.info('Pubnub::Client') do
"Created new Pubnub::Client instance. Version: #{Pubnub::VERSION}"
end
end
# Returns:
# ========
# True if client is subscribed to at least one channel or channel group, otherwise false.
def subscribed?
if @env[:subscription_pool].empty?
false
else
!@env[:subscription_pool].map(&:empty?).index(false)
end
end
# Parameters:
# ===========
#
# - origin
# - Domain name where connection should be connected.
#
# - event_type
# - Keyword. :subscribe_event or :single_event.
#
# - sync
# - Boolean. True if we want dispatcher for sync or sync event, otherwise false.
#
#
# Returns:
# ========
# Appropriate RequestDispatcher.
#
# It returns always new RequestDispatcher for sync events.
# For async events it checks if there's already RequestDispatcher
# created and returns it if created, otherwise creates it, assigns
# it in @env and returns newly created dispatcher.
def request_dispatcher(origin, event_type, sync)
Pubnub.logger.debug('Pubnub::Client') do
"Looking for requester for #{event_type}"
end
if sync
@env[:req_dispatchers_pool][:sync][origin] ||= {}
@env[:req_dispatchers_pool][:sync][origin][event_type] ||=
setup_httpclient(event_type)
else
@env[:req_dispatchers_pool][:async][origin] ||= {}
@env[:req_dispatchers_pool][:async][origin][event_type] ||=
setup_httpclient(event_type)
end
end
# Parameters:
# ===========
#
# - origin
# - Domain name where connection should be connected.
#
# - event_type
# - Keyword. :subscribe_event or :single_event.
#
#
# Functionality:
# ==============
# Terminates request dispatcher for given origin and event type. Usable while restarting subscription.
def kill_request_dispatcher(origin, event_type)
Pubnub.logger.debug('Pubnub::Client') { 'Killing requester' }
# @env[:req_dispatchers_pool][origin][event_type].async.terminate
@env[:req_dispatchers_pool][:async][origin][event_type].reset_all
@env[:req_dispatchers_pool][:async][origin][event_type] = nil
rescue
Pubnub.logger.debug('Pubnub::Client') { 'There\'s no requester' }
end
# Parameters:
# ===========
#
# - uuid
# - New uuid to be set.
#
#
# Returns:
# ========
# New uuid.
#
# Functionality:
# ==============
# Can't change uuid while subscribed. You have to leave every subscribed channel.
def change_uuid(uuid)
Pubnub.logger.debug('Pubnub::Client') { 'Changing uuid' }
if subscribed?
fail('Cannot change UUID while subscribed.')
else
@env[:uuid] = uuid
end
end
alias_method :session_uuid=, :change_uuid
alias_method :uuid=, :change_uuid
alias_method :set_uuid=, :change_uuid
# Returns:
# ========
# Current origin.
def current_origin
@env[:origins_pool].first
end
alias_method :origin, :current_origin
# Returns:
# ========
# Current client timetoken
def timetoken
@env[:timetoken]
end
# Parameters:
# ===========
#
# - timetoken
# - New timetoken.
#
# Returns:
# ========
# New timetoken.
def timetoken=(timetoken)
@env[:timetoken] = timetoken
end
# Returns:
# ========
# Current uuid.
def uuid
@env[:uuid]
end
# Returns:
# ========
# Array of all current events.
def events
@env[:events]
end
def heartbeat
@env[:heartbeat]
end
def heartbeat=(value)
@env[:heartbeat] = value
end
def apply_state(event)
Pubnub.logger.debug('Pubnub::Client') { 'Apply state' }
create_state_pools(event)
return unless event.state
event.channel.each do |channel|
@env[:state][event.origin][:channel][channel] = event.state
end
event.group.each do |group|
@env[:state][event.origin][:group][group] = event.state
end
end
def empty_state?
return true unless @env[:state]
totally_empty @env[:state]
end
private
def create_state_pools(event)
@env[:state] ||= {}
@env[:state][event.origin] ||= {}
@env[:state][event.origin][:channel] ||= {}
@env[:state][event.origin][:group] ||= {}
end
def setup_httpclient(event_type)
if ENV['HTTP_PROXY']
hc = HTTPClient.new(ENV['HTTP_PROXY'])
else
hc = HTTPClient.new
end
case event_type
when :subscribe_event
hc.receive_timeout = 310
when :single_event
hc.receive_timeout = 5
end
hc
end
def validate!(env)
Validator::Client.validate! env
end
def setup_app(options)
Pubnub.logger = options[:logger] || Logger.new('pubnub.log')
Celluloid.logger = Pubnub.logger
@subscriber = Subscriber.new(self)
@env = options
end
def prepare_env
assign_defaults
generate_uuid if @env[:uuid].blank?
setup_pools
end
def generate_uuid
@env[:uuid] = UUID.generate
end
def assign_defaults
default_values.each do |k, v|
@env[k] = v unless @env[k]
end
@env[:timetoken] = 0
end
def symbolize_options_keys(options)
symbolized_options = {}
options.each_key do |k|
symbolized_options.merge!(k.to_sym => options[k])
end
symbolized_options
end
end
end