Contents

class KuberKit::Core::ImageStore
  NotFoundError = Class.new(KuberKit::Error)
  AlreadyAddedError = Class.new(KuberKit::Error)

  include KuberKit::Import[
    "core.image_factory",
    "core.image_definition_factory",
    "shell.local_shell",
    "tools.logger"
  ]

  def define(image_name, image_dir = nil)
    definition = image_definition_factory.create(image_name, image_dir)
    add_definition(definition)
    definition
  end

  def add_definition(image_definition)
    @@image_definitions ||= {}

    unless @@image_definitions[image_definition.image_name].nil?
      raise AlreadyAddedError, "image #{image_definition.image_name} was already added"
    end

    @@image_definitions[image_definition.image_name] = image_definition
  end

  Contract Symbol => Any
  def get_definition(image_name)
    @@image_definitions ||= {}

    if @@image_definitions[image_name].nil?
      raise NotFoundError, "image #{image_name} not found"
    end

    @@image_definitions[image_name]
  end

  Contract Symbol => Any
  def get_image(image_name)
    definition = get_definition(image_name)

    image_factory.create(definition)
  end

  def load_definitions(dir_path)
    files = local_shell.recursive_list_files(dir_path, name: "image.rb").each do |path|
      load_definition(path)
    end
  rescue KuberKit::Shell::AbstractShell::DirNotFoundError
    logger.warn("Directory with images not found: #{dir_path}")
    []
  end

  def load_definition(file_path)
    require(file_path)
  end

  def reset!
    @@image_definitions = {}
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
kuber_kit-0.1.8	lib/kuber_kit/core/image_store.rb
kuber_kit-0.1.7	lib/kuber_kit/core/image_store.rb
kuber_kit-0.1.6	lib/kuber_kit/core/image_store.rb
kuber_kit-0.1.5	lib/kuber_kit/core/image_store.rb
kuber_kit-0.1.4	lib/kuber_kit/core/image_store.rb