/**************************************************************
 * 
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 * 
 *   http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 * 
 *************************************************************/


// MARKER(update_precomp.py): autogen include statement, do not remove
#include "precompiled_idlc.hxx"

#include "idlc/options.hxx"

#include "osl/diagnose.h"
#include "rtl/string.hxx"
#include "rtl/strbuf.hxx"

#include "rtl/ustring.hxx"
#include "osl/file.hxx"

#ifdef WNT
#   include <windows.h>
#endif

/*
#ifndef WIN32_LEAN_AND_MEAN
#   define WIN32_LEAN_AND_MEAN
# ifdef _MSC_VER
#   pragma warning(push,1) 
# endif
#   include <windows.h>
# ifdef _MSC_VER
#   pragma warning(pop)
# endif
#   include <tchar.h>
#   undef WIN32_LEAN_AND_MEAN
#endif
*/

#include <stdio.h>
#include <string.h>

using rtl::OString;
using rtl::OStringBuffer;

#ifdef SAL_UNX
#define SEPARATOR '/'
#else
#define SEPARATOR '\\'
#endif

Options::Options(char const * progname)
    : m_program(progname), m_stdin(false), m_verbose(false), m_quiet(false)
{
}	

Options::~Options()
{
}

// static
bool Options::checkArgument (std::vector< std::string > & rArgs, char const * arg, size_t len)
{
    bool result = ((arg != 0) && (len > 0));
    OSL_PRECOND(result, "idlc::Options::checkArgument(): invalid arguments");
    if (result)
    {
        switch(arg[0])
        {
        case '@':
            if ((result = (len > 1)) == true)
            {
                // "@<cmdfile>"
                result = Options::checkCommandFile (rArgs, &(arg[1]));
            }
            break;
        case '-':
            if ((result = (len > 1)) == true)
            {
                // "-<option>"
                switch (arg[1])
                {
                case 'O':
                case 'I':
                case 'D':
                {
                    // "-<option>[<param>]
                    std::string option(&(arg[0]), 2);
                    rArgs.push_back(option);
                    if (len > 2)
                    {
                        // "-<option><param>"
                        std::string param(&(arg[2]), len - 2);
                        rArgs.push_back(param);
                    }
                    break;
                }
                default:
                    // "-<option>" ([long] option, w/o param)
                    rArgs.push_back(std::string(arg, len));
                    break;
                }
            }
            break;
        default:
            // "<param>"
            rArgs.push_back(std::string(arg, len));
            break;
        }
    }
    return (result);
}

// static
bool Options::checkCommandFile (std::vector< std::string > & rArgs, char const * filename)
{
    FILE * fp = fopen(filename, "r");
    if (fp == 0)
    {
        fprintf(stderr, "ERROR: can't open command file \"%s\"\n", filename);
        return (false);
    }
    
    std::string buffer;
    buffer.reserve(256);
    
    bool quoted = false;
    int c = EOF;
    while ((c = fgetc(fp)) != EOF)
    {
        switch(c)
        {
        case '\"':
            quoted = !quoted;
            break;
        case ' ':
        case '\t':
        case '\r':
        case '\n':
            if (!quoted)
            {
                if (!buffer.empty())
                {
                    // append current argument.
                    if (!Options::checkArgument(rArgs, buffer.c_str(), buffer.size()))
                    {
                        (void) fclose(fp);
                        return (false);
                    }
                    buffer.clear();
                }
                break;
            }
        default:
            // quoted white-space fall through
            buffer.push_back(sal::static_int_cast<char>(c));
            break;
        }
    }
    if (!buffer.empty())
    {
        // append unterminated argument.
        if (!Options::checkArgument(rArgs, buffer.c_str(), buffer.size()))
        {
            (void) fclose(fp);
            return (false);
        }
        buffer.clear();
    }
    return (fclose(fp) == 0);
}

bool Options::badOption(char const * reason, std::string const & rArg) throw(IllegalArgument)
{
    OStringBuffer message;
    if (reason != 0)
    {
        message.append(reason); message.append(" option '"); message.append(rArg.c_str()); message.append("'");
        throw IllegalArgument(message.makeStringAndClear());
    }
    return false;
}

bool Options::setOption(char const * option, std::string const & rArg)
{
    bool result = (0 == strcmp(option, rArg.c_str()));
    if (result)
        m_options[rArg.c_str()] = OString(rArg.c_str(), rArg.size());
    return (result);
}

#ifdef WNT
/* Helper functiopn to convert windows paths including spaces, brackets etc. into
   a windows short Url. The ucpp preprocessor has problems with such paths and returns
   with error.
*/
OString convertIncPathtoShortWindowsPath(const OString& incPath) {
	rtl::OUString path = OStringToOUString(incPath, RTL_TEXTENCODING_UTF8);

	std::vector<sal_Unicode> vec(path.getLength() + 1);
    //GetShortPathNameW only works if the file can be found!
    const DWORD len = GetShortPathNameW(
		reinterpret_cast<LPCWSTR>(path.getStr()), reinterpret_cast<LPWSTR>(&vec[0]), path.getLength() + 1);

	rtl::OUString ret = rtl::OUString(&vec[0], len);

	if (len > 0)
		return OUStringToOString(ret, RTL_TEXTENCODING_UTF8);

	return incPath;
}
#endif

bool Options::initOptions(std::vector< std::string > & rArgs) throw(IllegalArgument)
{
    std::vector< std::string >::const_iterator first = rArgs.begin(), last = rArgs.end();
    for (; first != last; ++first)
    {
        if ((*first)[0] != '-')
        {
            OString filename((*first).c_str(), (*first).size());
            OString tmp(filename.toAsciiLowerCase());
            if (tmp.lastIndexOf(".idl") != (tmp.getLength() - 4))
            {
                throw IllegalArgument("'" + filename + "' is not a valid input file, only '*.idl' files will be accepted");
            }
            m_inputFiles.push_back(filename);
            continue;
        }
        
        std::string const option(*first);
        switch((*first)[1])
        {
        case 'O':
        {
            if (!((++first != last) && ((*first)[0] != '-')))
            {
                return badOption("invalid", option);
            }
            OString param((*first).c_str(), (*first).size());
            m_options["-O"] = param;
            break;
        }
        case 'I':
        {
            if (!((++first != last) && ((*first)[0] != '-')))
            {
                return badOption("invalid", option);
            }
            OString param((*first).c_str(), (*first).size());
            {
                // quote param token(s).
                OStringBuffer buffer;
                sal_Int32 k = 0;
                do
                {
                    OStringBuffer token;
                    token.append("-I");
#ifdef WNT
					rtl::OString incpath = convertIncPathtoShortWindowsPath(param.getToken(0, ';', k));
#else
					rtl::OString incpath = param.getToken(0, ';', k);
#endif
					token.append(incpath);
                    //token.append(param.getToken(0, ';', k));
                    if (buffer.getLength() > 0)
                        buffer.append(' ');
                    buffer.append(token);
                } while (k != -1);
                param = buffer.makeStringAndClear();
            }
            if (m_options.count("-I") > 0)
            {
                // append param.
                OStringBuffer buffer(m_options["-I"]);
                buffer.append(' '); buffer.append(param);
                param = buffer.makeStringAndClear();
            }
            m_options["-I"] = param;
            break;
        }
        case 'D':
        {
            if (!((++first != last) && ((*first)[0] != '-')))
            {
                return badOption("invalid", option);
            }
            OString param("-D"); param += OString((*first).c_str(), (*first).size());
            if (m_options.count("-D") > 0)
            {
                OStringBuffer buffer(m_options["-D"]);
                buffer.append(' '); buffer.append(param);
                param = buffer.makeStringAndClear();
            }
            m_options["-D"] = param;
            break;
        }
        case 'C':
        {
            if (!setOption("-C", option))
            {
                return badOption("invalid", option);
            }
            break;
        }
        case 'c':
        {
            if (!setOption("-cid", option))
            {
                return badOption("invalid", option);
            }
            break;
        }
        case 'q':
        {
            if (!setOption("-quiet", option))
            {
                return badOption("invalid", option);
            }
            m_quiet = true;
            break;
        }
        case 'v':
        {
            if (!setOption("-verbose", option))
            {
                return badOption("invalid", option);
            }
            m_verbose = true;
            break;
        }
        case 'w':
        {
            if (!(setOption("-w", option) || setOption("-we", option)))
            {
                return badOption("invalid", option);
            }
            break;
        }
        case 'h':
        case '?':
        {
            if (!(setOption("-h", option) || setOption("-?", option)))
            {
                return badOption("invalid", option);
            }
            {
                (void) fprintf(stdout, "%s", prepareHelp().getStr());
                return (false);
            }
            // break; // Unreachable
        }
        case 's':
        {
            if (!setOption("-stdin", option))
            {
                return badOption("invalid", option);
            }
            m_stdin = true;
            break;
        }
        default:
            return badOption("unknown", option);
        }
    }
    return (true);
}

OString	Options::prepareHelp()
{
	OString help("\nusing: ");
	help += m_program + " [-options] <file_1> ... <file_n> | @<filename> | -stdin\n";
	help += "    <file_n>    = file_n specifies one or more idl files.\n";
	help += "                  Only files with the extension '.idl' are valid.\n";
	help += "    @<filename> = filename specifies the name of a command file.\n";
    help += "    -stdin      = read idl file from standard input.\n";
	help += "  Options:\n";
	help += "    -O<path>    = path specifies the output directory.\n";
	help += "                  The generated output is a registry file with\n";
	help += "                  the same name as the idl input file (or 'stdin'\n";
    help += "                  for -stdin).\n";
	help += "    -I<path>    = path specifies a directory where include\n";
	help += "                  files will be searched by the preprocessor.\n";
	help += "                  Multiple directories can be combined with ';'.\n";
	help += "    -D<name>    = name defines a macro for the preprocessor.\n";
	help += "    -C          = generate complete type information, including\n";
	help += "                  documentation.\n";
	help += "    -cid        = check if identifiers fulfill the UNO naming\n";
    help += "                  requirements.\n";
	help += "    -w          = display warning messages.\n";
	help += "    -we         = treat warnings as errors.\n";
	help += "    -h|-?       = print this help message and exit.\n\n";
	help += prepareVersion();
	
	return help;
}	

OString	Options::prepareVersion()
{
	OString version(m_program);
	version += " Version 1.1\n\n";
	return version;
}	

const OString& Options::getProgramName() const
{
	return m_program;
}	

bool Options::isValid(const OString& option)
{
	return (m_options.count(option) > 0);	
}	

const OString& Options::getOption(const OString& option)
	throw( IllegalArgument )
{
	if (!isValid(option))
	{
		throw IllegalArgument("Option is not valid or currently not set.");
	}
    return m_options[option];
}
/* vi:set tabstop=4 shiftwidth=4 expandtab: */