kuber_kit 0.1.8 → 0.2.3

data/lib/kuber_kit/core/configuration_factory.rb

@@ -6,6 +6,7 @@ class KuberKit::Core::ConfigurationFactory
     "core.artifact_store",
     "core.env_file_store",
     "core.template_store",
+    "core.build_server_store",
     "configs"
   ]
 
@@ -16,6 +17,7 @@ class KuberKit::Core::ConfigurationFactory
     registries = fetch_registries(configuration_attrs.registries)
     env_files  = fetch_env_files(configuration_attrs.env_files)
     templates  = fetch_templates(configuration_attrs.templates)
+    build_servers = fetch_build_servers(configuration_attrs.build_servers)
 
     KuberKit::Core::Configuration.new(
       name:             configuration_attrs.name,
@@ -25,7 +27,9 @@ class KuberKit::Core::ConfigurationFactory
       templates:        templates,
       kubeconfig_path:  configuration_attrs.kubeconfig_path,
       deploy_strategy:  configuration_attrs.deploy_strategy || configs.deploy_strategy,
-      services_attributes: configuration_attrs.services_attributes
+      deploy_namespace: configuration_attrs.deploy_namespace,
+      services_attributes: configuration_attrs.services_attributes,
+      build_servers:    build_servers
     )
   end
 
@@ -61,4 +65,10 @@ class KuberKit::Core::ConfigurationFactory
       end
       result
     end
+
+    def fetch_build_servers(build_servers)
+      build_servers.map do |build_server_name|
+        build_server_store.get(build_server_name)
+      end
+    end
 end

data/lib/kuber_kit/core/configuration_store.rb

@@ -1,7 +1,4 @@
 class KuberKit::Core::ConfigurationStore
-  NotFoundError = Class.new(KuberKit::NotFoundError)
-  AlreadyAddedError = Class.new(KuberKit::Error)
-
   include KuberKit::Import[
     "core.configuration_factory",
     "core.configuration_definition_factory",
@@ -16,24 +13,12 @@ class KuberKit::Core::ConfigurationStore
   end
 
   def add_definition(configuration_definition)
-    @@configuration_definitions ||= {}
-
-    unless @@configuration_definitions[configuration_definition.configuration_name].nil?
-      raise AlreadyAddedError, "image #{configuration_definition.configuration_name} was already added"
-    end
-
-    @@configuration_definitions[configuration_definition.configuration_name] = configuration_definition
+    definitions_store.add(configuration_definition.configuration_name, configuration_definition)
   end
 
   Contract Symbol => Any
   def get_definition(configuration_name)
-    @@configuration_definitions ||= {}
-
-    if @@configuration_definitions[configuration_name].nil?
-      raise NotFoundError, "configuration #{configuration_name} not found"
-    end
-
-    @@configuration_definitions[configuration_name]
+    definitions_store.get(configuration_name)
   end
 
   Contract Symbol => Any
@@ -57,18 +42,23 @@ class KuberKit::Core::ConfigurationStore
   end
 
   def reset!
-    @@configuration_definitions = {}
-  end
-
-  def all_definitions
-    @@configuration_definitions ||= {}
+    definitions_store.reset!
   end
 
   def count
-    all_definitions.count
+    definitions_store.size
   end
 
   def exists?(configuration_name)
-    !all_definitions[configuration_name].nil?
+    definitions_store.exists?(configuration_name)
   end
+
+  def all_definitions
+    definitions_store.items
+  end
+
+  private
+    def definitions_store
+      @@definitions_store ||= KuberKit::Core::Store.new(KuberKit::Core::ConfigurationDefinition)
+    end
 end

data/lib/kuber_kit/core/env_files/env_file_store.rb

@@ -1,19 +1,6 @@
 class KuberKit::Core::EnvFiles::EnvFileStore
-  NotFoundError = Class.new(KuberKit::NotFoundError)
-  AlreadyAddedError = Class.new(KuberKit::Error)
-
   def add(env_file)
-    @@env_files ||= {}
-
-    if !env_file.is_a?(KuberKit::Core::EnvFiles::AbstractEnvFile)
-      raise ArgumentError.new("should be an instance of KuberKit::Core::EnvFiles::AbstractEnvFile, got: #{env_file.inspect}")
-    end
-
-    unless @@env_files[env_file.name].nil?
-      raise AlreadyAddedError, "env_file #{env_file.name} was already added"
-    end
-
-    @@env_files[env_file.name] = env_file
+    store.add(env_file.name, env_file)
   end
 
   def get(env_file_name)
@@ -24,14 +11,7 @@ class KuberKit::Core::EnvFiles::EnvFileStore
   end
 
   def get_global(env_file_name)
-    @@env_files ||= {}
-    env_file = @@env_files[env_file_name]
-
-    if env_file.nil?
-      raise NotFoundError, "env_file '#{env_file_name}' not found"
-    end
-    
-    env_file
+    store.get(env_file_name)
   end
 
   def get_from_configuration(env_file_name)
@@ -40,6 +20,11 @@ class KuberKit::Core::EnvFiles::EnvFileStore
   end
 
   def reset!
-    @@env_files = {}
+    store.reset!
   end
+
+  private
+    def store
+      @@store ||= KuberKit::Core::Store.new(KuberKit::Core::EnvFiles::AbstractEnvFile)
+    end
 end

data/lib/kuber_kit/core/image_store.rb

@@ -1,7 +1,4 @@
 class KuberKit::Core::ImageStore
-  NotFoundError = Class.new(KuberKit::Error)
-  AlreadyAddedError = Class.new(KuberKit::Error)
-
   include KuberKit::Import[
     "core.image_factory",
     "core.image_definition_factory",
@@ -16,24 +13,12 @@ class KuberKit::Core::ImageStore
   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
+    definitions_store.add(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]
+    definitions_store.get(image_name)
   end
 
   Contract Symbol => Any
@@ -57,6 +42,11 @@ class KuberKit::Core::ImageStore
   end
 
   def reset!
-    @@image_definitions = {}
+    definitions_store.reset!
   end
+
+  private
+    def definitions_store
+      @@definitions_store ||= KuberKit::Core::Store.new(KuberKit::Core::ImageDefinition)
+    end
 end

data/lib/kuber_kit/core/registries/registry_store.rb

@@ -1,19 +1,6 @@
 class KuberKit::Core::Registries::RegistryStore
-  NotFoundError = Class.new(KuberKit::NotFoundError)
-  AlreadyAddedError = Class.new(KuberKit::Error)
-
   def add(registry)
-    @@registries ||= {}
-
-    if !registry.is_a?(KuberKit::Core::Registries::AbstractRegistry)
-      raise ArgumentError.new("should be an instance of KuberKit::Core::Registries::AbstractRegistry, got: #{registry.inspect}")
-    end
-
-    unless @@registries[registry.name].nil?
-      raise AlreadyAddedError, "registry #{registry.name} was already added"
-    end
-
-    @@registries[registry.name] = registry
+    store.add(registry.name, registry)
   end
 
   def get(registry_name)
@@ -24,14 +11,7 @@ class KuberKit::Core::Registries::RegistryStore
   end
 
   def get_global(registry_name)
-    @@registries ||= {}
-    registry = @@registries[registry_name]
-
-    if registry.nil?
-      raise NotFoundError, "registry '#{registry_name}' not found"
-    end
-    
-    registry
+    store.get(registry_name)
   end
 
   def get_from_configuration(registry_name)
@@ -44,6 +24,11 @@ class KuberKit::Core::Registries::RegistryStore
   end
 
   def reset!
-    @@registries = {}
+    store.reset!
   end
+
+  private
+    def store
+      @@store ||= KuberKit::Core::Store.new(KuberKit::Core::Registries::AbstractRegistry)
+    end
 end

data/lib/kuber_kit/core/service_store.rb

@@ -1,7 +1,4 @@
 class KuberKit::Core::ServiceStore
-  NotFoundError = Class.new(KuberKit::Error)
-  AlreadyAddedError = Class.new(KuberKit::Error)
-
   include KuberKit::Import[
     "core.service_factory",
     "core.service_definition_factory",
@@ -16,24 +13,12 @@ class KuberKit::Core::ServiceStore
   end
 
   def add_definition(service_definition)
-    @@service_definitions ||= {}
-
-    unless @@service_definitions[service_definition.service_name].nil?
-      raise AlreadyAddedError, "service #{service_definition.service_name} was already added"
-    end
-
-    @@service_definitions[service_definition.service_name] = service_definition
+    definitions_store.add(service_definition.service_name, service_definition)
   end
 
   Contract Symbol => Any
   def get_definition(service_name)
-    @@service_definitions ||= {}
-
-    if @@service_definitions[service_name].nil?
-      raise NotFoundError, "service '#{service_name}' not found"
-    end
-
-    @@service_definitions[service_name]
+    definitions_store.get(service_name)
   end
 
   Contract Symbol => Any
@@ -57,18 +42,23 @@ class KuberKit::Core::ServiceStore
   end
 
   def reset!
-    @@service_definitions = {}
+    definitions_store.reset!
   end
 
-  def all_definitions
-    @@service_definitions ||= {}
+  def count
+    definitions_store.size
   end
 
-  def count
-    all_definitions.count
+  def all_definitions
+    definitions_store.items
   end
 
   def exists?(service_name)
-    !all_definitions[service_name].nil?
+    definitions_store.exists?(service_name)
   end
+
+  private
+    def definitions_store
+      @@definitions_store ||= KuberKit::Core::Store.new(KuberKit::Core::ServiceDefinition)
+    end
 end

data/lib/kuber_kit/core/store.rb

@@ -0,0 +1,48 @@
+class KuberKit::Core::Store
+  NotFoundError = Class.new(KuberKit::NotFoundError)
+  AlreadyAddedError = Class.new(KuberKit::Error)
+
+  attr_reader :object_class_name
+
+  def initialize(object_class_name)
+    @object_class_name = object_class_name
+  end
+
+  def add(item_name, item)
+    unless item.is_a?(object_class_name)
+      raise ArgumentError.new("#{self.object_class_name}: should be an instance of #{object_class_name}, got: #{item.inspect}")
+    end
+
+    unless items[item_name].nil?
+      raise AlreadyAddedError, "#{self.object_class_name}: item with name #{item_name} was already added"
+    end
+
+    items[item_name] = item
+  end
+
+  def get(item_name)
+    item = items[item_name]
+
+    if item.nil?
+      raise NotFoundError, "#{self.object_class_name}: item '#{item_name}' not found"
+    end
+
+    item
+  end
+
+  def items
+    @items ||= {}
+  end
+
+  def reset!
+    @items = {}
+  end
+
+  def size
+    items.count
+  end
+
+  def exists?(name)
+    !items[name].nil?
+  end
+end

data/lib/kuber_kit/core/templates/template_store.rb

@@ -1,19 +1,6 @@
 class KuberKit::Core::Templates::TemplateStore
-  NotFoundError = Class.new(KuberKit::NotFoundError)
-  AlreadyAddedError = Class.new(KuberKit::Error)
-
   def add(template)
-    @@templates ||= {}
-
-    if !template.is_a?(KuberKit::Core::Templates::AbstractTemplate)
-      raise ArgumentError.new("should be an instance of KuberKit::Core::Templates::AbstractTemplate, got: #{template.inspect}")
-    end
-
-    unless @@templates[template.name].nil?
-      raise AlreadyAddedError, "template #{template.name} was already added"
-    end
-
-    @@templates[template.name] = template
+    store.add(template.name, template)
   end
 
   def get(template_name)
@@ -24,14 +11,7 @@ class KuberKit::Core::Templates::TemplateStore
   end
 
   def get_global(template_name)
-    @@templates ||= {}
-    template = @@templates[template_name]
-
-    if template.nil?
-      raise NotFoundError, "template '#{template_name}' not found"
-    end
-    
-    template
+    store.get(template_name)
   end
 
   def get_from_configuration(template_name)
@@ -40,14 +20,15 @@ class KuberKit::Core::Templates::TemplateStore
   end
 
   def reset!
-    @@templates = {}
-  end
-
-  def all_definitions
-    @@templates ||= {}
+    store.reset!
   end
 
   def exists?(template_name)
-    !all_definitions[template_name].nil?
+    store.exists?(template_name)
   end
+
+  private
+    def store
+      @@store ||= KuberKit::Core::Store.new(KuberKit::Core::Templates::AbstractTemplate)
+    end
 end

data/lib/kuber_kit/image_compiler/build_server_pool.rb

@@ -0,0 +1,31 @@
+class KuberKit::ImageCompiler::BuildServerPool
+  attr_reader :ssh_shells, :local_shell
+  
+  def initialize(local_shell:, build_servers:, ssh_shell_class:)
+    @local_shell     = local_shell
+    @ssh_shell_class = ssh_shell_class
+    @build_servers   = build_servers
+    @ssh_shells      = []
+  end
+
+  def get_shell
+    if @build_servers.any?
+      shell = connect_to_ssh_shell(@build_servers.sample)
+      @ssh_shells << shell
+      shell
+    else
+      @local_shell
+    end
+  end
+
+  def disconnect_all
+    @ssh_shells.each(&:disconnect)
+  end
+
+  private
+    def connect_to_ssh_shell(bs)
+      shell = @ssh_shell_class.new
+      shell.connect(host: bs.host, user: bs.user, port: bs.port)
+      shell
+    end
+end