# frozen_string_literal: true

require "wsdirector/scenario_reader"
require "wsdirector/clients_holder"
require "wsdirector/results_holder"
require "wsdirector/result"
require "wsdirector/task"
require "wsdirector/ext/deep_dup"

module WSDirector
  def self.run(scenario, scale: 1, connection_options: {}, locals: {}, **options)
    scenario = ScenarioReader.parse(scenario, scale: scale, connection_options: connection_options, locals: locals)
    Runner.new(scenario, scale: scale, **options).execute
  end

  # Initiates all clients as a separate tasks (=threads)
  class Runner
    using WSDirector::Ext::DeepDup

    def initialize(scenario, url:, scale: 1, sync_timeout: 5, logger: nil, colorize: false)
      @scenario = scenario
      @url = url
      @scale = scale
      @logger = logger
      @colorize = colorize
      @total_count = scenario["total"]
      @global_holder = ClientsHolder.new(total_count, sync_timeout: sync_timeout)
      @results_holder = ResultsHolder.new
    end

    def execute
      Thread.abort_on_exception = true

      num = 0

      tasks = scenario["clients"].flat_map do |client_config|
        name = client_config.fetch("name")
        result = Result.new(name)
        results_holder << result

        Array.new(client_config.fetch("multiplier")) do
          num += 1
          id = "#{name}_#{num}"
          Thread.new do
            Task.new(client_config.deep_dup, id: id, colorize: colorize, global_holder: global_holder, result: result, scale: scale, logger: logger)
              .run(url)
          end
        end
      end

      tasks.each(&:join)

      results_holder
    end

    private

    attr_reader :scenario, :url, :scale, :total_count, :global_holder, :results_holder, :logger, :colorize
  end
end