vagrant-service-manager 1.1.0 → 1.2.0

data/lib/vagrant-service-manager/binary_handlers/adb_openshift_binary_handler.rb

@@ -0,0 +1,49 @@
+module VagrantPlugins
+  module ServiceManager
+    class ADBOpenshiftBinaryHandler < ADBBinaryHandler
+      OC_BINARY_BASE_URL = 'https://github.com/openshift/origin/releases'.freeze
+      OC_FILE_PREFIX = 'openshift-origin-client-tools'.freeze
+
+      def initialize(machine, env, options)
+        super(machine, env, options)
+      end
+
+      def build_download_url
+        download_base_path = "#{OC_BINARY_BASE_URL}/download/v#{@version}/"
+        file = "#{OC_FILE_PREFIX}-v#{@version}-#{version_sha}-#{archive_ext}"
+        @url = download_base_path + file
+      end
+
+      private
+
+      def version_sha
+        tag_url = "#{OC_BINARY_BASE_URL}/tag/v#{@version}"
+        data = Net::HTTP.get(URI(tag_url))
+        tokens = data.match("-v#{@version}-(.*)-#{archive_ext}").captures
+        tokens.first unless tokens.empty?
+      rescue StandardError => e
+        @ui.error e.message
+        exit 126
+      end
+
+      def os_type
+        if Vagrant::Util::Platform.windows?
+          'windows'
+        elsif Vagrant::Util::Platform.linux?
+          'linux'
+        elsif Vagrant::Util::Platform.darwin?
+          'mac'
+        end
+      end
+
+      def arch
+        arch = @machine.env.host.capability(:os_arch)
+        arch == 'x86_64' ? '64' : '32'
+      end
+
+      def archive_ext
+        os_type + (os_type == 'linux' ? "-#{arch}bit.tar.gz" : '.zip')
+      end
+    end
+  end
+end

data/lib/vagrant-service-manager/binary_handlers/binary_handler.rb

@@ -0,0 +1,119 @@
+require 'net/http'
+
+module VagrantPlugins
+  module ServiceManager
+    class URLValidationError < Vagrant::Errors::VagrantError
+      error_key(:url_validation_error)
+    end
+
+    class BinaryHandler
+      BINARY_ARCHIVE_FORMATS = ['.tgz', '.tar.gz', '.gz', '.zip'].freeze
+      BINARY_NAME = {
+        docker: 'docker', openshift: 'oc'
+      }.freeze
+      VERSION_CMD = {
+        docker: "docker version --format '{{.Server.Version}}'",
+        openshift: "oc version | grep 'oc' | grep -oE '[0-9.]+'"
+      }.freeze
+      BINARY_REGEX = {
+        windows: { docker: %r{\/docker.exe$}, openshift: /oc.exe$/ },
+        unix: { docker: %r{\/docker$}, openshift: /oc$/ }
+      }.freeze
+      ARCHIVE_MAP = {
+        '.tgz' => 'Tar', '.tar.gz' => 'Tar', '.gz' => 'Tar', '.zip' => 'Zip'
+      }.freeze
+      LABEL = 'servicemanager.commands.install_cli.message'.freeze
+
+      attr_accessor :path, :version, :type, :url,
+                    :binary_exists, :skip_download, :archive_file_path
+
+      def initialize(machine, env, options)
+        @machine = machine
+        @ui      = env.ui
+        @url = ''
+        @binary_exists = true
+        @skip_download = false
+        @archive_file_path = ''
+        @options = options
+        @type = options[:type]
+        @version = options['--cli-version'] || PluginUtil.execute_once(@machine, @ui, VERSION_CMD[@type])
+        @path = options['--path'] || binary_path
+        @temp_bin_dir = "#{ServiceManager.temp_dir}/#{@type}"
+      end
+
+      def install
+        build_download_url
+        validate_url
+        build_archive_path
+        ensure_binary_and_temp_directories
+        download_archive
+        prepare_binary
+      end
+
+      def build_download_url
+      end
+
+      def build_archive_path
+      end
+
+      def ensure_binary_and_temp_directories
+      end
+
+      def download_archive
+      end
+
+      def prepare_binary
+      end
+
+      def print_message
+      end
+
+      def archive_dir_name
+        @archive_file_path.sub(Regexp.new(BINARY_ARCHIVE_FORMATS.join('|')), '')
+      end
+
+      def archive_handler_class
+        Object.const_get("#{ServiceManager.name}::#{archive_handler_name}")
+      end
+
+      def binary_name
+        BINARY_NAME[@type] + binary_ext
+      end
+
+      def binary_ext
+        Vagrant::Util::Platform.windows? ? '.exe' : ''
+      end
+
+      def bin_dir
+        File.dirname(@path)
+      end
+
+      def file_regex
+        os_type = Vagrant::Util::Platform.windows? ? :windows : :unix
+        BINARY_REGEX[os_type][@type]
+      end
+
+      # Checks if url is accessible or not
+      def validate_url
+        url = URI.parse(@url)
+        req = Net::HTTP.new(url.host, url.port)
+        req.use_ssl = true if url.scheme == 'https'
+        res = req.request_head(url.path)
+
+        unless res.code == '200'
+          raise URLValidationError, I18n.t('servicemanager.commands.install_cli.url_validation_error')
+        end
+      end
+
+      private
+
+      def archive_handler_name
+        ARCHIVE_MAP[File.extname(@url)] + 'Handler'
+      end
+
+      def binary_path
+        "#{ServiceManager.bin_dir}/#{@type}/#{@version}/#{binary_name}"
+      end
+    end
+  end
+end

data/lib/vagrant-service-manager/binary_handlers/cdk_binary_handler.rb

@@ -0,0 +1,33 @@
+module VagrantPlugins
+  module ServiceManager
+    class CDKBinaryHandler < BinaryHandler
+      def initialize(machine, env, options = {})
+        super(machine, env, options)
+      end
+
+      # TODO
+      def build_download_url
+      end
+
+      # TODO
+      def build_archive_path
+      end
+
+      # TODO
+      def ensure_binary_and_temp_directories
+      end
+
+      # TODO
+      def download_archive
+      end
+
+      # TODO
+      def prepare_binary
+      end
+
+      # TODO
+      def print_message
+      end
+    end
+  end
+end

data/lib/vagrant-service-manager/binary_handlers/cdk_docker_binary_handler.rb

@@ -0,0 +1,9 @@
+module VagrantPlugins
+  module ServiceManager
+    class CDKDockerBinaryHandler < CDKBinaryHandler
+      def initialize(machine, env, options)
+        super(machine, env, options)
+      end
+    end
+  end
+end

data/lib/vagrant-service-manager/binary_handlers/cdk_openshift_binary_handler.rb

@@ -0,0 +1,9 @@
+module VagrantPlugins
+  module ServiceManager
+    class CDKOpenshiftBinaryHandler < CDKBinaryHandler
+      def initialize(machine, env, options)
+        super(machine, env, options)
+      end
+    end
+  end
+end

data/lib/vagrant-service-manager/command.rb

@@ -1,25 +1,50 @@
-require_relative 'os'
 require 'digest'
 require_relative 'plugin_util'
 require_relative 'plugin_logger'
+require_relative 'installer'
+require_relative 'archive_handlers/tar_handler'
+require_relative 'archive_handlers/zip_handler'
+require_relative 'binary_handlers/binary_handler'
+require_relative 'binary_handlers/adb_binary_handler'
+require_relative 'binary_handlers/cdk_binary_handler'
+require_relative 'binary_handlers/adb_docker_binary_handler'
+require_relative 'binary_handlers/adb_openshift_binary_handler'
+require_relative 'binary_handlers/cdk_docker_binary_handler'
+require_relative 'binary_handlers/cdk_openshift_binary_handler'
 
 module VagrantPlugins
   module ServiceManager
-    DOCKER_PATH = '/home/vagrant/.docker'
-    SUPPORTED_SERVICES = ['docker', 'openshift', 'kubernetes']
+    DOCKER_PATH = '/home/vagrant/.docker'.freeze
+    SUPPORTED_SERVICES = %w(docker openshift kubernetes).freeze
     KUBE_SERVICES = [
       'etcd', 'kube-apiserver', 'kube-controller-manager', 'kube-scheduler',
       'kubelet', 'kube-proxy', 'docker'
-    ]
+    ].freeze
     # NOTE: SERVICES_MAP[<service>] will give fully-qualified service class name
     # Eg: SERVICES_MAP['docker'] gives Vagrant::ServiceManager::Docker
     SERVICES_MAP = {
       'docker' => Docker, 'openshift' => OpenShift,
       'kubernetes' => Kubernetes
-    }
+    }.freeze
+
+    def self.bin_dir
+      @bin_dir
+    end
+
+    def self.temp_dir
+      @temp_dir
+    end
+
+    def self.bin_dir=(path)
+      @bin_dir = path
+    end
+
+    def self.temp_dir=(path)
+      @temp_dir = path
+    end
 
     class Command < Vagrant.plugin(2, :command)
-      OS_RELEASE_FILE = '/etc/os-release'
+      OS_RELEASE_FILE = '/etc/os-release'.freeze
 
       def self.synopsis
         I18n.t('servicemanager.synopsis')
@@ -37,11 +62,14 @@ module VagrantPlugins
       end
 
       def execute
+        ServiceManager.bin_dir = "#{@env.data_dir}/service-manager/bin"
+        ServiceManager.temp_dir = "#{@env.data_dir}/service-manager/tmp"
+
         argv = ARGV.dup
         # Don't propagate --debug argument to case operation
         if ARGV.include? '--debug'
           PluginLogger.enable_debug_mode
-          PluginLogger.set_logger(@logger)
+          PluginLogger.logger = @logger
           argv.delete('--debug')
         end
 
@@ -100,6 +128,8 @@ module VagrantPlugins
             case option
             when nil
               display_box_ip
+            when '--script-readable'
+              display_box_ip(true)
             when '--help', '-h'
               print_help(type: command)
             else
@@ -118,6 +148,19 @@ module VagrantPlugins
           else
             perform_service(command, subcommand)
           end
+        when 'install-cli'
+          exit_if_machine_not_running
+          # Transform hyphen into underscore which is valid char in method
+          command = command.tr('-', '_')
+          # Get options as Hash which are preceeded with -- like --version
+          options = Hash[ARGV.drop(3).each_slice(2).to_a]
+
+          case subcommand
+          when '--help', '-h'
+            print_help(type: command)
+          else
+            execute_install_cli(subcommand, options)
+          end
         when '--help', '-h', 'help'
           print_help
         else
@@ -127,7 +170,7 @@ module VagrantPlugins
 
       def execute_service(name, options = {})
         with_target_vms(nil, single_target: true) do |machine|
-          PluginUtil.service_class(name).info(machine, @env.ui, options)
+          PluginUtil.service_class(name).new(machine, @env).info(options)
         end
       end
 
@@ -142,16 +185,15 @@ module VagrantPlugins
       def execute_status_display(service = nil)
         with_target_vms(nil, single_target: true) do |machine|
           if service && SUPPORTED_SERVICES.include?(service)
-            PluginUtil.service_class(service).status(machine, @env.ui, service)
+            PluginUtil.service_class(service).new(machine, @env).status
           elsif service.nil?
             @env.ui.info I18n.t('servicemanager.commands.status.nil')
             SUPPORTED_SERVICES.each do |s|
-              PluginUtil.service_class(s).status(machine, @env.ui, s)
+              PluginUtil.service_class(s).new(machine, @env).status
             end
           else
             @env.ui.error I18n.t('servicemanager.commands.status.unsupported_service',
-                                  service: service,
-                                  services: SUPPORTED_SERVICES.join(', ') + ' etc')
+                                 service: service, services: SUPPORTED_SERVICES.join(', ') + ' etc')
             exit 126
           end
         end
@@ -169,7 +211,7 @@ module VagrantPlugins
             end
             # since we do not have feature to show the Kube connection information
             unless service == 'kubernetes'
-              service_class.info(machine, @env.ui, options)
+              service_class.new(machine, @env).info(options)
             end
           end
 
@@ -207,9 +249,42 @@ module VagrantPlugins
         end
       end
 
-      def display_box_ip
+      def display_box_ip(script_readable = false)
+        options = { script_readable: script_readable }
+
+        with_target_vms(nil, single_target: true) do |machine|
+          @env.ui.info machine.guest.capability(:machine_ip, options)
+        end
+      end
+
+      def execute_install_cli(service, options = {})
+        help_msg = I18n.t('servicemanager.commands.help.install_cli')
+
+        if service.nil?
+          service_missing_msg = I18n.t('servicemanager.commands.operation.service_missing')
+          @env.ui.error help_msg.gsub(/Install the client side tool for the service/, service_missing_msg)
+          exit 126
+        end
+
+        unless SUPPORTED_SERVICES.include?(service)
+          @env.ui.error "Unknown service '#{service}'."
+          @env.ui.error help_msg.gsub(/Install the client side tool for the service\n/, '')
+          exit 126
+        end
+
         with_target_vms(nil, single_target: true) do |machine|
-          @env.ui.info machine.guest.capability(:machine_ip)
+          args = [service.to_sym, machine, @env, options]
+
+          begin
+            args.last[:box_version] = machine.guest.capability(:os_variant)
+            Installer.new(*args).install
+          rescue Vagrant::Errors::GuestCapabilityNotFound
+            @env.ui.info 'Not a supported box.'
+            exit 126
+          rescue StandardError => e
+            @env.ui.error e.message
+            exit 126
+          end
         end
       end
     end

data/lib/vagrant-service-manager/config.rb

@@ -2,18 +2,18 @@ require 'set'
 
 module VagrantPlugins
   module ServiceManager
-    SERVICES = ['docker', 'openshift']
-    CONFIG_KEYS  = [
+    SERVICES = %w(docker openshift).freeze
+    CONFIG_KEYS = [
       :services, :openshift_docker_registry,
       :openshift_image_name, :openshift_image_tag
-    ]
+    ].freeze
 
     class Config < Vagrant.plugin('2', :config)
       attr_accessor(*CONFIG_KEYS)
 
       DEFAULTS = {
         services: ''
-      }
+      }.freeze
 
       def initialize
         super
@@ -28,10 +28,10 @@ module VagrantPlugins
         end
       end
 
-      def validate(machine)
+      def validate(_machine)
         errors = _detected_errors
         errors.concat(validate_services)
-        { "servicemanager" => errors }
+        { 'servicemanager' => errors }
       end
 
       private
@@ -39,7 +39,7 @@ module VagrantPlugins
       def validate_services
         errors = []
 
-        unless is_supported_services?
+        unless supported_services?
           errors << I18n.t('servicemanager.config.supported_devices',
                            services: SERVICES.inspect)
         end
@@ -47,7 +47,7 @@ module VagrantPlugins
         errors
       end
 
-      def is_supported_services?
+      def supported_services?
         @services.split(',').map(&:strip).to_set.subset?(SERVICES.to_set)
       end
     end