kuber_kit 0.1.5 → 0.2.0

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
+  attr_reader :name, :artifacts, :registries, :env_files, :templates, :kubeconfig_path, 
+              :deploy_strategy, :services_attributes, :build_servers
 
   Contract KeywordArgs[
     name:            Symbol,
@@ -7,10 +8,13 @@ class KuberKit::Core::Configuration
     registries:      Hash,
     env_files:       Hash,
     templates:       Hash,
-    kubeconfig_path:    Maybe[String],
-    deploy_strategy: Symbol
+    kubeconfig_path: Maybe[String],
+    deploy_strategy: Symbol,
+    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:)
+  def initialize(name:, artifacts:, registries:, env_files:, templates:, kubeconfig_path:, 
+                 deploy_strategy:, services_attributes:, build_servers:)
     @name             = name
     @artifacts        = artifacts
     @registries       = registries
@@ -18,5 +22,11 @@ class KuberKit::Core::Configuration
     @templates        = templates
     @kubeconfig_path  = kubeconfig_path
     @deploy_strategy  = deploy_strategy
+    @services_attributes = services_attributes
+    @build_servers    = build_servers
+  end
+
+  def service_attributes(service_name)
+    services_attributes[service_name.to_sym] || {}
   end
 end

data/lib/kuber_kit/core/configuration_definition.rb

@@ -10,6 +10,9 @@ class KuberKit::Core::ConfigurationDefinition
     @registries = {}
     @env_files  = {}
     @templates  = {}
+    @build_servers      = []
+    @enabled_services   = []
+    @services_attributes = {}
   end
 
   def to_attrs
@@ -20,7 +23,10 @@ class KuberKit::Core::ConfigurationDefinition
       env_files:        @env_files,
       templates:        @templates,
       kubeconfig_path:  @kubeconfig_path,
-      deploy_strategy:  @deploy_strategy
+      deploy_strategy:  @deploy_strategy,
+      enabled_services: @enabled_services,
+      build_servers:    @build_servers,
+      services_attributes: @services_attributes
     )
   end
 
@@ -60,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
 
@@ -71,4 +85,11 @@ class KuberKit::Core::ConfigurationDefinition
 
     self
   end
+
+  def enabled_services(services_hash)
+    @enabled_services += services_hash.keys.map(&:to_sym)
+    @services_attributes = @services_attributes.merge(services_hash)
+
+    self
+  end
 end

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,
@@ -24,7 +26,9 @@ class KuberKit::Core::ConfigurationFactory
       env_files:        env_files,
       templates:        templates,
       kubeconfig_path:  configuration_attrs.kubeconfig_path,
-      deploy_strategy:  configuration_attrs.deploy_strategy || configs.deploy_strategy
+      deploy_strategy:  configuration_attrs.deploy_strategy || configs.deploy_strategy,
+      services_attributes: configuration_attrs.services_attributes,
+      build_servers:    build_servers
     )
   end
 
@@ -60,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/context_helper/base_helper.rb

@@ -1,25 +1,30 @@
 class KuberKit::Core::ContextHelper::BaseHelper
   CONTRACT = RespondTo[:get_binding]
 
-  attr_reader :shell, :artifact_store, :image_store
+  attr_reader :shell, :artifact_store, :image_store, :env_file_reader
 
-  def initialize(image_store:, artifact_store:, shell:)
-    @image_store    = image_store
-    @artifact_store = artifact_store
-    @shell          = shell
+  def initialize(image_store:, artifact_store:, shell:, env_file_reader:)
+    @image_store      = image_store
+    @artifact_store   = artifact_store
+    @shell            = shell
+    @env_file_reader  = env_file_reader
   end
 
   def image_url(image_name)
-    image = @image_store.get_image(image_name)
+    image = @image_store.get_image(image_name.to_sym)
 
     image.remote_registry_url
   end
 
   def artifact_path(name, file_name = nil)
-    artifact = @artifact_store.get(name)
+    artifact = @artifact_store.get(name.to_sym)
     [artifact.cloned_path, file_name].compact.join("/")
   end
 
+  def env_file(env_file_name)
+    @env_file_reader.call(@shell, env_file_name)
+  end
+
   def configuration_name
     KuberKit.current_configuration.name
   end

data/lib/kuber_kit/core/context_helper/context_helper_factory.rb

@@ -1,23 +1,26 @@
 class KuberKit::Core::ContextHelper::ContextHelperFactory
   include KuberKit::Import[
     "core.image_store",
-    "core.artifact_store"
+    "core.artifact_store",
+    env_file_reader: "env_file_reader.action_handler"
   ]
 
   def build_image_context(shell)
     KuberKit::Core::ContextHelper::ImageHelper.new(
-      image_store:    image_store,
-      artifact_store: artifact_store,
-      shell:          shell
+      image_store:      image_store,
+      artifact_store:   artifact_store,
+      shell:            shell,
+      env_file_reader:  env_file_reader
     )
   end
 
   def build_service_context(shell, service)
     KuberKit::Core::ContextHelper::ServiceHelper.new(
-      image_store:    image_store,
-      artifact_store: artifact_store,
-      shell:          shell,
-      service:        service
+      image_store:      image_store,
+      artifact_store:   artifact_store,
+      shell:            shell,
+      env_file_reader:  env_file_reader,
+      service:          service,
     )
   end
 end

data/lib/kuber_kit/core/context_helper/service_helper.rb

@@ -1,9 +1,10 @@
 class KuberKit::Core::ContextHelper::ServiceHelper < KuberKit::Core::ContextHelper::BaseHelper
-  def initialize(image_store:, artifact_store:, shell:, service:)
+  def initialize(image_store:, artifact_store:, shell:, env_file_reader:, service:)
     super(
-      image_store:    image_store, 
-      artifact_store: artifact_store, 
-      shell:          shell
+      image_store:      image_store, 
+      artifact_store:   artifact_store, 
+      shell:            shell,
+      env_file_reader:  env_file_reader
     )
     @service = service
   end
@@ -16,7 +17,7 @@ class KuberKit::Core::ContextHelper::ServiceHelper < KuberKit::Core::ContextHelp
     @service.uri
   end
 
-  def attribute(attribute_name)
-    @service.attribute(attribute_name)
+  def attribute(attribute_name, default: nil)
+    @service.attribute(attribute_name, default: default)
   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