/**********************************************************************
 * Premake - watcom.c
 * The Open Watcom Code::Blocks target, version 1.7a, 1.8a (Open Watcom)
 *
 * Copyright (c) 2002-2005 Jason Perkins and the Premake project
 * OW version 2008 T N Channon strongly based upon existing Premake code
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License in the file LICENSE.txt for details.
 **********************************************************************/

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "../premake.h"
#include "ow.h"

static const char* filterLinks(const char* name);
static const char* filterLinksForPaths(const char* name);
static const char* listFiles(const char* name);
static void print_opt(const char* opt);


static int ow_cpp(void)
{
    int i;

    /* Write the file */
    if (!io_openfile(path_join(prj_get_pkgpath(), prj_get_pkgname(), "cbp")))
        return 0;

    io_print("<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\" ?>\n");
    io_print("<CodeBlocks_project_file>\n");
    io_print("\t<FileVersion major=\"1\" minor=\"6\" />\n");
    io_print("\t<Project>\n");
    io_print("\t\t<Option title=\"%s\" />\n", prj_get_pkgname());
    io_print("\t\t<Option pch_mode=\"2\" />\n");
    io_print("\t\t<Option compiler=\"ow\" />\n"); /* Open Watcom ID in CodeBlocks*/
    io_print("\t\t<Build>\n");

    for (i = 0; i < prj_get_numconfigs(); ++i)
    {
        int kindCode;

        prj_select_config(i);

        io_print("\t\t\t<Target title=\"%s\">\n", prj_get_cfgname());
        io_print("\t\t\t\t<Option output=\"%s\" prefix_auto=\"0\" extension_auto=\"0\" />\n", prj_get_target());
        io_print("\t\t\t\t<Option object_output=\"%s\" />\n", prj_get_objdir());

        if (prj_is_kind("winexe"))
            kindCode = 0;
        else if (prj_is_kind("exe"))
            kindCode = 1;
        else if (prj_is_kind("lib"))
            kindCode = 2;
        else if (prj_is_kind("dll"))
            kindCode = 3;
        else
        {
            printf("** Unsupported project kind %s\n", prj_get_kind());
            return 0;
        }
        io_print("\t\t\t\t<Option type=\"%d\" />\n", kindCode);

        io_print("\t\t\t\t<Option compiler=\"ow\" />\n");

        if (prj_is_kind("dll"))
        {
            io_print("\t\t\t\t<Option createDefFile=\"0\" />\n");
            if (prj_has_flag("no-import-lib"))
                io_print("\t\t\t\t<Option createStaticLib=\"0\" />\n");
            else
                io_print("\t\t\t\t<Option createStaticLib=\"1\" />\n");
        }

        io_print("\t\t\t\t<Compiler>\n");
        print_opt("-hw"); /* watcom debug format for Watcom debugger as tool under C::B , see OW wiki*/
        if (prj_has_flag("extra-warnings"))
            print_opt("-wx");
        if (prj_has_flag("fatal-warnings"))
            print_opt("-we");
        if (prj_has_flag("no-exceptions"))
            print_opt("-xd");
        /*                if (prj_has_flag("no-frame-pointer"))
                                print_opt("-fomit-frame-pointer");*/
        if (prj_has_flag("rtti"))
            print_opt("-xr");
        if (!prj_has_flag("no-symbols"))  /* debug on, not no-symbols, ie. not not therefore debug symbols */
            print_opt("-d2");
        else
            print_opt("-d0");
        if (prj_has_flag("optimize-size"))
            print_opt("-os");
        if (prj_has_flag("optimize-speed"))
            print_opt("-ot");
        if (prj_has_flag("optimize") && !prj_has_flag("optimize-size") && !prj_has_flag("optimize-speed"))
            print_opt("-ox");

        if (prj_has_pch())
        {
            /* Warns you if your pch file is out of date */
            print_opt("-Winvalid-pch");
            /* Force include the pch header so the user doesn't need to */
            io_print("\t\t\t\t\t<Add option=\"-include &quot;%s&quot;\" />\n", prj_get_pch_header());
        }

        print_list(prj_get_defines(), "\t\t\t\t\t<Add option=\"-D", "\" />\n", "", NULL);
        print_list(prj_get_buildoptions(), "\t\t\t\t\t<Add option=\"", "\" />\n", "", NULL);
        print_list(prj_get_incpaths(), "\t\t\t\t\t<Add directory=\"", "\" />\n", "", NULL);
        io_print("\t\t\t\t</Compiler>\n");

        io_print("\t\t\t\t<Linker>\n");
        if (prj_has_flag("no-symbols") == 0) /* inverted test re- gcc */
            print_opt("op symf");            /* linker generates separate symbol file, normal OW behaviour */
        print_list(prj_get_linkoptions(), "\t\t\t\t\t<Add option=\"", "\" />\n", "", NULL);

        io_print("\t\t\t\t\t<Add directory=\"%s\" />\n", prj_get_bindir());
        if (!matches(prj_get_bindir(), prj_get_libdir()))
            io_print("\t\t\t\t\t<Add directory=\"%s\" />\n", prj_get_libdir());
        print_list(prj_get_libpaths(), "\t\t\t\t\t<Add directory=\"", "\" />\n", "", NULL);
        print_list(prj_get_links(), "\t\t\t\t\t<Add directory=\"", "\" />\n", "", filterLinksForPaths);
        print_list(prj_get_links(), "\t\t\t\t\t<Add library=\"", "\" />\n", "", filterLinks);
        io_print("\t\t\t\t</Linker>\n");

        if (prj_find_filetype(".rc"))
        {
            io_print("\t\t\t\t<ResourceCompiler>\n");
            print_list(prj_get_respaths(), "\t\t\t\t\t<Add directory=\"", "\" />\n", "", NULL);
            io_print("\t\t\t\t</ResourceCompiler>\n");
        }

        if (prj_get_numprebuildcommands() > 0 || prj_get_numpostbuildcommands() > 0)
        {
            io_print("\t\t\t\t<ExtraCommands>\n");

            if (prj_get_numprebuildcommands() > 0)
                print_list(prj_get_prebuildcommands(), "\t\t\t\t\t<Add before=\"", "\" />\n", "", NULL);

            if (prj_get_numpostbuildcommands() > 0)
                print_list(prj_get_postbuildcommands(), "\t\t\t\t\t<Add after=\"", "\" />\n", "", NULL);

            io_print("\t\t\t\t</ExtraCommands>\n");
        }

        io_print("\t\t\t</Target>\n");
    }

    io_print("\t\t</Build>\n");

    print_list(prj_get_files(), "", "", "", listFiles);

    io_print("\t\t<Extensions />\n");
    io_print("\t</Project>\n");
    io_print("</CodeBlocks_project_file>\n");

    io_closefile();
    return 1;
}


static void print_opt(const char* opt)
{
    io_print("\t\t\t\t\t<Add option=\"%s\" />\n", opt);
}


/************************************************************************
 * Output a description block for each source file
 ***********************************************************************/

static const char* listFiles(const char* filename)
{
    int i;

    io_print("\t\t<Unit filename=\"%s\">\n", filename);

    if (matches(path_getextension(filename), ".rc"))
    {
        io_print("\t\t\t<Option compilerVar=\"WINDRES\" />\n");
    }
    else
    {
        /* Default is C++ (compilerVar=CC++) */
        if (prj_is_lang("c"))
            io_print("\t\t\t<Option compilerVar=\"CC\" />\n");
        else
            io_print("\t\t\t<Option compilerVar=\"CC++\" />\n");

        if (!is_cpp(filename))
        {
            /* Look for the specified pch header */
            if (prj_has_pch() && matches(filename, prj_get_pch_header()))
            {
                io_print("\t\t\t<Option compile=\"1\" />\n");
                io_print("\t\t\t<Option weight=\"0\" />\n");
            }
        }
    }

    for (i = 0; i < prj_get_numconfigs(); ++i)
    {
        Package* pkg = prj_get_package();
        io_print("\t\t\t<Option target=\"%s\" />\n", pkg->configs[i]->prjConfig->name);
    }

    io_print("\t\t</Unit>\n");
    return NULL;
}



/************************************************************************
 * Checks each entry in the list of package links. If the entry refers
 * to a sibling package, returns the path to that package's output
 ***********************************************************************/

static const char* filterLinks(const char* name)
{
    int i = prj_find_package(name);
    if (i >= 0)
    {
		const char* kind =  prj_get_kind_for(i);
        const char* lang = prj_get_language_for(i);
        if ((matches(lang, "c++") || matches(lang, "c")) &&
			(matches(kind, "dll") || matches(kind, "lib")))
        {
            const char* target = prj_get_targetname_for(i);
            return path_getname(target);
        }
        else
        {
            return NULL;
        }
    }
    else
    {
        return name;
    }
}


static const char* filterLinksForPaths(const char* name)
{
    int i = prj_find_package(name);
    if (i >= 0)
    {
        const char* lang = prj_get_language_for(i);
        if (matches(lang, "c++") || matches(lang, "c"))
        {
            const char* target = prj_get_target_for(i);
            const char* dir = path_getdir(target);
            if (!prj_has_libpath(dir))
            {
                return dir;
            }
        }
    }
    return NULL;
}

//==============

/************************************************************************
 * List callback: scans the list of links for a package. If a link is
 * found to a sibling package, prints a dependency string for the
 * workspace file.
 ***********************************************************************/
static const char* printProjectDependencies(const char* name)
{
    int i;
    for (i = 0; i < prj_get_numpackages(); ++i)
    {
        if (matches(prj_get_pkgname_for(i), name))
        {
            const char* filename = prj_get_pkgfilename_for(i, "cbp");
            if (startsWith(filename, "./")) filename += 2;
            io_print("\t\t\t<Depends filename=\"%s\" />\n", filename);
        }
    }

    return NULL;
}

/******************************/

static int writeWorkspace(void)
{
    int i;

    if (!io_openfile(path_join(prj_get_path(), prj_get_name(), "workspace")))
        return 0;

    io_print("<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\" ?>\n");
    io_print("<CodeBlocks_workspace_file>\n");
    io_print("\t<Workspace title=\"%s\">\n", prj_get_name());

    for (i = 0; i < prj_get_numpackages(); ++i)
    {
        const char* filename;

        prj_select_package(i);

        filename = prj_get_pkgfilename("cbp");
        if (startsWith(filename, "./")) filename += 2;
        io_print("\t\t<Project filename=\"%s\"", filename);
        if (i == 0) io_print(" active=\"1\"");
        io_print(">\n");

        /* Write project dependencies */
        prj_select_config(0);
        print_list(prj_get_links(), "", "", "", printProjectDependencies);

        io_print("\t\t</Project>\n");
    }

    io_print("\t</Workspace>\n");
    io_print("</CodeBlocks_workspace_file>\n");
    io_closefile();
    return 1;
}

int ow_cb_generate(void)
{
    int i;

    puts("Generating Code::Blocks workspace and project files:");

    if (!writeWorkspace())
        return 0;

    for (i = 0; i < prj_get_numpackages(); ++i)
    {
        prj_select_package(i);
        printf("...%s\n", prj_get_pkgname());

        if (prj_is_lang("c++") || prj_is_lang("c"))
        {
            if (!ow_cpp())
                return 0;
        }
        else
        {
            printf("** Error: unsupported language '%s'\n", prj_get_language());
            return 0;
        }
    }

    return 1;
}