require 'choria/colt'
require 'choria/colt/cli/formatter'
require 'choria/colt/cli/thor'

require 'json'
require 'tty/logger'

module Choria
  class Colt
    class CLI < Thor
      class Tasks < Thor
        def self.define_targets_and_filters_options
          option :targets,
                 aliases: ['--target', '-t'],
                 desc: 'Identifies the targets of the command.'
          option :targets_with_classes,
                 aliases: ['--targets-with-class', '-C'],
                 desc: 'Select the targets which have the specified Puppet classes.'
        end

        class_option :log_level,
                     desc: 'Set log level for CLI',
                     default: 'info'
        # BOLT: desc 'run <task name> [parameters] {--targets TARGETS | --query QUERY | --rerun FILTER} [options]', 'Run a Bolt task'
        desc 'run <task name> [parameters] --targets TARGETS [options]', 'Run a Bolt task'
        long_desc <<~DESC
          Run a task on the specified targets.

          Parameters take the form parameter=value.
        DESC
        define_targets_and_filters_options
        option :environment,
               aliases: ['-E'],
               desc: 'Puppet environment to grab tasks from',
               default: 'production'
        def run(*args)
          raise Thor::Error, 'Task name is required' if args.empty?

          input = extract_task_parameters_from_args(args)
          raise Thor::Error, "Too many arguments: #{args}" unless args.count == 1

          task_name = args.shift
          targets, targets_with_classes = extract_targets_and_filters_from_options

          environment = options['environment']
          results = colt.run_bolt_task task_name, input: input, targets: targets, targets_with_classes: targets_with_classes, environment: environment do |result|
            $stdout.puts formatter.format(result)
          end

          File.write 'last_run.json', JSON.pretty_generate(results)
        rescue Choria::Orchestrator::Error
          # This error is already logged and displayed.
        end

        desc 'show [task name] [options]', 'Show available tasks and task documentation'
        long_desc <<~DESC
          Show available tasks and task documentation.

          Omitting the name of a task will display a list of tasks available
          in the Bolt project.

          Providing the name of a task will display detailed documentation for
          the task, including a list of available parameters.
        DESC
        option :environment,
               aliases: ['-E'],
               desc: 'Puppet environment to grab tasks from',
               default: 'production'
        def show(*tasks_names)
          environment = options['environment']

          require 'active_support/cache/file_store'
          cache = ActiveSupport::Cache::FileStore.new File.expand_path('~/.cache/colt/tasks')
          tasks = colt.tasks(environment: environment, cache: cache)

          if tasks_names.empty?
            show_tasks_summary(tasks)
          else
            tasks_names.each { |task_name| show_task_details(task_name, tasks) }
          end
        end

        desc 'status <task id>', 'Show task results'
        long_desc <<~DESC
          Show results from a previously ran task.

          A task ID is required to request Choria services and retrieve results.
        DESC
        option :style,
               aliases: ['-S'],
               desc: "Output style; can be 'continuous' or 'summary'",
               default: 'summary'
        define_targets_and_filters_options
        def status(task_id)
          supported_styles = %i[summary continous]
          raise Thor::Error, "Invalid style: '#{options['style']}' (available: #{supported_styles})" unless supported_styles.include? options['style'].to_sym

          targets, targets_with_classes = extract_targets_and_filters_from_options

          case options['style']
          when 'continous'
            results = colt.wait_bolt_task(task_id, targets: targets, targets_with_classes: targets_with_classes) do |result|
              $stdout.puts formatter.format(result)
            end
          when 'summary'
            show_summarized_status(task_id, targets: targets, targets_with_classes: targets_with_classes)
          end

          File.write 'last_run.json', JSON.pretty_generate(results)
        rescue Choria::Orchestrator::Error
          # This error is already logged and displayed.
        end

        no_commands do # rubocop:disable Metrics/BlockLength
          def colt
            @colt ||= Choria::Colt.new logger: logger
          end

          def logger
            @logger ||= TTY::Logger.new do |config|
              config.handlers = [
                [:console, { output: $stderr, level: options['log_level'].to_sym }],
                [:stream, { output: File.open('colt-debug.log', 'a'), level: :debug }],
              ]
              config.metadata = %i[date time]
              Choria::Colt::Debugger.enabled = true
            end
          end

          def formatter
            @formatter ||= Formatter.new(colored: $stdout.tty?)
          end

          def extract_task_parameters_from_args(args)
            parameters = args.grep(/^\w+=/)
            args.reject! { |arg| arg =~ /^\w+=/ }

            parameters.map do |parameter|
              key, value = parameter.split('=', 2)

              # TODO: Convert to boolean only if the expected type of parameter is boolean
              # TODO: Support String to integer convertion
              # TODO: Support @notation from parameter and/or whole input
              value = true if value == 'true'
              value = false if value == 'false'

              [key, value]
            end.to_h
          end

          def extract_targets_and_filters_from_options
            raise Thor::Error, 'Flag --targets or --targets-with-class is required' if options['targets'].nil? && options['targets_with_classes'].nil?

            targets = extract_targets_from_options
            targets_with_classes = options['targets_with_classes']&.split(',')

            [targets, targets_with_classes]
          end

          def extract_targets_from_options
            return nil if options['targets'] == ':all'

            targets = options['targets']&.split(',')
            targets&.map do |t|
              t = File.read(t.sub(/^@/, '')).lines.map(&:strip) if t.start_with? '@'
              t
            end&.flatten
          end

          def show_tasks_summary(tasks)
            tasks.reject! { |_task, metadata| metadata['metadata']['private'] }

            puts <<~OUTPUT
              #{pastel.title 'Tasks'}
              #{tasks.map { |task, metadata| "#{task}#{' ' * (60 - task.size)}#{metadata['metadata']['description']}" }.join("\n").gsub(/^/, '  ')}
            OUTPUT
          end

          def show_task_details(task_name, tasks)
            metadata = tasks[task_name]
            puts <<~OUTPUT
              #{pastel.title "Task: #{task_name}"}
                #{metadata['metadata']['description']}

              #{pastel.title 'Parameters'}
              #{format_task_parameters(metadata['metadata']['parameters']).gsub(/^/, '  ')}
            OUTPUT
          end

          def format_task_parameters(parameters)
            parameters.map do |parameter, metadata|
              output = <<~OUTPUT
                #{pastel.parameter(parameter)}  #{pastel.parameter_type metadata['type']}
                  #{metadata['description']}
              OUTPUT
              output += "  Default: #{metadata['default']}" unless metadata['default'].nil?
              output
            end.join "\n"
          end

          def summarize(results)
            results_grouped_by_result_content = results.group_by { |result| result[:result] }
            results_grouped_by_result_content.each do |_content, grouped_results|
              # Display each host
              grouped_results.each do |result|
                $stdout.puts formatter.format(result).host
              end
              # Display the result content
              content = formatter.format(grouped_results.first).content
              content = pastel.bright_white '(no output)' if content.nil? || content.empty?
              $stdout.puts content
            end
          end

          def show_summarized_status(task_id, targets:, targets_with_classes:)
            require 'tty-progressbar'

            bar = TTY::ProgressBar.new('[:bar] :current/:total ET::elapsed ETA::eta :rate/s')
            results = colt.wait_bolt_task(task_id, targets: targets, targets_with_classes: targets_with_classes) do |_result, count, total_count|
              bar.update total: total_count
              bar.current = count
            end

            puts "\nSummary:"
            summarize(results)
          end

          def pastel
            @pastel ||= _pastel
          end

          def _pastel
            pastel = Pastel.new(enabled: $stdout.tty?)
            pastel.alias_color(:title, :cyan)
            pastel.alias_color(:parameter, :yellow)
            pastel.alias_color(:parameter_type, :bright_white)
            pastel
          end
        end
      end

      desc 'tasks', 'Show and run Bolt tasks.'
      subcommand 'tasks', CLI::Tasks
    end
  end
end