kuber_kit 0.1.8 → 0.1.9

data/lib/kuber_kit/core/artifacts/artifact_store.rb

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

data/lib/kuber_kit/core/build_servers/abstract_build_server.rb

@@ -0,0 +1,21 @@
+class KuberKit::Core::BuildServers::AbstractBuildServer
+  include KuberKit::Extensions::Inspectable
+
+  attr_reader :name
+
+  def initialize(build_server_name)
+    @name = build_server_name
+  end
+
+  def host
+    raise KuberKit::NotImplementedError, "must be implemented"
+  end
+
+  def user
+    raise KuberKit::NotImplementedError, "must be implemented"
+  end
+
+  def port
+    raise KuberKit::NotImplementedError, "must be implemented"
+  end
+end

data/lib/kuber_kit/core/build_servers/build_server.rb

@@ -0,0 +1,24 @@
+class KuberKit::Core::BuildServers::BuildServer < KuberKit::Core::BuildServers::AbstractBuildServer
+  def setup(host:, user:, port:)
+    @host = host
+    @user = user
+    @port = port
+
+    self
+  end
+
+  def host
+    raise ArgumentError, "host is not set, please use #setup method" if @host.nil?
+    @host
+  end
+
+  def user
+    raise ArgumentError, "user is not set, please use #setup method" if @user.nil?
+    @user
+  end
+
+  def port
+    raise ArgumentError, "port is not set, please use #setup method" if @port.nil?
+    @port
+  end
+end

data/lib/kuber_kit/core/build_servers/build_server_store.rb

@@ -0,0 +1,18 @@
+class KuberKit::Core::BuildServers::BuildServerStore
+  def add(build_server)
+    store.add(build_server.name, build_server)
+  end
+
+  def get(build_server_name)
+    store.get(build_server_name)
+  end
+
+  def reset!
+    store.reset!
+  end
+
+  private
+    def store
+      @@store ||= KuberKit::Core::Store.new(KuberKit::Core::BuildServers::AbstractBuildServer)
+    end
+end

data/lib/kuber_kit/core/configuration.rb

@@ -1,5 +1,6 @@
 class KuberKit::Core::Configuration
-  attr_reader :name, :artifacts, :registries, :env_files, :templates, :kubeconfig_path, :deploy_strategy, :services_attributes
+  attr_reader :name, :artifacts, :registries, :env_files, :templates, :kubeconfig_path, 
+              :deploy_strategy, :services_attributes, :build_servers
 
   Contract KeywordArgs[
     name:            Symbol,
@@ -9,9 +10,11 @@ class KuberKit::Core::Configuration
     templates:       Hash,
     kubeconfig_path: Maybe[String],
     deploy_strategy: Symbol,
-    services_attributes: HashOf[Symbol => Hash]
+    services_attributes: HashOf[Symbol => Hash],
+    build_servers:   ArrayOf[KuberKit::Core::BuildServers::AbstractBuildServer]
   ] => Any
-  def initialize(name:, artifacts:, registries:, env_files:, templates:, kubeconfig_path:, deploy_strategy:, services_attributes:)
+  def initialize(name:, artifacts:, registries:, env_files:, templates:, kubeconfig_path:, 
+                 deploy_strategy:, services_attributes:, build_servers:)
     @name             = name
     @artifacts        = artifacts
     @registries       = registries
@@ -20,6 +23,7 @@ class KuberKit::Core::Configuration
     @kubeconfig_path  = kubeconfig_path
     @deploy_strategy  = deploy_strategy
     @services_attributes = services_attributes
+    @build_servers    = build_servers
   end
 
   def service_attributes(service_name)

data/lib/kuber_kit/core/configuration_definition.rb

@@ -10,6 +10,7 @@ class KuberKit::Core::ConfigurationDefinition
     @registries = {}
     @env_files  = {}
     @templates  = {}
+    @build_servers      = []
     @enabled_services   = []
     @services_attributes = {}
   end
@@ -24,6 +25,7 @@ class KuberKit::Core::ConfigurationDefinition
       kubeconfig_path:  @kubeconfig_path,
       deploy_strategy:  @deploy_strategy,
       enabled_services: @enabled_services,
+      build_servers:    @build_servers,
       services_attributes: @services_attributes
     )
   end
@@ -64,6 +66,14 @@ class KuberKit::Core::ConfigurationDefinition
     self
   end
 
+  def use_build_server(build_server_name)
+    unless @build_servers.include?(build_server_name)
+      @build_servers.push(build_server_name)
+    end
+
+    self
+  end
+
   def kubeconfig_path(path)
     @kubeconfig_path = path
 

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,8 @@ 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
+      services_attributes: configuration_attrs.services_attributes,
+      build_servers:    build_servers
     )
   end
 
@@ -61,4 +64,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