peatio 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3697d323f43f4c9387766cae5aeb0a77acb29d47f933f84f95fa68ad95a4cf97
-  data.tar.gz: f345354fdaa1eb1660394c7d2373ced0d34aa80e09a336cf4541fd6ce58a5511
+  metadata.gz: e0654fc5cd21783e8e22ca88db407d01f1c268032f09b347dfed2f49ca3aae8e
+  data.tar.gz: 3346353884ce7fd78c47d8b6f8bad7f2a3d360a4366ba1b1f635e8eb750b87ba
 SHA512:
-  metadata.gz: fc2e62d12ef1bf3b58cbc378fe1af25dfc0fcd6a79270eff490bde22cf5deff4b9674d44e786be4fe943d699bb5ac16e062ba45a6d6b2a822a73c32213a6b182
-  data.tar.gz: 18d134f48a68fcbdfb2598d5dd7da95c844668ec1c550fcd9535e1193631f83f2ace108a96257cb392e22c4b9cf17d99d160f465f8eac3bf144900c3dce0d16f
+  metadata.gz: d5d538caf05569e94c3f812d5f15cfa49a5cbcfa0d23c1ccbc2f76a3c66fbb56c8c4f0e83e61dba96c1550bdc9b7a816bd7c3093e5b0e50b6fc0f5cce47f34e7
+  data.tar.gz: 9a5304ac15012c2488aadb4990bc62c331bb7e7f0f8cbd3f044ad3dd7ae1a0cc51f524f062b0abd46f7df31a8003486bcd70b0d0d2b9690547ecc4b8bfb0916e

data/.gitignore

@@ -6,6 +6,7 @@
 /pkg/
 /spec/reports/
 /tmp/
+/*.gem
 
 # rspec failure tracking
 .rspec_status

data/Gemfile.lock

@@ -3,9 +3,11 @@ PATH
   specs:
     peatio (0.2.0)
       amqp
+      bunny
       clamp
       em-websocket
       eventmachine
+      jwt
       mysql2
 
 GEM
@@ -16,49 +18,59 @@ GEM
       amq-protocol (>= 2.2.0)
       eventmachine
     ast (2.4.0)
+    bunny (2.11.0)
+      amq-protocol (~> 2.3.0)
+    bunny-mock (1.7.0)
+      bunny (>= 1.7)
     clamp (1.3.0)
     diff-lcs (1.3)
     docile (1.3.1)
+    em-spec (0.2.7)
+      eventmachine
     em-websocket (0.5.1)
       eventmachine (>= 0.12.9)
       http_parser.rb (~> 0.6.0)
+    em-websocket-client (0.1.2)
+      eventmachine
+      websocket
     eventmachine (1.2.7)
     http_parser.rb (0.6.0)
     jaro_winkler (1.5.1)
     json (2.1.0)
+    jwt (2.1.0)
     mysql2 (0.5.2)
     parallel (1.12.1)
-    parser (2.5.1.0)
+    parser (2.5.1.2)
       ast (~> 2.4.0)
     powerpack (0.1.2)
     rainbow (3.0.0)
     rake (10.5.0)
-    rspec (3.7.0)
-      rspec-core (~> 3.7.0)
-      rspec-expectations (~> 3.7.0)
-      rspec-mocks (~> 3.7.0)
-    rspec-core (3.7.1)
-      rspec-support (~> 3.7.0)
-    rspec-expectations (3.7.0)
+    rspec (3.8.0)
+      rspec-core (~> 3.8.0)
+      rspec-expectations (~> 3.8.0)
+      rspec-mocks (~> 3.8.0)
+    rspec-core (3.8.0)
+      rspec-support (~> 3.8.0)
+    rspec-expectations (3.8.1)
       diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.7.0)
-    rspec-mocks (3.7.0)
+      rspec-support (~> 3.8.0)
+    rspec-mocks (3.8.0)
       diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.7.0)
-    rspec-support (3.7.1)
+      rspec-support (~> 3.8.0)
+    rspec-support (3.8.0)
     rspec_junit_formatter (0.4.1)
       rspec-core (>= 2, < 4, != 2.12.0)
-    rubocop (0.57.2)
+    rubocop (0.58.2)
       jaro_winkler (~> 1.5.1)
       parallel (~> 1.10)
-      parser (>= 2.5)
+      parser (>= 2.5, != 2.5.1.1)
       powerpack (~> 0.1)
       rainbow (>= 2.2.2, < 4.0)
       ruby-progressbar (~> 1.7)
       unicode-display_width (~> 1.0, >= 1.0.1)
     rubocop-github (0.10.0)
       rubocop (~> 0.51)
-    ruby-progressbar (1.9.0)
+    ruby-progressbar (1.10.0)
     simplecov (0.16.1)
       docile (~> 1.1)
       json (>= 1.8, < 3)
@@ -68,12 +80,16 @@ GEM
       json
       simplecov
     unicode-display_width (1.4.0)
+    websocket (1.2.8)
 
 PLATFORMS
   ruby
 
 DEPENDENCIES
   bundler (~> 1.16)
+  bunny-mock
+  em-spec
+  em-websocket-client
   peatio!
   rake (~> 10.0)
   rspec (~> 3.0)

data/lib/peatio.rb

@@ -1,15 +1,19 @@
 require "logger"
 require "json"
 require "mysql2"
-require "amqp"
+require "bunny"
 require "eventmachine"
 require "em-websocket"
 
-require "peatio/logger"
-require "peatio/version"
-require "peatio/sql/client"
-require "peatio/sql/schema"
-require "peatio/mq/client"
-require "peatio/mq/events"
-require "peatio/ranger"
-require "peatio/injectors/peatio_events"
+module Peatio
+  require_relative "peatio/error"
+  require_relative "peatio/logger"
+  require_relative "peatio/version"
+  require_relative "peatio/sql/client"
+  require_relative "peatio/sql/schema"
+  require_relative "peatio/mq/client"
+  require_relative "peatio/mq/events"
+  require_relative "peatio/ranger"
+  require_relative "peatio/injectors/peatio_events"
+  require_relative "peatio/auth/jwt_authenticator"
+end

data/lib/peatio/auth/error.rb

@@ -0,0 +1,19 @@
+module Peatio::Auth
+  # Error repesent all errors that can be returned from Auth module.
+  class Error < Peatio::Error
+    # @return [String, JWT::*] Reason store underlying reason for given error.
+    #
+    # @see https://github.com/jwt/ruby-jwt/blob/master/lib/jwt/error.rb List of JWT::* errors.
+    attr_reader :reason
+
+    def initialize(reason = nil)
+      @reason = reason
+
+      super(
+        code: 2001,
+        text: "Authorization failed: #{reason}",
+        status: 401,
+      )
+    end
+  end
+end

data/lib/peatio/auth/jwt_authenticator.rb

@@ -0,0 +1,127 @@
+require "jwt"
+
+require_relative "error"
+
+module Peatio::Auth
+  # JWTAuthenticator used to authenticate user using JWT.
+  #
+  # It allows configuration of JWT verification through following ENV
+  # variables (all optional):
+  # * JWT_ISSUER
+  # * JWT_AUDIENCE
+  # * JWT_ALGORITHM (default: RS256)
+  # * JWT_DEFAULT_LEEWAY
+  # * JWT_ISSUED_AT_LEEWAY
+  # * JWT_EXPIRATION_LEEWAY
+  # * JWT_NOT_BEFORE_LEEWAY
+  #
+  # @see https://github.com/jwt/ruby-jwt JWT validation parameters.
+  #
+  # @example Token validation
+  #   rsa_private = OpenSSL::PKey::RSA.generate(2048)
+  #   rsa_public = rsa_private.public_key
+  #
+  #   payload = {
+  #     iat: Time.now.to_i,
+  #     exp: (Time.now + 60).to_i,
+  #     sub: "session",
+  #     iss: "barong",
+  #     aud: [
+  #       "peatio",
+  #       "barong",
+  #     ],
+  #     jti: "BEF5617B7B2762DDE61702F5",
+  #     uid: "TEST123",
+  #     email: "user@example.com",
+  #     role: "admin",
+  #     level: 4,
+  #     state: "active",
+  #   }
+  #
+  #   token = JWT.encode(payload, rsa_private, "RS256")
+  #
+  #   auth = Peatio::Auth::JWTAuthenticator.new(rsa_public)
+  #   auth.authenticate!("Bearer #{token}")
+  class JWTAuthenticator
+    @@verify_options = {
+      verify_expiration: true,
+      verify_not_before: true,
+      iss: ENV["JWT_ISSUER"],
+      verify_iss: !ENV["JWT_ISSUER"].nil?,
+      verify_iat: true,
+      verify_jti: true,
+      aud: ENV["JWT_AUDIENCE"].to_s.split(",").reject(&:empty?),
+      verify_aud: !ENV["JWT_AUDIENCE"].nil?,
+      sub: "session",
+      verify_sub: true,
+      algorithms: [ENV["JWT_ALGORITHM"] || "RS256"],
+      leeway: ENV["JWT_DEFAULT_LEEWAY"].yield_self { |n| n.to_i unless n.nil? },
+      iat_leeway: ENV["JWT_ISSUED_AT_LEEWAY"].yield_self { |n| n.to_i unless n.nil? },
+      exp_leeway: ENV["JWT_EXPIRATION_LEEWAY"].yield_self { |n| n.to_i unless n.nil? },
+      nbf_leeway: ENV["JWT_NOT_BEFORE_LEEWAY"].yield_self { |n| n.to_i unless n.nil? },
+    }.compact
+
+    @@encode_options = {
+      algorithm: @@verify_options[:algorithms].first,
+    }.compact
+
+    # Creates new authenticator with given public key.
+    #
+    # @param public_key [OpenSSL::PKey::PKey] Public key object to verify
+    #   signature.
+    # @param private_key [OpenSSL::PKey::PKey] Optional private key that used
+    #   only to encode new tokens.
+    def initialize(public_key, private_key = nil)
+      @public_key = public_key
+      @private_key = private_key
+    end
+
+    # Decodes and verifies JWT.
+    # Returns payload from JWT or raises an exception
+    #
+    # @param token [String] Token string. Must start from <tt>"Bearer "</tt>.
+    # @return [Hash] Payload Hash from JWT without any changes.
+    #
+    # @raise [Peatio::Auth::Error] If token is invalid or can't be verified.
+    def authenticate!(token)
+      token_type, token_value = token.to_s.split(" ")
+
+      unless token_type == "Bearer"
+        raise(Peatio::Auth::Error, "Token type is not provided or invalid.")
+      end
+
+      decode_and_verify_token(token_value)
+    rescue => error
+      case error
+      when Peatio::Auth::Error
+        raise(error)
+      else
+        raise(Peatio::Auth::Error, e.inspect)
+      end
+    end
+
+    # Encodes given payload and produces JWT.
+    #
+    # @param payload [String, Hash] Payload to encode.
+    # @return [String] JWT token string.
+    #
+    # @raise [ArgumentError] If no private key was passed to constructor.
+    def encode(payload)
+      raise(::ArgumentError, "No private key given.") if @private_key.nil?
+
+      JWT.encode(payload, @private_key, @@encode_options[:algorithm])
+    end
+
+    private
+
+    def decode_and_verify_token(token)
+      payload, header = JWT.decode(token, @public_key, true, @@verify_options)
+
+      payload.keys.each { |k| payload[k.to_sym] = payload.delete(k) }
+
+      payload
+    rescue JWT::DecodeError => e
+      raise(Peatio::Auth::Error, "Failed to decode and verify JWT: #{e.inspect}.")
+    end
+  end
+end

data/lib/peatio/command/service.rb

@@ -1,9 +1,19 @@
 module Peatio::Command::Service
-
   class Start < Peatio::Command::Base
     class Ranger < Peatio::Command::Base
       def execute
-        ::Peatio::Ranger.run!
+        if ENV["JWT_PUBLIC_KEY"].nil?
+          raise ArgumentError, "JWT_PUBLIC_KEY was not specified."
+        end
+
+        key_decoded = Base64.urlsafe_decode64(ENV["JWT_PUBLIC_KEY"])
+
+        jwt_public_key = OpenSSL::PKey.read(key_decoded)
+        if jwt_public_key.private?
+          raise ArgumentError, "JWT_PUBLIC_KEY was set to private key, however it should be public."
+        end
+
+        ::Peatio::Ranger.run!(jwt_public_key)
       end
     end
 
@@ -13,5 +23,4 @@ module Peatio::Command::Service
   class Root < Peatio::Command::Base
     subcommand "start", "Start a service", Start
   end
-
 end

data/lib/peatio/error.rb

@@ -0,0 +1,27 @@
+class Peatio::Error < ::StandardError
+  @@default_code = 2000
+  @@default_status = 400
+
+  attr :code, :text
+
+  def initialize(opts = {})
+    @code = opts[:code] || @@default_code
+    @text = opts[:text] || ""
+
+    @status = opts[:status] || @@default_status
+    @message = {error: {code: @code, message: @text}}
+
+    if @text != ""
+      super("#{@code}: #{text}")
+    else
+      super("#{@code}")
+    end
+  end
+
+  def inspect
+    message = @text
+    message += " (#{@reason})" unless @reason.nil?
+
+    %[#<#{self.class.name}: #{message}>]
+  end
+end

data/lib/peatio/injectors/peatio_events.rb

@@ -13,21 +13,18 @@ module Peatio::Injectors
 
       EventMachine.run do
         Peatio::MQ::Client.new
-        AMQP::Exchange.new(Peatio::MQ::Client.channel, :topic, @exchange_name) do |exchange, declare_ok|
-          logger.info "Exchange #{exchange.name} is ready to go"
-          next_message(exchange)
-        end
+        Peatio::MQ::Client.connect!
+        Peatio::MQ::Client.create_channel!
+        inject_message
       end
     end
 
-    def next_message(exchange)
+    def inject_message
       if message = @messages.shift
-        event_name, data = message
-        serialized_data = JSON.dump(data)
-        exchange.publish(serialized_data, routing_key: event_name) do
-          logger.debug { "event #{event_name} sent with data: #{serialized_data}" }
-          next_message(exchange)
-        end
+        type, id, event, data = message
+        Peatio::MQ::Events.publish(type, id, event, data) {
+          inject_message
+        }
       else
         Peatio::MQ::Client.disconnect { EventMachine.stop }
       end
@@ -35,6 +32,7 @@ module Peatio::Injectors
 
     def create_messages
       [
+        private_trade,
         order_created,
         order_canceled,
         order_completed,
@@ -50,141 +48,162 @@ module Peatio::Injectors
     def updated_at
       Time.now
     end
+
     alias :completed_at :updated_at
-    alias :canceled_at  :updated_at
+    alias :canceled_at :updated_at
+
+    def private_trade
+      [
+        "private",
+        "debug_user",
+        "trade",
+        {
+          trade: "some-data",
+        },
+      ]
+    end
 
     def order_created
       [
-        "#{market}.order_created",
+        "public",
+        market,
+        "order_created",
         {
-          market:                 "#{market}",
-          type:                   "buy",
-          trader_uid:             buyer_uid,
-          income_unit:            "btc",
-          income_fee_type:        "relative",
-          income_fee_value:       "0.0015",
-          outcome_unit:           "usd",
-          outcome_fee_type:       "relative",
-          outcome_fee_value:      "0.0",
-          initial_income_amount:  "14.0",
-          current_income_amount:  "14.0",
+          market: "#{market}",
+          type: "buy",
+          trader_uid: buyer_uid,
+          income_unit: "btc",
+          income_fee_type: "relative",
+          income_fee_value: "0.0015",
+          outcome_unit: "usd",
+          outcome_fee_type: "relative",
+          outcome_fee_value: "0.0",
+          initial_income_amount: "14.0",
+          current_income_amount: "14.0",
           initial_outcome_amount: "0.42",
           current_outcome_amount: "0.42",
-          strategy:               "limit",
-          price:                  "0.03",
-          state:                  "open",
-          trades_count:           0,
-          created_at:             created_at.iso8601
-        }
+          strategy: "limit",
+          price: "0.03",
+          state: "open",
+          trades_count: 0,
+          created_at: created_at.iso8601,
+        },
       ]
     end
 
     def order_canceled
       [
-        "#{market}.order_canceled",
+        "public",
+        market,
+        "order_canceled",
         {
-          market:                  "#{market}",
-          type:                    "sell",
-          trader_uid:              seller_uid,
-          income_unit:             "usd",
-          income_fee_type:         "relative",
-          income_fee_value:        "0.0015",
-          outcome_unit:            "btc",
-          outcome_fee_type:        "relative",
-          outcome_fee_value:       "0.0",
-          initial_income_amount:   "3.0",
-          current_income_amount:   "3.0",
-          initial_outcome_amount:  "100.0",
-          current_outcome_amount:  "100.0",
-          strategy:                "limit",
-          price:                   "0.03",
-          state:                   "canceled",
-          trades_count:            0,
-          created_at:              created_at.iso8601,
-          canceled_at:             canceled_at.iso8601
-        }
+          market: "#{market}",
+          type: "sell",
+          trader_uid: seller_uid,
+          income_unit: "usd",
+          income_fee_type: "relative",
+          income_fee_value: "0.0015",
+          outcome_unit: "btc",
+          outcome_fee_type: "relative",
+          outcome_fee_value: "0.0",
+          initial_income_amount: "3.0",
+          current_income_amount: "3.0",
+          initial_outcome_amount: "100.0",
+          current_outcome_amount: "100.0",
+          strategy: "limit",
+          price: "0.03",
+          state: "canceled",
+          trades_count: 0,
+          created_at: created_at.iso8601,
+          canceled_at: canceled_at.iso8601,
+        },
       ]
     end
 
     def order_completed
       [
-        "#{market}.order_completed", {
-          market:                  "#{market}",
-          type:                    "sell",
-          trader_uid:              seller_uid,
-          income_unit:             "usd",
-          income_fee_type:         "relative",
-          income_fee_value:        "0.0015",
-          outcome_unit:            "btc",
-          outcome_fee_type:        "relative",
-          outcome_fee_value:       "0.0",
-          initial_income_amount:   "3.0",
-          current_income_amount:   "0.0",
-          previous_income_amount:  "3.0",
-          initial_outcome_amount:  "100.0",
-          current_outcome_amount:  "0.0",
+        "public",
+        market,
+        "order_completed", {
+          market: "#{market}",
+          type: "sell",
+          trader_uid: seller_uid,
+          income_unit: "usd",
+          income_fee_type: "relative",
+          income_fee_value: "0.0015",
+          outcome_unit: "btc",
+          outcome_fee_type: "relative",
+          outcome_fee_value: "0.0",
+          initial_income_amount: "3.0",
+          current_income_amount: "0.0",
+          previous_income_amount: "3.0",
+          initial_outcome_amount: "100.0",
+          current_outcome_amount: "0.0",
           previous_outcome_amount: "100.0",
-          strategy:                "limit",
-          price:                   "0.03",
-          state:                   "completed",
-          trades_count:            1,
-          created_at:              created_at.iso8601,
-          completed_at:            completed_at.iso8601
-        }
+          strategy: "limit",
+          price: "0.03",
+          state: "completed",
+          trades_count: 1,
+          created_at: created_at.iso8601,
+          completed_at: completed_at.iso8601,
+        },
       ]
     end
 
     def order_updated
       [
-        "#{market}.order_updated", {
-          market:                  "#{market}",
-          type:                    "sell",
-          trader_uid:              seller_uid,
-          income_unit:             "usd",
-          income_fee_type:         "relative",
-          income_fee_value:        "0.0015",
-          outcome_unit:            "btc",
-          outcome_fee_type:        "relative",
-          outcome_fee_value:       "0.0",
-          initial_income_amount:   "3.0",
-          current_income_amount:   "2.4",
-          previous_income_amount:  "3.0",
-          initial_outcome_amount:  "100.0",
-          current_outcome_amount:  "80.0",
+        "public",
+        market,
+        "order_updated", {
+          market: "#{market}",
+          type: "sell",
+          trader_uid: seller_uid,
+          income_unit: "usd",
+          income_fee_type: "relative",
+          income_fee_value: "0.0015",
+          outcome_unit: "btc",
+          outcome_fee_type: "relative",
+          outcome_fee_value: "0.0",
+          initial_income_amount: "3.0",
+          current_income_amount: "2.4",
+          previous_income_amount: "3.0",
+          initial_outcome_amount: "100.0",
+          current_outcome_amount: "80.0",
           previous_outcome_amount: "100.0",
-          strategy:                "limit",
-          price:                   "0.03",
-          state:                   "open",
-          trades_count:            1,
-          created_at:              created_at.iso8601,
-          updated_at:              updated_at.iso8601
-        }
+          strategy: "limit",
+          price: "0.03",
+          state: "open",
+          trades_count: 1,
+          created_at: created_at.iso8601,
+          updated_at: updated_at.iso8601,
+        },
       ]
     end
 
     def trade_completed
       [
-        "#{market}.trade_completed", {
-          market:                "#{market}",
-          price:                 "0.03",
-          buyer_uid:             buyer_uid,
-          buyer_income_unit:     "btc",
-          buyer_income_amount:   "14.0",
-          buyer_income_fee:      "0.021",
-          buyer_outcome_unit:    "usd",
-          buyer_outcome_amount:  "0.42",
-          buyer_outcome_fee:     "0.0",
-          seller_uid:            seller_uid,
-          seller_income_unit:    "usd",
-          seller_income_amount:  "0.42",
-          seller_income_fee:     "0.00063",
-          seller_outcome_unit:   "btc",
+        "public",
+        market,
+        "trade_completed", {
+          market: "#{market}",
+          price: "0.03",
+          buyer_uid: buyer_uid,
+          buyer_income_unit: "btc",
+          buyer_income_amount: "14.0",
+          buyer_income_fee: "0.021",
+          buyer_outcome_unit: "usd",
+          buyer_outcome_amount: "0.42",
+          buyer_outcome_fee: "0.0",
+          seller_uid: seller_uid,
+          seller_income_unit: "usd",
+          seller_income_amount: "0.42",
+          seller_income_fee: "0.00063",
+          seller_outcome_unit: "btc",
           seller_outcome_amount: "14.0",
-          seller_outcome_fee:    "0.0",
-          completed_at:          completed_at.iso8601
-        }
+          seller_outcome_fee: "0.0",
+          completed_at: completed_at.iso8601,
+        },
       ]
     end
-
   end
 end

data/lib/peatio/mq/client.rb

@@ -1,23 +1,29 @@
 module Peatio::MQ
   class Client
     class << self
-      attr_reader :channel, :connection
+      attr_accessor :channel, :connection
 
       def new
-        options = {
+        @options = {
           host: ENV["RABBITMQ_HOST"] || "0.0.0.0",
           port: ENV["RABBITMQ_PORT"] || "5672",
           username: ENV["RABBITMQ_USER"],
           password: ENV["RABBITMQ_PASSWORD"],
         }
-        @connection = AMQP.connect(options)
-        @channel = AMQP::Channel.new(@connection)
+      end
+
+      def connect!
+        @connection = Bunny.new(@options)
+        @connection.start
+      end
+
+      def create_channel!
+        @channel = @connection.create_channel
       end
 
       def disconnect
-        connection.close do
-          yield
-        end
+        @connection.close
+        yield if block_given?
       end
     end
   end

data/lib/peatio/mq/events.rb

@@ -1,26 +1,47 @@
+require "socket"
+
 module Peatio::MQ::Events
   def self.subscribe!
     ranger = RangerEvents.new
     ranger.subscribe
   end
 
-  def self.start!
+  def self.publish(type, id, event, payload)
+    @@client ||= begin
+      ranger = RangerEvents.new
+      ranger.connect!
+      ranger
+    end
+
+    @@client.publish(type, id, event, payload) do
+      yield if block_given?
+    end
   end
 
-  class SocketHandler
-    attr_accessor :event
+  class Client
+    attr_accessor :streams, :authorized, :user
 
     @@all = []
 
-    class << self
-      def all
-        @@all
+    def self.all
+      @@all
+    end
+
+    def self.user(user)
+      @@all.each do |handler|
+        if handler.user == user
+          yield handler
+        end
       end
     end
 
-    def initialize(socket, event)
+    def initialize(socket, streams)
       @socket = socket
-      @event = event
+      @streams = streams
+
+      @user = ""
+      @authorized = false
+
       @@all << self
     end
 
@@ -36,9 +57,22 @@ module Peatio::MQ::Events
       @exchange_name = "peatio.events.market"
     end
 
-    def subscribe
-      require "socket"
+    def connect!
+      @exchange = Peatio::MQ::Client.channel.topic(@exchange_name)
+    end
+
+    def publish(type, id, event, payload)
+      routing_key = [type, id, event].join(".")
+      serialized_data = JSON.dump(payload)
+
+      @exchange.publish(serialized_data, routing_key: routing_key)
+
+      Peatio::Logger::debug { "published event to #{routing_key} " }
 
+      yield if block_given?
+    end
+
+    def subscribe
       exchange = Peatio::MQ::Client.channel.topic(@exchange_name)
 
       suffix = "#{Socket.gethostname.split(/-/).last}#{Random.rand(10_000)}"
@@ -47,19 +81,42 @@ module Peatio::MQ::Events
 
       Peatio::MQ::Client.channel
         .queue(queue_name, durable: false, auto_delete: true)
-        .bind(exchange, routing_key: "#").subscribe do |metadata, payload|
+        .bind(exchange, routing_key: "#").subscribe do |delivery_info, metadata, payload|
+
+        # type@id@event
+        # type can be public|private
+        # id can be user id or market
+        # event can be anything like order_completed or just trade
 
-        Peatio::Logger.debug { "event received: #{payload}" }
+        routing_key = delivery_info.routing_key
+        if routing_key.count(".") != 2
+          Peatio::Logger::error do
+            "got invalid routing key from amqp: #{routing_key}"
+          end
+
+          next
+        end
 
-        event = metadata.routing_key
+        type, id, event = routing_key.split(".")
 
-        SocketHandler.all.each do |handler|
-          if event == handler.event
+        if type == "private"
+          Client.user(id) do |client|
+            if client.streams.include?(event)
+              client.send_payload payload
+            end
+          end
+
+          next
+        end
+
+        stream = [id, event].join(".")
+
+        Client.all.each do |handler|
+          if handler.streams.include?(stream)
             handler.send_payload payload
           end
         end
       end
     end
   end
-
 end

data/lib/peatio/ranger.rb

@@ -1,31 +1,106 @@
 module Peatio::Ranger
-  def run!
+  class Connection
+    def initialize(authenticator, socket, logger)
+      @authenticator = authenticator
+      @socket = socket
+      @logger = logger
+      @streams = []
+    end
+
+    def send(method, data)
+      payload = JSON.dump(method => data)
+      @logger.debug { payload }
+      @socket.send payload
+    end
+
+    def handle(msg)
+      authorized = false
+      begin
+        data = JSON.parse(msg)
+
+        token = data["jwt"]
+
+        payload = @authenticator.authenticate!(token)
+
+        authorized = true
+      rescue JSON::ParserError
+      rescue => error
+        @logger.error error.message
+      end
+
+      if !authorized
+        send :error, message: "Authentication failed."
+        return
+      end
+
+      @client.user = payload[:uid]
+      @client.authorized = true
+
+      @logger.info "ranger: user #{@client.user} authenticated #{@streams}"
+
+      send :success, message: "Authenticated."
+    end
+
+    def handshake(handshake)
+      query = URI::decode_www_form(handshake.query_string)
+
+      @streams = query.map do |item|
+        if item.first == "stream"
+          item.last
+        end
+      end
+
+      @logger.info "ranger: WebSocket connection openned, streams: #{@streams}"
+
+      @client = Peatio::MQ::Events::Client.new(
+        @socket, @streams,
+      )
+
+      @socket.instance_variable_set(:@connection_handler, @client)
+    end
+  end
+
+  def self.run!(jwt_public_key)
+    host = ENV["RANGER_HOST"] || "0.0.0.0"
+    port = ENV["RANGER_PORT"] || "8081"
+
+    authenticator = Peatio::Auth::JWTAuthenticator.new(jwt_public_key)
+
     logger = Peatio::Logger.logger
-    port = 8081
     logger.info "Starting the server on port #{port}"
 
     EM.run do
       Peatio::MQ::Client.new
+      Peatio::MQ::Client.connect!
+      Peatio::MQ::Client.create_channel!
+
       Peatio::MQ::Events.subscribe!
 
-      EM::WebSocket.start(host: "0.0.0.0", port: port) do |ws|
-        ws.onopen do |id|
-          logger.info "ranger: WebSocket connection openned"
+      EM::WebSocket.start(
+        host: host,
+        port: port,
+        secure: true,
+      ) do |socket|
+        connection = Connection.new(authenticator, socket, logger)
+
+        socket.onopen do |handshake|
+          connection.handshake(handshake)
+        end
 
-          ws.instance_variable_set(
-            :@connection_handler,
-            Peatio::MQ::Events::SocketHandler.new(ws, "eurusd.order_created")
-          )
+        socket.onmessage do |msg|
+          connection.handle(msg)
         end
 
-        ws.onclose { logger.info "ranger: WebSocket connection closed" }
+        socket.onclose do
+          logger.info "ranger: WebSocket connection closed"
+        end
 
-        ws.onerror { |e|
-          puts "ranger: WebSocket Error: #{e.message}"
-        }
+        socket.onerror do |e|
+          logger.error "ranger: WebSocket Error: #{e.message}"
+        end
       end
+
+      yield if block_given?
     end
   end
-
-  module_function :run!
 end

data/lib/peatio/version.rb

@@ -1,3 +1,3 @@
 module Peatio
-  VERSION = '0.2.0'
+  VERSION = "0.3.0"
 end

data/peatio.gemspec

@@ -4,23 +4,22 @@ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 require "peatio/version"
 
 Gem::Specification.new do |spec|
-  spec.name          = "peatio"
-  spec.version       = Peatio::VERSION
-  spec.authors       = ["Louis B.", "Camille M."]
-  spec.email         = ["lbellet@heliostech.fr"]
-
-  spec.summary       = %q{Peatio is a gem for running critical core services}
-  spec.description   = %q{Peatio gem contains microservices and command line tools}
-  spec.homepage      = "https://www.peatio.tech"
+  spec.name = "peatio"
+  spec.version = Peatio::VERSION
+  spec.authors = ["Louis B.", "Camille M."]
+  spec.email = ["lbellet@heliostech.fr"]
 
+  spec.summary = %q{Peatio is a gem for running critical core services}
+  spec.description = %q{Peatio gem contains microservices and command line tools}
+  spec.homepage = "https://www.peatio.tech"
 
   # Specify which files should be added to the gem when it is released.
   # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
-  spec.files         = Dir.chdir(File.expand_path("..", __FILE__)) do
+  spec.files = Dir.chdir(File.expand_path("..", __FILE__)) do
     `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
   end
-  spec.bindir        = "bin"
-  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.bindir = "bin"
+  spec.executables = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
   spec.require_paths = ["lib"]
 
   spec.add_dependency "clamp"
@@ -28,9 +27,14 @@ Gem::Specification.new do |spec|
   spec.add_dependency "eventmachine"
   spec.add_dependency "em-websocket"
   spec.add_dependency "mysql2"
+  spec.add_dependency "jwt"
+  spec.add_dependency "bunny"
   spec.add_development_dependency "bundler", "~> 1.16"
   spec.add_development_dependency "rake", "~> 10.0"
   spec.add_development_dependency "rspec", "~> 3.0"
+  spec.add_development_dependency "em-spec"
+  spec.add_development_dependency "em-websocket-client"
+  spec.add_development_dependency "bunny-mock"
   spec.add_development_dependency "simplecov"
   spec.add_development_dependency "simplecov-json"
   spec.add_development_dependency "rspec_junit_formatter"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: peatio
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Louis B.
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-08-09 00:00:00.000000000 Z
+date: 2018-08-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: clamp
@@ -81,6 +81,34 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: jwt
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bunny
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -123,6 +151,48 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: em-spec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: em-websocket-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bunny-mock
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: simplecov
   requirement: !ruby/object:Gem::Requirement
@@ -202,12 +272,15 @@ files:
 - bin/peatio
 - bin/setup
 - lib/peatio.rb
+- lib/peatio/auth/error.rb
+- lib/peatio/auth/jwt_authenticator.rb
 - lib/peatio/command/amqp.rb
 - lib/peatio/command/base.rb
 - lib/peatio/command/db.rb
 - lib/peatio/command/inject.rb
 - lib/peatio/command/root.rb
 - lib/peatio/command/service.rb
+- lib/peatio/error.rb
 - lib/peatio/executor.rb
 - lib/peatio/injectors/peatio_events.rb
 - lib/peatio/logger.rb