selective-ruby-core 0.1.2-aarch64-linux → 0.1.4-aarch64-linux

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f46c8813512297c86dcc7d42a197c19c69261610793c6476b0ef863e94eac062
-  data.tar.gz: 59b7700e822f4810a468013ac74701976bcdc09426b718b35bc4350fdde06f14
+  metadata.gz: b5500d7971e466047c33f3ed28463a5662b330d90cb3c77812a26155de248319
+  data.tar.gz: f391897d219fb94dc0653ad86b1add32a0e7e973dd68fedee18aeaaabcf0081e
 SHA512:
-  metadata.gz: 87c2116863c1db55e887569178d8c5b109ee4c50fd9b64413410e9a274da99999ff0756da57b22d80a9ab019761fedc6d53231f878d5eca3f4a52ef1b9b5385f
-  data.tar.gz: 7e78f5ee5c42ed2d92cb76a2a38225703b3d13d8cb4b2439b117baaf5319933c94139634cde9fde5bab49fbac50b9c369916397f0cdf9ca2005da9a458013c06
+  metadata.gz: 65ebb5f6917cd574a8d089b88db70aed8ab9113a144b4895a325947f26bf158ad6f82bd3d18d185932b3851196bc7b292cfad217040c27c9b8c399a65f87a7ef
+  data.tar.gz: da6944fa1661503dbf9b341e0f35d568778e38c0bc879ce48aa977f4e21a757abc69c9986796551f43a423a3faac194922631bce097f5a668686d90a60e8f43a

data/lib/bin/build_env.sh

@@ -4,11 +4,13 @@
 if [ -n "$GITHUB_ACTIONS" ]; then
   # Get environment variables
   platform=github_actions
-  branch=${GITHUB_HEAD_REF:-$GITHUB_REF_NAME}
+  branch=${SELECTIVE_BRANCH:-${GITHUB_HEAD_REF:-$GITHUB_REF_NAME}}
   pr_title=$SELECTIVE_PR_TITLE
-  target_branch=${GITHUB_BASE_REF}
+  target_branch=${SELECTIVE_TARGET_BRANCH:-$GITHUB_BASE_REF}
   actor=$GITHUB_ACTOR
-  sha=$GITHUB_SHA
+  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)
 else
   platform=$SELECTIVE_PLATFORM
@@ -17,6 +19,8 @@ else
   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)
 fi
 
@@ -29,6 +33,8 @@ cat <<EOF
     "target_branch": "$target_branch",
     "actor": "$actor",
     "sha": "$sha",
+    "run_id": "$run_id",
+    "run_attempt": "$run_attempt",
     "commit_message": "$commit_message"
   }
 EOF

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

@@ -14,7 +14,7 @@ module Selective
           @debug = debug
           @runner = runner
           @retries = 0
-          @runner_id = ENV.fetch("SELECTIVE_RUNNER_ID", generate_runner_id)
+          @runner_id = safe_filename(ENV.fetch("SELECTIVE_RUNNER_ID", generate_runner_id))
           @logger = init_logger(log)
         end
 
@@ -23,6 +23,7 @@ module Selective
           @transport_pid = spawn_transport_process(reconnect ? transport_url + "&reconnect=true" : transport_url)
 
           handle_termination_signals(transport_pid)
+          wait_for_connectivity
           run_main_loop
         rescue NamedPipe::PipeClosedError
           retry!
@@ -66,8 +67,6 @@ module Selective
         def run_main_loop
           loop do
             message = pipe.read
-            next sleep(0.1) if message.nil? || message.empty?
-
             response = JSON.parse(message, symbolize_names: true)
 
             @logger.info("Received Command: #{response}")
@@ -101,13 +100,15 @@ module Selective
         def transport_url
           @transport_url ||= begin
             api_key = ENV.fetch("SELECTIVE_API_KEY")
-            run_id = ENV.fetch("SELECTIVE_RUN_ID")
-            run_attempt = ENV.fetch("SELECTIVE_RUN_ATTEMPT", SecureRandom.uuid)
             host = ENV.fetch("SELECTIVE_HOST", "wss://app.selective.ci")
 
             # Validate that host is a valid websocket url(starts with ws:// or wss://)
             raise "Invalid host: #{host}" unless host.match?(/^wss?:\/\//)
 
+            run_id = build_env.delete("run_id")
+            run_attempt = build_env.delete("run_attempt")
+            run_attempt = SecureRandom.uuid if run_attempt.nil? || run_attempt.empty?
+
             params = {
               "run_id" => run_id,
               "run_attempt" => run_attempt,
@@ -151,6 +152,31 @@ module Selective
           end
         end
 
+        def wait_for_connectivity
+          @connectivity = false
+
+          Thread.new do
+            sleep(5)
+            unless @connectivity
+              puts "Transport process failed to start. Exiting..."
+              kill_transport
+              exit(1)
+            end
+          end
+
+          loop do
+            message = pipe.read
+
+            # The message is nil until the transport opens the pipe
+            # for writing. So, we must handle that here.
+            next sleep(0.1) if message.nil?
+            
+            response = JSON.parse(message, symbolize_names: true)
+            @connectivity = true if response[:command] == "connected"
+            break
+          end
+        end
+
         def handle_termination_signals(pid)
           ["INT", "TERM"].each do |signal|
             Signal.trap(signal) do
@@ -173,7 +199,7 @@ module Selective
 
               sleep(1)
             end
-          rescue NamedPipe::PipeClosedError
+          rescue NamedPipe::PipeClosedError, IOError
             # If the pipe is close, move straight to killing
             # it forcefully.
           end
@@ -270,6 +296,13 @@ module Selective
           @debug
         end
 
+        def safe_filename(filename)
+          filename
+            .gsub(/[\/\\:*?"<>|\n\r]+/, '_')
+            .gsub(/^\.+|\.+$/, '')
+            .strip[0, 255]
+        end
+
         def with_error_handling(include_header: true)
           yield
         rescue => e

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

@@ -3,7 +3,7 @@
 module Selective
   module Ruby
     module Core
-      VERSION = "0.1.2"
+      VERSION = "0.1.4"
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: selective-ruby-core
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.4
 platform: aarch64-linux
 authors:
 - Benjamin Wood
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2023-11-14 00:00:00.000000000 Z
+date: 2023-12-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: zeitwerk