Contents

class KuberKit::ImageCompiler::ImageDependencyResolver
  CircularDependencyError = Class.new(KuberKit::Error)
  DependencyNotFoundError = Class.new(KuberKit::NotFoundError)

  include KuberKit::Import[
    "core.image_store",
    "configs"
  ]

  Contract Or[Symbol, ArrayOf[Symbol]], Proc => Any
  def each_with_deps(image_names, &block)
    compile_limit = configs.compile_simultaneous_limit

    resolved_dependencies = []
    next_dependencies = get_next(image_names, limit: compile_limit)

    while (next_dependencies - resolved_dependencies).any?
      block.call(next_dependencies)
      resolved_dependencies += next_dependencies
      next_dependencies = get_next(image_names, resolved: resolved_dependencies, limit: compile_limit)
    end

    (image_names - resolved_dependencies).each_slice(compile_limit) do |group|
      block.call(group)
    end
  end
  
  Contract Or[Symbol, ArrayOf[Symbol]], KeywordArgs[
    resolved: Optional[ArrayOf[Symbol]],
    limit:    Optional[Maybe[Num]]
  ] => Any
  def get_next(image_names, resolved: [], limit: nil)
    deps = Array(image_names).map { |i| get_recursive_deps(i) }.flatten.uniq

    ready_to_resolve = deps.select do |dep_name|
      unresolved_deps = get_deps(dep_name) - resolved
      unresolved_deps.empty?
    end
    unresolved_deps = ready_to_resolve - resolved
    unresolved_deps = unresolved_deps.take(limit) if limit
    unresolved_deps
  end

  def get_recursive_deps(image_name, dependency_tree: [])
    deps = get_deps(image_name)

    if dependency_tree.include?(image_name)
      raise CircularDependencyError, "Circular dependency found for #{image_name}. Dependency tree: #{dependency_tree.inspect}"
    end

    child_deps = []
    deps.each do |i| 
      child_deps += get_recursive_deps(i, dependency_tree: dependency_tree + [image_name])
    end

    (deps + child_deps).uniq
  end

  def get_deps(image_name)
    image_store.get_definition(image_name).dependencies
  end
end

Version data entries

9 entries across 9 versions & 1 rubygems

Version	Path
kuber_kit-0.5.1	lib/kuber_kit/image_compiler/image_dependency_resolver.rb
kuber_kit-0.5.0	lib/kuber_kit/image_compiler/image_dependency_resolver.rb
kuber_kit-0.4.9	lib/kuber_kit/image_compiler/image_dependency_resolver.rb
kuber_kit-0.4.8	lib/kuber_kit/image_compiler/image_dependency_resolver.rb
kuber_kit-0.4.7	lib/kuber_kit/image_compiler/image_dependency_resolver.rb
kuber_kit-0.4.6	lib/kuber_kit/image_compiler/image_dependency_resolver.rb
kuber_kit-0.4.5	lib/kuber_kit/image_compiler/image_dependency_resolver.rb
kuber_kit-0.4.4	lib/kuber_kit/image_compiler/image_dependency_resolver.rb
kuber_kit-0.4.3	lib/kuber_kit/image_compiler/image_dependency_resolver.rb