vpsadmin-client 3.0.0.master.20190517.pre.0.3ab5ddfe → 3.0.0.master.20220121.pre.0.b5a81113

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e88be4aa43616fcca48c5a44a608568bbe5a17a820b601a4750b559e306eb789
-  data.tar.gz: 93a4a8389faa06c619633113e3992deb7cbea545e0b5ee1814344d3d01f5fd83
+  metadata.gz: ce5635948bc213752357b413d4e92f8b7f2089306b9a3e02194b76f44f3617c3
+  data.tar.gz: 4304acd24bbecf0389a954d61ce680fece999e1128651f84e75306cf61889195
 SHA512:
-  metadata.gz: 431461ad20a69847e7d4c52d0731b0691b9745e4f16e448e1bbcfbe43c44cfd8bbe16ca7c9fe2257a9fc285b1b97fbebff8821f610fb8bc409cb3ec7fc5c7443
-  data.tar.gz: 9adafa55fa0e00c99159f60a049014b866eb41a4a7bad8f4a4b84efd9bddede771d4d7bfbeeb9ed5aabd20d035165b79269df800a53e017920c895038bda8717
+  metadata.gz: b3c0c9029f6d14639adb9f901f4b1cc133d1518360be85d8e506e8aa285fa5a1b3e8be5cfa475ae87d8147fadf78dd55bd168593e945f59c1dc237412f714705
+  data.tar.gz: 35e2c3b730bf246d1a14c0a6a7cdf77c08910efa4d460aeb878a52a9d4a0210058a7450c556c7e30a08199b011464a45b17b155b13d965c2da057e14a3878309

data/README.md

@@ -1,6 +1,15 @@
 # vpsAdmin client
 
-TODO: Write a gem description
+vpsAdmin client is a Ruby CLI and client library for vpsAdmin API. It is based
+on [haveapi-client](https://github.com/vpsfreecz/haveapi/tree/master/clients/ruby).
+
+vpsAdmin client extends `haveapi-client` with several command-line operations
+specific to vpsAdmin, including:
+
+- VPS remote console
+- Snapshot downloads, either ZFS streams or tar archives
+- Automated utility for local backups using ZFS
+- Live IP traffic monitor
 
 ## Installation
 
@@ -17,13 +26,6 @@ Or install it yourself as:
     $ gem install vpsadmin-client
 
 ## Usage
-
-TODO: Write usage instructions here
-
-## Contributing
-
-1. Fork it ( https://github.com/[my-github-username]/vpsadminctl/fork )
-2. Create your feature branch (`git checkout -b my-new-feature`)
-3. Commit your changes (`git commit -am 'Add some feature'`)
-4. Push to the branch (`git push origin my-new-feature`)
-5. Create a new Pull Request
+See
+[haveapi-client](https://github.com/vpsfreecz/haveapi/tree/master/clients/ruby)
+for usage information.

data/lib/vpsadmin/cli/commands/vps_remote_console.rb

@@ -1,5 +1,4 @@
-require 'eventmachine'
-require 'em-http'
+require 'thread'
 require 'time'
 require 'json'
 require 'base64'
@@ -11,16 +10,35 @@ module VpsAdmin::CLI::Commands
     args 'VPS_ID'
     desc 'Open VPS remote console'
 
-    class InputHandler < EventMachine::Connection
-      attr_accessor :buffer
-
-      def initialize
+    class InputHandler
+      def initialize(http_client)
+        @http_client = http_client
         @private_buffer = ''
-        @buffer = ''
         @end_seq = ["\r", "\e", "."]
         @end_i = 0
+        @stop = false
+      end
+
+      def read_from(io)
+        begin
+          data = io.read_nonblock(4096)
+        rescue IO::WaitReadable
+          return
+        rescue Errno::EIO
+          stop
+          return
+        end
+
+        write(data)
       end
 
+      def stop?
+        @stop
+      end
+
+      protected
+      attr_reader :http_client
+
       # Data is checked on the presence of the end sequence. The first character
       # in the sequence (ENTER) can be read multiple times in a row and it is
       # to be forwarded.
@@ -29,38 +47,160 @@ module VpsAdmin::CLI::Commands
       # but stored in a private buffer. If the sequence is later broken, the private
       # buffer is forwarded and reset.
       #
-      # If the whole end sequence is read, EM event loop is stopped.
-      def receive_data(data)
+      # If the whole end sequence is read, we exit.
+      def write(data)
+        buffer = ''
+
         data.each_char do |char|
           if char == @end_seq[ @end_i ]
-            if @end_i == @end_seq.size-1
-              EM.stop
+            if @end_i == @end_seq.size - 1
+              @stop = true
               return
             end
 
             @end_i += 1
 
             if @end_i == 1
-              @buffer += char
-
+              buffer += char
             else
               @private_buffer += char
             end
 
           elsif char == @end_seq.first
-            @buffer += char
-
+            buffer += char
           else
             @end_i = 0
 
             unless @private_buffer.empty?
-              @buffer += @private_buffer
+              buffer += @private_buffer
               @private_buffer.clear
             end
 
-            @buffer += char
+            buffer += char
           end
         end
+
+        http_client.write(buffer) unless buffer.empty?
+      end
+    end
+
+    class HttpClient
+      def initialize(vps, token, rate)
+        @vps = vps
+        @token = token
+        @rate = rate
+        @mutex = Mutex.new
+        @write_buffer = ''
+        @stop = false
+        @error = false
+      end
+
+      def start
+        @thread = Thread.new { run }
+      end
+
+      def stop
+        @stop = true
+      end
+
+      def stop?
+        @stop
+      end
+
+      def error?
+        @error != false
+      end
+
+      def error
+        @error
+      end
+
+      def join
+        stop
+        thread.join
+      end
+
+      def write(data)
+        mutex.synchronize do
+          write_buffer << data
+        end
+      end
+
+      def resize(width, height)
+        @width = width
+        @height = height
+      end
+
+      protected
+      attr_reader :vps, :token, :rate, :width, :height, :write_buffer,
+        :mutex, :thread
+
+      def run
+        uri = URI("#{vps.node.location.remote_console_server}/console/feed/#{vps.id}")
+        start_args = [uri.host, uri.port, nil, nil, nil, nil, {
+          use_ssl: uri.scheme == 'https',
+        }]
+
+        Net::HTTP.start(*start_args) do |http|
+          loop do
+            break if stop? || error?
+
+            rate_limit { send_request(http, uri) }
+          end
+        end
+      end
+
+      def send_request(http, uri)
+        req = Net::HTTP::Post.new(uri)
+
+        keys =
+          mutex.synchronize do
+            s = write_buffer.dup
+            write_buffer.clear
+            s
+          end
+
+        req.set_form_data(
+          'session' => token,
+          'keys' => keys,
+          'width' => width,
+          'height' => height,
+        )
+
+        res = http.request(req)
+
+        unless res.is_a?(Net::HTTPSuccess)
+          set_error(
+            "Console server returned error: "+
+            "HTTP #{res.code} - #{res.message}\n\n#{res.body}"
+          )
+          stop
+          return
+        end
+
+        ret = JSON.parse(res.body, symbolize_names: true)
+
+        unless ret[:session]
+          $stdout.write(ret[:data])
+          set_error("Session closed.")
+          stop
+          return
+        end
+
+        $stdout.write(Base64.decode64(ret[:data]))
+      end
+
+      def rate_limit
+        t1 = Time.now
+        yield
+        t2 = Time.now
+
+        cooldown = rate - (t2 - t1)
+        sleep(cooldown) if cooldown > 0
+      end
+
+      def set_error(str)
+        @error = str
       end
     end
 
@@ -93,7 +233,7 @@ module VpsAdmin::CLI::Commands
       end
 
       puts "  VPS is on #{vps.node.domain_name}, located in #{vps.node.location.label}."
-      puts "Console router URL is #{vps.node.location.remote_console_server}"
+      puts "Console server URL is #{vps.node.location.remote_console_server}"
       write "Obtaining authentication token..."
 
       begin
@@ -105,30 +245,18 @@ module VpsAdmin::CLI::Commands
         exit(false)
       end
 
-      @token = t.token
-
       puts
-      puts "Connecting to remote console..."
+      puts "Connecting to the remote console..."
       puts "Press ENTER ESC . to exit"
       puts
 
       raw_mode do
-        EventMachine.run do
-          @input = EventMachine.open_keyboard(InputHandler)
-
-          @http = EventMachine::HttpRequest.new(
-              "#{vps.node.location.remote_console_server}/console/feed/#{vps_id}"
-          )
-          communicate
-        end
+        run_console(vps, t.token)
       end
     end
 
     protected
-    def write(s)
-      $stdout.write(s)
-      $stdout.flush
-    end
+    attr_reader :client
 
     def raw_mode
       state = `stty -g`
@@ -139,6 +267,7 @@ module VpsAdmin::CLI::Commands
 
         Signal.trap('WINCH') do
           @size = Terminal.size!
+          client.resize(@size[:width], @size[:height]) if client
         end
 
         yield
@@ -150,38 +279,35 @@ module VpsAdmin::CLI::Commands
       puts
     end
 
-    def communicate
-      post = @http.post(
-        body: {
-          session: @token,
-          keys: @input.buffer,
-          width: @size[:width],
-          height: @size[:height],
-        },
-        keepalive: true
-      )
+    def run_console(vps, token)
+      Thread.abort_on_exception = true
 
-      @input.buffer = ''
+      @client = HttpClient.new(vps, token, @opts[:rate])
+      client.resize(@size[:width], @size[:height])
 
-      post.errback do
-        puts "Error: connection to console router failed"
-        EventMachine.stop
-      end
+      input = InputHandler.new(client)
+      client.start
 
-      post.callback do
-        ret = JSON.parse(post.response, symbolize_names: true)
+      loop do
+        res = IO.select([$stdin], [], [], 1)
+        input.read_from($stdin) if res
 
-        unless ret[:session]
-          $stdout.write(ret[:data])
-          puts "\nSession closed."
-          EM.stop
-          next
-        end
+        if input.stop? || client.stop?
+          client.join
 
-        $stdout.write(Base64.decode64(ret[:data]))
+          if client.error?
+            write("\n")
+            write(client.error)
+          end
 
-        EM.add_timer(@opts[:rate]) { communicate }
+          return
+        end
       end
     end
+
+    def write(s)
+      $stdout.write(s)
+      $stdout.flush
+    end
   end
 end

data/lib/vpsadmin/client/version.rb

@@ -1,5 +1,5 @@
 module VpsAdmin
   module Client
-    VERSION = '3.0.0.master.20190517-0.3ab5ddfe'
+    VERSION = '3.0.0.master.20220121-0.b5a81113'
   end
 end

data/vpsadmin-client.gemspec

@@ -21,9 +21,7 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency 'bundler'
   spec.add_development_dependency 'rake'
 
-  spec.add_runtime_dependency 'haveapi-client', '~> 0.13.0'
-  spec.add_runtime_dependency 'eventmachine', '~> 1.0.3'
-  spec.add_runtime_dependency 'em-http-request', '~> 1.1.3'
+  spec.add_runtime_dependency 'haveapi-client', '~> 0.15.1'
   spec.add_runtime_dependency 'json'
   spec.add_runtime_dependency 'curses'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: vpsadmin-client
 version: !ruby/object:Gem::Version
-  version: 3.0.0.master.20190517.pre.0.3ab5ddfe
+  version: 3.0.0.master.20220121.pre.0.b5a81113
 platform: ruby
 authors:
 - Jakub Skokan
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2019-05-17 00:00:00.000000000 Z
+date: 1980-01-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -44,42 +44,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.13.0
+        version: 0.15.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.13.0
-- !ruby/object:Gem::Dependency
-  name: eventmachine
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 1.0.3
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 1.0.3
-- !ruby/object:Gem::Dependency
-  name: em-http-request
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 1.1.3
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 1.1.3
+        version: 0.15.1
 - !ruby/object:Gem::Dependency
   name: json
   requirement: !ruby/object:Gem::Requirement
@@ -142,7 +114,7 @@ homepage: ''
 licenses:
 - GPL
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -157,9 +129,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: 1.3.1
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.7.6.2
-signing_key: 
+rubygems_version: 3.2.26
+signing_key:
 specification_version: 4
 summary: Ruby API and CLI for vpsAdmin API
 test_files: []