/*
 * (c) 2017 Jan Horáček <me@apophis.cz>
 * (c) 2017-2019 Zuzana Baranová <xbaranov@fi.muni.cz>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <divcc/cc/cc1.hpp>
#include <divcc/cc/driver.hpp>
#include <divcc/cc/native.hpp>

#include <divine/ui/version.hpp> /* FIXME */

#include <clang/Driver/Compilation.h>
#include <clang/Driver/ToolChain.h>

/* This is the user-facing tool that forms the frontend for DivCC. It's a
 * C/C++ compiler based on Clang whose main goal is to aid LLVM-based
 * verification. To this end, it is capable of generating a whole-program binary
 * containing an additional .llvmbc section that is used to store LLVM bitcode
 * representation of the binary. It also handles linking of compilation units,
 * appending the bitcode sections where necessary.
 *
 * One of the aims of the project is to be a drop-in replacement for a more
 * traditional C/C++ toolchain such as GCC or Clang/LLVM. Therefore, the
 * user-facing CLI is basically the same. */

using namespace divine;

// Insert system include paths and C++ standard library include paths to 'out'
/* Fetched in form of options {"-isystem", "PATH", etc.} */

template< typename IOVector >
void get_cpp_header_paths( IOVector out )
{
        using clang::driver::Compilation;
        divcc::clang_driver drv;

        std::unique_ptr< Compilation > c( drv.BuildCompilation( { "divcc", "-x", "c++", "-" } ) );

        if ( drv.diag.engine.hasErrorOccurred() )
            throw divcc::compile_error( "failed to get cpp header paths" );

        llvm::opt::InputArgList drv_args;
        llvm::opt::ArgStringList cc1_args;

        c->getDefaultToolChain().AddClangCXXStdlibIncludeArgs( drv_args, cc1_args );
        c->getDefaultToolChain().AddClangSystemIncludeArgs( drv_args, cc1_args );

        for( auto arg : cc1_args )
            *out++ = arg;
}

/* usage: same as gcc */
int main( int argc, char **argv )
{
    try
    {
        // We choose whether to act as a C or C++ compiler depending on the name
        // the tool was called by. 'divcc' for C, 'divc++' for C++

        divcc::Native nativeCC( { argv + 1, argv + argc } );
        nativeCC.set_cxx( brq::ends_with( argv[ 0 ], "divc++" ) );
        auto& po = nativeCC._po; // Parsed command line options

        if ( nativeCC._cxx )
        {
            po.opts.push_back( "-D_GNU_SOURCE=1" );
            get_cpp_header_paths( std::back_inserter( po.cc1_args ) );
        }

        if ( po.hasHelp || po.hasVersion )
        {
            nativeCC.print_info( ui::version() );
            return 0;
        }

        return nativeCC.run();

    }
    catch ( divcc::compile_error &err )
    {
        std::cerr << err.what() << std::endl;
        return 1;
    }
    catch ( std::runtime_error &err )
    {
        std::cerr << "compilation failed: " << err.what() << std::endl;
        return 2;
    }
}