helm-wrapper 0.1.0

data/lib/helm-wrapper/shared/latest.rb

@@ -0,0 +1,79 @@
+###############################################################################
+
+require 'json'
+require 'net/http'
+require 'singleton'
+require 'uri'
+
+###############################################################################
+
+module HelmWrapper
+
+###############################################################################
+
+  module Shared
+
+###############################################################################
+
+    class Latest
+
+###############################################################################
+
+      include Singleton
+
+###############################################################################
+
+      include HelmWrapper::Shared::Logging
+
+###############################################################################
+
+      @version
+
+###############################################################################
+
+      def version
+        @version ||= refresh
+
+        return @version
+      end
+
+###############################################################################
+
+      private
+
+###############################################################################
+
+      def refresh
+        logger.info("Finding latest available Helm release...")
+
+        response = Net::HTTP.get_response(URI("https://api.github.com/repos/helm/helm/releases/latest"))
+
+        logger.fatal("GitHub API did not return status 200 for latest version check!") if response.code != "200"
+        logger.fatal("Response body from GitHub API is not permitted!")                if not response.class.body_permitted?
+        logger.fatal("Response body from GitHub API is empty!")                        if response.body.nil?
+
+        body = JSON.parse(response.body)
+
+        logger.fatal("GitHub API JSON response did not include latest available Helm tag name!") if not body.key?("tag_name")
+        logger.fatal("GitHub API indicated the latest available tag name for Helm is blank!")    if body["tag_name"].empty?
+
+        version = body["tag_name"]
+
+        logger.success("Latest available Helm release found: #{version}")
+
+        return version
+      end
+
+###############################################################################
+
+    end
+
+###############################################################################
+
+  end
+
+###############################################################################
+
+end
+
+###############################################################################

data/lib/helm-wrapper/shared/logger.rb

@@ -0,0 +1,80 @@
+###############################################################################
+
+require 'logger'
+
+###############################################################################
+
+module HelmWrapper
+
+###############################################################################
+
+  module Shared
+
+###############################################################################
+
+    class Logger < ::Logger
+
+###############################################################################
+
+      @colour = false
+
+###############################################################################
+
+      def colour()
+        @colour
+      end
+
+###############################################################################
+
+      def colour=(enabled)
+        @colour = [ true, false ].include?(enabled) ? enabled : false
+      end
+
+###############################################################################
+
+      def success(message)
+        info(format(colour: 32, message: message))
+      end
+
+###############################################################################
+
+      def warn(message)
+        super(format(colour: 33, message: message))
+      end
+
+###############################################################################
+
+      def error(message)
+        super(format(colour: 31, message: message))
+      end
+
+###############################################################################
+
+      def fatal(message)
+        super(format(colour: 31, message: message))
+        exit(1)
+      end
+
+###############################################################################
+
+      private
+
+###############################################################################
+
+      def format(colour: 32, message:)
+        return @colour ? "\e[" + colour.to_s + "m" + message + "\e[0m" : message
+      end
+
+###############################################################################
+
+    end
+
+###############################################################################
+
+  end
+
+###############################################################################
+
+end
+
+###############################################################################

data/lib/helm-wrapper/shared/logging.rb

@@ -0,0 +1,77 @@
+###############################################################################
+
+require 'logger'
+
+###############################################################################
+
+module HelmWrapper
+
+###############################################################################
+
+  module Shared
+
+###############################################################################
+
+    module Logging
+
+###############################################################################
+
+      private
+
+###############################################################################
+
+      def logger
+        @logger ||= Logging.logger_for(self.class.name)
+      end
+
+###############################################################################
+
+      @loggers = {}
+
+###############################################################################
+
+      class << self
+
+###############################################################################
+
+        def logger_for(classname)
+          @loggers[classname] ||= configure_logger_for(classname)
+        end
+
+###############################################################################
+
+        def configure_logger_for(classname)
+          colour = ENV["HELM_WRAPPER_LOG_COLOUR"] || "true"
+          level  = ENV["HELM_WRAPPER_LOG_LEVEL"]  || "INFO"
+
+          logger = ::HelmWrapper::Shared::Logger.new(STDOUT)
+
+          logger.colour   = colour.downcase == "true"
+          logger.level    = level.upcase
+          logger.progname = classname
+
+          logger.formatter = proc do |severity, datetime, progname, msg|
+            sevId = severity.chars.first.upcase
+            "[#{sevId}] [#{progname}] #{msg}\n"
+          end
+
+          logger
+        end
+
+###############################################################################
+
+      end
+
+###############################################################################
+
+    end
+
+###############################################################################
+
+  end
+
+###############################################################################
+
+end
+
+###############################################################################

data/lib/helm-wrapper/shared/runner.rb

@@ -0,0 +1,250 @@
+###############################################################################
+
+require 'open3'
+
+###############################################################################
+
+module HelmWrapper
+
+###############################################################################
+
+  module Shared
+
+###############################################################################
+
+    class Runner
+
+###############################################################################
+
+      include HelmWrapper::Shared::Logging
+
+###############################################################################
+
+      @@repo_actions = [ "registry", "repo" ]
+
+###############################################################################
+
+      @auths_attempted = false
+
+###############################################################################
+
+      attr_reader :auths
+      attr_reader :binary
+      attr_reader :chart
+      attr_reader :config
+      attr_reader :repos
+
+###############################################################################
+
+      def initialize(binary:, chart:, config:)
+        @binary = binary
+        @chart  = chart
+        @config = config
+
+        @auths = false
+        @repos = false
+      end
+
+###############################################################################
+
+      def init_auths()
+        @auths_attempted = true
+
+        config.auths.map(&:auth)
+
+        @auths = true
+      end
+
+###############################################################################
+
+      def init_repos(force: false)
+        ENV["HELM_EXPERIMENTAL_OCI"] = "true"
+
+        @chart.oci.each_with_index do |oci_repo, oci_index|
+          logger.info("Logging into OCI repository: #{oci_repo["name"]}")
+          @chart.oci_active(active: true, index: oci_index)
+
+          parameters = Array.new
+          parameters.append("login")
+          parameters.append("\"#{oci_repo["url"]}\"")
+          parameters.append("--username=\"#{oci_repo["username"]}\"")
+          parameters.append("--password-stdin")
+          logger.fatal("Failed to login to Helm OCI repository: #{oci_repo["name"]}, url: #{oci_repo["url"]}") unless run(action: "registry", parameters: parameters, stdin: oci_repo["password"])
+        end
+
+        if @chart.artefact.length > 0 then
+
+          @chart.artefact.each_with_index do |artefact_repo, artefact_index|
+            logger.info("Adding artefact repository: #{artefact_repo["name"]}")
+            @chart.artefact_active(active: true, index: artefact_index)
+
+            parameters = Array.new
+            parameters.append("add")
+            parameters.append("\"#{artefact_repo["name"]}\"")
+            parameters.append("\"#{artefact_repo["url"]}\"")
+            parameters.append("--username=\"#{artefact_repo["username"]}\"") unless artefact_repo["username"].nil?
+            parameters.append("--password=\"#{artefact_repo["password"]}\"") unless artefact_repo["password"].nil?
+            parameters.append("--force-update") if force
+            logger.fatal("Failed to add Helm repository: #{artefact_repo["name"]}, url: #{artefact_repo["url"]}") unless run(action: "repo", parameters: parameters)
+          end
+
+          logger.info("Updating artefact repositories...")
+          logger.fatal("Failed to update Helm artefact repositories!") unless run(action: "repo", parameters: Array.new([ "update" ]))
+        end
+
+        @repos = true
+      end
+
+###############################################################################
+
+      def clean(repos: true)
+        clean_repos if repos
+        clean_auths if @auths_attempted
+      end
+
+###############################################################################
+
+      def delete()
+        logger.fatal("Cannot Helm delete before initialising authenticators!") unless auths
+
+        parameters = Array.new
+        parameters.append("--namespace \"#{@config.namespace}\"")
+        parameters.append(@config.release)
+
+        logger.fatal("Helm delete failed!") unless run(action: "delete", parameters: parameters)
+      end
+
+###############################################################################
+
+      def upgrade(install: true)
+        logger.fatal("Cannot Helm upgrade before initialising authenticators!") unless auths
+        logger.fatal("Cannot Helm upgrade before initialising repositories!") unless repos
+
+        parameters = Array.new
+        parameters.append("--namespace \"#{@config.namespace}\"")
+        parameters.append("--install") if install
+        parameters.append(@config.release)
+        parameters.append(@chart.name)
+        parameters.concat(variable_files)
+        parameters.concat(variable_strings)
+
+        logger.fatal("Helm upgrade failed!") unless run(action: "upgrade", parameters: parameters)
+      end
+
+###############################################################################
+
+    private
+
+###############################################################################
+
+      def clean_auths
+        @auths = false
+        config.auths.map(&:clear)
+        @auths_attempted = false
+      end
+
+###############################################################################
+
+      def clean_repos
+        @repos = false
+
+        @chart.oci.each do |oci_repo|
+          if oci_repo["active"] then
+            logger.info("Logging out of OCI repository: #{oci_repo["name"]}")
+
+            parameters = Array.new
+            parameters.append("logout")
+            parameters.append("\"#{oci_repo["url"]}\"")
+            logger.error("Failed to logout of Helm OCI repository: #{oci_repo["name"]}, url: #{oci_repo["url"]}") unless run(action: "registry", parameters: parameters)
+          end
+        end
+
+        @chart.artefact.each do |artefact_repo|
+          if artefact_repo["active"] then
+            if artefact_repo["password"].nil? then
+              logger.info("Not removing artefact repository: #{artefact_repo["name"]}, no credentials stored.")
+            else
+              logger.info("Removing artefact repository: #{artefact_repo["name"]}")
+
+              parameters = Array.new
+              parameters.append("remove")
+              parameters.append("\"#{artefact_repo["name"]}\"")
+              logger.error("Failed to remove Helm repository: #{artefact_repo["name"]}, url: #{artefact_repo["url"]}") unless run(action: "repo", parameters: parameters)
+            end
+          end
+        end
+
+        ENV.delete("HELM_EXPERIMENTAL_OCI")
+      end
+
+###############################################################################
+
+      def variable_files
+        result = Array.new
+
+        @config.variable_files.each do |variable_file|
+          result.append("--values=\"#{variable_file}\"")
+        end
+
+        return result
+      end
+
+###############################################################################
+
+      def variable_strings
+        result = Array.new
+
+        result.append("--set=\"config=#{@config.name}\"")
+
+        @config.variables.values.each do |key, value|
+          result.append("--set=\"#{key.to_s}=#{value}\"")
+        end
+
+        return result
+      end
+
+###############################################################################
+
+      def run(action:, parameters: Array.new, stdin: nil)
+        result = false
+
+        parameters.reject! { |item| not item.kind_of?(String) or item.strip.empty? }
+
+        cmdline = [ "\"#{@binary.path}\"", action ].concat(parameters).join(" ")
+
+        if @@repo_actions.include?(action)
+          begin
+            out, status = Open3.capture2e(cmdline, :stdin_data=>stdin)
+            logger.debug("Helm output:")
+            logger.debug(out)
+            result = status.success?
+          rescue
+            result = false
+          end
+        else
+          logger.info("Starting Helm, action: #{action}")
+
+          fatal("Running Helm with data for stdin can only be used when configuring repos!") unless stdin.nil?
+          puts("\n" + ('#' * 80) + "\n\n")
+
+          result = system(cmdline) || false
+
+          puts("\n" + ('#' * 80) + "\n\n")
+        end
+
+        return result
+      end
+
+###############################################################################
+
+    end
+
+###############################################################################
+
+  end
+
+###############################################################################
+
+end
+
+###############################################################################