1
# frozen_string_literal: true
2

3
require 'English'
4
require 'open3'
5
require 'rake'
6
require 'io/wait'
7
require_relative 'selenium_rake/checks'
8

9
module Bazel
10
  def self.execute(kind, args, target, &block)
11
    verbose = Rake::FileUtilsExt.verbose_flag
12

13
    if target.end_with?(':run')
14
      kind = 'run'
15
      target = target[0, target.length - 4]
16
    end
17

18
    cmd = %w[bazel] + [kind, target] + (args || [])
19
    cmd_out = ''
20
    cmd_exit_code = 0
21

22
    if SeleniumRake::Checks.windows?
23
      cmd += ['2>&1']
24
      cmd_line = cmd.join(' ')
25
      cmd_out = `#{cmd_line}`.encode('UTF-8', 'binary', invalid: :replace, undef: :replace, replace: '')
26
      puts cmd_out if verbose
27
      cmd_exit_code = $CHILD_STATUS
28
    else
29
      Open3.popen2e(*cmd) do |stdin, stdouts, wait|
30
        is_running = true
31
        stdin.close
32
        while is_running
33
          begin
34
            stdouts.wait_readable
35
            line = stdouts.readpartial(512)
36
            cmd_out += line
37
            $stdout.print line if verbose
38
          rescue EOFError
39
            is_running = false
40
          end
41
        end
42
        cmd_exit_code = wait.value.exitstatus
43
      end
44
    end
45

46
    raise "#{cmd.join(' ')} failed with exit code: #{cmd_exit_code}\nOutput: #{cmd_out}" if cmd_exit_code != 0
47

48
    block&.call(cmd_out)
49
    return unless cmd_out =~ %r{\s+(bazel-bin/\S+)}
50

51
    out_artifact = Regexp.last_match(1)
52
    puts "#{target} -> #{out_artifact}" if out_artifact
53
    out_artifact
54
  end
55
end
56

57