Contents

module Vidar
  module K8s
    class PodSet
      def initialize(namespace:, filter: nil)
        @namespace = namespace
        @filter = filter
      end

      def deployed?
        if items.empty?
          Log.error "Could not fetch pod list"
          return false
        end

        Log.line

        container_statuses.each(&:print)

        Log.line

        container_statuses.all?(&:deployed?)
      end

      def success?
        container_statuses.all?(&:success?)
      end

      private

      attr_reader :namespace, :filter

      def items
        @items ||= begin
          json = JSON.parse(kubectl_get.strip)
          json["items"] || []
        end
      end

      def kubectl_get
        if namespace == "all"
          `kubectl get pods --all-namespaces -o json`
        else
          `kubectl get pods -n #{namespace} -o json`
        end
      end

      def container_statuses
        return all_container_statuses unless filter

        all_container_statuses.select { |cs| cs.name.to_s.include?(filter) }
      end

      def all_container_statuses
        @all_container_statuses ||= container_statuses_data.map { |status| ContainerStatus.new(status) }
      end

      def container_statuses_data
        items.map do |i|
          owner_references = i.dig("metadata", "ownerReferences") || []
          kind             = (owner_references[0] || {})["kind"]
          namespace        = i.dig("metadata", "namespace")
          statuses         = i.dig("status", "containerStatuses") || []
          statuses.each do |s|
            s["namespace"] = namespace
            s["kind"] = kind
          end
          statuses
        end.flatten
      end
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
vidar-0.4.3	lib/vidar/k8s/pod_set.rb
vidar-0.4.1	lib/vidar/k8s/pod_set.rb
vidar-0.4.0	lib/vidar/k8s/pod_set.rb