class KuberKit::Actions::ImageCompiler
  include KuberKit::Import[
    "image_compiler.compiler",
    "image_compiler.image_dependency_resolver",
    "shell.local_shell",
    "tools.logger",
    "configs",
    "ui"
  ]

  Contract ArrayOf[Symbol], Hash => Any
  def call(image_names, options)
    build_id = generate_build_id

    resolved_dependencies = []
    dependencies = image_dependency_resolver.get_next(image_names)

    while (dependencies - resolved_dependencies).any?
      compile_simultaneously(dependencies, build_id)
      resolved_dependencies += dependencies
      dependencies = image_dependency_resolver.get_next(image_names, resolved: resolved_dependencies)
    end

    compile_simultaneously(image_names - resolved_dependencies, build_id)
  end

  private
    def compile_simultaneously(image_names, build_id)
      task_group = ui.create_task_group
      image_names.map do |image_name|

        logger.info("Started compiling: #{image_name.to_s.green}")
        task_group.add("Compiling #{image_name.to_s.yellow}") do |task|
          compile_image(image_name, build_id)

          task.update_title("Compiled #{image_name.to_s.green}")
          logger.info("Finished compiling: #{image_name.to_s.green}")
        end
        
      end
      task_group.wait
    end

    def compile_image(image_name, build_id)
      compile_dir = generate_compile_dir(build_id: build_id)
      image_compiler.compile(local_shell, image_name, compile_dir)
    end

    def generate_build_id
      Time.now.strftime("%H%M%S")
    end

    def generate_compile_dir(build_id:)
      File.join(configs.image_compile_dir, build_id)
    end
end