/**

 * vim: set ts=4 :

 * ================================================================

 * SourceMod SQLite Extension

 * Copyright (C) 2004-2007 AlliedModders LLC.  All rights reserved.

 * ================================================================

 *

 * This program is free software; you can redistribute it and/or 

 * modify it under the terms of the GNU General Public License, 

 * version 3.0, as published by the Free Software Foundation.

 * 

 * 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 for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program.  If not, see <http://www.gnu.org/licenses/>.

 *

 * As a special exception, AlliedModders LLC gives you permission to 

 * link the code of this program (as well as its derivative works) to 

 * "Half-Life 2," the "Source Engine," the "SourcePawn JIT," and any 

 * Game MODs that run on software by the Valve Corporation.  You must 

 * obey the GNU General Public License in all respects for all other 

 * code used. Additionally, AlliedModders LLC grants this exception 

 * to all derivative works. AlliedModders LLC defines further 

 * exceptions, found in LICENSE.txt (as of this writing, version 

 * JULY-31-2007), or <http://www.sourcemod.net/license.php>.

 *

 * Version: $Id$

 */

#include "SqQuery.h"

SqQuery::SqQuery(SqDatabase *parent, sqlite3_stmt *stmt) : 

 m_pParent(parent), m_pStmt(stmt), m_pResults(NULL), m_AffectedRows(0), m_InsertID(0)

{

	m_ParamCount = sqlite3_bind_parameter_count(m_pStmt);

	m_ColCount = sqlite3_column_count(m_pStmt);

	m_pParent->IncReferenceCount();

}

SqQuery::~SqQuery()

{

	delete m_pResults;

	sqlite3_finalize(m_pStmt);

	m_pParent->Close();

}

IResultSet *SqQuery::GetResultSet()

{

	return m_pResults;

}

bool SqQuery::FetchMoreResults()

{

	/* We never have multiple result sets */

	return false;

}

void SqQuery::Destroy()

{

	delete this;

}

bool SqQuery::BindParamFloat(unsigned int param, float f)

{

	/* SQLite is 1 indexed */

	param++;

	if (param > m_ParamCount)

	{

		return false;

	}

	return (sqlite3_bind_double(m_pStmt, param, (double)f) == SQLITE_OK);

}

bool SqQuery::BindParamNull(unsigned int param)

{

	/* SQLite is 1 indexed */

	param++;

	if (param > m_ParamCount)

	{

		return false;

	}

	return (sqlite3_bind_null(m_pStmt, param) == SQLITE_OK);

}

bool SqQuery::BindParamString(unsigned int param, const char *text, bool copy)

{

	/* SQLite is 1 indexed */

	param++;

	if (param > m_ParamCount)

	{

		return false;

	}

	return (sqlite3_bind_text(m_pStmt, param, text, -1, copy ? SQLITE_TRANSIENT : SQLITE_STATIC) == SQLITE_OK);

}

bool SqQuery::BindParamInt(unsigned int param, int num, bool signd/* =true */)

{

	/* SQLite is 1 indexed */

	param++;

	if (param > m_ParamCount)

	{

		return false;

	}

	return (sqlite3_bind_int(m_pStmt, param, num) == SQLITE_OK);

}

bool SqQuery::BindParamBlob(unsigned int param, const void *data, size_t length, bool copy)

{

	/* SQLite is 1 indexed */

	param++;

	if (param > m_ParamCount)

	{

		return false;

	}

	return (sqlite3_bind_blob(m_pStmt, param, data, length, copy ? SQLITE_TRANSIENT : SQLITE_STATIC) == SQLITE_OK);

}

sqlite3_stmt *SqQuery::GetStmt()

{

	return m_pStmt;

}

bool SqQuery::Execute()

{

	int rc;

	/* If we don't have a result set and we have a column count, 

	 * create a result set pre-emptively.  This is in case there

	 * are no rows in the upcoming result set.

	 */

	if (!m_pResults && m_ColCount)

	{

		m_pResults = new SqResults(this);

	}

	/* If we've got results, throw them away */

	if (m_pResults)

	{

		m_pResults->ResetResultCount();

	}

	/* Fetch each row, if any */

	while ((rc = sqlite3_step(m_pStmt)) == SQLITE_ROW)

	{

		/* This should NEVER happen but we're being safe. */

		if (!m_pResults)

		{

			m_pResults = new SqResults(this);

		}

		m_pResults->PushResult();

	}

	sqlite3 *db = m_pParent->GetDb();

	if (rc != SQLITE_OK && rc != SQLITE_DONE && rc == sqlite3_errcode(db))

	{

		/* Something happened... */

		m_LastErrorCode = rc;

		m_LastError.assign(sqlite3_errmsg(db));

		m_AffectedRows = 0;

		m_InsertID = 0;

	} else {

		m_LastErrorCode = SQLITE_OK;

		m_AffectedRows = (unsigned int)sqlite3_changes(db);

		m_InsertID = (unsigned int)sqlite3_last_insert_rowid(db);

	}

	/* Reset everything for the next execute */

	sqlite3_reset(m_pStmt);

	sqlite3_clear_bindings(m_pStmt);

	return (m_LastErrorCode == SQLITE_OK);

}

const char *SqQuery::GetError(int *errCode/* =NULL */)

{

	if (errCode)

	{

		*errCode = m_LastErrorCode;

	}

	return m_LastError.c_str();

}

unsigned int SqQuery::GetAffectedRows()

{

	return m_AffectedRows;

}

unsigned int SqQuery::GetInsertID()

{

	return m_InsertID;

}