require 'json'
require 'omnitest/reporters'
require 'omnitest/documentation_generator'

module Omnitest
  module Command
    class Generate
      class Code2Doc < Thor::Group
        include Core::DefaultLogger
        include Omnitest::Core::Logging
        include Thor::Actions
        include Omnitest::Core::FileSystem
        include Omnitest::Core::Util::String

        class_option :log_level,
                     aliases: '-l',
                     desc: 'Set the log level (debug, info, warn, error, fatal)'
        class_option :file,
                     aliases: '-f',
                     desc: 'The Omnitest project set file',
                     default: 'omnitest.yaml'
        class_option :skeptic,
                     aliases: '-s',
                     desc: 'The Skeptic test manifest file',
                     default: 'skeptic.yaml'
        class_option :format,
                     aliases: '-f',
                     enum: %w(md rst),
                     default: 'md',
                     desc: 'Target documentation format'
        class_option :target_dir,
                     aliases: '-d',
                     default: 'docs/',
                     desc: 'The target directory where documentation for generated documentation.'
        class_option :source_files, type: :array

        class_option :destination, default: 'docs/'

        argument :project_regexp, default: 'all'
        argument :scenario_regexp, default: 'all'

        def setup
          Omnitest.update_config!(options)
          Omnitest.setup
          @scenarios = Omnitest.scenarios(project_regexp, scenario_regexp, options)
          abort "No scenarios for regex `#{scenario_regexp}', try running `omnitest list'" if @scenarios.empty?
        end

        def set_destination_root
          self.destination_root = options[:destination]
        end

        def source_files
          @source_files = @scenarios.map do |scenario|
            [scenario.slug, scenario.absolute_source_file]
          end
        end

        def code2doc
          @source_files.each do |slug, source_file|
            if source_file.nil?
              warn "No code sample available for #{slug}, no documentation will be generated."
              next
            end

            target_file_name = slug + ".#{options[:format]}"

            begin
              doc = Omnitest::DocumentationGenerator.new.code2doc(source_file)
              create_file(target_file_name, doc)
            rescue Omnitest::Psychic::Code2Doc::CommentStyles::UnknownStyleError
              warn "Could not generated documentation for #{source_file}, because the language couldn't be detected."
            end
          end
        end
      end
    end
  end
end