# frozen_string_literal: true

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

ENV["helm_upgrade_logs_log_start"] ||= "1"
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} --field-selector 'status.phase!=Failed' -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

def options_with_args
  %w[--set --ca-file --cert-file --description --history-max --key-file --keyring --output --password --post-renderer
     --repo --set --set-file --set-string --timeout --username --values --version --kube-apiserver --kube-as-group
     --kube-as-user --kube-as-file --kube-context --kube-token --kubeconfig --namespace --registry-config
     --repository-cache --repository-config -n -f -o]
end

# @return Release name from command line arguments
def release_name_from_args(args)
  arg_list = args.clone

  loop do
    option_with_val = false
    arg_list.each_with_index do |arg, index|
      if options_with_args.include?(arg_list[index - 1])
        arg_list.delete_at(index - 1)
        arg_list.delete_at(index - 1)
        option_with_val = true
        break
      end
    end
    break if option_with_val == false
  end
  arg_list.find { |arg| !arg.start_with?("-") }
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
end