kuber_kit 0.1.5 → 0.2.0

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

data/lib/kuber_kit/service_deployer/strategy_detector.rb

@@ -0,0 +1,6 @@
+class KuberKit::ServiceDeployer::StrategyDetector
+  Contract KuberKit::Core::Service => Symbol
+  def call(service)
+    KuberKit.current_configuration.deploy_strategy
+  end
+end

data/lib/kuber_kit/service_reader/action_handler.rb

@@ -0,0 +1,13 @@
+class KuberKit::ServiceReader::ActionHandler
+  include KuberKit::Import[
+    "service_reader.reader",
+    "core.service_store",
+  ]
+
+  Contract KuberKit::Shell::AbstractShell, Symbol => Any
+  def call(shell, service_name)
+    service = service_store.get_service(service_name)
+
+    reader.read(shell, service)
+  end
+end

data/lib/kuber_kit/{service_deployer/service_reader.rb → service_reader/reader.rb}

@@ -1,4 +1,4 @@
-class KuberKit::ServiceDeployer::ServiceReader
+class KuberKit::ServiceReader::Reader
   include KuberKit::Import[
     "core.template_store",
     "core.context_helper_factory",

data/lib/kuber_kit/shell/abstract_shell.rb

@@ -17,4 +17,8 @@ class KuberKit::Shell::AbstractShell
   def recursive_list_files(path, name: nil)
     raise KuberKit::NotImplementedError, "must be implemented"
   end
+
+  def sync(local_path, remote_path, exclude: nil)
+    raise KuberKit::NotImplementedError, "must be implemented"
+  end
 end

data/lib/kuber_kit/shell/{bash_commands.rb → commands/bash_commands.rb}

@@ -1,4 +1,4 @@
-class KuberKit::Shell::BashCommands
+class KuberKit::Shell::Commands::BashCommands
   def rm(shell, path)
     shell.exec!(%Q{rm "#{path}"})
   end

data/lib/kuber_kit/shell/{docker_commands.rb → commands/docker_commands.rb}

@@ -1,4 +1,4 @@
-class KuberKit::Shell::DockerCommands
+class KuberKit::Shell::Commands::DockerCommands
   def build(shell, build_dir, args = [])
     default_args = ["--rm=true"]
     args_list = (default_args + args).join(" ")

data/lib/kuber_kit/shell/{git_commands.rb → commands/git_commands.rb}

@@ -1,4 +1,4 @@
-class KuberKit::Shell::GitCommands
+class KuberKit::Shell::Commands::GitCommands
   def get_remote_url(shell, git_repo_path, remote_name: "origin")
     shell.exec! [
       "cd #{git_repo_path}",

data/lib/kuber_kit/shell/{kubectl_commands.rb → commands/kubectl_commands.rb}

@@ -1,7 +1,7 @@
 require 'json'
 require 'shellwords'
 
-class KuberKit::Shell::KubectlCommands
+class KuberKit::Shell::Commands::KubectlCommands
   def apply_file(shell, file_path, kubeconfig_path: nil)
     command_parts = []
     if kubeconfig_path
@@ -13,6 +13,30 @@ class KuberKit::Shell::KubectlCommands
     shell.exec!(command_parts.join(" "))
   end
 
+  def exec(shell, pod_name, command, args: nil, kubeconfig_path: nil, interactive: false)
+    command_parts = []
+
+    if kubeconfig_path
+      command_parts << "KUBECONFIG=#{kubeconfig_path}"
+    end
+
+    command_parts << "kubectl exec"
+
+    if args
+      command_parts << args
+    end
+
+    command_parts << pod_name
+    command_parts << "-- #{command}"
+
+    # TODO: investigate how to do it with shell.
+    if interactive
+      system(command_parts.join(" "))
+    else
+      shell.exec!(command_parts.join(" "))
+    end
+  end
+
   def rolling_restart(shell, deployment_name, kubeconfig_path: nil)
     patch_deployment(shell, deployment_name, {
       spec: {

data/lib/kuber_kit/shell/commands/rsync_commands.rb

@@ -0,0 +1,32 @@
+class KuberKit::Shell::Commands::RsyncCommands
+  def rsync(shell, source_path, target_path, target_host: nil, exclude: nil, delete: true)
+    # Add a trailing slash to directory to have behavior similar to CP command
+    if path_is_directory?(source_path) && !source_path.end_with?("/")
+      source_path = "#{source_path}/"
+    end
+
+    if target_host
+      destination = "#{target_host}:#{target_path}"
+    else
+      destination = target_path
+    end
+
+    args = [source_path, destination]
+    if exclude
+      Array(exclude).each do |e|
+        args << "--exclude=#{e}"
+      end
+    end
+
+    if delete
+      args << "--delete"
+    end
+
+    shell.exec!(%Q{rsync -a #{args.join(' ')}})
+  end
+
+  private
+    def path_is_directory?(path)
+      File.directory?(path)
+    end
+end

data/lib/kuber_kit/shell/local_shell.rb

@@ -3,21 +3,24 @@ require 'fileutils'
 class KuberKit::Shell::LocalShell < KuberKit::Shell::AbstractShell
   include KuberKit::Import[
     "tools.logger",
-    "shell.command_counter"
+    "shell.command_counter",
+    "shell.rsync_commands",
   ]
 
-  def exec!(command)
+  def exec!(command, log_command: true)
     command_number = command_counter.get_number.to_s.rjust(2, "0")
     
-    logger.info("Executed command [#{command_number}]: #{command.to_s.cyan}")
+    if log_command
+      logger.info("Execute: [#{command_number}]: #{command.to_s.cyan}")
+    end
 
     result = nil
     IO.popen(command, err: [:child, :out]) do |io|
       result = io.read.chomp.strip
     end
 
-    if result && result != ""
-      logger.info("Finished command [#{command_number}] with result: \n#{result.grey}")
+    if result && result != "" && log_command
+      logger.info("Finished [#{command_number}] with result: \n#{result.grey}")
     end
 
     if $?.exitstatus != 0
@@ -27,6 +30,10 @@ class KuberKit::Shell::LocalShell < KuberKit::Shell::AbstractShell
     result
   end
 
+  def sync(local_path, remote_path, exclude: nil)
+    rsync_commands.rsync(self, local_path, remote_path, exclude: exclude)
+  end
+
   def read(file_path)
     File.read(file_path)
   end
@@ -41,6 +48,18 @@ class KuberKit::Shell::LocalShell < KuberKit::Shell::AbstractShell
     true
   end
 
+  def delete(file_path)
+    exec!("rm #{file_path}")
+  end
+
+  def file_exists?(file_path)
+    exec!("test -f #{file_path} && echo 'true' || echo 'false'", log_command: false) == 'true'
+  end
+
+  def dir_exists?(dir_path)
+    exec!("test -d #{dir_path} && echo 'true' || echo 'false'", log_command: false) == 'true'
+  end
+
   def recursive_list_files(path, name: nil)
     command = %Q{find -L #{path}  -type f}
     command += " -name #{name}" if name

data/lib/kuber_kit/shell/ssh_session.rb

@@ -0,0 +1,60 @@
+require 'net/ssh'
+
+class KuberKit::Shell::SshSession
+  SshSessionError = Class.new(KuberKit::Error)
+
+  attr_reader :session, :host, :user, :port
+
+  def initialize(host:, user:, port:)
+    @host = host
+    @user = user
+    @port = port
+    @session = Net::SSH.start(host, user, {port: port})
+  end
+
+  def connected?
+    !!@session
+  end
+  
+  def disconnect
+    return unless connected?
+    @session.close
+    @session = nil
+  end
+
+  def exec!(command)
+    stdout_data = ''
+    stderr_data = ''
+    exit_code = nil
+    channel = session.open_channel do |ch|
+      ch.exec(command) do |ch, success|
+        if !success
+          raise SshSessionError, "Shell command failed: #{command}\r\n#{stdout_data}\r\n#{stderr_data}"
+        end
+
+        channel.on_data do |ch,data|
+          stdout_data += data
+        end
+
+        channel.on_extended_data do |ch,type,data|
+          stderr_data += data
+        end
+
+        channel.on_request('exit-status') do |ch,data|
+          exit_code = data.read_long
+        end
+      end
+    end
+
+    channel.wait
+    session.loop
+
+    stdout_data = stdout_data.chomp.strip
+
+    if exit_code != 0
+      raise SshSessionError, "Shell command failed: #{command}\r\n#{stdout_data}\r\n#{stderr_data}"
+    end
+
+    stdout_data
+  end
+end

data/lib/kuber_kit/shell/ssh_shell.rb

@@ -0,0 +1,77 @@
+require 'tempfile'
+
+class KuberKit::Shell::SshShell < KuberKit::Shell::LocalShell
+  include KuberKit::Import[
+    "tools.logger",
+    "shell.command_counter",
+    "shell.rsync_commands",
+    "shell.local_shell"
+  ]
+
+  def connect(host:, user:, port:)
+    @ssh_session = KuberKit::Shell::SshSession.new(host: host, user: user, port: port)
+  end
+  
+  def connected?
+    @ssh_session && @ssh_session.connected?
+  end
+
+  def disconnect
+    @ssh_session.disconnect if @ssh_session
+  end
+
+  def exec!(command, log_command: true)
+    command_number = command_counter.get_number.to_s.rjust(2, "0")
+    
+    if log_command
+      logger.info("#{ssh_session.host.green} > Execute: [#{command_number}]: #{command.to_s.cyan}")
+    end
+
+    result = ssh_session.exec!(command)
+
+    if result && result != "" && log_command
+      logger.info("#{ssh_session.host.green} > Finished [#{command_number}] with result: \n#{result.grey}")
+    end
+
+    result
+  rescue KuberKit::Shell::SshSession::SshSessionError => e
+    raise ShellError.new(e.message)
+  end
+
+  def sync(local_path, remote_path, exclude: nil)
+    rsync_commands.rsync(
+      local_shell, local_path, remote_path, 
+      target_host: "#{ssh_session.user}@#{ssh_session.host}",
+      exclude:     exclude
+    )
+  end
+
+  def read(file_path)
+    exec!("cat #{file_path}")
+  end
+
+  def write(file_path, content)
+    Tempfile.create do |file| 
+      file << content
+      file.flush
+      sync(file.path, file_path)
+    end
+
+    logger.info("Created file #{file_path.to_s.cyan}\r\n#{content.grey}")
+
+    true
+  end
+
+  private
+    def ssh_session
+      unless connected?
+        raise ArgumentError, "ssh session is not created, please call #connect"
+      end
+
+      @ssh_session
+    end
+
+    def ensure_directory_exists(file_path)
+      exec!("mkdir -p #{file_path}")
+    end
+end