lxd-common 0.4.1

data/lib/nexussw/lxd/transport/mixins/local.rb

@@ -0,0 +1,71 @@
+require 'open3'
+require 'nio/websocket'
+
+module NexusSW
+  module LXD
+    class Transport
+      module Mixins
+        module Local
+          def initialize(config = {})
+            @config = config
+          end
+
+          attr_reader :config
+
+          include ExecuteMixin
+
+          def execute_chunked(command, options)
+            NIO::WebSocket::Reactor.start
+            LXD.with_timeout_and_retries options do
+              # Let's borrow the NIO::WebSocket reactor
+              Open3.popen3(command) do |_stdin, stdout, stderr, th|
+                mon_out = mon_err = nil
+                NIO::WebSocket::Reactor.queue_task do
+                  mon_out = NIO::WebSocket::Reactor.selector.register(stdout, :r)
+                  mon_out.value = proc do
+                    data = read(mon_out) # read regardless of block_given? so that we don't spin out on :r availability
+                    yield(data) if data && block_given?
+                  end
+                end
+                NIO::WebSocket::Reactor.queue_task do
+                  mon_err = NIO::WebSocket::Reactor.selector.register(stderr, :r)
+                  mon_err.value = proc do
+                    data = read(mon_err) # read regardless of block_given? so that we don't spin out on :r availability
+                    yield(nil, data) if data && block_given?
+                  end
+                end
+                th.join
+                loop do
+                  return LXDExecuteResult.new(command, options, th.value.exitstatus) if th.value.exited? && mon_out && mon_err && mon_out.closed? && mon_err.closed?
+                  Thread.pass
+                end
+              end
+            end
+          end
+
+          def read_file(path)
+            return '' unless File.exist? path
+            File.read path
+          end
+
+          def write_file(path, content)
+            File.open path, 'w' do |f|
+              f.write content
+            end
+          end
+
+          private
+
+          def read(monitor)
+            monitor.io.read_nonblock(16384)
+          rescue IO::WaitReadable # rubocop:disable Lint/ShadowedException
+            return nil
+          rescue Errno::ECONNRESET, EOFError, IOError
+            monitor.close
+            return nil
+          end
+        end
+      end
+    end
+  end
+end

data/lib/nexussw/lxd/transport/mixins/rest.rb

@@ -0,0 +1,120 @@
+require 'nio/websocket'
+
+module NexusSW
+  module LXD
+    class Transport
+      module Mixins
+        module Rest
+          def initialize(driver, container_name, config = {})
+            @container_name = container_name
+            @config = config
+            raise "The rest transport requires the Rest Driver.  You supplied #{driver}" unless driver.respond_to?(:hk) && driver.respond_to?(:rest_endpoint) # driver.is_a? NexusSW::LXD::Driver::Rest
+            @rest_endpoint = driver.rest_endpoint
+            @driver_options = driver.driver_options
+            @hk = driver.hk
+          end
+
+          attr_reader :hk, :rest_endpoint, :container_name, :config
+
+          include ExecuteMixin
+
+          def execute_chunked(command, options = {}, &block)
+            opid = nil
+            backchannel = nil
+            if block_given? # Allow for an optimized case that doesn't require the support of 3 new websocket connections
+              retval = hk.execute_command(container_name, command, wait_for_websocket: true, interactive: false, sync: false)
+              opid = retval[:id]
+              backchannel = ws_connect opid, retval[:metadata][:fds], &block
+            else
+              opid = hk.execute_command(container_name, command, sync: false)[:id]
+            end
+            LXD.with_timeout_and_retries({ timeout: 0 }.merge(options)) do
+              begin
+                retval = hk.wait_for_operation opid
+                backchannel.exit if backchannel.respond_to? :exit
+                return LXDExecuteResult.new command, options, retval[:metadata][:return].to_i
+              rescue Faraday::TimeoutError => e
+                raise Timeout::Retry.new e # rubocop:disable Style/RaiseArgs
+              end
+            end
+          end
+
+          def read_file(path)
+            hk.read_file container_name, path
+          rescue ::Hyperkit::NotFound
+            return ''
+          end
+
+          def write_file(path, content)
+            hk.write_file container_name, path, content: content
+          end
+
+          def download_file(path, local_path)
+            hk.pull_file container_name, path, local_path
+          end
+
+          def upload_file(local_path, path)
+            hk.push_file local_path, container_name, path
+          end
+
+          protected
+
+          class WSWrapper
+            def initialize(waitlist)
+              @waitlist = waitlist.compact
+            end
+            attr_reader :waitlist
+
+            def exit
+              loop do
+                allclosed = true
+                waitlist.each do |driver|
+                  allclosed = false unless driver.state == :closed
+                end
+                break if allclosed
+                Thread.pass
+                sleep 0.1
+              end
+            end
+          end
+
+          def ws_connect(opid, endpoints)
+            # NIO::WebSocket.log_traffic = true
+            verify_ssl = OpenSSL::SSL::VERIFY_NONE if @driver_options[:verify_ssl] == false
+            ws_options = { ssl_context: { verify_mode: verify_ssl } } unless verify_ssl.nil?
+            ws_options ||= {}
+            baseurl = rest_endpoint.sub(%r{^http([s]?://)}, 'ws\1')
+            baseurl += '/' unless baseurl.end_with? '/'
+            baseurl += "1.0/operations/#{opid}/websocket?secret="
+
+            pipes = {}
+            NIO::WebSocket.connect(baseurl + endpoints[:control], ws_options) do |driver|
+              driver.on :io_error do # usually I get an EOF
+                pipes.each { |_, v| v.close if v.respond_to? :close }
+              end
+              driver.on :close do # but on occasion I get a legit close
+                pipes.each { |_, v| v.close if v.respond_to? :close }
+              end
+            end
+            pipes[:'1'] = NIO::WebSocket.connect(baseurl + endpoints[:'1'], ws_options) do |driver|
+              driver.on :message do |ev|
+                data = ev.data.is_a?(String) ? ev.data : ev.data.pack('U*')
+                yield data
+              end
+            end
+            endpoints.each do |fd, secret|
+              next if [:control, :'1'].include? fd
+              pipes[fd] = NIO::WebSocket.connect(baseurl + secret, ws_options) do |driver|
+                driver.on :message do |ev|
+                  data = ev.data.is_a?(String) ? ev.data : ev.data.pack('U*')
+                  yield nil, data
+                end
+              end
+            end
+            WSWrapper.new [pipes[:'1'], pipes[:'2']]
+          end
+        end
+      end
+    end
+  end
+end

data/lib/nexussw/lxd/transport/rest.rb

@@ -0,0 +1,12 @@
+require 'nexussw/lxd/transport'
+require 'nexussw/lxd/transport/mixins/rest'
+
+module NexusSW
+  module LXD
+    class Transport
+      class Rest < Transport
+        include Mixins::Rest
+      end
+    end
+  end
+end

data/lib/nexussw/lxd/version.rb

@@ -0,0 +1,5 @@
+module NexusSW
+  module LXD
+    VERSION = '0.4.1'.freeze
+  end
+end

data/lxd-common.gemspec

@@ -0,0 +1,35 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'nexussw/lxd/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'lxd-common'
+  spec.version       = NexusSW::LXD::VERSION
+  spec.authors       = ['Sean Zachariasen']
+  spec.email         = ['thewyzard@hotmail.com']
+
+  spec.summary       = 'Shared LXD Container Access Library'
+  # spec.description   = %q{TODO: Write a longer description or delete this line.}
+  spec.homepage      = 'http://github.com/NexusSW/lxd-common'
+
+  # Prevent pushing this gem to RubyGems.org. To allow pushes either set the 'allowed_push_host'
+  # to allow pushing to a single host or delete this section to allow pushing to any host.
+
+  # spec.metadata['allowed_push_host'] = "TODO: Set to 'http://mygemserver.com'" if spec.respond_to?(:metadata)
+
+  spec.files = `git ls-files -z`.split("\x0").reject do |f|
+    f.match(%r{^(test|spec|features)/})
+  end
+  spec.bindir = 'exe'
+  spec.executables = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ['lib']
+
+  spec.add_dependency 'hyperkit', '~> 1.1'
+  spec.add_dependency 'nio4r-websocket', '~> 0.6'
+
+  spec.add_development_dependency 'bundler'
+  spec.add_development_dependency 'rake'
+  spec.add_development_dependency 'rspec'
+  spec.add_development_dependency 'simplecov'
+end

metadata

@@ -0,0 +1,152 @@
+--- !ruby/object:Gem::Specification
+name: lxd-common
+version: !ruby/object:Gem::Version
+  version: 0.4.1
+platform: ruby
+authors:
+- Sean Zachariasen
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2017-11-08 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: hyperkit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
+- !ruby/object:Gem::Dependency
+  name: nio4r-websocket
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.6'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.6'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: 
+email:
+- thewyzard@hotmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- Vagrantfile
+- bin/console
+- bin/setup
+- lib/nexussw/lxd.rb
+- lib/nexussw/lxd/driver.rb
+- lib/nexussw/lxd/driver/cli.rb
+- lib/nexussw/lxd/driver/mixins/cli.rb
+- lib/nexussw/lxd/driver/mixins/rest.rb
+- lib/nexussw/lxd/driver/rest.rb
+- lib/nexussw/lxd/transport.rb
+- lib/nexussw/lxd/transport/cli.rb
+- lib/nexussw/lxd/transport/local.rb
+- lib/nexussw/lxd/transport/mixins/cli.rb
+- lib/nexussw/lxd/transport/mixins/local.rb
+- lib/nexussw/lxd/transport/mixins/rest.rb
+- lib/nexussw/lxd/transport/rest.rb
+- lib/nexussw/lxd/version.rb
+- lxd-common.gemspec
+homepage: http://github.com/NexusSW/lxd-common
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.6.13
+signing_key: 
+specification_version: 4
+summary: Shared LXD Container Access Library
+test_files: []