/*
 *  Created by Phil on 31/10/2010.
 *  Copyright 2010 Two Blue Cubes Ltd. All rights reserved.
 *
 *  Distributed under the Boost Software License, Version 1.0. (See accompanying
 *  file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
 */
#ifndef TWOBLUECUBES_CATCH_RUNNER_HPP_INCLUDED
#define TWOBLUECUBES_CATCH_RUNNER_HPP_INCLUDED

#include "internal/catch_commandline.hpp"
#include "internal/catch_list.hpp"
#include "internal/catch_runner_impl.hpp"
#include "internal/catch_test_spec.h"
#include "internal/catch_version.h"
#include "internal/catch_text.h"

#include <fstream>
#include <stdlib.h>
#include <limits>

namespace Catch {

    class Runner {

    public:
        Runner( Ptr<Config> const& config )
        :   m_config( config )
        {
            openStream();
            makeReporter();
        }

        Totals runTests() {

            std::vector<TestCaseFilters> filterGroups = m_config->filters();
            if( filterGroups.empty() ) {
                TestCaseFilters filterGroup( "" );
                filterGroups.push_back( filterGroup );
            }

            RunContext context( m_config.get(), m_reporter );

            Totals totals;

            for( std::size_t i=0; i < filterGroups.size() && !context.aborting(); ++i ) {
                context.testGroupStarting( filterGroups[i].getName(), i, filterGroups.size() );
                totals += runTestsForGroup( context, filterGroups[i] );
                context.testGroupEnded( filterGroups[i].getName(), totals, i, filterGroups.size() );
            }
            return totals;
        }

        Totals runTestsForGroup( RunContext& context, const TestCaseFilters& filterGroup ) {
            Totals totals;
            std::vector<TestCase>::const_iterator it = getRegistryHub().getTestCaseRegistry().getAllTests().begin();
            std::vector<TestCase>::const_iterator itEnd = getRegistryHub().getTestCaseRegistry().getAllTests().end();
            int testsRunForGroup = 0;
            for(; it != itEnd; ++it ) {
                if( filterGroup.shouldInclude( *it ) ) {
                    testsRunForGroup++;
                    if( m_testsAlreadyRun.find( *it ) == m_testsAlreadyRun.end() ) {

                        if( context.aborting() )
                            break;

                        totals += context.runTest( *it );
                        m_testsAlreadyRun.insert( *it );
                    }
                }
            }
            if( testsRunForGroup == 0 && !filterGroup.getName().empty() )
                m_reporter->noMatchingTestCases( filterGroup.getName() );
            return totals;
            
        }

    private:
        void openStream() {
            // Open output file, if specified
            if( !m_config->getFilename().empty() ) {
                m_ofs.open( m_config->getFilename().c_str() );
                if( m_ofs.fail() ) {
                    std::ostringstream oss;
                    oss << "Unable to open file: '" << m_config->getFilename() << "'";
                    throw std::domain_error( oss.str() );
                }
                m_config->setStreamBuf( m_ofs.rdbuf() );
            }
        }
        void makeReporter() {
            std::string reporterName = m_config->getReporterName().empty()
                ? "console"
                : m_config->getReporterName();

            m_reporter = getRegistryHub().getReporterRegistry().create( reporterName, m_config.get() );
            if( !m_reporter ) {
                std::ostringstream oss;
                oss << "No reporter registered with name: '" << reporterName << "'";
                throw std::domain_error( oss.str() );
            }
        }
        
    private:
        Ptr<Config> m_config;
        std::ofstream m_ofs;
        Ptr<IStreamingReporter> m_reporter;
        std::set<TestCase> m_testsAlreadyRun;
    };

    inline int Main( Ptr<Config> const& config ) {
        int result = 0;
        try
        {
            Runner runner( config );

            // Handle list request
            if( list( config ) ) {
                Catch::cleanUp();
                return 0;
            }

            result = static_cast<int>( runner.runTests().assertions.failed );

        }
        catch( std::exception& ex ) {
            std::cerr << ex.what() << std::endl;
            result = (std::numeric_limits<int>::max)();
        }

        Catch::cleanUp();
        return result;
    }

    inline void showHelp( std::string const& processName ) {
        Clara::CommandLine<ConfigData> cli = makeCommandLineParser();

        std::cout << "\nCatch v"    << libraryVersion.majorVersion << "."
                                    << libraryVersion.minorVersion << " build "
                                    << libraryVersion.buildNumber;
        if( libraryVersion.branchName != "master" )
            std::cout << " (" << libraryVersion.branchName << " branch)";
        std::cout << "\n";

        cli.usage( std::cout, processName );
        std::cout << "\nFor more detail usage please see: https://github.com/philsquared/Catch/wiki/Command-line\n" << std::endl;
    }
    inline Ptr<Config> processConfig( int argc, char* const argv[], ConfigData& configData ) {
        Clara::CommandLine<ConfigData> cli = makeCommandLineParser();
        std::vector<Clara::Parser::Token> unused = cli.parseInto( argc, argv, configData );
        if( !unused.empty() ) {
            std::vector<Clara::Parser::Token>::const_iterator
                it = unused.begin(),
                itEnd = unused.end();
            std::string msg;
            for(; it != itEnd; ++it )
                msg += "  unrecognised option: " + it->data + "\n";
            throw std::runtime_error( msg.substr( 0, msg.size()-1 ) );
        }
        Ptr<Config> config = new Config( configData );
        return config;        
    }
    inline Ptr<Config> processConfig( int argc, char* const argv[] ) {
        ConfigData configData;
        return processConfig( argc, argv, configData );
    }

    inline int Main( int argc, char* const argv[], ConfigData configData = ConfigData() ) {

        Ptr<Config> config;

        try {
            config = processConfig( argc, argv, configData );
            if( config->showHelp() ) {
                showHelp( config->getProcessName() );
                Catch::cleanUp();        
                return 0;
            }
        }
        catch( std::exception& ex ) {
            std::cerr   << "\nError in input:\n"
                        << "  " << ex.what() << "\n\n";
            makeCommandLineParser().usage( std::cout, configData.processName );
            Catch::cleanUp();
            return (std::numeric_limits<int>::max)();
        }
        return Main( config );
    }
    
} // end namespace Catch

#endif // TWOBLUECUBES_CATCH_RUNNER_HPP_INCLUDED