/*
* Copyright (c) 2008-2009 Nokia Corporation and/or its subsidiary(-ies).
* All rights reserved.
* This component and the accompanying materials are made available
* under the terms of "Eclipse Public License v1.0"
* which accompanies this distribution, and is available
* at the URL "http://www.eclipse.org/legal/epl-v10.html".
*
* Initial Contributors:
* Nokia Corporation - initial contribution.
*
* Contributors:
*
* Description:  StatementUtils
*
*/


#include "javacommonutils.h"
#include "logger.h"
#include "statementutils.h"

using namespace java::storage;
using namespace java::util;
using namespace std;

StatementUtils::StatementUtils()
{
}

StatementUtils::~StatementUtils()
{
}

void StatementUtils::createWhereStatement(
    const JavaStorageApplicationEntry_t& aEntry,
    wstring& aStatement)
{
    aStatement.append(L" WHERE ");

    wstring insertion = L" AND ";
    JavaStorageApplicationEntry_t::const_iterator iter;
    for (iter = aEntry.begin(); iter != aEntry.end(); iter++)
    {
        aStatement.append((*iter).entryName()).append(L"=");
        addValueByType(aStatement, (*iter));
        appendCond(aStatement, insertion, iter != --(aEntry.end()));
    }
}

void StatementUtils::appendCond(
    wstring& aStatement, const wstring& aValue, bool aCond)
{
    if (aCond)
    {
        aStatement.append(aValue);
    }
}

void StatementUtils::createWriteStatement(
    const JavaStorageApplicationEntry_t& aEntry,
    const string& aTableName,
    const wstring& aInsertion,
    wstring& aStatement)
{
    aStatement.append(L"INSERT INTO ")
    .append(JavaCommonUtils::utf8ToWstring(aTableName.c_str()))
    .append(L"(");

    JavaStorageApplicationEntry_t::const_iterator iter;

    for (iter = aEntry.begin(); iter != aEntry.end(); iter++)
    {
        aStatement.append((*iter).entryName());
        appendCond(aStatement, aInsertion, iter != --(aEntry.end()));
    }

    aStatement.append(L") VALUES (");

    for (iter = aEntry.begin(); iter != aEntry.end(); iter++)
    {
        addValueByType(aStatement, (*iter));
        appendCond(aStatement, aInsertion, iter != --(aEntry.end()));
    }

    aStatement.append(L");");
}

void StatementUtils::createTableStatement(
    const JavaStorageApplicationEntry_t& aEntry,
    const string& aTableName,
    const wstring& aInsertion,
    wstring& aStatement)
{
    aStatement.append(L"CREATE TABLE ")
    .append(JavaCommonUtils::utf8ToWstring(aTableName.c_str()))
    .append(L"(");

    JavaStorageApplicationEntry_t::const_iterator colIter;
    for (colIter = aEntry.begin(); colIter != aEntry.end(); colIter++)
    {
        if ((*colIter).entryName() == L"")
        {
            throw JavaStorageException(EInvalidDataStructure,
                                       "Column name not defined",
                                       __FILE__, __FUNCTION__, __LINE__);
        }

        aStatement.append((*colIter).entryName());

        if (JavaStorageEntry::STRING == (*colIter).entryType())
        {
            aStatement.append(L" varchar");
        }
        else if (JavaStorageEntry::INT == (*colIter).entryType())
        {
            aStatement.append(L" int");
        }
        else
        {
            ELOG(EJavaStorage, "Unknown column type");
            throw JavaStorageException(EInvalidDataStructure,
                                       "Unknown column type",
                                       __FILE__, __FUNCTION__, __LINE__);
        }

        appendCond(aStatement, aInsertion, colIter != --(aEntry.end()));
    }

    aStatement.append(L");");
}

void StatementUtils::createSearchStatement(
    const JavaStorageApplicationEntry_t& aEntry,
    const std::string& aTableName,
    std::wstring& aStatement)
{
    wstring selectStatement;
    wstring searchPattern;

    JavaStorageApplicationEntry_t::const_iterator colIter;

    for (colIter = aEntry.begin(); colIter != aEntry.end(); colIter++)
    {
        // Value is used as search query.
        if (((*colIter).entryValue()).size() > 0)
        {
            searchPattern.append((*colIter).entryName()).append(L"=");
            addValueByType(searchPattern, (*colIter));
            searchPattern += L" AND ";
        }
        else
        {
            selectStatement.append((*colIter).entryName()).append(L", ");
        }
    }

    if (selectStatement.size() > 0)
    {
        // Remove trailing ',' string. Preserve white space
        selectStatement.erase((selectStatement.size() - 2), 1);
    }
    else
    {
        selectStatement = L"*";  // Select * search;
    }

    aStatement.append(L"SELECT ").append(selectStatement).append(L" FROM ")
    .append(JavaCommonUtils::utf8ToWstring(aTableName.c_str()));

    if (searchPattern.size() > 0)
    {
        // Remove railing ' AND' string.
        searchPattern.erase((searchPattern.size() - 5), 5);

        // Define search pattern
        aStatement.append(L" WHERE " + searchPattern + L";");
    }
    else
    {
        aStatement.append(L";");
    }
}

void StatementUtils::createUpdateStatement(
    const JavaStorageApplicationEntry_t& aUpdate,
    const JavaStorageApplicationEntry_t& aMatch,
    const std::string& aTableName,
    std::wstring& aStatement)
{
    aStatement.append(L"UPDATE ")
    .append(JavaCommonUtils::utf8ToWstring(aTableName.c_str()))
    .append(L" SET ");

    JavaStorageApplicationEntry_t::const_iterator colIter;
    wstring insertion(L", ");

    for (colIter = aUpdate.begin();
            colIter != aUpdate.end();
            colIter++)
    {
        aStatement.append((*colIter).entryName()).append(L"=");
        addValueByType(aStatement, (*colIter));
        appendCond(
            aStatement, insertion, colIter != --(aUpdate.end()));
    }

    createWhereStatement(aMatch, aStatement);
}

void StatementUtils::addValueByType(wstring& aStatement,
                                    const JavaStorageEntry& aEntry)
{
    if (aEntry.entryType() == JavaStorageEntry::STRING
            || aEntry.entryType() == JavaStorageEntry::NOT_DEFINED)
    {
        aStatement.append(L"'");
        addValue(aStatement, aEntry);
        aStatement.append(L"'");
    }
    else if (aEntry.entryType() == JavaStorageEntry::INT)
    {
        addValue(aStatement, aEntry);
    }
    else
    {
        aStatement.append(L"NULL");
    }
}

void StatementUtils::addValue(wstring& aStatement,
                              const JavaStorageEntry& aEntry)
{
    wstring val = aEntry.entryValue();
    escape(val);
    aStatement.append(val);
}

void StatementUtils::escape(wstring& aUnescaped)
{
    wstring escapedChar = L"\'";
    wstring::size_type idx = 0;
    if ((idx = aUnescaped.find(escapedChar)) != string::npos)
    {
        do
        {
            aUnescaped.insert(idx, escapedChar);
            idx += escapedChar.size() + 1;
        }
        while ((idx = aUnescaped.find(escapedChar, idx)) != string::npos);
    }
}