#include "opdatabase.h"
#include <string>
#include <hv/hlog.h>
#include <sqlite3.h>
#include "openjson.h"


OpDatabase::OpDatabase()
	:m_pSqliteDb(nullptr)
{
}

OpDatabase::~OpDatabase()
{
	CloseDatabase();
}


bool OpDatabase::OpenDatabase(const std::string& db_file_path)
{
	// 打开数据库
	if (sqlite3_open(db_file_path.c_str(), &m_pSqliteDb) != SQLITE_OK)
	{
		hloge("Error opening database.");
		return false;
	}

	return true;
}

// 关闭数据库连接
void OpDatabase::CloseDatabase()
{
	sqlite3_close(m_pSqliteDb);
}

void OpDatabase::InsertMessage(const std::string& ts, const std::string& msg_type,const std::string& fsu, const std::string& content, int topic, int dev_id)
{
	const char* insertDataSQL = "INSERT INTO tbl_data (data_timestamp, data_type, FsuCode, data_content, topic, device_id) VALUES (?,?,?,?,?,?);";
	sqlite3_stmt* statement;
	if (sqlite3_prepare_v2(m_pSqliteDb, insertDataSQL, -1, &statement, nullptr) == SQLITE_OK) 
	{
		// 绑定参数
		sqlite3_bind_text(statement, 1, ts.c_str(), -1, SQLITE_STATIC);
		sqlite3_bind_text(statement, 2, msg_type.c_str(), -1, SQLITE_STATIC);
		sqlite3_bind_text(statement, 3, fsu.c_str(), -1, SQLITE_STATIC);
		sqlite3_bind_blob(statement, 4, content.c_str(), content.size(), SQLITE_STATIC);
		//sqlite3_bind_text(statement, 4, content.c_str(), -1, SQLITE_STATIC);
		sqlite3_bind_int(statement, 5, topic);
		sqlite3_bind_int(statement, 6, dev_id);

		// 执行语句
		if (sqlite3_step(statement) != SQLITE_DONE) 
		{
			hloge("Error inserting data.");
		}

		// 释放资源
		sqlite3_finalize(statement);
	}
}

bool OpDatabase::QueryUser(const std::string& uname, const std::string& passwd, std::string& resultJson)
{
	const char* queryDataSQL = "SELECT * FROM tbl_user WHERE uid=?;";
	sqlite3_stmt * statement;
	bool ret = false;
	
	int rc = sqlite3_prepare_v2(m_pSqliteDb, queryDataSQL, -1, &statement, nullptr);
	if (rc != SQLITE_OK)
	{
		hloge("SQL error: %s ", sqlite3_errmsg(m_pSqliteDb));
		return ret;
	}

	if ( rc == SQLITE_OK)
	{
		// 绑定参数
		sqlite3_bind_text(statement, 1, uname.c_str(), -1, SQLITE_STATIC);
		
		// 执行语句
		if (sqlite3_step(statement) != SQLITE_DONE)
		{
			hloge("Error query user %s", uname.c_str());
		}
		else
		{
			OpenJson json;
			auto& nodeRoot = json["users"];
			size_t i = 0;
			// 遍历结果集
			while (sqlite3_step(statement) == SQLITE_ROW)
			{
				auto& node = nodeRoot[i];
				node["id"] = sqlite3_column_text(statement, 0);
				node["user"] = sqlite3_column_text(statement, 1);
				node["salt"] = sqlite3_column_text(statement, 2);
				node["passwd"] = sqlite3_column_text(statement, 3);
			}

			resultJson = json.encode();
			ret = true;
		}

		// 释放资源
		sqlite3_finalize(statement);

		return ret;
	}

	return ret;
}