//#pragma comment(lib, "sqlite3.lib")
#ifndef _TQND_SQLITEDB_
#define _TQND_SQLITEDB_ 
//#include "sqlite3ext.h"
#include "sqlite3.h"
#include "IChatLogDB.h"

#ifndef NAMESPACE_SQLITE_BEGIN
#define NAMESPACE_SQLITE_BEGIN				namespace sqlite {
#define NAMESPACE_SQLITE_END				}
#endif

NAMESPACE_SQLITE_BEGIN
using namespace std;

class CResult
{
friend class SQLiteDB;
public:
	CResult() : m_ppRes(NULL), m_nRow(0), m_nCol(0) {}
	~CResult()		{sqlite3_free_table(m_ppRes);}

	//獲得行數(shù)
	int GetRowNum()			{return m_nRow+1;}		//0行是字段名

	//獲得列數(shù)
	int GetColNum()			{return m_nCol;}

	//獲得結(jié)果數(shù)目
	int GetResultNum()		{return m_nRow;}

	//獲取對應(yīng)行列的結(jié)果
	char* GetAt(int nRow,int nCol)	
	{
		if(nRow<=m_nRow+1&&nCol<=m_nCol)
			return m_ppRes[nRow*m_nCol+nCol];
		else return NULL;
	}

private:
	char** m_ppRes;
	int m_nRow;
	int m_nCol;	
};

class SQLiteDB
{	
public:
	SQLiteDB() : m_pDB(NULL), m_bOpenflag(false), m_pRes(NULL) 
	{
		m_strKey = "";
	}

	//功能說明  :連接數(shù)據(jù)庫。如果數(shù)據(jù)庫已經(jīng)存在，則打開現(xiàn)有數(shù)據(jù)庫，如果不存在，則新創(chuàng)建一個(gè)數(shù)據(jù)庫
	//參數(shù)說明  :string DBName 數(shù)據(jù)庫名稱
	//返回值說明:bool類型  true 數(shù)據(jù)庫打開成功STRIN
	//                     false 數(shù)據(jù)庫打開失敗
	bool OpenDB(const char* pszPath, const char* pszKey = NULL);
	
	//功能說明  :在指定的數(shù)據(jù)庫中執(zhí)行sql語句（非select語句?。?	//參數(shù)說明  :string sql 執(zhí)行的sql語句
	//返回值說明:int類型 	-1 sql語句執(zhí)行失敗，同時(shí)輸出錯(cuò)誤信息
	//                      0 sql語句執(zhí)行成功，返回結(jié)果
	int Execute(const char* pszStatement);
	
	//功能說明  :在指定的數(shù)據(jù)庫中執(zhí)行文件中的sql語句（文件中不能含有select語句!）
	//參數(shù)說明  :string sql 執(zhí)行的sql語句
	//返回值說明:int類型 	-1 sql語句執(zhí)行失敗，同時(shí)輸出錯(cuò)誤信息
	//                      0 sql語句執(zhí)行成功，返回結(jié)果
	int ExecuteSQLFromFile(const char* pszFile);
	
	//功能說明  :在指定的數(shù)據(jù)庫中執(zhí)行select的sql語句
	//參數(shù)說明  :string sql 執(zhí)行的sql語句
	//返回值說明:Result類型 NULL sql語句執(zhí)行失敗，同時(shí)輸出錯(cuò)誤信息
	//                      rs   sql語句執(zhí)行成功，返回Result類對象rs
	CResult* ExecuteSelect(const char* pszSelect);


	//////////////////////////////////////////////////////////////////////////
	bool AddInfoContainBinary(string& strSql, string& strTxt);
	bool GetInfoContainBinary(string& strSql, ROW_INFO_VEC& vecContent, bool bSys);

	//功能說明   :關(guān)閉數(shù)據(jù)庫。
	//參數(shù)說明   :void 無
	//返回值說明 :bool類型 true 數(shù)據(jù)庫關(guān)閉成功
	//                     false 數(shù)據(jù)庫關(guān)閉失敗
	bool CloseDB();
	~SQLiteDB();

private:
	string	Decode(string& strContent);		
	string	Encode(string& strContent);
	string	Process(string& strContent);

private:
	sqlite3*	m_pDB;
	bool		m_bOpenflag;
	CResult*	m_pRes;
	string		m_strKey;
};

#endif
NAMESPACE_SQLITE_END