module BenchmarkDriver::RidkUse

def self.parse_spec(full_spec)
  name, spec = full_spec.split('::', 2)
  spec ||= name # if `::` is not given, regard whole string as spec
  version, *args = spec.shellsplit
  BenchmarkDriver::Config::Executable.new(
    name: name,
    command: [BenchmarkDriver::RidkUse.ruby_path(version), *args],
  )
end

def self.ridk_use_list
  ruby_list = []
  regex = /(\d+)\s-\s([^\s]+)\s\truby\s([^\s]+)\s/
  cmd = "ridk use list"
  stdout, _status = Open3.capture2e(cmd)
  stdout.each_line do |line|
    if matched = regex.match(line)
      idx, rubypath, rubyver = matched[1..3]
      ruby_list << rubyver << [idx, "#{rubypath}/bin/ruby.exe"]
    else
      abort "Failed to execute 'ridk use list'"
    end
  end
  Hash[*ruby_list]
end

def self.ruby_path(version)
  ruby_list = BenchmarkDriver::RidkUse.ridk_use_list
  regex = Regexp.new(version)
  matched = ruby_list.keys.find {|k| k =~ regex}
  if ruby_list.has_key?(version)
    ruby_list[version][1]
  elsif matched
    ruby_list[matched][1]
  else
    abort "version #{version} not found"
  end
end