127 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			Ruby
		
	
	
	
			
		
		
	
	
			127 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			Ruby
		
	
	
	
# frozen_string_literal: true
 | 
						|
 | 
						|
module Gitlab
 | 
						|
  module Kubernetes
 | 
						|
    module Helm
 | 
						|
      class API
 | 
						|
        def initialize(kubeclient)
 | 
						|
          @kubeclient = kubeclient
 | 
						|
          @namespace = Gitlab::Kubernetes::Namespace.new(
 | 
						|
            Gitlab::Kubernetes::Helm::NAMESPACE,
 | 
						|
            kubeclient,
 | 
						|
            labels: Gitlab::Kubernetes::Helm::NAMESPACE_LABELS
 | 
						|
          )
 | 
						|
        end
 | 
						|
 | 
						|
        def install(command)
 | 
						|
          namespace.ensure_exists!
 | 
						|
 | 
						|
          create_service_account(command)
 | 
						|
          create_cluster_role_binding(command)
 | 
						|
          create_config_map(command)
 | 
						|
 | 
						|
          delete_pod!(command.pod_name)
 | 
						|
          kubeclient.create_pod(command.pod_resource)
 | 
						|
        end
 | 
						|
 | 
						|
        alias_method :update, :install
 | 
						|
 | 
						|
        def uninstall(command)
 | 
						|
          namespace.ensure_exists!
 | 
						|
          create_config_map(command)
 | 
						|
 | 
						|
          delete_pod!(command.pod_name)
 | 
						|
          kubeclient.create_pod(command.pod_resource)
 | 
						|
        end
 | 
						|
 | 
						|
        ##
 | 
						|
        # Returns Pod phase
 | 
						|
        #
 | 
						|
        # https://kubernetes.io/docs/concepts/workloads/pods/pod-lifecycle/#pod-phase
 | 
						|
        #
 | 
						|
        # values: "Pending", "Running", "Succeeded", "Failed", "Unknown"
 | 
						|
        #
 | 
						|
        def status(pod_name)
 | 
						|
          kubeclient.get_pod(pod_name, namespace.name).status.phase
 | 
						|
        end
 | 
						|
 | 
						|
        def log(pod_name)
 | 
						|
          kubeclient.get_pod_log(pod_name, namespace.name).body
 | 
						|
        end
 | 
						|
 | 
						|
        def delete_pod!(pod_name)
 | 
						|
          kubeclient.delete_pod(pod_name, namespace.name)
 | 
						|
        rescue ::Kubeclient::ResourceNotFoundError
 | 
						|
          # no-op
 | 
						|
        end
 | 
						|
 | 
						|
        def get_config_map(config_map_name)
 | 
						|
          namespace.ensure_exists!
 | 
						|
 | 
						|
          kubeclient.get_config_map(config_map_name, namespace.name)
 | 
						|
        end
 | 
						|
 | 
						|
        private
 | 
						|
 | 
						|
        attr_reader :kubeclient, :namespace
 | 
						|
 | 
						|
        def create_config_map(command)
 | 
						|
          command.config_map_resource.tap do |config_map_resource|
 | 
						|
            break unless config_map_resource
 | 
						|
 | 
						|
            if config_map_exists?(config_map_resource)
 | 
						|
              kubeclient.update_config_map(config_map_resource)
 | 
						|
            else
 | 
						|
              kubeclient.create_config_map(config_map_resource)
 | 
						|
            end
 | 
						|
          end
 | 
						|
        end
 | 
						|
 | 
						|
        def update_config_map(command)
 | 
						|
          command.config_map_resource.tap do |config_map_resource|
 | 
						|
            kubeclient.update_config_map(config_map_resource)
 | 
						|
          end
 | 
						|
        end
 | 
						|
 | 
						|
        def create_service_account(command)
 | 
						|
          command.service_account_resource.tap do |service_account_resource|
 | 
						|
            break unless service_account_resource
 | 
						|
 | 
						|
            if service_account_exists?(service_account_resource)
 | 
						|
              kubeclient.update_service_account(service_account_resource)
 | 
						|
            else
 | 
						|
              kubeclient.create_service_account(service_account_resource)
 | 
						|
            end
 | 
						|
          end
 | 
						|
        end
 | 
						|
 | 
						|
        def create_cluster_role_binding(command)
 | 
						|
          command.cluster_role_binding_resource.tap do |cluster_role_binding_resource|
 | 
						|
            break unless cluster_role_binding_resource
 | 
						|
 | 
						|
            kubeclient.update_cluster_role_binding(cluster_role_binding_resource)
 | 
						|
          end
 | 
						|
        end
 | 
						|
 | 
						|
        def config_map_exists?(resource)
 | 
						|
          kubeclient.get_config_map(resource.metadata.name, resource.metadata.namespace)
 | 
						|
        rescue ::Kubeclient::ResourceNotFoundError
 | 
						|
          false
 | 
						|
        end
 | 
						|
 | 
						|
        def service_account_exists?(resource)
 | 
						|
          kubeclient.get_service_account(resource.metadata.name, resource.metadata.namespace)
 | 
						|
        rescue ::Kubeclient::ResourceNotFoundError
 | 
						|
          false
 | 
						|
        end
 | 
						|
 | 
						|
        def cluster_role_binding_exists?(resource)
 | 
						|
          kubeclient.get_cluster_role_binding(resource.metadata.name)
 | 
						|
        rescue ::Kubeclient::ResourceNotFoundError
 | 
						|
          false
 | 
						|
        end
 | 
						|
      end
 | 
						|
    end
 | 
						|
  end
 | 
						|
end
 |