# frozen_string_literal: true

module MuxTf
  module Cli
    module Mux
      extend PiotrbCliUtils::Util
      extend PiotrbCliUtils::ShellHelpers

      class << self
        def run(_args)
          Dotenv.load(".env.mux")

          log "Enumerating folders ..."
          dirs = enumerate_terraform_dirs

          fail_with "Error: - no subfolders detected! Aborting." if dirs.empty?

          tasks = dirs.map { |dir|
            {
              name: dir,
              cwd: dir,
              cmd: File.expand_path(File.join(__dir__, "..", "..", "..", "exe", "tf_current"))
            }
          }

          project = File.basename(Dir.getwd)

          if ENV["MUX_TF_AUTH_WRAPPER"]
            log "Warming up AWS connection ..."
            words = Shellwords.shellsplit(ENV["MUX_TF_AUTH_WRAPPER"])
            result = capture_shell([*words, "aws", "sts", "get-caller-identity"], raise_on_error: true)
            p JSON.parse(result)
          end

          if Tmux.session_running?(project)
            log "Killing existing session ..."
            Tmux.kill_session(project)
          end

          log "Starting new session ..."
          Tmux.new_session project
          Tmux.select_pane "initial"

          Tmux.set_hook "pane-exited", "select-layout tiled"
          Tmux.set_hook "window-pane-changed", "select-layout tiled"

          Tmux.set "mouse", "on"

          window_id = Tmux.list_windows.first[:id]

          unless tasks.empty?
            tasks.each do |task|
              log "launching task: #{task[:name]} ...", depth: 2
              Tmux.split_window :horizontal, "#{project}:#{window_id}", cmd: task[:cmd], cwd: task[:cwd]
              Tmux.select_pane task[:name]
              Tmux.tile!
              task[:commands]&.each do |cmd|
                Tmux.send_keys cmd, enter: true
              end
            end
          end

          log "Almost done ..."

          initial_pane = Tmux.find_pane("initial")
          Tmux.kill_pane initial_pane[:id]
          Tmux.tile!

          puts "\e]0;tmux: #{project}\007"

          log "Attaching ..."
          Tmux.attach(project, cc: !!ENV["MUXP_CC_MODE"])
          log "Done!"
        end

        private

        def enumerate_terraform_dirs
          ignored = []

          ignored += ENV["MUX_IGNORE"].split(",") if ENV["MUX_IGNORE"]

          dirs = Dir["**/*/.terraform"].map { |n| n.gsub(%r{/\.terraform}, "") }
          dirs.reject! { |d| d.in?(ignored) }

          dirs
        end
      end
    end
  end
end