# frozen_string_literal: true

require "open3"
require 'English'
require_relative "helm_upgrade_logs/version"

ENV["helm_upgrade_logs_log_start"] ||= "90"
ENV["helm_upgrade_logs_pod_start"] ||= "35"

# Approach not ideal as it will for all containers to be ready and want to stream logs before that
def wait_for_container_ready
  wait_pid = Process.spawn 'kubectl wait --for=condition=ContainersReady pod --selector "app.kubernetes.io/managed-by=Helm" --timeout=30s'
  Process.wait wait_pid
end

# Wait for very first pods with logs to be present
# Not ideal due to https://github.com/kubernetes/kubernetes/issues/28746
def wait_for_pod_to_log
  ENV["helm_upgrade_logs_log_start"].to_i.times do |i|
    break unless Process.waitpid(@helm_pid, Process::WNOHANG).nil?

    sleep 1
    stdout, stderr, = Open3.capture3(add_ns("kubectl logs -lapp.kubernetes.io/instance=#{@release_name}"))
    if stderr.empty? && !stdout.strip.empty?
      puts "[INFO] Pods with logs found"
      break
    elsif (i % 3).zero?
      puts "[INFO] Waiting for pod logs: #{stderr}"
    end
  end
end

# Wait for logs from a specific pod
def wait_for_specific_pod_to_log(pod_name)
  ENV["helm_upgrade_logs_pod_start"].to_i.times do |i|
    sleep 1
    stdout, stderr, = Open3.capture3(add_ns("kubectl logs #{pod_name}"))
    if stderr.empty? && !stdout.strip.empty?
      puts "[INFO] Pod #{pod_name} with logs found"
      break
    elsif i.even?
      puts "[INFO] Waiting for pod #{pod_name} logs: #{stderr}"
    end
  end
end

# Get pods
def read_pods
  stdout, stderr, = Open3.capture3(add_ns("kubectl get pods -lapp.kubernetes.io/instance=#{@release_name} -o name"))
  if stderr.empty?
    stdout.lines.collect(&:strip)
  else
    []
  end
end

# @param [Array] args
def namespace_from_args(args)
  match_index = args.find_index { |arg| %w[-n --namespace].include?(arg) }
  return nil unless match_index

  args[match_index + 1]
end

# Add namespace to kube query
def add_ns(kube_query)
  kube_query += " -n #{@namespace}" if @namespace
  kube_query
end

# Add namespace to kube query
def add_ns_file(kube_query, pod)
  kube_query += " -n #{@namespace}" if @namespace
  kube_query += "> #{pod}.log"
  kube_query
end

module HelmUpgradeLogs
  class Error < StandardError; end
  # Your code goes here...
end