vagrant-k3s 0.1.0

data/lib/vagrant-k3s/command.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+require 'vagrant'
+
+module VagrantPlugins
+  module K3s
+    class Command < Vagrant.plugin('2', :command)
+      require_relative 'command/kubectl'
+
+      def self.synopsis
+        'Lightweight Kubernetes'
+      end
+
+      def execute
+        @logger.debug("ITS K3S!")
+        0
+      end
+    end
+  end
+end

data/lib/vagrant-k3s/config.rb

@@ -0,0 +1,148 @@
+# frozen_string_literal: true
+
+module VagrantPlugins
+  module K3s
+    class Config < Vagrant.plugin(2, :config)
+      DEFAULT_CONFIG_PATH = "/etc/rancher/k3s/conf.yaml".freeze
+      DEFAULT_ENV_PATH = "/etc/rancher/k3s/install.env".freeze
+      DEFAULT_INSTALLER_URL = "https://get.k3s.io".freeze
+
+      # string or array
+      # @return [Array<String]
+      attr_accessor :args
+
+      # string (.yaml) or hash
+      # @return [Hash]
+      attr_accessor :config
+
+      # Defaults to `/etc/rancher/k3s/conf.yaml`
+      # @return [String]
+      attr_accessor :config_path
+
+      # string (.env), array, or hash
+      # @return [Array<String>]
+      attr_accessor :env
+
+      # Defaults to `/etc/rancher/k3s/install.env`
+      # @return [String]
+      attr_accessor :env_path
+
+      # Defaults to `https://get.k3s.io`
+      # @return [String]
+      attr_accessor :installer_url
+
+      # # INSTALL_K3S_BIN_DIR
+      # # @return [String]
+      # attr_accessor :install_bin_dir
+      #
+      # # INSTALL_K3S_BIN_DIR_READ_ONLY
+      # # @return [Boolean]
+      # attr_accessor :install_bin_dir_ro
+      #
+      # # INSTALL_K3S_CHANNEL
+      # # @return [String]
+      # attr_accessor :install_channel
+      #
+      # # INSTALL_K3S_CHANNEL_URL
+      # # @return [String]
+      # attr_accessor :install_channel_url
+      #
+      # # INSTALL_K3S_COMMIT
+      # # @return [String]
+      # attr_accessor :install_commit
+      #
+      # # INSTALL_K3S_EXEC
+      # # @return [String]
+      # attr_accessor :install_exec
+      #
+      # # INSTALL_K3S_NAME
+      # # @return [String]
+      # attr_accessor :install_name
+      #
+      # # INSTALL_K3S_SYSTEMD_DIR
+      # # @return [String]
+      # attr_accessor :install_systemd_dir
+      #
+      # # INSTALL_K3S_TYPE
+      # # @return [String]
+      # attr_accessor :install_systemd_type
+      #
+      # # INSTALL_K3S_VERSION
+      # # @return [String]
+      # attr_accessor :install_version
+
+      def initialize
+        @args = UNSET_VALUE
+        @config = UNSET_VALUE
+        @config_path = UNSET_VALUE
+        @env = UNSET_VALUE
+        @env_path = UNSET_VALUE
+        @installer_url = UNSET_VALUE
+      end
+
+      def finalize!
+        @args = [] if @args == UNSET_VALUE
+        @config = "" if @config == UNSET_VALUE
+        @config_path = DEFAULT_CONFIG_PATH if @config_path == UNSET_VALUE
+        @env = [] if @env == UNSET_VALUE
+        @env_path = DEFAULT_ENV_PATH if @env_path == UNSET_VALUE
+        @installer_url = DEFAULT_INSTALLER_URL if @installer_url == UNSET_VALUE
+
+        if @args && args_valid?
+          @args = @args.is_a?(Array) ? @args.map { |a| a.to_s } : @args.to_s
+        end
+      end
+
+      def validate(machine)
+        errors = _detected_errors
+
+        unless args_valid?
+          errors << "K3s provisioner `args` must be an array or string."
+        end
+
+        unless config_valid?
+          errors << "K3s provisioner `config` must be a hash or string (yaml)."
+        end
+
+        unless env_valid?
+          errors << "K3s provisioner `env` must be an array, hash, or string."
+        end
+
+        { "k3s provisioner" => errors }
+      end
+
+      def args_valid?
+        return true unless args
+        return true if args.is_a?(String)
+        return true if args.is_a?(Integer)
+        if args.is_a?(Array)
+          args.each do |a|
+            return false if !a.kind_of?(String) && !a.kind_of?(Integer)
+          end
+          return true
+        end
+        false
+      end
+
+      def config_valid?
+        return true unless args
+        return true if config.is_a?(String)
+        return true if config.is_a?(Hash)
+        false
+      end
+
+      def env_valid?
+        return true unless env
+        return true if env.is_a?(String)
+        return true if env.is_a?(Hash)
+        if env.is_a?(Array)
+          env.each do |a|
+            return false unless a.kind_of?(String)
+          end
+          return true
+        end
+        false
+      end
+    end
+  end
+end

data/lib/vagrant-k3s/plugin.rb

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+require_relative 'version'
+
+module VagrantPlugins
+  module K3s
+    class Plugin < Vagrant.plugin(2)
+      name 'k3s'
+      description 'Lightweight Kubernetes'
+
+      config(:k3s, :provisioner) do
+        require_relative 'config'
+        Config
+      end
+
+      provisioner(:k3s) do
+        require_relative 'provisioner'
+        Provisioner
+      end
+
+      command(:k3s, primary: true) do
+        require_relative 'command'
+        Command
+      end
+
+      guest_capability(:linux, :k3s_installed) do
+        require_relative "cap/linux/k3s_installed"
+        Cap::Linux::K3sInstalled
+      end
+
+      guest_capability(:linux, :curl_installed) do
+        require_relative "cap/linux/curl_installed"
+        Cap::Linux::CurlInstalled
+      end
+
+      guest_capability(:alpine, :curl_install) do
+        require_relative "cap/alpine/curl_install"
+        Cap::Alpine::CurlInstall
+      end
+
+      guest_capability(:debian, :curl_install) do
+        require_relative "cap/debian/curl_install"
+        Cap::Debian::CurlInstall
+      end
+
+      guest_capability(:redhat, :curl_install) do
+        require_relative "cap/redhat/curl_install"
+        Cap::Redhat::CurlInstall
+      end
+
+      guest_capability(:suse, :curl_install) do
+        require_relative "cap/suse/curl_install"
+        Cap::Suse::CurlInstall
+      end
+    end
+  end
+end

data/lib/vagrant-k3s/provisioner.rb

@@ -0,0 +1,144 @@
+# frozen_string_literal: false
+
+require 'multi_json/convertible_hash_keys'
+require 'vagrant'
+require 'vagrant/errors'
+require 'yaml'
+
+module VagrantPlugins
+  module K3s
+    class Provisioner < Vagrant.plugin('2', :provisioner)
+      include MultiJson::ConvertibleHashKeys
+      def initialize(machine,config)
+        super(machine,config)
+        @logger = Log4r::Logger.new("vagrant::provisioners::k3s")
+      end
+
+      def configure(root_config)
+      end
+
+      def cleanup
+      end
+
+      def provision
+        args = ""
+        if config.args.is_a?(String)
+          args = " #{config.args.to_s}"
+        elsif config.args.is_a?(Array)
+          args = config.args.map { |a| quote_and_escape(a) }
+          args = " #{args.join(" ")}"
+        end
+
+        @machine.ui.info 'Ensuring cURL ...'
+        unless @machine.guest.capability(:curl_installed)
+          @machine.guest.capability(:curl_install)
+        end
+
+        @machine.ui.info 'Installing K3s ...'
+        config_file = config.config_path.to_s
+        config_yaml = config.config.is_a?(String) ? config.config : stringify_keys(config.config).to_yaml
+        with_file_upload "k3s-config.yaml".freeze, config_file, config_yaml
+
+        install_env_file = config.env_path.to_s
+        install_env_text = ""
+        if config.env.is_a?(String)
+          install_env_text = config.env.to_s
+        end
+        if config.env.is_a?(Array)
+          config.env.each {|line| install_env_text << "#{line.to_s}\n"}
+        end
+        if config.env.is_a?(Hash)
+          config.env.each {|key,value| install_env_text << "#{key.to_s}=#{quote_and_escape(value.to_s)}\n"}
+        end
+        with_file_upload "k3s-install.env".freeze, install_env_file, install_env_text
+
+        install_sh = "/tmp/vagrant-k3s-provisioner-install.sh".freeze
+        with_file_upload "k3s-install.sh", install_sh, <<~EOF
+          #/usr/bin/env bash
+          set -eu -o pipefail
+          mkdir -p $(dirname #{config.config_path}) $(dirname #{config.env_path})
+          chown  root:root #{config.config_path} #{config.env_path}
+          set -o allexport
+          source #{config.env_path}
+          set +o allexport
+          curl -fsL #{config.installer_url} | sh -s - #{args}
+        EOF
+        outputs, handler = build_outputs
+        begin
+          @machine.communicate.sudo("sh #{install_sh}", error_key: :ssh_bad_exit_status_muted, &handler)
+        ensure
+          outputs.values.map(&:close)
+        end
+
+        @machine.guest.capability(:k3s_installed)
+
+        begin
+          exe = "k3s"
+          @machine.ui.info 'Checking the K3s version ...'
+          @machine.communicate.execute("/usr/bin/which k3s", :error_key => :ssh_bad_exit_status_muted) do |type, data|
+            exe = data.chomp if type == :stdout
+          end
+        rescue Vagrant::Errors::VagrantError => e
+          @machine.ui.detail "#{e.extra_data[:stderr].chomp}", :color => :red
+        else
+          outputs, handler = build_outputs
+          begin
+            @machine.communicate.execute("#{exe} --version", :error_key => :ssh_bad_exit_status_muted, &handler)
+          ensure
+            outputs.values.map(&:close)
+          end
+        end
+      end
+
+      def build_outputs
+        outputs = {
+          stdout: Vagrant::Util::LineBuffer.new { |line| handle_comm(:stdout, line) },
+          stderr: Vagrant::Util::LineBuffer.new { |line| handle_comm(:stderr, line) },
+        }
+        block = proc { |type, data|
+          outputs[type] << data if outputs[type]
+        }
+        [outputs, block]
+      end
+
+      # This handles outputting the communication line back to the UI
+      def handle_comm(type, data)
+        if [:stderr, :stdout].include?(type)
+          # Output the line with the proper color based on the stream.
+          options = {}
+          options[:color] = type == :stdout ? :green : :red
+
+          @machine.ui.detail(data.chomp, **options)
+        end
+      end
+
+      def with_file(name='vagrant-k3s-provisioner', content)
+        file = Tempfile.new([name])
+        file.binmode
+        begin
+          file.write(content)
+          file.fsync
+          file.close
+          yield file.path
+        ensure
+          file.close
+          file.unlink
+        end
+      end
+
+      def with_file_upload(name='vagrant-k3s-provisioner', to, content)
+        with_file(name, content) do |from|
+          tmpdir = @machine.guest.capability :create_tmp_path, {:type => :directory}
+          tmpfile = [tmpdir, File.basename(to)].join('/')
+          @machine.communicate.upload(from, tmpfile)
+          @machine.communicate.sudo("mv -f #{tmpfile} #{to}")
+        end
+        to
+      end
+
+      def quote_and_escape(text, quote = '"')
+        "#{quote}#{text.to_s.gsub(/#{quote}/) { |m| "#{m}\\#{m}#{m}" }}#{quote}"
+      end
+    end
+  end
+end

data/lib/vagrant-k3s/version.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+module VagrantPlugins
+  module K3s
+    VERSION = '0.1.0'
+  end
+end

data/lib/vagrant-k3s.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+begin
+  require 'vagrant'
+rescue LoadError
+  raise 'The Vagrant K3s plugin must be run within Vagrant.'
+end
+
+require_relative 'vagrant-k3s/plugin'
+require_relative 'vagrant-k3s/version'

data/vagrant-k3s.gemspec

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+$:.unshift File.expand_path("../lib", __FILE__)
+require 'vagrant-k3s/version'
+
+Gem::Specification.new do |spec|
+  spec.name         = 'vagrant-k3s'
+  spec.version      = VagrantPlugins::K3s::VERSION
+  spec.platform     = Gem::Platform::RUBY
+  spec.authors      = ['Jacob Blain Christen']
+  spec.email        = ['dweomer5@gmail.com']
+
+  spec.license      = 'Apache 2.0'
+
+  spec.summary      = 'Manage k3s installations on Vagrant guests.'
+  spec.description  = spec.summary
+  spec.homepage     = 'https://github.com/dweomer/vagrant-k3s'
+
+  spec.metadata['homepage_uri'] = spec.homepage
+  spec.metadata['source_code_uri'] = spec.homepage
+
+  spec.required_ruby_version     = ">= 2.5", "< 3.1"
+  spec.required_rubygems_version = ">= 1.3.6"
+
+  spec.add_development_dependency "rake", "~> 13.0"
+  spec.add_development_dependency "rspec", "~> 3.10.0"
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files = Dir.chdir(File.expand_path(__dir__)) do
+    `git ls-files -z`.split("\x0").reject do |f|
+      (f == __FILE__) || f.match(%r{\A(?:(?:test|spec|features)/|\.(?:git|travis|circleci)|appveyor)})
+    end
+  end
+  spec.bindir        = 'exe'
+  spec.executables   = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
+  spec.require_paths = ['lib']
+end

metadata

@@ -0,0 +1,141 @@
+--- !ruby/object:Gem::Specification
+name: vagrant-k3s
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Jacob Blain Christen
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2021-10-04 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.10.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.10.0
+description: Manage k3s installations on Vagrant guests.
+email:
+- dweomer5@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- CHANGELOG.md
+- CODE_OF_CONDUCT.md
+- Gemfile
+- Gemfile.lock
+- LICENSE
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- lib/vagrant-k3s.rb
+- lib/vagrant-k3s/cap/alpine/curl_install.rb
+- lib/vagrant-k3s/cap/debian/curl_install.rb
+- lib/vagrant-k3s/cap/linux/curl_installed.rb
+- lib/vagrant-k3s/cap/linux/k3s_installed.rb
+- lib/vagrant-k3s/cap/redhat/curl_install.rb
+- lib/vagrant-k3s/cap/suse/curl_install.rb
+- lib/vagrant-k3s/command.rb
+- lib/vagrant-k3s/command/kubectl.rb
+- lib/vagrant-k3s/config.rb
+- lib/vagrant-k3s/plugin.rb
+- lib/vagrant-k3s/provisioner.rb
+- lib/vagrant-k3s/version.rb
+- vagrant-k3s.gemspec
+- vendor/cache/bcrypt_pbkdf-1.1.0.gem
+- vendor/cache/builder-3.2.4.gem
+- vendor/cache/childprocess-4.1.0.gem
+- vendor/cache/concurrent-ruby-1.1.9.gem
+- vendor/cache/diff-lcs-1.4.4.gem
+- vendor/cache/ed25519-1.2.4.gem
+- vendor/cache/erubi-1.10.0.gem
+- vendor/cache/excon-0.86.0.gem
+- vendor/cache/ffi-1.15.4.gem
+- vendor/cache/gssapi-1.3.1.gem
+- vendor/cache/gyoku-1.3.1.gem
+- vendor/cache/hashicorp-checkpoint-0.1.5.gem
+- vendor/cache/httpclient-2.8.3.gem
+- vendor/cache/i18n-1.8.10.gem
+- vendor/cache/listen-3.7.0.gem
+- vendor/cache/little-plugger-1.1.4.gem
+- vendor/cache/log4r-1.1.10.gem
+- vendor/cache/logging-2.3.0.gem
+- vendor/cache/mime-types-3.3.1.gem
+- vendor/cache/mime-types-data-3.2021.0901.gem
+- vendor/cache/multi_json-1.15.0.gem
+- vendor/cache/net-scp-3.0.0.gem
+- vendor/cache/net-sftp-3.0.0.gem
+- vendor/cache/net-ssh-6.1.0.gem
+- vendor/cache/nori-2.6.0.gem
+- vendor/cache/rake-13.0.6.gem
+- vendor/cache/rb-fsevent-0.11.0.gem
+- vendor/cache/rb-inotify-0.10.1.gem
+- vendor/cache/rb-kqueue-0.2.7.gem
+- vendor/cache/rexml-3.2.5.gem
+- vendor/cache/rspec-3.10.0.gem
+- vendor/cache/rspec-core-3.10.1.gem
+- vendor/cache/rspec-expectations-3.10.1.gem
+- vendor/cache/rspec-mocks-3.10.2.gem
+- vendor/cache/rspec-support-3.10.2.gem
+- vendor/cache/rubyntlm-0.6.3.gem
+- vendor/cache/rubyzip-2.3.2.gem
+- vendor/cache/vagrant_cloud-3.0.5.gem
+- vendor/cache/wdm-0.1.1.gem
+- vendor/cache/winrm-2.3.6.gem
+- vendor/cache/winrm-elevated-1.2.3.gem
+- vendor/cache/winrm-fs-1.3.5.gem
+homepage: https://github.com/dweomer/vagrant-k3s
+licenses:
+- Apache 2.0
+metadata:
+  homepage_uri: https://github.com/dweomer/vagrant-k3s
+  source_code_uri: https://github.com/dweomer/vagrant-k3s
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '2.5'
+  - - "<"
+    - !ruby/object:Gem::Version
+      version: '3.1'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 1.3.6
+requirements: []
+rubygems_version: 3.1.6
+signing_key: 
+specification_version: 4
+summary: Manage k3s installations on Vagrant guests.
+test_files: []