#include "building_model.h"

#include "core/database/database.h"
#include "core/database/database_manager.h"
#include "core/database/query_builder.h"
#include "core/database/query_result.h"
#include "core/database/table_builder.h"

#include "core/hash/sha256.h"

#include "building.h"

#define BUILDING_TABLE_NAME "buildings"

#define BUILDING_TABLE_COLUMNS "id, name, description, icon, rank, next_rank, time_to_build, creates, num_creates, score, defense, ability, cost_food, cost_wood, cost_stone, cost_iron, cost_mana, mod_max_food, mod_max_wood, mod_max_stone, mod_max_iron, mod_max_mana, mod_rate_food, mod_rate_wood, mod_rate_stone, mod_rate_iron, mod_rate_mana, mod_percent_food, mod_percent_wood, mod_percent_stone, mod_percent_iron, mod_percent_mana, assignment1, assignment2, assignment3, assignment4, assignment5, req_tech, tech_group, tech_secondary_group"
#define BUILDING_TABLE_COLUMNS_NOID "name, description, icon, rank, next_rank, time_to_build, creates, num_creates, score, defense, ability, cost_food, cost_wood, cost_stone, cost_iron, cost_mana, mod_max_food, mod_max_wood, mod_max_stone, mod_max_iron, mod_max_mana, mod_rate_food, mod_rate_wood, mod_rate_stone, mod_rate_iron, mod_rate_mana, mod_percent_food, mod_percent_wood, mod_percent_stone, mod_percent_iron, mod_percent_mana, assignment1, assignment2, assignment3, assignment4, assignment5, req_tech, tech_group, tech_secondary_group"

Ref<Building> BuildingModel::get_building(const int id) {
	if (id == 0) {
		return Ref<Building>();
	}

	Ref<QueryBuilder> b = DatabaseManager::get_singleton()->ddb->get_query_builder();

	b->select(BUILDING_TABLE_COLUMNS);
	b->from(BUILDING_TABLE_NAME);

	b->where()->wp("id", id);

	b->end_command();

	Ref<QueryResult> r = b->run();

	if (!r->next_row()) {
		return Ref<Building>();
	}

	Ref<Building> building;
	building.instance();

	parse_row(r, building);

	return building;
}

Vector<Ref<Building> > BuildingModel::get_all() {
	Ref<QueryBuilder> b = DatabaseManager::get_singleton()->ddb->get_query_builder();

	b->select(BUILDING_TABLE_COLUMNS);
	b->from(BUILDING_TABLE_NAME);
	b->end_command();
	//b->print();

	Vector<Ref<Building> > buildings;

	Ref<QueryResult> r = b->run();

	while (r->next_row()) {
		Ref<Building> building;
		building.instance();

		parse_row(r, building);

		buildings.push_back(building);
	}

	return buildings;
}

void BuildingModel::save_building(Ref<Building> &building) {
	Ref<QueryBuilder> b = DatabaseManager::get_singleton()->ddb->get_query_builder();

	if (building->id == 0) {
		b->insert(BUILDING_TABLE_NAME, BUILDING_TABLE_COLUMNS_NOID);

		b->values();

		b->val(building->name);
		b->val(building->description);
		b->val(building->icon);

		b->val(building->rank);
		b->val(building->next_rank);
		b->val(building->time_to_build);
		b->val(building->creates);
		b->val(building->num_creates);
		b->val(building->score);
		b->val(building->defense);
		b->val(building->ability);

		b->val(building->cost_food);
		b->val(building->cost_wood);
		b->val(building->cost_stone);
		b->val(building->cost_iron);
		b->val(building->cost_mana);

		b->val(building->mod_max_food);
		b->val(building->mod_max_wood);
		b->val(building->mod_max_stone);
		b->val(building->mod_max_iron);
		b->val(building->mod_max_mana);

		b->vald(building->mod_rate_food);
		b->vald(building->mod_rate_wood);
		b->vald(building->mod_rate_stone);
		b->vald(building->mod_rate_iron);
		b->vald(building->mod_rate_mana);

		b->val(building->mod_percent_food);
		b->val(building->mod_percent_wood);
		b->val(building->mod_percent_stone);
		b->val(building->mod_percent_iron);
		b->val(building->mod_percent_mana);

		b->val(building->assignment1);
		b->val(building->assignment2);
		b->val(building->assignment3);
		b->val(building->assignment4);
		b->val(building->assignment5);

		b->val(building->req_tech);
		b->val(building->tech_group);
		b->val(building->tech_secondary_group);

		b->cvalues();

		b->end_command();
		b->select_last_insert_id();
		//b->print();

		Ref<QueryResult> r = b->run();

		building->id = r->get_last_insert_rowid();
	} else {
		b->update(BUILDING_TABLE_NAME);
		b->set();

		b->setp("name", building->name);
		b->setp("description", building->description);
		b->setp("icon", building->icon);

		b->setp("userankrname", building->rank);
		b->setp("next_rank", building->next_rank);
		b->setp("time_to_build", building->time_to_build);
		b->setp("creates", building->creates);
		b->setp("num_creates", building->num_creates);
		b->setp("score", building->score);
		b->setp("defense", building->defense);
		b->setp("ability", building->ability);

		b->setp("cost_food", building->cost_food);
		b->setp("cost_wood", building->cost_wood);
		b->setp("cost_stone", building->cost_stone);
		b->setp("cost_iron", building->cost_iron);
		b->setp("cost_mana", building->cost_mana);

		b->setp("mod_max_food", building->mod_max_food);
		b->setp("mod_max_wood", building->mod_max_wood);
		b->setp("mod_max_stone", building->mod_max_stone);
		b->setp("mod_max_iron", building->mod_max_iron);
		b->setp("mod_max_mana", building->mod_max_mana);

		b->setpd("mod_rate_food", building->mod_rate_food);
		b->setpd("mod_rate_wood", building->mod_rate_wood);
		b->setpd("mod_rate_stone", building->mod_rate_stone);
		b->setpd("mod_rate_iron", building->mod_rate_iron);
		b->setpd("mod_rate_mana", building->mod_rate_mana);

		b->setp("mod_percent_food", building->mod_percent_food);
		b->setp("mod_percent_wood", building->mod_percent_wood);
		b->setp("mod_percent_stone", building->mod_percent_stone);
		b->setp("mod_percent_iron", building->mod_percent_iron);
		b->setp("mod_percent_mana", building->mod_percent_mana);

		b->setp("assignment1", building->assignment1);
		b->setp("assignment2", building->assignment2);
		b->setp("assignment3", building->assignment3);
		b->setp("assignment4", building->assignment4);
		b->setp("assignment5", building->assignment5);

		b->setp("req_tech", building->req_tech);
		b->setp("tech_group", building->tech_group);
		b->setp("tech_secondary_group", building->tech_secondary_group);

		b->cset();
		b->where()->wp("id", building->id);

		//b->print();

		b->run_query();
	}
}

void BuildingModel::parse_row(Ref<QueryResult> &result, Ref<Building> &building) {

	building->id = result->get_cell_int(0);

	building->name = result->get_cell(1);
	building->description = result->get_cell(2);
	building->icon = result->get_cell(3);

	building->rank = result->get_cell_int(4);
	building->next_rank = result->get_cell_int(5);
	building->time_to_build = result->get_cell_int(6);
	building->creates = result->get_cell_int(7);
	building->num_creates = result->get_cell_int(8);
	building->score = result->get_cell_int(9);
	building->defense = result->get_cell_int(10);
	building->ability = result->get_cell_int(11);

	building->cost_food = result->get_cell_int(12);
	building->cost_wood = result->get_cell_int(13);
	building->cost_stone = result->get_cell_int(14);
	building->cost_iron = result->get_cell_int(15);
	building->cost_mana = result->get_cell_int(16);

	building->mod_max_food = result->get_cell_int(17);
	building->mod_max_wood = result->get_cell_int(18);
	building->mod_max_stone = result->get_cell_int(19);
	building->mod_max_iron = result->get_cell_int(20);
	building->mod_max_mana = result->get_cell_int(21);

	building->mod_rate_food = result->get_cell_double(22);
	building->mod_rate_wood = result->get_cell_double(23);
	building->mod_rate_stone = result->get_cell_double(24);
	building->mod_rate_iron = result->get_cell_double(25);
	building->mod_rate_mana = result->get_cell_double(26);

	building->mod_percent_food = result->get_cell_int(27);
	building->mod_percent_wood = result->get_cell_int(28);
	building->mod_percent_stone = result->get_cell_int(29);
	building->mod_percent_iron = result->get_cell_int(30);
	building->mod_percent_mana = result->get_cell_int(31);

	building->assignment1 = result->get_cell_int(32);
	building->assignment2 = result->get_cell_int(33);
	building->assignment3 = result->get_cell_int(34);
	building->assignment4 = result->get_cell_int(35);
	building->assignment5 = result->get_cell_int(36);

	building->req_tech = result->get_cell_int(37);
	building->tech_group = result->get_cell_int(38);
	building->tech_secondary_group = result->get_cell_int(39);
}

void BuildingModel::create_table() {
	Ref<TableBuilder> tb = DatabaseManager::get_singleton()->ddb->get_table_builder();

	tb->create_table(BUILDING_TABLE_NAME);
	tb->integer("id", 11)->auto_increment()->next_row();
	tb->varchar("name", 200)->not_null()->next_row();
	tb->varchar("description", 500)->not_null()->next_row();
	tb->varchar("icon", 500)->not_null()->next_row();
	tb->integer("rank", 11)->not_null()->next_row();
	tb->integer("next_rank", 11)->not_null()->next_row();
	tb->integer("time_to_build", 11)->not_null()->next_row();
	tb->integer("creates", 11)->not_null()->defval("0")->next_row();
	tb->integer("num_creates", 11)->not_null()->defval("0")->next_row();
	tb->integer("score", 11)->not_null()->next_row();
	tb->integer("defense", 11)->not_null()->next_row();
	tb->integer("ability", 11)->not_null()->defval("0")->next_row();

	tb->integer("cost_food", 11)->not_null()->next_row();
	tb->integer("cost_wood", 11)->not_null()->next_row();
	tb->integer("cost_stone", 11)->not_null()->next_row();
	tb->integer("cost_iron", 11)->not_null()->next_row();
	tb->integer("cost_mana", 11)->not_null()->next_row();

	tb->integer("mod_max_food", 11)->not_null()->defval("0")->next_row();
	tb->integer("mod_max_wood", 11)->not_null()->defval("0")->next_row();
	tb->integer("mod_max_stone", 11)->not_null()->defval("0")->next_row();
	tb->integer("mod_max_iron", 11)->not_null()->defval("0")->next_row();
	tb->integer("mod_max_mana", 11)->not_null()->defval("0")->next_row();

	tb->real_double("mod_rate_food")->not_null()->defval("0")->next_row();
	tb->real_double("mod_rate_wood")->not_null()->defval("0")->next_row();
	tb->real_double("mod_rate_stone")->not_null()->defval("0")->next_row();
	tb->real_double("mod_rate_iron")->not_null()->defval("0")->next_row();
	tb->real_double("mod_rate_mana")->not_null()->defval("0")->next_row();

	tb->integer("mod_percent_food", 11)->not_null()->defval("0")->next_row();
	tb->integer("mod_percent_wood", 11)->not_null()->defval("0")->next_row();
	tb->integer("mod_percent_stone", 11)->not_null()->defval("0")->next_row();
	tb->integer("mod_percent_iron", 11)->not_null()->defval("0")->next_row();
	tb->integer("mod_percent_mana", 11)->not_null()->defval("0")->next_row();

	tb->integer("assignment1", 11)->not_null()->defval("0")->next_row();
	tb->integer("assignment2", 11)->not_null()->defval("0")->next_row();
	tb->integer("assignment3", 11)->not_null()->defval("0")->next_row();
	tb->integer("assignment4", 11)->not_null()->defval("0")->next_row();
	tb->integer("assignment5", 11)->not_null()->defval("0")->next_row();

	tb->integer("req_tech", 11)->not_null()->defval("0")->next_row();
	tb->integer("tech_group", 11)->not_null()->defval("0")->next_row();
	tb->integer("tech_secondary_group", 11)->not_null()->defval("0")->next_row();

	tb->primary_key("id");
	tb->ccreate_table();

	tb->run_query();
	//tb->print();
}
void BuildingModel::drop_table() {
	Ref<TableBuilder> tb = DatabaseManager::get_singleton()->ddb->get_table_builder();

	tb->drop_table_if_exists(BUILDING_TABLE_NAME)->cdrop_table();

	tb->run_query();
}
void BuildingModel::migrate() {
	drop_table();
	create_table();
}

void BuildingModel::add_default_data() {
	String table_columns = "id, name, description, icon, rank, next_rank, time_to_build, creates, num_creates, score, defense, ability, cost_food, cost_wood, cost_stone, cost_iron, cost_mana, mod_max_food, mod_max_wood, mod_max_stone, mod_max_iron, mod_max_mana, mod_rate_food, mod_rate_wood, mod_rate_stone, mod_rate_iron, mod_rate_mana, mod_percent_food, mod_percent_wood, mod_percent_stone, mod_percent_iron, mod_percent_mana, assignment1, assignment2, assignment3, assignment4, assignment5, req_tech, tech_group, tech_secondary_group";

	Ref<QueryBuilder> qb = DatabaseManager::get_singleton()->ddb->get_query_builder();

	qb->begin_transaction()->nl();
	qb->insert(BUILDING_TABLE_NAME, table_columns)->nl()->w("VALUES(1, 'empty', '', 'empty/empty.png', 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0)")->end_command()->nl();
	qb->insert(BUILDING_TABLE_NAME, table_columns)->nl()->w("VALUES(2, 'Build in Progress', '', 'bip/bip.png', 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0)")->end_command()->nl();
	qb->insert(BUILDING_TABLE_NAME, table_columns)->nl()->w("VALUES(3, 'Corn Field', 'Produces food.', 'corn_field/r1.png', 1, 7, 20, 0, 0, 20, 1, 0, 60, 100, 10, 5, 0, 0, 0, 0, 0, 0, 0.01, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 2, 4, 3, 0, 0, 2, 3)")->end_command()->nl();
	qb->insert(BUILDING_TABLE_NAME, table_columns)->nl()->w("VALUES(4, 'Lumber Mill', 'Your main wood producing building.', 'lumber_mill/r1.png', 1, 0, 1000, 0, 0, 20, 0, 0, 30, 40, 50, 10, 0, 0, 0, 0, 0, 0, 0, 0.01, 0, 0, 0, 0, 0, 0, 0, 0, 3, 0, 0, 0, 0, 0, 1, 2)")->end_command()->nl();
	qb->insert(BUILDING_TABLE_NAME, table_columns)->nl()->w("VALUES(5, 'Stone Mine', 'Your main stone producing building.', 'stone_mine/r1.png', 1, 0, 1000, 2, 20, 0, 0, 0, 30, 50, 20, 30, 0, 0, 0, 0, 0, 0, 0, 0, 0.01, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0)")->end_command()->nl();
	qb->insert(BUILDING_TABLE_NAME, table_columns)->nl()->w("VALUES(6, 'House', 'Can create villagers.', 'house/r1.png', 1, 0, 20, 1, 10, 0, 0, 0, 50, 70, 30, 5, 0, 0, 0, 0, 0, 0, -0.005, -0.001, -0.001, -0.001, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0)")->end_command()->nl();
	qb->insert(BUILDING_TABLE_NAME, table_columns)->nl()->w("VALUES(7, 'Corn Field', '', 'corn_field/r2.png', 2, 0, 20, 0, 0, 0, 0, 0, 40, 60, 20, 10, 0, 0, 0, 0, 0, 0, 0.2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 2, 0, 0, 0, 2, 0, 0)")->end_command()->nl();
	qb->insert(BUILDING_TABLE_NAME, table_columns)->nl()->w("VALUES(8, 'Farm', 'Creates villagers.', 'farm/r1.png', 1, 0, 80, 1, 20, 0, 0, 0, 50, 60, 10, 10, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0)")->end_command()->nl();
	qb->insert(BUILDING_TABLE_NAME, table_columns)->nl()->w("VALUES(9, 'Iron Mine', 'Your main iron producing building.', 'iron_mine/r1.png', 1, 0, 1000, 2, 100000, 0, 0, 0, 70, 30, 70, 20, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0.01, 0, 0, 0, 0, 0, 0, 3, 0, 0, 0, 0, 0, 0, 0)")->end_command()->nl();
	qb->insert(BUILDING_TABLE_NAME, table_columns)->nl()->w("VALUES(10, 'School', 'School', 'school/r1.png', 1, 0, 60, 2, 60, 0, 0, 0, 300, 300, 300, 300, 20, 0, 0, 0, 0, 0, 0.001, 0.001, 0.001, 0.001, 0.001, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1, 1, 2)")->end_command()->nl();
	qb->commit();

	qb->run_query();
	//qb->print();
}

BuildingModel *BuildingModel::get_singleton() {
	return _self;
}

BuildingModel::BuildingModel() :
		Object() {

	if (_self) {
		printf("BuildingModel::BuildingModel(): Error! self is not null!/n");
	}

	_self = this;
}

BuildingModel::~BuildingModel() {
	if (_self == this) {
		_self = nullptr;
	}
}

BuildingModel *BuildingModel::_self = nullptr;