rookout 0.1.15 → 0.1.20

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a79d1186aec6390fdce4b56f19a1a2586e91837e51e1e40d70b3290284f8ca0a
-  data.tar.gz: bc0400c6406c42355a9aec84ad0655c37ae7546464e5b80bbc9d3d8a7ae1c330
+  metadata.gz: 481ca24b7cd172e3416b29fee18905c6282d589db6cb316b2aa4d5adab26da1a
+  data.tar.gz: 5e391b43a1e7177ef5c020bc64d133fb9e62810bc4596301b2ed7298d18182d6
 SHA512:
-  metadata.gz: 181118c9440177b284e30570420d4b12bb5d9e5885605c2081695b07e128828460653fc74bfa8123cc0c87f1d5ec9d210eda05a553b64fd7cf19610cceb2ecb5
-  data.tar.gz: 52f5f5a5c62b617f8936c5ee5e9ff6b3a5c2bad83dce5ea912275dcc0a762e636dae84b37cfa78803208247cc5d560ae24002d67b74452e91430f3e9e8280568
+  metadata.gz: 30d63371644e94dba222b0e63650158d92f1de63e96d4731dd4eaa300c6a8e2d8220cd6f93fdcc0fdee5aaf390d3170bb52ebbcb7e82d2611228a7341c15faed
+  data.tar.gz: 5d0c24c4f132464d38b547d68f221983500fb76d7a6a4b438378abdb5d6f3bfb7f924e8784ac64b9da9642f7f84b2bf5c9facaa835a2f3d75d103d4e72df615c

data/lib/rookout/com_ws/pinger.rb

@@ -17,9 +17,14 @@ module Rookout
           if Time.now - @last_ping > @interval
             Logger.instance.debug "Sending Ping"
             @last_ping = Time.now
-            @connection.ping Time.now.to_s do
-              Logger.instance.debug "Got Ping reply"
-              @last_pong = Time.now
+            begin
+              @connection.ping Time.now.to_s do
+                Logger.instance.debug "Got Ping reply"
+                @last_pong = Time.now
+              end
+            rescue RuntimeError, Errno::EPIPE
+              Logger.instance.debug "Failed to send ping"
+              break
             end
           end
 

data/lib/rookout/com_ws/websocket_client.rb

@@ -2,6 +2,7 @@ module Rookout
   module ComWs
     require "openssl"
     require "websocket/driver"
+    require "uri"
 
     require_relative "../config"
     require_relative "../logger"
@@ -11,14 +12,12 @@ module Rookout
       def initialize url, proxy, token
         @token = token
         @connection = WebsocketConnection.new url, proxy
+        @proxy = proxy
         @driver = nil
-        @error = nil
-        @last_ping = nil
       end
 
       def connect
-        # TODO: add proxy support
-        @connection.connect
+        connection_error = nil
         @driver = WebSocket::Driver.client @connection
 
         headers.each do |key, value|
@@ -26,18 +25,20 @@ module Rookout
         end
 
         @driver.on :error do |error|
-          @error = error
+          connection_error = error
         end
 
         # Connect to the remote server
-        @driver.start
         # TODO: ADD CONNECT TIMEOUT
+        @connection.connect @driver
+        @driver.start
+
         while @driver.state == :connecting
           recv_data = @connection.read_char
           @driver.parse recv_data
         end
 
-        raise Exceptions::RookWebsocketException, @error if @driver.state != :open
+        raise Exceptions::RookWebsocketException, connection_error if @driver.state != :open
       end
 
       def connection_pump message_handler
@@ -62,7 +63,12 @@ module Rookout
       def close
         return if @driver.nil?
 
-        @driver.close
+        begin
+          @driver.close
+        rescue RuntimeError, Errno::EPIPE
+          # Protocol close may fail if the connection is already closed
+          nil
+        end
         @connection.close
       end
 
@@ -88,14 +94,36 @@ module Rookout
         @secure = %w[https wss].include? @uri.scheme
 
         @socket = nil
+
+        @proxy_connected = false
       end
 
-      def connect
-        if @secure
-          @socket = ssl_connect
-        else
-          @socket = tcp_connect
+      def connect driver
+        @socket = tcp_connect
+
+        proxy_connect driver if @proxy
+
+        @socket = ssl_connect @socket if @secure
+      end
+
+      def proxy_connect driver
+        connection_error = nil
+        @proxy_connected = false
+        proxy = driver.proxy @proxy
+
+        proxy.on :connect do
+          @proxy_connected = true
         end
+
+        proxy.on :error do |error|
+          connection_error = error
+        end
+
+        proxy.start
+
+        proxy.parse read_char until @proxy_connected == true || !connection_error.nil?
+
+        raise Exceptions::RookProxyException, connection_error unless connection_error.nil?
       end
 
       attr_reader :url
@@ -118,13 +146,16 @@ module Rookout
       private
 
       def tcp_connect
-        TCPSocket.new @uri.host,
-                      @uri.port || (@secure ? 443 : 80)
+        if @proxy
+          uri = URI(@proxy)
+          TCPSocket.new uri.host, uri.port
+        else
+          TCPSocket.new @uri.host,
+                        @uri.port || (@secure ? 443 : 80)
+        end
       end
 
-      def ssl_connect
-        tcp_socket = tcp_connect
-
+      def ssl_connect tcp_socket
         ctx = ::OpenSSL::SSL::SSLContext.new
         ctx.min_version = ::OpenSSL::SSL::TLS1_2_VERSION
         cert_store = ::OpenSSL::X509::Store.new

data/lib/rookout/commit.rb

@@ -1,3 +1,3 @@
 module Rookout
-  COMMIT = "c6e9d699d0dd8a6d17e8c466c18ccc082a29e5aa".freeze
+  COMMIT = "f63d232fa7d56db9e3a82bdbbe8a1eafaea49ab3".freeze
 end

data/lib/rookout/exceptions.rb

@@ -144,6 +144,12 @@ module Rookout
       end
     end
 
+    class RookInvalidLabel < ToolException
+      def initialize label_name
+        super "Invalid label: must not start with the '$' character (#{label_name})"
+      end
+    end
+
     class RookCrcMismatchException < ToolException
       def initialize filepath, expected, calculated
         super "Line CRC32s do not match! path: #{filepath}, expected: #{expected}, calculated:#{calculated}",
@@ -178,10 +184,22 @@ module Rookout
       end
     end
 
+    class RookProxyException < ToolException
+      def initialize error
+        super "Error from proxy #{error}", { "error" => error }
+      end
+    end
+
     class RookBadProtobuf < ToolException
       def initialize
         super 'Bad protobuf version. Please execute "bundle config force_ruby_platform true" before "bundler install".'
       end
     end
+
+    class RookBadProtobufPlatform < ToolException
+      def initialize platform
+        super "Bad protobuf platform: #{platform}"
+      end
+    end
   end
 end

data/lib/rookout/interface.rb

@@ -12,11 +12,18 @@ module Rookout
       @start_options = nil
     end
 
+    def print_debug_messages
+      puts "[Rookout] Running in debug mode"
+      puts "[Rookout] Rookout SDK for ruby: " + Config.rookout_version
+    end
+
     def start options = {}
       return unless @rook.nil?
       throw_errors = options[:throw_errors] == true
       Config.debug = evaluate_flag options[:debug], "ROOKOUT_DEBUG"
 
+      print_debug_messages if Config.debug
+
       begin
         verify_env
 
@@ -75,7 +82,11 @@ module Rookout
       return unless File.exist? "/etc/alpine-release"
 
       protobuf = Gem::Specification.find_by_path "google/protobuf"
-      raise RookBadProtobuf if protobuf.nil? || protobuf.platform != "ruby"
+      STDERR.puts RookBadProtobuf.new.message if protobuf.nil?
+      return unless protobuf.platform != "ruby"
+
+      error = RookBadProtobufPlatform.new protobuf.platform
+      STDERR.puts error.message
     end
 
     def configure_globals options
@@ -103,7 +114,7 @@ module Rookout
       proxy = evaluate_config options[:proxy], "ROOKOUT_PROXY"
       token = evaluate_config options[:token], "ROOKOUT_TOKEN"
 
-      raise RookMissingToken if token.nil? && host == "wss://control.rookout.com"
+      raise RookMissingToken if token.nil? && !(host_specified options)
       verify_token token if token
 
       labels = stringify_labels(options[:labels]) || parse_labels(ENV["ROOKOUT_LABELS"])
@@ -115,6 +126,10 @@ module Rookout
       { host: host, port: port, proxy: proxy, token: token, labels: labels, async_start: async_start, fork: fork }
     end
 
+    def host_specified options
+      !options[:host].nil? || !ENV["ROOKOUT_CONTROLLER_HOST"].nil? || !ENV["ROOKOUT_AGENT_HOST"].nil?
+    end
+
     def evaluate_flag argument, env_var_name
       return true? argument unless argument.nil?
       true? ENV[env_var_name]

data/lib/rookout/logger.rb

@@ -12,7 +12,8 @@ module Rookout
 
     def initialize
       # Detect unit tests
-      if ($PROGRAM_NAME.end_with?("minitest_runner.rb") ||
+      if (Config.debug ||
+          $PROGRAM_NAME.end_with?("minitest_runner.rb") ||
           $PROGRAM_NAME.end_with?("tunit_or_minitest_in_folder_runner.rb")) &&
          Dir.pwd.end_with?("ruby-sdk")
         Config.logger_log_level = :DEBUG

data/lib/rookout/version.rb

@@ -1,3 +1,3 @@
 module Rookout
-  VERSION = "0.1.15".freeze
+  VERSION = "0.1.20".freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rookout
 version: !ruby/object:Gem::Version
-  version: 0.1.15
+  version: 0.1.20
 platform: ruby
 authors:
 - Liran Haimovitch
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-01-04 00:00:00.000000000 Z
+date: 2021-02-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: binding_of_caller
@@ -84,14 +84,14 @@ dependencies:
   name: google-style
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - '='
       - !ruby/object:Gem::Version
         version: 1.24.0
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - '='
       - !ruby/object:Gem::Version
         version: 1.24.0
 - !ruby/object:Gem::Dependency
@@ -268,7 +268,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.2
+rubygems_version: 3.1.4
 signing_key: 
 specification_version: 4
 summary: rookout is the Ruby SDK for the Rookout Debugging Platform