require 'bundler/setup'
require 'celluloid_pubsub'
require 'logger'

debug_enabled = ENV['DEBUG'].present? && ENV['DEBUG'].to_s == 'true'

if debug_enabled == true
  log_file_path = File.join(File.expand_path(File.dirname(__FILE__)), 'log', 'celluloid_pubsub.log')
  puts log_file_path
  FileUtils.mkdir_p(File.dirname(log_file_path))
  log_file = File.open(log_file_path, 'w')
  log_file.sync = true
  Celluloid.logger = ::Logger.new(log_file_path)
end

# actor that subscribes to a channel
class Subscriber
  include Celluloid
  include Celluloid::Logger

  def initialize(options = {})
    @client = CelluloidPubsub::Client.connect({ actor: Actor.current }.merge(options)) do |ws|
      ws.subscribe('test_channel') # this will execute after the connection is opened
    end
  end

  def on_message(message)
    if @client.succesfull_subscription?(message)
      puts "subscriber got successful subscription #{message.inspect}"
      @client.publish('test_channel2', 'data' => ' subscriber got successfull subscription') # the message needs to be a Hash
    else
      puts "subscriber got message #{message.inspect}"
      @client.unsubscribe('test_channel')
    end
  end

  def on_close(code, reason)
    puts "websocket connection closed: #{code.inspect}, #{reason.inspect}"
    terminate
  end
end

# actor that publishes a message in a channel
class Publisher
  include Celluloid
  include Celluloid::Logger

  def initialize(options = {})
    @client = CelluloidPubsub::Client.connect({ actor: Actor.current }.merge(options)) do |ws|
      ws.subscribe('test_channel2') # this will execute after the connection is opened
    end
    @client.publish('test_channel', 'data' => 'my_message') # the message needs to be a Hash
  end

  def on_message(message)
    puts " publisher got #{message.inspect}"
    @client.unsubscribe('test_channel2')
  end

  def on_close(code, reason)
    puts "websocket connection closed: #{code.inspect}, #{reason.inspect}"
    terminate
  end
end

CelluloidPubsub::WebServer.supervise_as(:web_server, enable_debug: debug_enabled)
Subscriber.supervise_as(:subscriber, enable_debug: debug_enabled)
Publisher.supervise_as(:publisher, enable_debug: debug_enabled)
sleep