kuber_kit 0.1.7 → 0.2.2

data/lib/kuber_kit/actions/service_reader.rb

@@ -10,5 +10,11 @@ class KuberKit::Actions::ServiceReader
     result = service_reader.call(local_shell, service_name)
 
     ui.print_info(service_name.to_s, result)
+
+    true
+  rescue KuberKit::Error => e
+    ui.print_error("Error", e.message)
+
+    false
   end
 end

data/lib/kuber_kit/actions/template_reader.rb

@@ -13,5 +13,10 @@ class KuberKit::Actions::TemplateReader
     result = reader.read(local_shell, template)
 
     ui.print_info(template_name.to_s, result)
+
+    true
+  rescue KuberKit::Error => e
+    ui.print_error("Error", e.message)
+    false
   end
 end

data/lib/kuber_kit/cli.rb

@@ -14,56 +14,90 @@ class KuberKit::CLI < Thor
     
     image_names = image_names_str.split(",").map(&:strip).map(&:to_sym)
 
-    KuberKit::Container['actions.configuration_loader'].call(options)
-    KuberKit::Container['actions.image_compiler'].call(image_names, options)
+    if KuberKit::Container['actions.configuration_loader'].call(options)
+      result = KuberKit::Container['actions.image_compiler'].call(image_names, options)
+    end
 
     logger = KuberKit::Container['tools.logger']
-    logger.info("---------------------------")
-    logger.info("Image compilation finished!")
-    logger.info("---------------------------")
+    if result
+      logger.info("---------------------------")
+      logger.info("Image compilation finished!")
+      logger.info("---------------------------")
+    else
+      logger.info("-------------------------".red)
+      logger.info("Image compilation failed!".red)
+      logger.info("-------------------------".red)
+    end
+  end
+
+  desc "deploy CONTEXT_NAME", "Deploy CONTEXT_NAME with kubectl"
+  method_option :services,   :type => :array, aliases: ["-s"]
+  method_option :tags,       :type => :array, aliases: ["-t"]
+  def deploy
+    KuberKit.set_debug_mode(options[:debug])
+
+    if KuberKit::Container['actions.configuration_loader'].call(options)
+      result = KuberKit::Container['actions.service_deployer'].call(
+        services: options[:services] || [], 
+        tags:     options[:tags] || []
+      )
+    end
+
+    logger = KuberKit::Container['tools.logger']
+    if result
+      logger.info("---------------------------")
+      logger.info("Service deployment finished!")
+      logger.info("---------------------------")
+    else
+      logger.info("-------------------------".red)
+      logger.info("Service deployment failed!".red)
+      logger.info("-------------------------".red)
+    end
   end
 
   desc "env ENV_FILE_NAME", "Return content of Env File ENV_FILE_NAME"
   def env(env_file_name)
     KuberKit.set_debug_mode(options[:debug])
 
-    KuberKit::Container['actions.configuration_loader'].call(options)
-    KuberKit::Container['actions.env_file_reader'].call(env_file_name.to_sym, options)
+    if KuberKit::Container['actions.configuration_loader'].call(options)
+      KuberKit::Container['actions.env_file_reader'].call(env_file_name.to_sym, options)
+    end
   end
 
   desc "template TEMPLATE_NAME", "Return content of Template TEMPLATE_NAME"
   def template(template_name)
     KuberKit.set_debug_mode(options[:debug])
 
-    KuberKit::Container['actions.configuration_loader'].call(options)
-    KuberKit::Container['actions.template_reader'].call(template_name.to_sym, options)
+    if KuberKit::Container['actions.configuration_loader'].call(options)
+      KuberKit::Container['actions.template_reader'].call(template_name.to_sym, options)
+    end
   end
 
   desc "service SERVICE_NAME", "Return content of Service SERVICE_NAME"
   def service(service_name)
     KuberKit.set_debug_mode(options[:debug])
 
-    KuberKit::Container['actions.configuration_loader'].call(options)
-    KuberKit::Container['actions.service_reader'].call(service_name.to_sym, options)
+    if KuberKit::Container['actions.configuration_loader'].call(options)
+      KuberKit::Container['actions.service_reader'].call(service_name.to_sym, options)
+    end
   end
 
   desc "apply FILE_PATH", "Apply FILE_PATH with kubectl"
   def apply(file_path)
     KuberKit.set_debug_mode(options[:debug])
 
-    KuberKit::Container['actions.configuration_loader'].call(options)
-    KuberKit::Container['actions.kubectl_applier'].call(File.expand_path(file_path), options)
+    if KuberKit::Container['actions.configuration_loader'].call(options)
+      KuberKit::Container['actions.kubectl_applier'].call(File.expand_path(file_path), options)
+    end
   end
 
-  desc "deploy CONTEXT_NAME", "Deploy CONTEXT_NAME with kubectl"
-  method_option :services,   :type => :array, aliases: ["-s"]
-  method_option :tags,       :type => :array, aliases: ["-t"]
-  def deploy
+  desc "attach POD_NAME", "Attach to POD_NAME with kubectl"
+  def attach(pod_name)
     KuberKit.set_debug_mode(options[:debug])
 
-    KuberKit::Container['actions.configuration_loader'].call(options)
-
-    KuberKit::Container['actions.service_deployer'].call(services: options[:services], tags: options[:tags])
+    if KuberKit::Container['actions.configuration_loader'].call(options)
+      KuberKit::Container['actions.kubectl_attacher'].call(pod_name, options)
+    end
   end
 
   def self.exit_on_failure?

data/lib/kuber_kit/configs.rb

@@ -14,33 +14,35 @@ class KuberKit::Configs
     'tmp',
     'logs'
   ]
-  KUBER_KIT_DIRNAME      = "kuber_kit".freeze
-  IMAGES_DIRNAME         = "images".freeze
-  SERVICES_DIRNAME       = "services".freeze
-  INFRA_DIRNAME          = "infrastructure".freeze
-  CONFIGURATIONS_DIRNAME = "configurations".freeze
-  ARTIFACT_CLONE_DIR     = "/tmp/kuber_kit/artifacts"
-  SERVICE_CONFIG_DIR     = "/tmp/kuber_kit/services"
-  DEPLOY_STRATEGY        = :kubernetes
+  KUBER_KIT_DIRNAME           = "kuber_kit".freeze
+  IMAGES_DIRNAME              = "images".freeze
+  SERVICES_DIRNAME            = "services".freeze
+  INFRA_DIRNAME               = "infrastructure".freeze
+  CONFIGURATIONS_DIRNAME      = "configurations".freeze
+  ARTIFACT_CLONE_DIR          = "/tmp/kuber_kit/artifacts"
+  SERVICE_CONFIG_DIR          = "/tmp/kuber_kit/services"
+  DEPLOY_STRATEGY             = :kubernetes
+  COMPILE_SIMULTANEOUS_LIMIT  = 5
 
   attr_accessor :image_dockerfile_name, :image_build_context_dir, :image_tag,
                 :docker_ignore_list, :image_compile_dir, 
                 :kuber_kit_dirname, :images_dirname, :services_dirname, :infra_dirname, :configurations_dirname,
-                :artifact_clone_dir, :service_config_dir, :deploy_strategy
+                :artifact_clone_dir, :service_config_dir, :deploy_strategy, :compile_simultaneous_limit
 
   def initialize
-    @image_dockerfile_name   = IMAGE_DOCKERFILE_NAME
-    @image_build_context_dir = IMAGE_BUILD_CONTEXT_DIR
-    @image_tag               = IMAGE_TAG
-    @image_compile_dir       = IMAGE_COMPILE_DIR
-    @docker_ignore_list      = DOCKER_IGNORE_LIST
-    @kuber_kit_dirname       = KUBER_KIT_DIRNAME
-    @images_dirname          = IMAGES_DIRNAME
-    @services_dirname        = SERVICES_DIRNAME
-    @infra_dirname           = INFRA_DIRNAME
-    @configurations_dirname  = CONFIGURATIONS_DIRNAME
-    @artifact_clone_dir      = ARTIFACT_CLONE_DIR
-    @service_config_dir      = SERVICE_CONFIG_DIR
-    @deploy_strategy         = DEPLOY_STRATEGY
+    @image_dockerfile_name      = IMAGE_DOCKERFILE_NAME
+    @image_build_context_dir    = IMAGE_BUILD_CONTEXT_DIR
+    @image_tag                  = IMAGE_TAG
+    @image_compile_dir          = IMAGE_COMPILE_DIR
+    @docker_ignore_list         = DOCKER_IGNORE_LIST
+    @kuber_kit_dirname          = KUBER_KIT_DIRNAME
+    @images_dirname             = IMAGES_DIRNAME
+    @services_dirname           = SERVICES_DIRNAME
+    @infra_dirname              = INFRA_DIRNAME
+    @configurations_dirname     = CONFIGURATIONS_DIRNAME
+    @artifact_clone_dir         = ARTIFACT_CLONE_DIR
+    @service_config_dir         = SERVICE_CONFIG_DIR
+    @deploy_strategy            = DEPLOY_STRATEGY
+    @compile_simultaneous_limit = COMPILE_SIMULTANEOUS_LIMIT
   end
 end

data/lib/kuber_kit/container.rb

@@ -29,6 +29,10 @@ class KuberKit::Container
     KuberKit::Actions::KubectlApplier.new
   end
 
+  register "actions.kubectl_attacher" do
+    KuberKit::Actions::KubectlAttacher.new
+  end
+
   register "configs" do
     KuberKit::Configs.new
   end
@@ -69,18 +73,22 @@ class KuberKit::Container
     KuberKit::Core::ConfigurationStore.new
   end
 
-  register "core.registry_store" do
-    KuberKit::Core::Registries::RegistryStore.new
-  end
-
   register "core.artifact_store" do
     KuberKit::Core::Artifacts::ArtifactStore.new
   end
 
+  register "core.build_server_store" do
+    KuberKit::Core::BuildServers::BuildServerStore.new
+  end
+
   register "core.env_file_store" do
     KuberKit::Core::EnvFiles::EnvFileStore.new
   end
 
+  register "core.registry_store" do
+    KuberKit::Core::Registries::RegistryStore.new
+  end
+
   register "core.template_store" do
     KuberKit::Core::Templates::TemplateStore.new
   end
@@ -102,23 +110,23 @@ class KuberKit::Container
   end
 
   register "shell.bash_commands" do
-    KuberKit::Shell::BashCommands.new
+    KuberKit::Shell::Commands::BashCommands.new
   end
 
   register "shell.docker_commands" do
-    KuberKit::Shell::DockerCommands.new
+    KuberKit::Shell::Commands::DockerCommands.new
   end
 
   register "shell.git_commands" do
-    KuberKit::Shell::GitCommands.new
+    KuberKit::Shell::Commands::GitCommands.new
   end
 
   register "shell.rsync_commands" do
-    KuberKit::Shell::RsyncCommands.new
+    KuberKit::Shell::Commands::RsyncCommands.new
   end
 
   register "shell.kubectl_commands" do
-    KuberKit::Shell::KubectlCommands.new
+    KuberKit::Shell::Commands::KubectlCommands.new
   end
 
   register "shell.local_shell" do
@@ -137,14 +145,14 @@ class KuberKit::Container
     KuberKit::Preprocessing::FilePreprocessor.new
   end
 
-  register "preprocessing.dir_preprocessor" do
-    KuberKit::Preprocessing::DirPreprocessor.new
-  end
-
   register "image_compiler.action_handler" do
     KuberKit::ImageCompiler::ActionHandler.new
   end
 
+  register "image_compiler.build_server_pool_factory" do
+    KuberKit::ImageCompiler::BuildServerPoolFactory.new
+  end
+
   register "image_compiler.compiler" do
     KuberKit::ImageCompiler::Compiler.new
   end

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,6 +20,15 @@ class KuberKit::Core::Artifacts::ArtifactStore
   end
 
   def reset!
-    @@artifacts = {}
+    store.reset!
+  end
+
+  def exists?(artifact_name)
+    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, :deploy_namespace, :services_attributes, :build_servers
 
   Contract KeywordArgs[
     name:            Symbol,
@@ -8,10 +9,13 @@ class KuberKit::Core::Configuration
     env_files:       Hash,
     templates:       Hash,
     kubeconfig_path: Maybe[String],
-    deploy_strategy: Symbol,
-    services_attributes: HashOf[Symbol => Hash]
+    deploy_strategy:  Symbol,
+    deploy_namespace: Maybe[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:, services_attributes:)
+  def initialize(name:, artifacts:, registries:, env_files:, templates:, kubeconfig_path:, 
+                 deploy_strategy:, deploy_namespace:, services_attributes:, build_servers:)
     @name             = name
     @artifacts        = artifacts
     @registries       = registries
@@ -19,7 +23,9 @@ class KuberKit::Core::Configuration
     @templates        = templates
     @kubeconfig_path  = kubeconfig_path
     @deploy_strategy  = deploy_strategy
+    @deploy_namespace = deploy_namespace
     @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
@@ -23,8 +24,10 @@ class KuberKit::Core::ConfigurationDefinition
       templates:        @templates,
       kubeconfig_path:  @kubeconfig_path,
       deploy_strategy:  @deploy_strategy,
+      deploy_namespace: @deploy_namespace,
       enabled_services: @enabled_services,
-      services_attributes: @services_attributes
+      build_servers:    @build_servers,
+      services_attributes: @services_attributes,
     )
   end
 
@@ -64,12 +67,26 @@ 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
 
     self
   end
 
+  def deploy_namespace(namespace)
+    @deploy_namespace = namespace
+
+    self
+  end
+
   def deploy_strategy(path)
     @deploy_strategy = path