#include "db.h"
#include <iostream>
#include <string>

using namespace std;

Database::Database(string filename)
{
	database = NULL;
	open(filename);
}

Database::~Database()
{
}

bool Database::open(string filename)
{
	if(sqlite3_open(filename.c_str(), &database) == SQLITE_OK)
		return true;
		
	return false;   
}

/*vector<vector<string> > Database::qstream
{
	return query(query.str());
}*/

vector<vector<string> > Database::query(string query)
{
	sqlite3_stmt *statement;
	vector<vector<string> > results;
	if(sqlite3_prepare_v2(database, query.c_str(), -1, &statement, 0) == SQLITE_OK)
	{
		int cols = sqlite3_column_count(statement);
		int result = 0;
		while(true)
		{
			result = sqlite3_step(statement);
			
			if(result == SQLITE_ROW)
			{
				vector<string> values;
				for(int col = 0; col < cols; col++)
				{
					values.push_back((char*)sqlite3_column_text(statement, col));
				}
				results.push_back(values);
			}
			else
			{
				break;   
			}
		}
	   
		sqlite3_finalize(statement);
	}
	
	string error = sqlite3_errmsg(database);
	if(error != "not an error") cout << query << " " << error << endl;
	
	return results;  
}

void Database::close()
{
	sqlite3_close(database);   
}