/* ScummVM - Graphic Adventure Engine
 *
 * ScummVM is the legal property of its developers, whose names
 * are too numerous to list here. Please refer to the COPYRIGHT
 * file distributed with this source distribution.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

#include "ultima/ultima1/u1dialogs/king.h"
#include "ultima/ultima1/game.h"
#include "ultima/ultima1/core/resources.h"
#include "ultima/ultima1/maps/map.h"
#include "ultima/shared/gfx/visual_surface.h"

namespace Ultima {
namespace Ultima1 {
namespace U1Dialogs {

BEGIN_MESSAGE_MAP(King, Dialog)
	ON_MESSAGE(ShowMsg)
	ON_MESSAGE(CharacterInputMsg)
	ON_MESSAGE(TextInputMsg)
END_MESSAGE_MAP()

King::King(Ultima1Game *game, uint kingIndex) : Dialog(game), _kingIndex(kingIndex), _mode(SELECT) {
	_bounds = Rect(31, 23, 287, 127);
}

bool King::ShowMsg(CShowMsg *msg) {
	addInfoMsg(_game->_res->KING_TEXT[0], false);
	getKeypress();
	return true;
}

void King::draw() {
	Dialog::draw();

	if (_mode != SERVICE)
		return;

	Shared::Gfx::VisualSurface s = getSurface();

	// Draw the background and frame
	s.clear();
	s.frameRect(Rect(3, 3, _bounds.width() - 3, _bounds.height() - 3), getGame()->_borderColor);

	if (_kingIndex % 1) {
		// Kill a monster
		centerText(_game->_res->KING_TEXT[8], 2);

		Common::String name;
		switch ((_kingIndex - 1) / 2) {
		case 0:
			// Gelatinous Cube
			name = _game->_res->DUNGEON_MONSTER_NAMES[9];
			break;
		case 1:
			// Carrion Creeper
			name = _game->_res->DUNGEON_MONSTER_NAMES[14];
			break;
		case 2:
			// Lich
			name = _game->_res->DUNGEON_MONSTER_NAMES[18];
			break;
		case 3:
			// Balron
			name = _game->_res->DUNGEON_MONSTER_NAMES[23];
			break;
		default:
			break;
		}

		centerText(name, 4);
	} else {
		// Find a location
		centerText(_game->_res->KING_TEXT[9], 2);

		switch (_kingIndex / 2) {
		case 0:
			centerText(_game->_res->LOCATION_NAMES[47], 4);
			break;
		case 1:
			centerText(_game->_res->LOCATION_NAMES[45], 4);
			break;
		case 2:
			centerText(_game->_res->LOCATION_NAMES[43], 4);
			break;
		case 3:
			centerText(_game->_res->LOCATION_NAMES[41], 4);
			break;
		}
	}

	centerText(_game->_res->KING_TEXT[10], 6);
	centerText(_game->_res->KING_TEXT[11], 7);
}

void King::setMode(KingMode mode) {
	_mode = mode;

	switch (_mode) {
	case PENCE:
		addInfoMsg(_game->_res->KING_TEXT[2]);			// Pence
		addInfoMsg(_game->_res->KING_TEXT[4], false);	// How much?
		getInput();
		break;

	case SERVICE:
		addInfoMsg(_game->_res->KING_TEXT[3]);

		if (_game->_quests[_kingIndex].isInProgress()) {
			alreadyOnQuest();
			return;
		} else {
			_game->_quests[_kingIndex].start();
			addInfoMsg(_game->_res->PRESS_SPACE_TO_CONTINUE, false);
			getKeypress();
		}
		break;

	default:
		break;
	}

	setDirty();
}

bool King::CharacterInputMsg(CCharacterInputMsg *msg) {
	switch (_mode) {
	case SELECT:
		if (msg->_keyState.keycode == Common::KEYCODE_p)
			setMode(PENCE);
		else if (msg->_keyState.keycode == Common::KEYCODE_s)
			setMode(SERVICE);
		else
			neither();
		break;

	case SERVICE:
		addInfoMsg("");
		_game->endOfTurn();
		hide();
		break;

	default:
		break;
	}

	return true;
}

bool King::TextInputMsg(CTextInputMsg *msg) {
	assert(_mode == PENCE);
	const Shared::Character &c = *_game->_party;
	uint amount = atoi(msg->_text.c_str());

	if (msg->_escaped || !amount) {
		none();
	} else if (amount > c._coins) {
		notThatMuch();
	} else {
		addInfoMsg(Common::String::format("%u", amount));
		giveHitPoints(amount * 3 / 2);
	}

	return true;
}

void King::neither() {
	addInfoMsg(_game->_res->KING_TEXT[1]);
	_game->endOfTurn();
	hide();
}

void King::none() {
	addInfoMsg(_game->_res->NONE);
	_game->endOfTurn();
	hide();
}

void King::notThatMuch() {
	addInfoMsg(_game->_res->KING_TEXT[5]);
	_game->endOfTurn();
	hide();
}

void King::alreadyOnQuest() {
	addInfoMsg(_game->_res->KING_TEXT[7]);
	_game->endOfTurn();
	hide();
}

void King::giveHitPoints(uint amount) {
	Shared::Character &c = *_game->_party;
	assert(amount <= c._coins);
	c._coins -= amount;
	c._hitPoints += amount;

	addInfoMsg(Common::String::format(_game->_res->KING_TEXT[6], amount));
	_game->endOfTurn();
	hide();
}

} // End of namespace U1Dialogs
} // End of namespace Gfx
} // End of namespace Ultima