=begin
    Copyright 2010-2014 Tasos Laskos <tasos.laskos@arachni-scanner.com>

    This file is part of the Arachni Framework project and is subject to
    redistribution and commercial restrictions. Please see the Arachni Framework
    web site for more information on licensing and terms of use.
=end

require_relative '../option_parser'

module Arachni
module UI::CLI

class Reporter

# @author Tasos "Zapotek" Laskos <tasos.laskos@arachni-scanner.com>
class OptionParser < UI::CLI::OptionParser

    attr_reader :framework

    # @return   [Hash{<String, Symbol> => Hash{String => String}}]
    #   Reports to load, by name, as keys and their options as values.
    #
    # @see Reporters
    # @see Reporter::Base
    # @see Reporter::Manager
    attr_accessor :reporters

    attr_accessor :report_path

    def initialize(*)
        super

        @framework = Arachni::Framework.new
        @reporters = {}
    end

    def reporter
        separator ''
        separator 'Reporters'

        on( '--reporters-list [PATTERN]', Regexp,
            'List available reporters based on the provided pattern.',
            '(If no pattern is provided all will be listed.)'
        ) do |pattern|
            list_reporters( framework.list_reporters( pattern ) )
            exit
        end

        on( "--reporter 'REPORTER:OPTION=VALUE,OPTION2=VALUE2'",
            "REPORTER is the name of the reporter as displayed by '--reporters-list'.",
            "(Reporters are referenced by their filename without the '.rb' " +
                "extension, use '--reporters-list' to list all.)",
            '(Default: stdout)',
            '(Can be used multiple times.)'
        ) do |reporter|
            prepare_component_options( reporters, reporter )
        end
    end

    def after_parse
        @report_path = ARGV.shift
    end

    def validate
        if !@report_path
            print_error 'No report file provided.'
            exit 1
        end

        @report_path = File.expand_path( @report_path )

        if !File.exists?( @report_path )
            print_error "Report does not exist: #{@report_path}"
            exit 1
        end

        if reporters.any?
            begin
                framework.reporters.load( reporters.keys )
            rescue Component::Error::NotFound => e
                print_error e
                print_info 'Available reporters are:'
                print_info framework.reporters.available.join( ', ' )
                print_line
                print_info 'Use the \'--reporters-list\' parameter to see a' <<
                               ' detailed list of all available reports.'
                exit 1
            ensure
                framework.reporters.clear
            end
        end
    end

    def banner
        "Usage: #{$0} REPORT"
    end

end
end
end
end
