require_relative 'dispatch'
require_relative 'message'
require_relative 'net_dispatcher'
require_relative 'repos'

module Kamerling
  class TaskDispatcher
    def initialize(net_dispatcher: NetDispatcher, repos: Repos)
      @net_dispatcher = net_dispatcher
      @repos          = repos
    end

    def dispatch_all
      repos.projects.each do |project|
        repos.free_clients_for(project).each do |client|
          task = repos.next_task_for(project)
          dispatch_task client: client, project: project, task: task if task
        end
      end
    end

    attr_reader :net_dispatcher, :repos
    private     :net_dispatcher, :repos

    private

    def dispatch_task(client:, project:, task:)
      message = Message.build(client: client, payload: task.data,
                              project: project, task: task, type: :DATA)
      dispatch = Dispatch.new(addr: client.addr, client: client,
                              project: project, task: task)
      net_dispatcher.dispatch client.addr, message
      client.busy = true
      repos << client
      repos << dispatch
    end
  end
end