selective-ruby-core 0.1.5-x86_64-linux → 0.1.6-x86_64-linux

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ced267905bdcf1596a248cdcd2018cd2e628422e2e27aec445311b5d9afd4857
-  data.tar.gz: f4fbeb39edca06a95e3bf982f7b28c612a4d04d0165556ec341aef304eeb5718
+  metadata.gz: 0ec7acf96ff854b3c63a2c89560440378444ec9544694bb4c2107b456c067bcb
+  data.tar.gz: c607a7497f234d12f5d7d5011e125a9734f31e0a8ef0070767149bb7dc48a059
 SHA512:
-  metadata.gz: 937a82d41f7b323f925d7a5a761a7045617decf18073fd1aedd482a0c613f24adbbbff3ff53e2b2b068d584cfc7da1071f1b101378a4bbb3575a2f625904d58c
-  data.tar.gz: e93d8037c5af50b1d61700b68dea98d368b56db507480831b5b0be4b704b7e2684614571ef9d789f4c04409464e94b965db1676127a8e7368c0fc0e78aa3b1a9
+  metadata.gz: 3c653ea2a4a1cf52ff46e27715619fa70ff20d26fa6f3b363486fef7832f25140a033475af9050d8da912cf5c0e10e69501bb1f040b1766e999ccc0527d7c99c
+  data.tar.gz: 86060d98bb070045ba93ebc1d9cd2949d5d5c5cf0f09a99598e732a1c610164f3a9548d69a2d965e90d360d31255b13567f1fbf10e3ac2e8ef922474d046ddd0

data/lib/bin/build_env.sh

@@ -2,26 +2,32 @@
 
 # Detect the platform (only GitHub Actions in this case)
 if [ -n "$GITHUB_ACTIONS" ]; then
-  # Get environment variables
   platform=github_actions
   branch=${SELECTIVE_BRANCH:-${GITHUB_HEAD_REF:-$GITHUB_REF_NAME}}
-  pr_title=$SELECTIVE_PR_TITLE
   target_branch=${SELECTIVE_TARGET_BRANCH:-$GITHUB_BASE_REF}
   actor=$GITHUB_ACTOR
   sha=${SELECTIVE_SHA:-$GITHUB_SHA}
   run_id=${SELECTIVE_RUN_ID:-$GITHUB_RUN_ID}
   run_attempt=${SELECTIVE_RUN_ATTEMPT:-$GITHUB_RUN_ATTEMPT}
-  commit_message=$(git log --format=%s -n 1 $sha)
+  runner_id=$SELECTIVE_RUNNER_ID
+elif [ -n "$CIRCLECI" ]; then
+  platform=circleci
+  branch=${SELECTIVE_BRANCH:-$CIRCLE_BRANCH}
+  target_branch=$SELECTIVE_TARGET_BRANCH
+  actor=${SELECTIVE_ACTOR:-${CIRCLE_USERNAME:-$CIRCLE_PR_USERNAME}}
+  sha=${SELECTIVE_SHA:-$CIRCLE_SHA1}
+  run_id=$SELECTIVE_RUN_ID
+  run_attempt=${SELECTIVE_RUN_ATTEMPT:-$CIRCLE_BUILD_NUM}
+  runner_id=${SELECTIVE_RUNNER_ID:-$CIRCLE_NODE_INDEX}
 else
   platform=$SELECTIVE_PLATFORM
   branch=$SELECTIVE_BRANCH
-  pr_title=$SELECTIVE_PR_TITLE
   target_branch=$SELECTIVE_TARGET_BRANCH
   actor=$SELECTIVE_ACTOR
   sha=$SELECTIVE_SHA
   run_id=$SELECTIVE_RUN_ID
   run_attempt=$SELECTIVE_RUN_ATTEMPT
-  commit_message=$(git log --format=%s -n 1 $sha)
+  runner_id=$SELECTIVE_RUNNER_ID
 fi
 
 # Output the JSON
@@ -29,12 +35,13 @@ cat <<EOF
   {
     "platform": "$platform",
     "branch": "$branch",
-    "pr_title": "$pr_title",
+    "pr_title": "$SELECTIVE_PR_TITLE",
     "target_branch": "$target_branch",
     "actor": "$actor",
     "sha": "$sha",
     "run_id": "$run_id",
     "run_attempt": "$run_attempt",
-    "commit_message": "$commit_message"
+    "commit_message": "$(git log --format=%s -n 1 $sha)",
+    "runner_id": "$runner_id"
   }
 EOF

data/lib/selective/ruby/core/controller.rb

@@ -14,13 +14,13 @@ module Selective
           @debug = debug
           @runner = runner
           @retries = 0
-          @runner_id = safe_filename(ENV.fetch("SELECTIVE_RUNNER_ID", generate_runner_id))
+          @runner_id = safe_filename(get_runner_id)
           @logger = init_logger(log)
         end
 
         def start(reconnect: false)
           @pipe = NamedPipe.new("/tmp/#{runner_id}_2", "/tmp/#{runner_id}_1")
-          @transport_pid = spawn_transport_process(reconnect ? transport_url + "&reconnect=true" : transport_url)
+          @transport_pid = spawn_transport_process(reconnect: reconnect)
 
           handle_termination_signals(transport_pid)
           wait_for_connectivity
@@ -53,7 +53,14 @@ module Selective
 
         attr_reader :runner, :pipe, :transport_pid, :retries, :logger, :runner_id
 
-        BUILD_ENV_SCRIPT_PATH = "../../../bin/build_env.sh".freeze
+        ROOT_GEM_PATH = Gem.loaded_specs["selective-ruby-core"].full_gem_path
+
+        def get_runner_id
+          runner_id = build_env.delete("runner_id")
+          return generate_runner_id if runner_id.nil? || runner_id.empty?
+
+          runner_id
+        end
 
         def init_logger(enabled)
           if enabled
@@ -70,9 +77,7 @@ module Selective
             response = JSON.parse(message, symbolize_names: true)
 
             @logger.info("Received Command: #{response}")
-            next if handle_command(response)
-
-            break
+            break if handle_command(response) == :break
           end
         end
 
@@ -97,7 +102,7 @@ module Selective
           "selgen-#{SecureRandom.hex(4)}"
         end
 
-        def transport_url
+        def transport_url(reconnect: false)
           @transport_url ||= begin
             api_key = ENV.fetch("SELECTIVE_API_KEY")
             host = ENV.fetch("SELECTIVE_HOST", "wss://app.selective.ci")
@@ -113,9 +118,16 @@ module Selective
               "run_id" => run_id,
               "run_attempt" => run_attempt,
               "api_key" => api_key,
-              "runner_id" => runner_id
+              "runner_id" => runner_id,
+              "language" => "ruby",
+              "core_version" => Selective::Ruby::Core::VERSION,
+              "framework" => runner.framework,
+              "framework_version" => runner.framework_version,
+              "framework_wrapper_version" => runner.wrapper_version,
             }.merge(metadata: build_env.to_json)
 
+            prams[:reconnect] = true if reconnect
+
             query_string = URI.encode_www_form(params)
 
             "#{host}/transport/websocket?#{query_string}"
@@ -124,15 +136,14 @@ module Selective
 
         def build_env
           @build_env ||= begin
-            result = `#{Pathname.new(__dir__) + BUILD_ENV_SCRIPT_PATH}`
+            result = `#{File.join(ROOT_GEM_PATH, "lib", "bin", "build_env.sh")}`
             JSON.parse(result)
           end
         end
 
-        def spawn_transport_process(url)
-          root_path = Gem.loaded_specs["selective-ruby-core"].full_gem_path
-          transport_path = File.join(root_path, "lib", "bin", "transport")
-          get_transport_path = File.join(root_path, "bin", "get_transport")
+        def spawn_transport_process(reconnect: false)
+          transport_path = File.join(ROOT_GEM_PATH, "lib", "bin", "transport")
+          get_transport_path = File.join(ROOT_GEM_PATH, "bin", "get_transport")
 
           # The get_transport script is not released with the gem, so this
           # code is intended for development/CI purposes.
@@ -147,7 +158,7 @@ module Selective
             end
           end
 
-          Process.spawn(transport_path, url, runner_id).tap do |pid|
+          Process.spawn(transport_path, transport_url(reconnect: reconnect), runner_id).tap do |pid|
             Process.detach(pid)
           end
         end
@@ -211,39 +222,23 @@ module Selective
           # Process already gone noop
         end
 
-        def handle_command(response)
-          case response[:command]
-          when "print_notice"
-            print_notice(response[:message])
-          when "test_manifest"
-            handle_test_manifest
-          when "run_test_cases"
-            handle_run_test_cases(response[:test_case_ids])
-          when "remove_failed_test_case_result"
-            handle_remove_failed_test_case_result(response[:test_case_id])
-          when "reconnect"
-            handle_reconnect
-          when "print_message"
-            handle_print_message(response[:message])
-          when "close"
-            handle_close(response[:exit_status])
-            # This return is here for the sake of test where
-            # we cannot exit but we need to break the loop
-            return false
-          else
-            raise "Unknown command received: #{response[:command]}" if debug?
-          end
+        def handle_command(data)
+          send("handle_#{data[:command]}", data)
+        rescue NoMethodError
+          raise "Unknown command received: #{data[:command]}" if debug?
+        end
 
-          true
+        def handle_print_notice(data)
+          print_notice(data[:message])
         end
 
-        def handle_reconnect
+        def handle_reconnect(_data)
           kill_transport
           pipe.reset!
           start(reconnect: true)
         end
 
-        def handle_test_manifest
+        def handle_test_manifest(_data)
           self.class.restore_reporting!
           @logger.info("Sending Response: test_manifest")
           data = {test_cases: runner.manifest["examples"]}
@@ -251,8 +246,29 @@ module Selective
           write({type: "test_manifest", data: data})
         end
 
-        def handle_run_test_cases(test_cases)
-          runner.run_test_cases(test_cases, method(:test_case_callback))
+        def handle_run_test_cases(data)
+          runner.run_test_cases(data[:test_case_ids], method(:test_case_callback))
+        end
+
+        def handle_remove_failed_test_case_result(data)
+          runner.remove_failed_test_case_result(data[:test_case_id])
+        end
+
+        def handle_print_message(data)
+          print_warning(data[:message])
+        end
+
+        def handle_close(data)
+          exit_status = data[:exit_status]
+          self.class.restore_reporting!
+          runner.finish unless exit_status.is_a?(Integer)
+
+          kill_transport
+          pipe.delete_pipes
+          exit(exit_status || runner.exit_status)
+          # This :break is here for the sake of test where
+          # we cannot exit but we need to break the loop
+          :break
         end
 
         def test_case_callback(test_case)
@@ -260,10 +276,6 @@ module Selective
           write({type: "test_case_result", data: test_case})
         end
 
-        def handle_remove_failed_test_case_result(test_case_id)
-          runner.remove_failed_test_case_result(test_case_id)
-        end
-
         def modified_test_files
           @modified_test_files ||= begin
             target_branch = build_env["target_branch"]
@@ -279,19 +291,6 @@ module Selective
           end
         end
 
-        def handle_print_message(message)
-          print_warning(message)
-        end
-
-        def handle_close(exit_status = nil)
-          self.class.restore_reporting!
-          runner.finish unless exit_status.is_a?(Integer)
-
-          kill_transport
-          pipe.delete_pipes
-          exit(exit_status || runner.exit_status)
-        end
-
         def debug?
           @debug
         end

data/lib/selective/ruby/core/version.rb

@@ -3,7 +3,7 @@
 module Selective
   module Ruby
     module Core
-      VERSION = "0.1.5"
+      VERSION = "0.1.6"
     end
   end
 end

data/lib/selective-ruby-core.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require "zeitwerk"
+require "#{__dir__}/selective/ruby/core/version"
 
 loader = Zeitwerk::Loader.for_gem(warn_on_extra_files: false)
 loader.ignore("#{__dir__}/selective-ruby-core.rb")

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: selective-ruby-core
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.6
 platform: x86_64-linux
 authors:
 - Benjamin Wood
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2023-12-08 00:00:00.000000000 Z
+date: 2024-01-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: zeitwerk