/*
 * This file is a part of morm
 * and is distributed under the 2-Clause BSD licence.
 * Author: Tomasz Sowa <t.sowa@ttmath.org>
 */

/*
 * Copyright (c) 2018, Tomasz Sowa
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 *
 */

#include "model.h"


namespace morm
{

Model::Model()
{
	model_connector = nullptr;
	model_data = nullptr;
	save_mode = DO_INSERT_ON_SAVE;
}

Model::~Model()
{
}

void Model::set_save_mode(SaveMode save_mode)
{
	this->save_mode = save_mode;
}


Model::SaveMode Model::get_save_mode()
{
	return save_mode;
}


void Model::mark_to_delete()
{
	save_mode = DO_DELETE_ON_SAVE;
}

void Model::mark_to_remove()
{
	save_mode = DO_DELETE_ON_SAVE;
}

void Model::mark_to_insert()
{
	save_mode = DO_INSERT_ON_SAVE;
}


void Model::mark_to_update()
{
	save_mode = DO_UPDATE_ON_SAVE;
}


bool Model::was_db_error()
{
	if( model_connector )
	{
		return model_connector->was_db_error();
	}

	return false;
}


std::wstring Model::get_db_error()
{
	if( model_connector )
	{
		return model_connector->get_db_error();
	}

	return std::wstring();
}



void Model::table_name(PT::TextStream & stream)
{
}


void Model::set_connector(ModelConnector & connector)
{
	model_connector = &connector;
}

void Model::set_connector(ModelConnector * connector)
{
	model_connector = connector;
}


bool Model::object_exists()
{
	return save_mode == DO_UPDATE_ON_SAVE;
}


bool Model::found()
{
	return save_mode == DO_UPDATE_ON_SAVE;
}



void Model::to_text(PT::TextStream & stream, bool clear_stream)
{
	if( clear_stream )
	{
		stream.clear();
	}

	if( model_connector )
	{
		model_connector->to_text(stream, *this);
	}
}


void Model::to_text(std::string & str, bool clear_string)
{
	PT::TextStream stream;

	to_text(stream, false);
	stream.to_string(str, clear_string);
}


std::string Model::to_text()
{
	std::string str;

	to_text(str);
	return str;
}


std::string Model::to_string()
{
	return to_text();
}


void Model::generate_insert_query(PT::TextStream & stream)
{
	if( model_connector )
	{
		model_connector->generate_insert_query(stream, *this);
	}
}


bool Model::insert(ModelData * model_data)
{
	bool result = false;

	if( model_connector )
	{
		this->model_data = model_data;

		if( model_connector->insert(*this) )
		{
			save_mode = DO_UPDATE_ON_SAVE; // IMPROVE ME check if there is a primary key
			result = true;
		}

		this->model_data = nullptr;
	}

	return result;
}


bool Model::insert(ModelData & model_data)
{
	return insert(&model_data);
}


void Model::generate_update_query(PT::TextStream & stream)
{
	if( model_connector )
	{
		model_connector->generate_update_query(stream, *this);
	}
}


bool Model::update(ModelData * model_data)
{
	bool result = false;

	if( model_connector )
	{
		this->model_data = model_data;
		result = model_connector->update(*this);
		this->model_data = nullptr;
	}

	return result;
}


bool Model::update(ModelData & model_data)
{
	return update(&model_data);
}


void Model::generate_remove_query(PT::TextStream & stream)
{
	if( model_connector )
	{
		model_connector->generate_remove_query(stream, *this);
	}
}


bool Model::remove(ModelData * model_data)
{
	bool result = false;

	if( model_connector )
	{
		this->model_data = model_data;

		if( model_connector->remove(*this) )
		{
			save_mode = DO_INSERT_ON_SAVE;
			result = true;
		}

		this->model_data = nullptr;
	}

	return result;
}


bool Model::remove(ModelData & model_data)
{
	return remove(&model_data);
}


bool Model::save(ModelData * model_data)
{
	bool result = false;

	switch( save_mode )
	{
	case DO_DELETE_ON_SAVE:
		result = remove(model_data);
		break;

	case DO_INSERT_ON_SAVE:
		result = insert(model_data);
		break;

	case DO_UPDATE_ON_SAVE:
		result = update(model_data);
		break;

	case DO_NOTHING_ON_SAVE:
		// may return true?
		break;
	}

	return result;
}


bool Model::save(ModelData & model_data)
{
	return save(&model_data);
}


void Model::clear()
{
	if( model_connector )
	{
		model_connector->clear_values(*this);
	}

	save_mode = DO_INSERT_ON_SAVE;
	model_data = nullptr;
}


void Model::before_select()
{
}

void Model::before_insert()
{
}

void Model::before_update()
{
}

void Model::before_remove()
{
}

void Model::after_select()
{
}

void Model::after_insert()
{
}

void Model::after_update()
{
}

void Model::after_remove()
{
}

void Model::after_select_failure()
{
}

void Model::after_insert_failure()
{
}

void Model::after_update_failure()
{
}

void Model::after_remove_failure()
{
}

int Model::get_connector_mode()
{
	if( model_connector )
	{
		return model_connector->get_connector_mode();
	}

	return MORM_MODEL_CONNECTOR_MODE_NONE;
}


} // namespace