#pragma once

#include <vector>

#include <string>

#include "../RecordManager/RecordManager.h"

#include "../catalog_manager/CatalogManager.h"

#include "../IndexManager/IndexManager.h"

#include "../Public/Base.h"

#include "../Public/SelectCondition.h"

#include "../Public/Attribute.h"

#include "../Public/Tuple.h"

using namespace std;

struct InsertVal // insert values

{

string val; // value

NumType given_type; // type

InsertVal() : val(""), given_type(DEFAULT) {}

InsertVal(string given_val, NumType given_t)

: val(given_val), given_type(given_t) {}

};

class InsertQuery

{

public:

InsertQuery(){};

~InsertQuery(){};

void SetTableName(string tablename);

void Clear(); // clear all saved data

void Insert(string insert_val); // insert value

Result Query(CatalogManager& catalog_manager, RecordManager& record_manager, IndexManager& index_manager); // main process function

private:

string insert_table_name;

vector<InsertVal> insert_values;

};

class CreateTable

{

public:

CreateTable(){};

~CreateTable(){};

void SetTableName(string &given_table_name);

void InsertAttr(string &attr_name);

void InsertType(NumType given_type);

void InsertSize(int given_size);

void InsertUnique();

void InsertPrimary(string& attr_name);

void Clear();

int Query(CatalogManager& catalog_manager, IndexManager& index_manager);

private:

string tmp_table_name;

vector<Attribute> attr_list;

vector<string> primary_list;

};

class SelectQuery

{

public:

SelectQuery();

~SelectQuery();

Result Query(CatalogManager& catalog_manager, RecordManager& record_manager, IndexManager& index_manager);

void Insert(string &attr);

void Clear();

void SetSelectAll();

void SetSelectTable(string &table);

void SetWhereExpr(string& expr2);

void SetCmp(CMP in_cmp);

void SetExpr1(string& in_expr1);

private:

vector<string> select_attr_list;

string select_table_name;

bool select_all;

vector<WhereExpr> where_expr_list;

string tmp_expr1;

CMP tmp_cmp;

};

class DeleteQuery

{

public:

DeleteQuery();

~DeleteQuery();

Result Query(CatalogManager& catalog_manager, RecordManager& record_manager, IndexManager& index_manager);

void Clear();

void SetDeleteTable(string& table);

void SetExpr1(string& in_expr1);

void SetCmp(CMP in_cmp);

void SetWhereExpr(string& expr2);

private:

string delete_table_name;

vector<WhereExpr> where_expr_list;

string tmp_expr1;

CMP tmp_cmp;

};

class API {

public:

InsertQuery insert_values_query;

CreateTable create_table;

SelectQuery select_query;

DeleteQuery delete_query;

RecordManager record_manager;

IndexManager index_manager;

CatalogManager catalog_manager;

void DropIndex(string drop_index){

DropIndex(drop_index, catalog_manager, index_manager);

}

void DropIndex(string drop_index, CatalogManager& catalog_manager, IndexManager& index_manager);

void DropTable(string drop_table){

DropTable(drop_table, catalog_manager);

}

void DropTable(string drop_table, CatalogManager& catalog_manager);

void CreateIndex(string index_name, string table_name, string attr_name);

};

Result WhereExpr_To_SelectCondition(CatalogManager& catalog_manager, string& tableName, vector<WhereExpr>& wheres, vector<SelectCondition>& selectConditions);