#include "scriptingqtdbproxy.h"
#include "db/db.h"
#include "db/sqlquery.h"
#include <QScriptContext>
#include <QScriptEngine>

ScriptingQtDbProxy::ScriptingQtDbProxy(QObject *parent) :
    QObject(parent)
{
}
Db* ScriptingQtDbProxy::getDb() const
{
    return db;
}

void ScriptingQtDbProxy::setDb(Db* value)
{
    db = value;
}
bool ScriptingQtDbProxy::getUseDbLocking() const
{
    return useDbLocking;
}

void ScriptingQtDbProxy::setUseDbLocking(bool value)
{
    useDbLocking = value;
}

QHash<QString, QVariant> ScriptingQtDbProxy::mapToHash(const QMap<QString, QVariant>& map)
{
    QHash<QString, QVariant> hash;
    QMapIterator<QString, QVariant> it(map);
    while (it.hasNext())
    {
        it.next();
        hash[it.key()] = it.value();
    }
    return hash;
}

QVariant ScriptingQtDbProxy::evalInternal(const QString& sql, const QList<QVariant>& listArgs, const QMap<QString, QVariant>& mapArgs,
                                          bool singleCell, const QScriptValue* funcPtr)
{
    if (!db)
    {
        QString funcName = singleCell ? QStringLiteral("db.onecolumn()") : QStringLiteral("db.eval()");
        context()->throwError(tr("No database available in current context, while called QtScript's %1 command.").arg(funcName));
        return evalInternalErrorResult(singleCell);
    }

    Db::Flags flags;
    if (!useDbLocking)
        flags |= Db::Flag::NO_LOCK;

    SqlQueryPtr results;
    if (listArgs.size() > 0)
        results = db->exec(sql, listArgs, flags);
    else
        results = db->exec(sql, mapToHash(mapArgs), flags);

    if (results->isError())
    {
        QString funcName = singleCell ? QStringLiteral("db.onecolumn()") : QStringLiteral("db.eval()");
        context()->throwError(tr("Error from %1: %2").arg(funcName, results->getErrorText()));
        return evalInternalErrorResult(singleCell);
    }

    if (singleCell)
    {
        return results->getSingleCell();
    }
    else if (funcPtr)
    {
        QScriptValue func(*funcPtr);
        SqlResultsRowPtr row;
        QScriptValue funcArgs;
        QScriptValue funcResult;
        while (results->hasNext())
        {
            row = results->next();
            funcArgs = context()->engine()->toScriptValue(row->valueList());
            funcResult = func.call(context()->thisObject(), funcArgs);
            if (!funcResult.isUndefined())
                break;
        }
        return funcResult.toVariant();
    }
    else
    {
        QList<QVariant> evalResults;
        SqlResultsRowPtr row;
        while (results->hasNext())
        {
            row = results->next();
            evalResults << QVariant(row->valueList());
        }
        return evalResults;
    }
}

QVariant ScriptingQtDbProxy::evalInternalErrorResult(bool singleCell)
{
    QList<QVariant> result;
    if (singleCell)
        result << QVariant();

    return result;
}

QVariant ScriptingQtDbProxy::eval(const QString& sql)
{
    return evalInternal(sql, QList<QVariant>(), QMap<QString, QVariant>(), false);
}

QVariant ScriptingQtDbProxy::eval(const QString& sql, const QList<QVariant>& args)
{
    return evalInternal(sql, args, QMap<QString, QVariant>(), false);
}

QVariant ScriptingQtDbProxy::eval(const QString& sql, const QMap<QString, QVariant>& args)
{
    return evalInternal(sql, QList<QVariant>(), args, false);
}

QVariant ScriptingQtDbProxy::eval(const QString& sql, const QList<QVariant>& args, const QScriptValue& func)
{
    return evalInternal(sql, args, QMap<QString, QVariant>(), false, &func);
}

QVariant ScriptingQtDbProxy::eval(const QString& sql, const QMap<QString, QVariant>& args, const QScriptValue& func)
{
    return evalInternal(sql, QList<QVariant>(), args, false, &func);
}

QVariant ScriptingQtDbProxy::onecolumn(const QString& sql, const QList<QVariant>& args)
{
    return evalInternal(sql, args, QMap<QString, QVariant>(), true);
}

QVariant ScriptingQtDbProxy::onecolumn(const QString& sql, const QMap<QString, QVariant>& args)
{
    return evalInternal(sql, QList<QVariant>(), args, true);
}