kuber_kit 1.1.4 → 1.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b70bdef220e67c9789122ab7eb7ee064dff475174ec1e497b0baf11566da6e1f
-  data.tar.gz: 004b9a6c4e8265f1f3f498b1b42fa35a0447567ef428abef4b571cce635c0cd9
+  metadata.gz: fffaf809577e0796e0b18bcc8d61894276308e5792ba1e3c67b7d792dce32d0a
+  data.tar.gz: 73470e476a996467fe60b6e6966b9821e14f75492d0b53aee231db21995dfeb8
 SHA512:
-  metadata.gz: 1e60b51c8108814fc41f89b41a28732e59a8f5bddb938ee94d8cb765ccc5f2ea3de260aed0a52a212649389814884c40734c7f60fe77502e78e3814821c31790
-  data.tar.gz: 5f165249b8d91cd8bc148f9cc5f3a3a9a57402355c371b30f7f8511fcb6bf8d0650a7ba7a08c072dcba550837e52aced3a1d9cec4959be98c18d5b526678fb29
+  metadata.gz: 13a911f81ba63ff223073ffa2d4363827a5334b08bd8d99ea027b5de0d135a4b4323b87ddaf650dc1e63e7515022cd4ca119c968a12b1c7adb6a2a3ef897feff
+  data.tar.gz: b26560006e1c3c27cbf4a25363a8c4e3a5ebf673f55ffc549e9bf4d415023eec8e6b18cc8052859bde4efb24fe9d0d1a82ff55d9e0ead9284c68a4959b02f225

data/CHANGELOG.md

@@ -1,3 +1,35 @@
+**1.3.3**
+- Support "partials" in the templates with "render" method.
+- Cleanup artifacts after deployment if cache_result: false
+- Improve fetching git artifact when branch name has changed
+
+**1.3.2**
+- Added an ability to generate helm templates using `kit generate` command
+
+**1.3.1**
+- Fix upgrade command for helm strategy
+
+**1.3.0**
+- Allow sending custom apply command for k8s deploy strategy
+- Added initial support for helm deploy strategy
+
+**1.2.7**
+- Added an option to skip deployment and only build images for `deploy` command
+
+**1.2.6**
+- Lock cli-ui version on 2.1.0 (2.2.x has some bugs)
+
+**1.2.5**
+- Improve handling ContextVars, ability to convert context var to OpenStruct
+
+**1.2.4**
+- Fix a mistake in setting env variables
+- Added an option to set "use local deployment" variable
+- Allow finding job resources in kit attach and kit log
+
+**1.2.2**
+- Support Ruby 3.2.0
+
 **1.2.1**
 - Update shell commands so that STDERR stream won't be merged for commands using the command result.
 - kit sh would also set current default configuration

data/README.md

@@ -19,7 +19,8 @@ Please install specific kuber_kit version, depending on Ruby version.
 | Ruby Version | KuberKit Version |
 | ------------ | ------------ |
 | 2.6 | 1.0.1 |
-| 2.7 | 1.1.4 |
+| 2.7 | 1.1.5 |
+| > 3.0 | 1.2.6 |
 
 ## Usage
 

data/kuber_kit.gemspec

@@ -25,11 +25,7 @@ Gem::Specification.new do |spec|
 
   spec.required_ruby_version = ">= 2.7.0"
 
-  if Gem::Version.new(RUBY_VERSION) >= Gem::Version.new('3.0.0')
-    spec.add_dependency "contracts", '0.17.0'
-  else
-    spec.add_dependency "contracts", '0.16.0'
-  end
+  spec.add_dependency "contracts", '0.16.0'
 
   spec.add_dependency "dry-auto_inject", "~> 0.9.0"
   spec.add_dependency "dry-core", "~> 0.8.1"
@@ -37,9 +33,10 @@ Gem::Specification.new do |spec|
   spec.add_dependency "dry-container", "~> 0.10.1"
 
   spec.add_dependency "thor"
-  spec.add_dependency "cli-ui"
+  spec.add_dependency "cli-ui", '2.1.0'
   spec.add_dependency "net-ssh"
   spec.add_dependency "tty-prompt"
+  spec.add_dependency 'ed25519'
 
   spec.add_development_dependency "bundler", "~> 2.2"
   spec.add_development_dependency "rake", "~> 10.0"

data/lib/kuber_kit/actions/configuration_loader.rb

@@ -20,6 +20,7 @@ class KuberKit::Actions::ConfigurationLoader
     configurations_path  = options[:configurations_path]  || File.join(root_path, configs.configurations_dirname)
     configuration_name   = options[:configuration] || ENV["KUBER_KIT_CONFIGURATION"]
     load_inventory = options.fetch(:load_inventory, true)
+    use_local_deploy = options.fetch(:use_local_deploy, false)
 
     ui.print_debug "ConfigurationLoader", "Launching kuber_kit with:"
     ui.print_debug "ConfigurationLoader", "  Root path: #{root_path.to_s.yellow}"
@@ -28,10 +29,11 @@ class KuberKit::Actions::ConfigurationLoader
     ui.print_debug "ConfigurationLoader", "  Infrastructure path: #{infra_path.to_s.yellow}"
     ui.print_debug "ConfigurationLoader", "  Configurations path: #{configurations_path.to_s.yellow}"
     ui.print_debug "ConfigurationLoader", "  Configuration name: #{configuration_name.to_s.yellow}"
+    ui.print_debug "ConfigurationLoader", "  Use local deploy: #{use_local_deploy.to_s.yellow}"
 
     ui.print_info("Logs", "See logs at: #{configs.log_file_path}")
 
-    unless File.exists?(root_path)
+    unless File.exist?(root_path)
       ui.print_warning "ConfigurationLoader", "KuberKit root path #{root_path} doesn't exist. You may want to pass it --path parameter."
     end
 
@@ -39,6 +41,10 @@ class KuberKit::Actions::ConfigurationLoader
       raise KuberKit::Error, "The minimal required kuber_kit version is #{configs.kuber_kit_min_version}"
     end
 
+    if use_local_deploy
+      ENV["KUBER_KIT_USE_LOCAL_DEPLOYMENT"] = "true"
+    end
+
     load_configurations(configurations_path, configuration_name)
     load_infrastructure(infra_path)
 

data/lib/kuber_kit/actions/kubectl_attacher.rb

@@ -7,17 +7,20 @@ class KuberKit::Actions::KubectlAttacher
   ]
 
   Contract Maybe[String], Hash => Any
-  def call(pod_name, options)
+  def call(resource_name, options)
     kubeconfig_path    = KuberKit.current_configuration.kubeconfig_path
     kubectl_entrypoint = KuberKit.current_configuration.kubectl_entrypoint
     deployer_namespace = KuberKit.current_configuration.deployer_namespace
 
-    if !pod_name
-      pod_name  = resource_selector.call("attach")
+    if !resource_name 
+      resource_name  = resource_selector.call("attach", additional_resources: [
+        KuberKit::Kubernetes::Resources::POD,
+        KuberKit::Kubernetes::Resources::JOB
+      ])
     end
 
     kubectl_commands.exec(
-      local_shell, pod_name, "bash", args: "-it", 
+      local_shell, resource_name, "bash", args: "-it", 
       kubeconfig_path: kubeconfig_path, 
       interactive:  true,
       namespace:    deployer_namespace,

data/lib/kuber_kit/actions/kubectl_logs.rb

@@ -7,12 +7,15 @@ class KuberKit::Actions::KubectlLogs
   ]
 
   Contract Maybe[String], Hash => Any
-  def call(pod_name, options)
+  def call(resource_name, options)
     kubeconfig_path = KuberKit.current_configuration.kubeconfig_path
     deployer_namespace = KuberKit.current_configuration.deployer_namespace
 
-    if !pod_name 
-      pod_name = resource_selector.call("attach")
+    if !resource_name 
+      resource_name  = resource_selector.call("attach", additional_resources: [
+        KuberKit::Kubernetes::Resources::POD,
+        KuberKit::Kubernetes::Resources::JOB
+      ])
     end
 
     args = nil
@@ -21,7 +24,7 @@ class KuberKit::Actions::KubectlLogs
     end
 
     kubectl_commands.logs(
-      local_shell, pod_name,
+      local_shell, resource_name,
       args: args,
       kubeconfig_path: kubeconfig_path, 
       namespace: deployer_namespace

data/lib/kuber_kit/actions/service_deployer.rb

@@ -29,11 +29,11 @@ class KuberKit::Actions::ServiceDeployer
       services, tags = deployment_options_selector.call()
     end
 
-    disabled_services = current_configuration.disabled_services.map(&:to_s)
-    disabled_services += skip_services if skip_services
-    default_services  = current_configuration.default_services.map(&:to_s) - disabled_services
-    initial_services  = current_configuration.initial_services.map(&:to_s) - disabled_services
-    initial_service_names = initial_services.map(&:to_sym)
+    disabled_services     = current_configuration.disabled_services.map(&:to_s)
+    disabled_services     += skip_services if skip_services
+    default_services      = current_configuration.default_services.map(&:to_s) - disabled_services
+    pre_deploy_services   = current_configuration.pre_deploy_services.map(&:to_s) - disabled_services
+    post_deploy_services  = current_configuration.post_deploy_services.map(&:to_s) - disabled_services
 
     service_names = service_list_resolver.resolve(
       services:         services || [],
@@ -50,17 +50,19 @@ class KuberKit::Actions::ServiceDeployer
       all_service_names = service_dependency_resolver.get_all(service_names)
     end
 
+    all_service_names_with_hooks  = (pre_deploy_services.map(&:to_sym) + all_service_names + post_deploy_services.map(&:to_sym)).uniq
+
     unless all_service_names.any?
       ui.print_warning "ServiceDeployer", "No service found with given options, nothing will be deployed."
       return false
     end
 
-    unless allow_deployment?(require_confirmation: require_confirmation, service_names: (initial_service_names + all_service_names).uniq)
+    unless allow_deployment?(require_confirmation: require_confirmation, service_names: all_service_names_with_hooks)
       return false
     end
 
     # Compile images for all services and dependencies
-    images_names = get_image_names(service_names: (initial_service_names + all_service_names).uniq)
+    images_names = get_image_names(service_names: all_service_names_with_hooks.uniq)
     unless skip_compile
       compilation_result = compile_images(images_names)
 
@@ -72,33 +74,31 @@ class KuberKit::Actions::ServiceDeployer
       return deployment_result
     end
 
-    # First deploy initial services. 
+    # First, deploy pre-deploy services.
     # This feature is used to deploy some services, required for deployment of other services, e.g. env files
     # Note: Initial services are deployed without dependencies
-    initial_services.map(&:to_sym).each_slice(configs.deploy_simultaneous_limit) do |batch_service_names|
-      ui.print_debug("ServiceDeployer", "Scheduling to compile: #{batch_service_names.inspect}. Limit: #{configs.deploy_simultaneous_limit}")
+    pre_deploy_services.map(&:to_sym).each_slice(configs.deploy_simultaneous_limit) do |batch_service_names|
+      deploy_simultaneously(batch_service_names, deployment_result)
+    end
 
-      if deployment_result.succeeded?
+    # Next, deploy all initializers simultaneously.
+    # Note: In earlier versions, KuberKit would deploy all dependencies in the order defined by dependency tree.
+    #       Now it would deploy all dependencies (initializers) at the same time, even if one initializer depends on another initializer.
+    unless skip_dependencies
+      initializers = service_dependency_resolver.get_all_deps(service_names)
+      initializers.map(&:to_sym).each_slice(configs.deploy_simultaneous_limit) do |batch_service_names|
         deploy_simultaneously(batch_service_names, deployment_result)
       end
     end
 
-    if skip_dependencies
-      service_names.each_slice(configs.deploy_simultaneous_limit) do |batch_service_names|
-        ui.print_debug("ServiceDeployer", "Scheduling to deploy: #{batch_service_names.inspect}. Limit: #{configs.deploy_simultaneous_limit}")
-
-        if deployment_result.succeeded?
-          deploy_simultaneously(batch_service_names, deployment_result)
-        end
-      end
-    else
-      service_dependency_resolver.each_with_deps(service_names) do |dep_service_names|
-        ui.print_debug("ServiceDeployer", "Scheduling to deploy: #{dep_service_names.inspect}. Limit: #{configs.deploy_simultaneous_limit}")
+    # Next, deploy all requested services.
+    service_names.each_slice(configs.deploy_simultaneous_limit) do |batch_service_names|
+      deploy_simultaneously(batch_service_names, deployment_result)
+    end
 
-        if deployment_result.succeeded?
-          deploy_simultaneously(dep_service_names, deployment_result)
-        end
-      end
+    # Last, deploy post-deploy services.
+    post_deploy_services.map(&:to_sym).each_slice(configs.deploy_simultaneous_limit) do |batch_service_names|
+      deploy_simultaneously(batch_service_names, deployment_result)
     end
 
     deployment_result
@@ -116,6 +116,13 @@ class KuberKit::Actions::ServiceDeployer
 
   private
     def deploy_simultaneously(service_names, deployment_result)
+      unless deployment_result.succeeded?
+        ui.print_debug("ServiceDeployer", "Deploymet already failed. Canceling: #{service_names.inspect}")
+        return
+      end
+
+      ui.print_debug("ServiceDeployer", "Scheduling to deploy: #{service_names.inspect}. Limit: #{configs.deploy_simultaneous_limit}")
+
       task_group = ui.create_task_group
 
       service_names.each do |service_name|

data/lib/kuber_kit/actions/service_generator.rb

@@ -0,0 +1,20 @@
+class KuberKit::Actions::ServiceGenerator
+  include KuberKit::Import[
+    "shell.local_shell",
+    "service_generator.action_handler",
+    "ui",
+  ]
+
+  Contract Symbol, String => Any
+  def call(service_name, path)
+    expanded_path = File.expand_path(path)
+    puts expanded_path
+    action_handler.call(local_shell, service_name, expanded_path)
+
+    true
+  rescue KuberKit::Error => e
+    ui.print_error("Error", e.message)
+    
+    false
+  end
+end

data/lib/kuber_kit/artifacts_sync/artifact_updater.rb

@@ -1,27 +1,37 @@
 class KuberKit::ArtifactsSync::ArtifactUpdater
-  ResolverNotFoundError = Class.new(KuberKit::NotFoundError)
+  StrategyNotFoundError = Class.new(KuberKit::NotFoundError)
 
   include KuberKit::Import[
     "ui"
   ]
 
-  def use_resolver(artifact_resolver, artifact_class:)
-    @@resolvers ||= {}
+  def use_strategy(strategy, artifact_class:)
+    @@strategies ||= {}
 
-    if !artifact_resolver.is_a?(KuberKit::ArtifactsSync::AbstractArtifactResolver)
-      raise ArgumentError.new("should be an instance of KuberKit::ArtifactsSync::AbstractArtifactResolver, got: #{artifact_resolver.inspect}")
+    if !strategy.is_a?(KuberKit::ArtifactsSync::Strategies::Abstract)
+      raise ArgumentError.new("should be an instance of KuberKit::ArtifactsSync::Strategies::Abstract, got: #{strategy.inspect}")
     end
 
-    @@resolvers[artifact_class] = artifact_resolver
+    @@strategies[artifact_class] = strategy
   end
 
   def update(shell, artifact)
-    resolver = @@resolvers[artifact.class]
+    strategy = @@strategies[artifact.class]
 
     ui.print_debug "ArtifactUpdater", "Updating artifact #{artifact.name.to_s.green}"
     
-    raise ResolverNotFoundError, "Can't find resolver for artifact #{artifact}" if resolver.nil?
+    raise StrategyNotFoundError, "Can't find strategy for artifact #{artifact}" if strategy.nil?
 
-    resolver.resolve(shell, artifact)
+    strategy.update(shell, artifact)
+  end
+
+  def cleanup(shell, artifact)
+    strategy = @@strategies[artifact.class]
+
+    ui.print_debug "ArtifactUpdater", "Cleaning artifact #{artifact.name.to_s.green}"
+    
+    raise StrategyNotFoundError, "Can't find strategy for artifact #{artifact}" if strategy.nil?
+
+    strategy.cleanup(shell, artifact)
   end
 end

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

@@ -0,0 +1,9 @@
+class KuberKit::ArtifactsSync::Strategies::Abstract
+  def update(shell, artifact)
+    raise KuberKit::NotImplementedError, "must be implemented"
+  end
+
+  def cleanup(shell, artifact)
+    raise KuberKit::NotImplementedError, "must be implemented"
+  end
+end

data/lib/kuber_kit/artifacts_sync/strategies/git_updater.rb

@@ -0,0 +1,46 @@
+class KuberKit::ArtifactsSync::Strategies::GitUpdater < KuberKit::ArtifactsSync::Strategies::Abstract
+
+  include KuberKit::Import[
+    "shell.git_commands",
+    "shell.bash_commands",
+  ]
+
+  Contract KuberKit::Shell::AbstractShell, KuberKit::Core::Artifacts::Git => Any
+  def update(shell, artifact)
+    already_cloned = artifact_already_cloned?(
+      shell:      shell,
+      repo_path:  artifact.cloned_path,
+      artifact:   artifact
+    )
+
+    if already_cloned
+      git_commands.force_pull_repo(shell, 
+        path: artifact.cloned_path, branch: artifact.branch
+      )
+    else
+      git_commands.download_repo(shell, 
+        remote_url: artifact.remote_url, path: artifact.cloned_path, branch: artifact.branch
+      )
+    end
+  end
+
+  Contract KuberKit::Shell::AbstractShell, KuberKit::Core::Artifacts::Git => Any
+  def cleanup(shell, artifact)
+    bash_commands.rm_rf(shell, artifact.cloned_path)
+  end
+
+  private
+    def artifact_already_cloned?(shell:, repo_path:, artifact:)
+      target_remote_url = git_commands.get_remote_url(shell, repo_path)
+      if target_remote_url != artifact.remote_url
+        return false
+      end
+
+      target_branch = git_commands.get_branch_name(shell, repo_path)
+      if target_branch != artifact.branch
+        return false
+      end
+
+      return true
+    end
+end

data/lib/kuber_kit/artifacts_sync/strategies/null_updater.rb

@@ -0,0 +1,12 @@
+class KuberKit::ArtifactsSync::Strategies::NullUpdater < KuberKit::ArtifactsSync::Strategies::Abstract
+
+  Contract KuberKit::Shell::AbstractShell, Any => Any
+  def update(shell, artifact)
+    return true
+  end
+
+  Contract KuberKit::Shell::AbstractShell, Any => Any
+  def cleanup(shell, artifact)
+    return true
+  end
+end

data/lib/kuber_kit/cli.rb

@@ -11,6 +11,7 @@ class KuberKit::CLI < Thor
   class_option :debug, :type => :boolean, aliases: ["-d"]
   class_option :configuration, :type => :string, aliases: ["-C"]
   class_option :user, :type => :string, aliases: ["-u"]
+  class_option :use_local_deploy, :type => :boolean, aliases: ["-l"]
 
   desc "compile IMAGE_NAMES", "Compile image with IMAGE_NAMES (comma-separated)"
   def compile(image_names_str)
@@ -23,6 +24,8 @@ class KuberKit::CLI < Thor
       action_result = KuberKit::Container['actions.image_compiler'].call(image_names, options)
     end
 
+    clean_artifacts(options)
+
     if action_result && action_result.succeeded?
       time = (Time.now.to_i - started_at)
       print_result("Image compilation finished! (#{time}s)", result: {
@@ -63,6 +66,8 @@ class KuberKit::CLI < Thor
       )
     end
 
+    clean_artifacts(options)
+
     if action_result && action_result.succeeded?
       time = (Time.now.to_i - started_at)
       print_result("Service deployment finished! (#{time}s)", result: {
@@ -110,6 +115,16 @@ class KuberKit::CLI < Thor
     end
   end
 
+
+  desc "generate SERVICE_NAME PATH_NAME", "Generates a template for a given service in a given path"
+  def generate(service_name, path)
+    setup(options)
+
+    if KuberKit::Container['actions.configuration_loader'].call(options)
+      KuberKit::Container['actions.service_generator'].call(service_name.to_sym, path)
+    end
+  end
+
   desc "apply FILE_PATH", "Apply FILE_PATH with kubectl"
   def apply(file_path)
     setup(options)
@@ -211,7 +226,7 @@ class KuberKit::CLI < Thor
       # We should load config before loading any bean, to make sure that bean won't be built with default config
       root_path = KuberKit::Container['tools.workdir_detector'].call(options)
       config_file_path = File.join(root_path, APP_CONFIG_FILENAME)
-      if File.exists?(config_file_path)
+      if File.exist?(config_file_path)
         require config_file_path
       end
     end
@@ -219,4 +234,17 @@ class KuberKit::CLI < Thor
     def print_result(message, data = {})
       KuberKit::Container['ui'].print_result(message, data)
     end
+
+    def clean_artifacts(options)
+      artifacts = KuberKit.current_configuration.artifacts.values
+      artifacts_to_clean = artifacts.select(&:cleanup_needed?)
+  
+      return unless artifacts_to_clean.any?
+
+      artifacts_to_clean.each do |artifact|
+        KuberKit::Container['artifacts_sync.artifact_updater'].cleanup(
+          KuberKit::Container['shell.local_shell'], artifact
+        )
+      end
+    end
 end

data/lib/kuber_kit/configs.rb

@@ -5,7 +5,8 @@ class KuberKit::Configs
     :image_dockerfile_name, :image_build_context_dir, :image_tag, :docker_ignore_list, :image_compile_dir, :remote_image_compile_dir,
     :kuber_kit_dirname, :kuber_kit_min_version, :images_dirname, :services_dirname, :infra_dirname, :configurations_dirname,
     :artifact_clone_dir, :service_config_dir, :deployer_strategy, :compile_simultaneous_limit, :deploy_simultaneous_limit,
-    :additional_images_paths, :deprecation_warnings_disabled, :log_file_path, :env_file_compile_dir, :shell_launcher_strategy, :shell_launcher_sets_configration
+    :additional_images_paths, :deprecation_warnings_disabled, :log_file_path, :env_file_compile_dir, :shell_launcher_strategy, 
+    :shell_launcher_sets_configration, :generator_strategy
   ]
   DOCKER_IGNORE_LIST = [
     'Dockerfile',
@@ -60,6 +61,7 @@ class KuberKit::Configs
     set :log_file_path,           File.join(absolute_kuber_kit_path, "deploy.log")
     set :env_file_compile_dir,    File.join(absolute_kuber_kit_path, "env_files")
     set :shell_launcher_sets_configration, true
+    set :generator_strategy,         :helm
   end
 
   def items

data/lib/kuber_kit/container.rb

@@ -25,6 +25,10 @@ class KuberKit::Container
     KuberKit::Actions::ServiceChecker.new
   end
 
+  register "actions.service_generator" do
+    KuberKit::Actions::ServiceGenerator.new
+  end
+
   register "actions.configuration_loader" do
     KuberKit::Actions::ConfigurationLoader.new
   end
@@ -181,6 +185,10 @@ class KuberKit::Container
     KuberKit::Shell::Commands::SystemCommands.new
   end
 
+  register "shell.helm_commands" do
+    KuberKit::Shell::Commands::HelmCommands.new
+  end
+
   register "shell.local_shell" do
     KuberKit::Shell::LocalShell.new
   end
@@ -229,12 +237,12 @@ class KuberKit::Container
     KuberKit::ArtifactsSync::ArtifactUpdater.new
   end
 
-  register "artifacts_sync.git_artifact_resolver" do
-    KuberKit::ArtifactsSync::GitArtifactResolver.new
+  register "artifacts_sync.strategies.git_updater" do
+    KuberKit::ArtifactsSync::Strategies::GitUpdater.new
   end
 
-  register "artifacts_sync.null_artifact_resolver" do
-    KuberKit::ArtifactsSync::NullArtifactResolver.new
+  register "artifacts_sync.strategies.null_updater" do
+    KuberKit::ArtifactsSync::Strategies::NullUpdater.new
   end
 
   register "env_file_reader.action_handler" do
@@ -269,6 +277,10 @@ class KuberKit::Container
     KuberKit::TemplateReader::Reader.new
   end
 
+  register "template_reader.renderer" do
+    KuberKit::TemplateReader::Renderer.new
+  end
+
   register "template_reader.strategies.artifact_file" do
     KuberKit::TemplateReader::Strategies::ArtifactFile.new
   end
@@ -297,6 +309,18 @@ class KuberKit::Container
     KuberKit::ServiceDeployer::ServiceDependencyResolver.new
   end
 
+  register "service_generator.action_handler" do
+    KuberKit::ServiceGenerator::ActionHandler.new
+  end
+
+  register "service_generator.strategy_detector" do
+    KuberKit::ServiceGenerator::StrategyDetector.new
+  end
+
+  register "service_generator.generator" do
+    KuberKit::ServiceGenerator::Generator.new
+  end
+
   register "service_reader.action_handler" do
     KuberKit::ServiceReader::ActionHandler.new
   end

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

@@ -14,4 +14,8 @@ class KuberKit::Core::Artifacts::AbstractArtifact
   def sync_description
     raise KuberKit::NotImplementedError, "must be implemented"
   end
+
+  def cleanup_needed?
+    raise KuberKit::NotImplementedError, "must be implemented"
+  end
 end

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

@@ -5,12 +5,13 @@ class KuberKit::Core::Artifacts::Git < KuberKit::Core::Artifacts::AbstractArtifa
   DEFAULT_REMOTE_NAME = "origin"
   DEFAULT_BRANCH = "master"
 
-  def setup(remote_url:, remote_name: DEFAULT_REMOTE_NAME, branch: DEFAULT_BRANCH, clone_path: nil, ssh_key: DEFAULT_SSH_KEY)
+  def setup(remote_url:, remote_name: DEFAULT_REMOTE_NAME, branch: DEFAULT_BRANCH, clone_path: nil, ssh_key: DEFAULT_SSH_KEY, cache_result: true)
     @remote_name = remote_name
     @remote_url = remote_url
     @branch = branch
     @clone_path = clone_path
     @ssh_key = ssh_key
+    @cache_result = cache_result
     self
   end
 
@@ -23,4 +24,8 @@ class KuberKit::Core::Artifacts::Git < KuberKit::Core::Artifacts::AbstractArtifa
   def sync_description
     "#{remote_url}:#{branch}"
   end
+
+  def cleanup_needed?
+    !@cache_result
+  end
 end

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

@@ -15,4 +15,8 @@ class KuberKit::Core::Artifacts::Local < KuberKit::Core::Artifacts::AbstractArti
   def sync_description
     "local"
   end
+
+  def cleanup_needed?
+    false
+  end
 end