Implement RBS trace

Author: aguspe

rb/Gemfile

@@ -10,3 +10,5 @@ gem 'activesupport', '~> 7.0', require: false, platforms: %i[mri mingw x64_mingw
 gem 'curb', '~> 1.0.5', require: false, platforms: %i[mri mingw x64_mingw]
 gem 'debug', '~> 1.7', require: false, platforms: %i[mri mingw x64_mingw]
 gem 'steep', '~> 1.5.0', require: false, platforms: %i[mri mingw x64_mingw]
+
+gem "rbs-trace", "~> 0.5.1"

rb/Gemfile.lock

@@ -100,6 +100,9 @@ GEM
       ffi (~> 1.0)
     rbs (3.9.2)
       logger
+    rbs-trace (0.5.1)
+      prism (>= 0.3.0)
+      rbs (>= 3.5.0)
     rchardet (1.9.0)
     rdoc (6.13.1)
       psych (>= 4.0.0)
@@ -199,6 +202,7 @@ DEPENDENCIES
   git (~> 1.19)
   rack (~> 2.0)
   rake (~> 13.0)
+  rbs-trace (~> 0.5.1)
   rspec (~> 3.0)
   rubocop (~> 1.75)
   rubocop-performance (~> 1.25)

rb/lib/selenium/webdriver.rb

@@ -95,6 +95,7 @@ def self.for(*args)
     # @return [Logger]
     #
 
+    # @rbs (**nil) -> Selenium::WebDriver::Logger
     def self.logger(**opts)
       level = $DEBUG || ENV.key?('DEBUG') ? :debug : :info
       @logger ||= WebDriver::Logger.new('Selenium', default_level: level, **opts)

rb/lib/selenium/webdriver/chrome.rb

@@ -31,6 +31,7 @@ def self.path=(path)
         @path = path
       end
 
+      # @rbs () -> nil
       def self.path
         @path ||= nil
       end

rb/lib/selenium/webdriver/chrome/driver.rb

@@ -30,11 +30,13 @@ module Chrome
       class Driver < Chromium::Driver
         include LocalDriver
 
+        # @rbs (?options: Selenium::WebDriver::Chrome::Options, ?service: Selenium::WebDriver::Chrome::Service, ?url: nil, **nil) -> void
         def initialize(options: nil, service: nil, url: nil, **opts)
           caps, url = initialize_local_driver(options, service, url)
           super(caps: caps, url: url, **opts)
         end
 
+        # @rbs () -> Symbol
         def browser
           :chrome
         end

rb/lib/selenium/webdriver/chrome/features.rb

@@ -35,10 +35,12 @@ module Features
           send_command: [:post, 'session/:session_id/goog/cdp/execute']
         }.freeze
 
+        # @rbs () -> Hash[untyped, untyped]
         def command_list
           CHROME_COMMANDS.merge(CHROMIUM_COMMANDS).merge(self.class::COMMANDS)
         end
 
+        # @rbs (Symbol) -> Array[untyped]
         def commands(command)
           command_list[command]
         end

rb/lib/selenium/webdriver/chrome/options.rb

@@ -32,6 +32,7 @@ def enable_logging(browser_options)
           browser_options['goog:loggingPrefs'] = @logging_prefs
         end
 
+        # @rbs () -> nil
         def binary_path
           Chrome.path
         end

rb/lib/selenium/webdriver/chrome/service.rb

@@ -26,6 +26,7 @@ class Service < WebDriver::Service
         SHUTDOWN_SUPPORTED = true
         DRIVER_PATH_ENV_KEY = 'SE_CHROMEDRIVER'
 
+        # @rbs () -> nil
         def log
           return @log unless @log.is_a? String
 

rb/lib/selenium/webdriver/chromium/features.rb

@@ -75,6 +75,7 @@ def delete_network_conditions
           execute :delete_network_conditions
         end
 
+        # @rbs (Hash[untyped, untyped]) -> Hash[untyped, untyped]
         def send_command(command_params)
           execute :send_command, {}, command_params
         end

rb/lib/selenium/webdriver/chromium/options.rb

@@ -67,6 +67,7 @@ class Options < WebDriver::Options
         # @option opts [Array<String>] window_types A list of window types to appear in the list of window handles
         #
 
+        # @rbs (?profile: nil, **Array[untyped] | Hash[untyped, untyped] | nil) -> void
         def initialize(profile: nil, **opts)
           super(**opts)
 
@@ -203,6 +204,7 @@ def enable_android(package: 'com.android.chrome', serial_number: nil, use_runnin
 
         protected
 
+        # @rbs (Hash[untyped, untyped]) -> void
         def process_browser_options(browser_options)
           enable_logging(browser_options) unless @logging_prefs.empty?
 
@@ -234,6 +236,7 @@ def validate_extension(path)
           @extensions << path
         end
 
+        # @rbs (Symbol | String) -> bool
         def camelize?(key)
           !%w[localState prefs].include?(key)
         end