kubeclient_exec 0.1.2 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5649bee867b5a1b4bee74f728bd9d4e20237f8147d1f2b322d1a5c837dc525dc
-  data.tar.gz: 2ccc8823b1f3184b975729385652003ac826e8cc3e4fa2b03c4d49cfb7cb16d7
+  metadata.gz: 274101b279088781eb0174768f8f276386c0c984ac8103a0b20e5324fb0d513c
+  data.tar.gz: 23ff5894068bd1e43e5df70345c7bc5c70f4b172054ee9483ef78e9efce351d6
 SHA512:
-  metadata.gz: 5016b2107c6efb54d6ad996e7b9fe24d5f3a4ec2a11b7f103c792ec0bd6e1170d89aebfad05537d557cb0ee247a5ca6a69f4cef4069d06c91136d16d8b0f1609
-  data.tar.gz: d8e74ca1a7bfd763c851393d89f548d012177a854ed16878969eb7db4f7a8a03c31fa9583d83e63130ec2f53a0ce03d93fef23babf324e898186da436338b98d
+  metadata.gz: 2187896745c3f28154ebf0de96b2cd0e3e97ba1c59c107c574f6667656c27ec011f636af7af314aaeb3a5a9c458f5d468164a426d881fcfea2fb49d134881c2a
+  data.tar.gz: 9e58927e75109caf0a0020957522106b21e9fa6c232ffc3610390fc10362466e74f0a81e25b8569e971c8bd3562a86485aa76f0c1568ed844f571f2ebbbba723

data/lib/kubeclient_exec/copy/copy.rb

@@ -12,7 +12,9 @@ module KubeclientExec
       suppress_errors: false,
       tls: {
         cert_chain_file: nil,
+        cert: nil,
         private_key_file: nil,
+        private_key: nil,
         verify_peer: true
       }
     }
@@ -42,13 +44,22 @@ module KubeclientExec
 
       if copy_file
         exec_pod("tar xf - -C #{remote_path.split('/')[0...-1].join('/')}", name, namespace, options: { tty: false }.merge!(options)) do |executor|
-          executor.write(tar_file.string)
+          chunks = split_string_into_chunks(tar_file.string, 1024 * 1024) # 1MB
+          chunks.each do |chunk|
+            executor.write(chunk)
+          end
 
           # Feels like there should be a better way for this
+          stopping = false
           EM.add_periodic_timer(0.1) do
-            if executor.done?
+            if executor.done? && !stopping
+              stopping = true
               executor.stop
             end
+
+            if executor.ready_state == 3
+              EM.stop_event_loop
+            end
           end
         end
       else
@@ -56,10 +67,16 @@ module KubeclientExec
           executor.write(tar_file.string)
 
           # Feels like there should be a better way for this
+          stopping = false
           EM.add_periodic_timer(0.1) do
-            if executor.done?
+            if executor.done? && !stopping
+              stopping = true
               executor.stop
             end
+
+            if executor.ready_state == 3
+              EM.stop_event_loop
+            end
           end
         end
       end
@@ -70,22 +87,40 @@ module KubeclientExec
 
       exec_pod("tar cf - #{remote_path}", name, namespace, options: { tty: false }.merge!(options)) do |executor|
         count = 0
+        content = ''
 
         executor.on_stdout do |data|
-          if count == 1
-            if local_path.is_a? String
-              untar(StringIO.new(data), local_path)
-            elsif local_path == :single_result
-              result = single_untar(StringIO.new(data))
-            end
-            executor.stop
+          if count >= 1
+            content += data
           end
 
           count += 1
         end
+
+        executor.on_close do
+          if local_path.is_a? String
+            untar(StringIO.new(content), local_path)
+          elsif local_path == :single_result
+            result = single_untar(StringIO.new(content))
+          end
+        end
       end
 
       result
     end
+
+    private
+    def split_string_into_chunks(string, chunk_size)
+      chunks = []
+      start_index = 0
+
+      while start_index < string.length
+        chunk = string[start_index, chunk_size]
+        chunks << chunk
+        start_index += chunk_size
+      end
+
+      chunks
+    end
   end
 end

data/lib/kubeclient_exec/execute/execute.rb

@@ -13,7 +13,9 @@ module KubeclientExec
       suppress_errors: false,
       tls: {
         cert_chain_file: nil,
+        cert: nil,
         private_key_file: nil,
+        private_key: nil,
         verify_peer: true
       }
     }

data/lib/kubeclient_exec/execute/executor.rb

@@ -55,8 +55,12 @@ module KubeclientExec
       end
 
       def stop
-        @ws.close if @ws
+        @ws.instance_variable_get(:@stream).close_connection_after_writing
         @on_close.call if @on_close
+      end
+
+      def stop!
+        stop
         EM.stop_event_loop
       end
 
@@ -64,6 +68,10 @@ module KubeclientExec
         @ws.instance_variable_get(:@driver).instance_variable_get(:@queue).empty?
       end
 
+      def ready_state
+        @ws.ready_state
+      end
+
       private
       def setup
         @ws = Faye::WebSocket::Client.new(@url, nil, {
@@ -71,14 +79,17 @@ module KubeclientExec
           headers: @kubeclient_options[:headers],
           tls: {
             cert_chain_file: @kubeclient_options[:tls][:cert_chain_file],
+            cert: @kubeclient_options[:tls][:cert],
             private_key_file: @kubeclient_options[:tls][:private_key_file],
+            private_key: @kubeclient_options[:tls][:private_key],
             verify_peer: @kubeclient_options[:tls][:verify_peer],
-          }
+          },
+          max_length: 2**32,
         })
 
         @ws.on(:message) do |msg|
           if msg.type == :close
-            stop
+            stop!
             return
           end
 
@@ -91,7 +102,7 @@ module KubeclientExec
             if @options[:mode] == :adhoc
               @last_stdout = 1 if type == EXEC_STDOUT
               @last_stderr = 1 if type == EXEC_STDERR || EXEC_DCKERR
-              stop
+              stop!
             end
           end
 
@@ -112,7 +123,7 @@ module KubeclientExec
         end
 
         @ws.on(:close) do
-          stop
+          stop!
         end
 
         @ws.on(:open) do

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: kubeclient_exec
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.2.0
 platform: ruby
 authors:
 - Wout Ceulemans
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-02-22 00:00:00.000000000 Z
+date: 2025-03-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: kubeclient
@@ -38,7 +38,7 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.11.3
-description: 
+description:
 email: me@wout.dev
 executables: []
 extensions: []
@@ -54,7 +54,7 @@ licenses:
 - MIT
 metadata:
   source_code_uri: https://github.com/WoutDev/kubeclient_exec
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -70,7 +70,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubygems_version: 3.5.3
-signing_key: 
+signing_key:
 specification_version: 4
 summary: An extension to the kubeclient gem that adds exec_pod and cp_pod functionality.
 test_files: []