dvander@1203: /**
dvander@1203:  * vim: set ts=4 :
dvander@1226:  * ================================================================
dvander@1226:  * SourceMod SQLite Extension
dvander@1226:  * Copyright (C) 2004-2007 AlliedModders LLC.  All rights reserved.
dvander@1226:  * ================================================================
dvander@1203:  *
dvander@1226:  * This program is free software; you can redistribute it and/or 
dvander@1226:  * modify it under the terms of the GNU General Public License, 
dvander@1226:  * version 3.0, as published by the Free Software Foundation.
dvander@1226:  * 
dvander@1203:  * This program is distributed in the hope that it will be useful,
dvander@1203:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
dvander@1203:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
dvander@1203:  * GNU General Public License for more details.
dvander@1226:  *
dvander@1203:  * You should have received a copy of the GNU General Public License
dvander@1226:  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@1226:  *
dvander@1226:  * As a special exception, AlliedModders LLC gives you permission to 
dvander@1226:  * link the code of this program (as well as its derivative works) to 
dvander@1226:  * "Half-Life 2," the "Source Engine," the "SourcePawn JIT," and any 
dvander@1226:  * Game MODs that run on software by the Valve Corporation.  You must 
dvander@1226:  * obey the GNU General Public License in all respects for all other 
dvander@1226:  * code used. Additionally, AlliedModders LLC grants this exception 
dvander@1226:  * to all derivative works. AlliedModders LLC defines further 
dvander@1226:  * exceptions, found in LICENSE.txt (as of this writing, version 
dvander@1226:  * JULY-31-2007), or <http://www.sourcemod.net/license.php>.
dvander@1203:  *
dvander@1203:  * Version: $Id$
dvander@1203:  */
dvander@1203: 
dvander@1203: #include "SqQuery.h"
dvander@1203: 
dvander@1203: SqQuery::SqQuery(SqDatabase *parent, sqlite3_stmt *stmt) : 
dvander@1203:  m_pParent(parent), m_pStmt(stmt), m_pResults(NULL), m_AffectedRows(0), m_InsertID(0)
dvander@1203: {
dvander@1203: 	m_ParamCount = sqlite3_bind_parameter_count(m_pStmt);
dvander@1210: 	m_ColCount = sqlite3_column_count(m_pStmt);
dvander@1203: 	m_pParent->IncReferenceCount();
dvander@1203: }
dvander@1203: 
dvander@1203: SqQuery::~SqQuery()
dvander@1203: {
dvander@1203: 	delete m_pResults;
dvander@1203: 	sqlite3_finalize(m_pStmt);
dvander@1203: 	m_pParent->Close();
dvander@1203: }
dvander@1203: 
dvander@1203: IResultSet *SqQuery::GetResultSet()
dvander@1203: {
dvander@1203: 	return m_pResults;
dvander@1203: }
dvander@1203: 
dvander@1203: bool SqQuery::FetchMoreResults()
dvander@1203: {
dvander@1203: 	/* We never have multiple result sets */
dvander@1203: 	return false;
dvander@1203: }
dvander@1203: 
dvander@1203: void SqQuery::Destroy()
dvander@1203: {
dvander@1203: 	delete this;
dvander@1203: }
dvander@1203: 
dvander@1203: bool SqQuery::BindParamFloat(unsigned int param, float f)
dvander@1203: {
dvander@1203: 	/* SQLite is 1 indexed */
dvander@1203: 	param++;
dvander@1203: 
dvander@1203: 	if (param > m_ParamCount)
dvander@1203: 	{
dvander@1203: 		return false;
dvander@1203: 	}
dvander@1203: 
dvander@1203: 	return (sqlite3_bind_double(m_pStmt, param, (double)f) == SQLITE_OK);
dvander@1203: }
dvander@1203: 
dvander@1203: bool SqQuery::BindParamNull(unsigned int param)
dvander@1203: {
dvander@1203: 	/* SQLite is 1 indexed */
dvander@1203: 	param++;
dvander@1203: 
dvander@1203: 	if (param > m_ParamCount)
dvander@1203: 	{
dvander@1203: 		return false;
dvander@1203: 	}
dvander@1203: 
dvander@1203: 	return (sqlite3_bind_null(m_pStmt, param) == SQLITE_OK);
dvander@1203: }
dvander@1203: 
dvander@1203: bool SqQuery::BindParamString(unsigned int param, const char *text, bool copy)
dvander@1203: {
dvander@1203: 	/* SQLite is 1 indexed */
dvander@1203: 	param++;
dvander@1203: 
dvander@1203: 	if (param > m_ParamCount)
dvander@1203: 	{
dvander@1203: 		return false;
dvander@1203: 	}
dvander@1203: 
dvander@1203: 	return (sqlite3_bind_text(m_pStmt, param, text, -1, copy ? SQLITE_TRANSIENT : SQLITE_STATIC) == SQLITE_OK);
dvander@1203: }
dvander@1203: 
dvander@1203: bool SqQuery::BindParamInt(unsigned int param, int num, bool signd/* =true */)
dvander@1203: {
dvander@1203: 	/* SQLite is 1 indexed */
dvander@1203: 	param++;
dvander@1203: 
dvander@1203: 	if (param > m_ParamCount)
dvander@1203: 	{
dvander@1203: 		return false;
dvander@1203: 	}
dvander@1203: 
dvander@1203: 	return (sqlite3_bind_int(m_pStmt, param, num) == SQLITE_OK);
dvander@1203: }
dvander@1203: 
dvander@1203: bool SqQuery::BindParamBlob(unsigned int param, const void *data, size_t length, bool copy)
dvander@1203: {
dvander@1203: 	/* SQLite is 1 indexed */
dvander@1203: 	param++;
dvander@1203: 
dvander@1203: 	if (param > m_ParamCount)
dvander@1203: 	{
dvander@1203: 		return false;
dvander@1203: 	}
dvander@1203: 
dvander@1203: 	return (sqlite3_bind_blob(m_pStmt, param, data, length, copy ? SQLITE_TRANSIENT : SQLITE_STATIC) == SQLITE_OK);
dvander@1203: }
dvander@1203: 
dvander@1203: sqlite3_stmt *SqQuery::GetStmt()
dvander@1203: {
dvander@1203: 	return m_pStmt;
dvander@1203: }
dvander@1203: 
dvander@1203: bool SqQuery::Execute()
dvander@1203: {
dvander@1203: 	int rc;
dvander@1203: 
dvander@1210: 	/* If we don't have a result set and we have a column count, 
dvander@1210: 	 * create a result set pre-emptively.  This is in case there
dvander@1210: 	 * are no rows in the upcoming result set.
dvander@1210: 	 */
dvander@1210: 	if (!m_pResults && m_ColCount)
dvander@1210: 	{
dvander@1210: 		m_pResults = new SqResults(this);
dvander@1210: 	}
dvander@1210: 
dvander@1203: 	/* If we've got results, throw them away */
dvander@1203: 	if (m_pResults)
dvander@1203: 	{
dvander@1203: 		m_pResults->ResetResultCount();
dvander@1203: 	}
dvander@1203: 
dvander@1210: 	/* Fetch each row, if any */
dvander@1203: 	while ((rc = sqlite3_step(m_pStmt)) == SQLITE_ROW)
dvander@1203: 	{
dvander@1210: 		/* This should NEVER happen but we're being safe. */
dvander@1203: 		if (!m_pResults)
dvander@1203: 		{
dvander@1203: 			m_pResults = new SqResults(this);
dvander@1203: 		}
dvander@1203: 		m_pResults->PushResult();
dvander@1203: 	}
dvander@1203: 
dvander@1203: 	sqlite3 *db = m_pParent->GetDb();
dvander@1203: 	if (rc != SQLITE_OK && rc != SQLITE_DONE && rc == sqlite3_errcode(db))
dvander@1203: 	{
dvander@1203: 		/* Something happened... */
dvander@1203: 		m_LastErrorCode = rc;
dvander@1203: 		m_LastError.assign(sqlite3_errmsg(db));
dvander@1203: 		m_AffectedRows = 0;
dvander@1203: 		m_InsertID = 0;
dvander@1203: 	} else {
dvander@1203: 		m_LastErrorCode = SQLITE_OK;
dvander@1203: 		m_AffectedRows = (unsigned int)sqlite3_changes(db);
dvander@1203: 		m_InsertID = (unsigned int)sqlite3_last_insert_rowid(db);
dvander@1203: 	}
dvander@1203: 
dvander@1203: 	/* Reset everything for the next execute */
dvander@1203: 	sqlite3_reset(m_pStmt);
dvander@1203: 	sqlite3_clear_bindings(m_pStmt);
dvander@1203: 
dvander@1203: 	return (m_LastErrorCode == SQLITE_OK);
dvander@1203: }
dvander@1203: 
dvander@1203: const char *SqQuery::GetError(int *errCode/* =NULL */)
dvander@1203: {
dvander@1203: 	if (errCode)
dvander@1203: 	{
dvander@1203: 		*errCode = m_LastErrorCode;
dvander@1203: 	}
dvander@1203: 	return m_LastError.c_str();
dvander@1203: }
dvander@1203: 
dvander@1203: unsigned int SqQuery::GetAffectedRows()
dvander@1203: {
dvander@1203: 	return m_AffectedRows;
dvander@1203: }
dvander@1203: 
dvander@1203: unsigned int SqQuery::GetInsertID()
dvander@1203: {
dvander@1203: 	return m_InsertID;
dvander@1203: }
dvander@1203: 
dvander@1203: