kuber_kit 0.1.4 → 0.1.9

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,6 +20,15 @@ class KuberKit::Core::Templates::TemplateStore
   end
 
   def reset!
-    @@templates = {}
+    store.reset!
+  end
+
+  def exists?(template_name)
+    store.exists?(template_name)
   end
+
+  private
+    def store
+      @@store ||= KuberKit::Core::Store.new(KuberKit::Core::Templates::AbstractTemplate)
+    end
 end

data/lib/kuber_kit/env_file_reader/action_handler.rb

@@ -0,0 +1,12 @@
+class KuberKit::EnvFileReader::ActionHandler
+  include KuberKit::Import[
+    "env_file_reader.reader",
+    "core.env_file_store",
+  ]
+
+  Contract KuberKit::Shell::AbstractShell, Symbol => Any
+  def call(shell, env_file_name)
+    env_file = env_file_store.get(env_file_name)
+    reader.read(shell, env_file)
+  end
+end

data/lib/kuber_kit/env_file_reader/reader.rb

@@ -2,14 +2,14 @@ class KuberKit::EnvFileReader::Reader
   ReaderNotFoundError = Class.new(KuberKit::NotFoundError)
 
   include KuberKit::Import[
-    "env_file_reader.artifact_file_reader",
+    "env_file_reader.strategies.artifact_file",
   ]
 
   def use_reader(env_file_reader, env_file_class:)
     @@readers ||= {}
 
-    if !env_file_reader.is_a?(KuberKit::EnvFileReader::AbstractEnvFileReader)
-      raise ArgumentError.new("should be an instance of KuberKit::EnvFileReader::AbstractEnvFileReader, got: #{env_file_reader.inspect}")
+    if !env_file_reader.is_a?(KuberKit::EnvFileReader::Strategies::Abstract)
+      raise ArgumentError.new("should be an instance of KuberKit::EnvFileReader::Strategies::Abstract, got: #{env_file_reader.inspect}")
     end
 
     @@readers[env_file_class] = env_file_reader
@@ -26,7 +26,7 @@ class KuberKit::EnvFileReader::Reader
   end
 
   def add_default_readers
-    use_reader(artifact_file_reader, env_file_class: KuberKit::Core::EnvFiles::ArtifactFile)
+    use_reader(artifact_file, env_file_class: KuberKit::Core::EnvFiles::ArtifactFile)
   end
 
   def reset!

data/lib/kuber_kit/env_file_reader/{abstract_env_file_reader.rb → strategies/abstract.rb}

@@ -1,4 +1,4 @@
-class KuberKit::EnvFileReader::AbstractEnvFileReader
+class KuberKit::EnvFileReader::Strategies::Abstract
   def read(shell, env_file)
     raise KuberKit::NotImplementedError, "must be implemented"
   end

data/lib/kuber_kit/env_file_reader/{artifact_file_reader.rb → strategies/artifact_file.rb}

@@ -1,4 +1,4 @@
-class KuberKit::EnvFileReader::ArtifactFileReader < KuberKit::EnvFileReader::AbstractEnvFileReader
+class KuberKit::EnvFileReader::Strategies::ArtifactFile < KuberKit::EnvFileReader::Strategies::Abstract
   include KuberKit::Import[
     "core.artifact_store"
   ]

data/lib/kuber_kit/image_compiler/action_handler.rb

@@ -0,0 +1,21 @@
+class KuberKit::ImageCompiler::ActionHandler
+  include KuberKit::Import[
+    "image_compiler.compiler",
+    "core.image_store",
+    "configs",
+  ]
+
+  Contract KuberKit::Shell::AbstractShell, Symbol, String => Any
+  def call(shell, image_name, build_id)
+    image = image_store.get_image(image_name)
+    
+    compile_dir = generate_compile_dir(build_id: build_id)
+
+    compiler.compile(shell, image, compile_dir)
+  end
+
+  private
+    def generate_compile_dir(build_id:)
+      File.join(configs.image_compile_dir, build_id)
+    end
+end

data/lib/kuber_kit/image_compiler/build_server_pool.rb

@@ -0,0 +1,30 @@
+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
+    @ssh_shells      = connect_to_ssh_shells(build_servers)
+  end
+
+  def get_shell
+    if @ssh_shells.any?
+      @ssh_shells.sample
+    else
+      @local_shell
+    end
+  end
+
+  def disconnect_all
+    @ssh_shells.each(&:disconnect)
+  end
+
+  private
+    def connect_to_ssh_shells(build_servers)
+      build_servers.map do |bs|
+        shell = @ssh_shell_class.new
+        shell.connect(host: bs.host, user: bs.user, port: bs.port)
+        shell
+      end
+    end
+end

data/lib/kuber_kit/image_compiler/build_server_pool_factory.rb

@@ -0,0 +1,13 @@
+class KuberKit::ImageCompiler::BuildServerPoolFactory
+  include KuberKit::Import[
+    "shell.local_shell",
+  ]
+
+  def create(ssh_shell_class: KuberKit::Shell::SshShell)
+    KuberKit::ImageCompiler::BuildServerPool.new(
+      local_shell:      local_shell,
+      build_servers:    KuberKit.current_configuration.build_servers,
+      ssh_shell_class:  ssh_shell_class,
+    )
+  end
+end

data/lib/kuber_kit/image_compiler/compiler.rb

@@ -3,13 +3,10 @@ class KuberKit::ImageCompiler::Compiler
     "image_compiler.image_build_dir_creator",
     "image_compiler.image_builder",
     "core.context_helper_factory",
-    "core.image_store"
   ]
 
-  Contract KuberKit::Shell::AbstractShell, Symbol, String => Any
-  def compile(shell, image_name, builds_dir)
-    image = image_store.get_image(image_name)
-
+  Contract KuberKit::Shell::AbstractShell, KuberKit::Core::Image, String => Any
+  def compile(shell, image, builds_dir)
     image_build_dir = File.join(builds_dir, image.name.to_s)
 
     context_helper = context_helper_factory.build_image_context(shell)

data/lib/kuber_kit/image_compiler/image_build_dir_creator.rb

@@ -1,8 +1,8 @@
 class KuberKit::ImageCompiler::ImageBuildDirCreator
   include KuberKit::Import[
-    "preprocessing.dir_preprocessor",
     "preprocessing.file_preprocessor",
     "shell.bash_commands",
+    "shell.local_shell",
     "configs"
   ]
 
@@ -14,16 +14,22 @@ class KuberKit::ImageCompiler::ImageBuildDirCreator
     bash_commands.mkdir_p(shell, build_dir)
 
     if image.build_context_dir
-      dir_preprocessor.compile(
-        shell, image.build_context_dir, build_dir,
-        context_helper: context_helper
-      )
+      # Sync build context and then preprocess
+      shell.sync(image.build_context_dir, build_dir)
+
+      shell.recursive_list_files(build_dir).each do |file_path|
+        file_preprocessor.compile(
+          shell, file_path, 
+          context_helper: context_helper
+        )
+      end
     end
 
+    # Sync dockerfile and then preprocess
     target_dockerfile = File.join(build_dir, configs.image_dockerfile_name)
+    shell.sync(image.dockerfile_path, target_dockerfile)
     file_preprocessor.compile(
-      shell, image.dockerfile_path, 
-      destination_path: target_dockerfile, 
+      shell, target_dockerfile,
       context_helper: context_helper
     )
 

data/lib/kuber_kit/image_compiler/image_dependency_resolver.rb

@@ -3,20 +3,40 @@ class KuberKit::ImageCompiler::ImageDependencyResolver
   DependencyNotFoundError = Class.new(KuberKit::NotFoundError)
 
   include KuberKit::Import[
-    "core.image_store"
+    "core.image_store",
+    "configs"
   ]
+
+  Contract Or[Symbol, ArrayOf[Symbol]], Proc => Any
+  def each_with_deps(image_names, &block)
+    compile_limit = configs.compile_simultaneous_limit
+
+    resolved_dependencies = []
+    next_dependencies = get_next(image_names, limit: compile_limit)
+
+    while (next_dependencies - resolved_dependencies).any?
+      block.call(next_dependencies)
+      resolved_dependencies += next_dependencies
+      next_dependencies = get_next(image_names, resolved: resolved_dependencies, limit: compile_limit)
+    end
+
+    block.call(image_names - resolved_dependencies)
+  end
   
-  Contract Any, KeywordArgs[
-    resolved: Optional[ArrayOf[Symbol]]
+  Contract Or[Symbol, ArrayOf[Symbol]], KeywordArgs[
+    resolved: Optional[ArrayOf[Symbol]],
+    limit:    Optional[Maybe[Num]]
   ] => Any
-  def get_next(image_names, resolved: [])
+  def get_next(image_names, resolved: [], limit: nil)
     deps = Array(image_names).map { |i| get_recursive_deps(i) }.flatten.uniq
 
     ready_to_resolve = deps.select do |dep_name|
       unresolved_deps = get_deps(dep_name) - resolved
       unresolved_deps.empty?
     end
-    ready_to_resolve - resolved
+    unresolved_deps = ready_to_resolve - resolved
+    unresolved_deps = unresolved_deps.take(limit) if limit
+    unresolved_deps
   end
 
   def get_recursive_deps(image_name, dependency_tree: [])

data/lib/kuber_kit/preprocessing/file_preprocessor.rb

@@ -1,6 +1,7 @@
 class KuberKit::Preprocessing::FilePreprocessor
   include KuberKit::Import[
-    "preprocessing.text_preprocessor"
+    "preprocessing.text_preprocessor",
+    "shell.bash_commands"
   ]
 
   PreprocessingError = Class.new(KuberKit::Error)
@@ -8,7 +9,7 @@ class KuberKit::Preprocessing::FilePreprocessor
   def compile(shell, source_path, destination_path: nil, context_helper: nil)
     destination_path ||= source_path
 
-    prepare_destination_dir(destination_path)
+    prepare_destination_dir(shell, destination_path)
 
     template = shell.read(source_path)
     content = text_preprocessor.compile(template, context_helper: context_helper)
@@ -27,7 +28,7 @@ class KuberKit::Preprocessing::FilePreprocessor
     raise PreprocessingError, "Error while processing template #{source_path}.\r\n#{message}"
   end
 
-  def prepare_destination_dir(destination_path)
-    FileUtils.mkdir_p(File.dirname(destination_path))
+  def prepare_destination_dir(shell, destination_path)
+    bash_commands.mkdir_p(shell, File.dirname(destination_path))
   end
 end

data/lib/kuber_kit/preprocessing/text_preprocessor.rb

@@ -2,6 +2,6 @@ require 'erb'
 
 class KuberKit::Preprocessing::TextPreprocessor
   def compile(template, context_helper: nil)
-    ERB.new(template).result(context_helper&.get_binding)
+    ERB.new(template, nil, '-').result(context_helper&.get_binding)
   end
 end

data/lib/kuber_kit/service_deployer/action_handler.rb

@@ -0,0 +1,16 @@
+class KuberKit::ServiceDeployer::ActionHandler
+  include KuberKit::Import[
+    "service_deployer.deployer",
+    "service_deployer.strategy_detector",
+    "core.service_store",
+  ]
+
+  Contract KuberKit::Shell::AbstractShell, Symbol => Any
+  def call(shell, service_name)
+    service = service_store.get_service(service_name)
+
+    strategy_name = strategy_detector.call(service)
+
+    deployer.deploy(shell, service, strategy_name)
+  end
+end

data/lib/kuber_kit/service_deployer/deployer.rb

@@ -1,15 +1,37 @@
 class KuberKit::ServiceDeployer::Deployer
+  StrategyNotFoundError = Class.new(KuberKit::NotFoundError)
+
   include KuberKit::Import[
-    "service_deployer.service_restarter",
     "core.service_store",
+    "service_deployer.strategies.kubernetes"
   ]
 
-  Contract KuberKit::Shell::AbstractShell, Symbol => Any
-  def deploy(shell, service_name)
-    service = service_store.get_service(service_name)
+  def register_strategy(strategy_name, strategy)
+    @@strategies ||= {}
+
+    if !strategy.is_a?(KuberKit::ServiceDeployer::Strategies::Abstract)
+      raise ArgumentError.new("should be an instance of KuberKit::ServiceDeployer::Strategies::Abstract, got: #{strategy.inspect}")
+    end
+
+    @@strategies[strategy_name] = strategy
+  end
+
+  Contract KuberKit::Shell::AbstractShell, KuberKit::Core::Service, Symbol => Any
+  def deploy(shell, service, strategy_name)
+    add_default_strategies
 
-    strategy_name = KuberKit.current_configuration.deploy_strategy
+    deployer = @@strategies[strategy_name]
+
+    raise StrategyNotFoundError, "Can't find strategy with name #{strategy_name}" if deployer.nil?
+
+    deployer.deploy(shell, service)
+  end
+
+  def add_default_strategies
+    register_strategy(:kubernetes, kubernetes)
+  end
 
-    service_restarter.restart(shell, service, strategy_name)
+  def reset!
+    @@strategies = {}
   end
 end

data/lib/kuber_kit/service_deployer/strategies/abstract.rb

@@ -1,5 +1,5 @@
 class KuberKit::ServiceDeployer::Strategies::Abstract
-  def restart(shell, service)
+  def deploy(shell, service)
     raise KuberKit::NotImplementedError, "must be implemented"
   end
 end

data/lib/kuber_kit/service_deployer/strategies/kubernetes.rb

@@ -1,18 +1,23 @@
 class KuberKit::ServiceDeployer::Strategies::Kubernetes < KuberKit::ServiceDeployer::Strategies::Abstract
   include KuberKit::Import[
-    "service_deployer.service_reader",
+    "service_reader.reader",
     "shell.kubectl_commands",
     "configs",
   ]
 
   Contract KuberKit::Shell::AbstractShell, KuberKit::Core::Service => Any
-  def restart(shell, service)
-    service_config = service_reader.read(shell, service)
+  def deploy(shell, service)
+    service_config = reader.read(shell, service)
     config_path    = "#{configs.service_config_dir}/#{service.name}.yml"
     shell.write(config_path, service_config)
 
     kubeconfig_path = KuberKit.current_configuration.kubeconfig_path
     kubectl_commands.apply_file(shell, config_path, kubeconfig_path: kubeconfig_path)
-    kubectl_commands.rolling_restart(shell, service.uri, kubeconfig_path: kubeconfig_path)
+
+    deployment_restart_enabled = service.attribute(:deployment_restart_enabled, default: true)
+    deployment_restart_name    = service.attribute(:deployment_restart_name, default: service.uri)
+    if deployment_restart_enabled
+      kubectl_commands.rolling_restart(shell, deployment_restart_name, kubeconfig_path: kubeconfig_path)
+    end
   end
 end