/*
   client/savegamemenu.cc
   This file is part of the Osirion project and is distributed under
   the terms of the GNU General Public License version 2
*/

#include <iomanip>
#include <cstdio>

#include "core/application.h"
#include "core/gameinterface.h"
#include "core/gameserver.h"
#include "client/savegamemenu.h"
#include "filesystem/filesystem.h"
#include "ui/ui.h"
#include "sys/sys.h"

namespace client
{

SaveGameMenu::SaveGameMenu(ui::Widget *parent, const char *label, const Mode mode) : 
	ui::Window(parent)
{
	set_label(label);
	set_border(true);
	set_background(true);
	set_font(ui::root()->font_small());
	
	savegamemenu_mode = mode;
	
	// window title
	savegamemenu_titlelabel = new ui::Label(this);
	savegamemenu_titlelabel->set_label("title");
	savegamemenu_titlelabel->set_background(false);
	savegamemenu_titlelabel->set_border(false);
	savegamemenu_titlelabel->set_font(ui::root()->font_large());
	savegamemenu_titlelabel->set_alignment(ui::AlignCenter);
	if (savegamemenu_mode == Save) {
		savegamemenu_titlelabel->set_text("SAVE GAME");
	} else {
		savegamemenu_titlelabel->set_text("LOAD GAME");
	}
	
	// close button
	savegamemenu_closebutton = new ui::IconButton(savegamemenu_titlelabel, "bitmaps/icons/window_close");
	savegamemenu_closebutton->set_label("close");
	
	// file listview
	savegamemenu_filelistview = new ui::ListView(this);
	savegamemenu_filelistview->set_label("files");
	
	// savegame desription label
	savegamemenu_descrlabel = new ui::Label(this);
	savegamemenu_descrlabel->set_label("description");
	
	// save button
	savegamemenu_savebutton = new ui::Button(this, "Save");
	//savegamemenu_descrinput = new ui::InputBox(this);
	
	// delete button
	savegamemenu_deletebutton = new ui::IconButton(this, "bitmaps/icons/button_delete");
	savegamemenu_deletebutton->set_label("delete");
}

SaveGameMenu::~SaveGameMenu()
{
}

void SaveGameMenu::resize()
{
	const float padding =  ui::root()->font_large()->height();
	const float icon_size = 24.0f; // small icons
	
	// resize title label
	savegamemenu_titlelabel->set_size(width() - padding * 2.0f, savegamemenu_titlelabel->font()->height());	
	savegamemenu_titlelabel->set_location(padding, padding);

	// resize close button
	savegamemenu_closebutton->set_size(savegamemenu_titlelabel->font()->height(), savegamemenu_titlelabel->font()->height());
	savegamemenu_closebutton->set_location(savegamemenu_titlelabel->width() - savegamemenu_closebutton->width(), 0);
	
	// resize files listview
	savegamemenu_filelistview->set_size(
		ui::UI::elementsize.width(), 
		height() - icon_size - padding * 3.0f - savegamemenu_titlelabel->bottom()
	);
	savegamemenu_filelistview->set_location(
		padding, 
		savegamemenu_titlelabel->bottom() + padding
	);
	
	// resize eject button
	savegamemenu_deletebutton->set_size(icon_size, icon_size);
	savegamemenu_deletebutton->set_location(
		savegamemenu_filelistview->right() - icon_size, 
		savegamemenu_filelistview->bottom() + padding
	);
	
	// resize description label
	savegamemenu_descrlabel->set_size(
		width() - savegamemenu_filelistview->right() - 2.0f * padding,
		savegamemenu_filelistview->height() - ui::UI::elementsize.height() - padding
	);
	savegamemenu_descrlabel->set_location(savegamemenu_filelistview->right() + padding, savegamemenu_filelistview->top());
	
	// resize save button
	savegamemenu_savebutton->set_size(ui::UI::elementsize);
	savegamemenu_savebutton->set_location(
		savegamemenu_descrlabel->left() + (savegamemenu_descrlabel->width() - ui::UI::elementsize.width()) * 0.5f,
		savegamemenu_descrlabel->bottom() + padding);
}

void SaveGameMenu::refresh()
{	
	ui::ListItem *listitem = 0;
	
	std::string savegamepath(filesystem::writedir());
	savegamepath.append("savegames");
	
	filesystem::Directory directory(savegamepath);

	savegamemenu_filelistview->clear();
	savegamemenu_deletebutton->disable();
	savegamemenu_savebutton->hide();
	savegamemenu_savebutton->enable();
	
	if (savegamemenu_mode == Save) {
		listitem = new ui::ListItem(savegamemenu_filelistview, "NEW SAVEGAME");
		listitem->set_label("new");
		listitem->set_sortkey("[new]");
		listitem->set_font(ui::root()->font_tiny());
		listitem->set_height(listitem->font()->height() * 2.0f);
		
		savegamemenu_savebutton->set_text("Save");
		if (!core::localcontrol()) {
			savegamemenu_savebutton->disable();
		}
	} else {
		savegamemenu_savebutton->set_text("Load");
	}
	
	for (filesystem::Directory::FileNames::const_iterator it = directory.filenames().begin();
	    	it != directory.filenames().end(); ++it) 
	{
		const std::string filename((*it).c_str());
		if (filename.size() < 5) {
			continue;
		}
		
		std::string ext = filename.substr(filename.size() - 4);
		aux::lowercase(ext);
		if (ext.compare(".ini")) {
			continue;
		}
		
		std::string savename = filename.substr(0, filename.size() - 4);
		
		std::string descr;
		std::string timestamp;
		get_savegame_info(savename, descr, timestamp);
			
		if (descr.size()) {
			std::ostringstream descrstr;
			descrstr << descr << '\n'<< timestamp;
			listitem = new ui::ListItem(savegamemenu_filelistview, descrstr.str().c_str());
			listitem->set_value(savename);
			listitem->set_sortkey(timestamp);
			
			listitem->set_font(ui::root()->font_tiny());
			listitem->set_height(listitem->font()->height() * 3.0f);
		}
	}
	
	savegamemenu_filelistview->sort_reverse();
	savegamemenu_filelistview->event_resize();
	
	show_file_info();
}

void SaveGameMenu::show()
{
	Window::show();
	event_resize();
	refresh();
}

void SaveGameMenu::show_file_info()
{
	savegamemenu_descrlabel->clear();
	savegamemenu_deletebutton->disable();

	if (savegamemenu_filelistview->selected()) {		
		std::string savename(savegamemenu_filelistview->selected()->value().c_str());
		savegamemenu_descrlabel->set_text(savegamemenu_filelistview->selected()->text()  + '\n' + '\n' + savename);		
		if (savename.size()) {
			savegamemenu_deletebutton->enable();
		}		
		savegamemenu_savebutton->show();
	}
}

bool SaveGameMenu::on_keypress(const int key, const unsigned int modifier)
{
	if (key == SDLK_ESCAPE) {
		if (visible()) {
			// hide the parent mainwindow
			parent()->hide();
			return true;
		}
	}

	return false;
}

bool SaveGameMenu::on_emit(ui::Widget *sender, const ui::Widget::Event event, void *data)
{
	if (sender == savegamemenu_closebutton) {
		if (event == ui::Widget::EventButtonClicked) {
			// hide the parent mainwindow
			parent()->hide();
			return true;
		}
	} else if (sender == savegamemenu_savebutton) {
		if (event == ui::Widget::EventButtonClicked) {
			if (savegamemenu_mode == Save) {
				if (savegamemenu_filelistview->selected()) {
					savegame(savegamemenu_filelistview->selected()->value());
					parent()->hide();
				}
			} else {
				if (savegamemenu_filelistview->selected()) {
					loadgame(savegamemenu_filelistview->selected()->value());
					parent()->hide();
				}
			}
			return true;
		}		
	} else if (sender == savegamemenu_deletebutton) {
		if (event == ui::Widget::EventButtonClicked) {
			if (savegamemenu_filelistview->selected()) {
				deletegame(savegamemenu_filelistview->selected()->value());
				refresh();
			}
			return true;
		}		
	} else if (sender == savegamemenu_filelistview) {
		if (event == ui::Widget::EventListViewChanged) {
			show_file_info();
			return true;
		}
	}
	
	return Window::on_emit(sender, event, data);
}

void SaveGameMenu::get_savegame_info(const std::string & savename, std::string &game_descr, std::string &game_timestamp)
{
	filesystem::IniFile inifile("savegames/" + savename);
	
	game_descr.clear();
	game_timestamp.clear();
	
	if (!inifile.is_open()) {
		return;
	}

	while(inifile.getline()) {
		if (inifile.got_key()) {

			if (inifile.in_section("savegame")) {

				if (inifile.got_key_string("description", game_descr)) {
					continue;
					
				} else if (inifile.got_key_string("timestamp", game_timestamp)) {
					continue;
				}
			}
		}
	}
	inifile.close();
}

void SaveGameMenu::deletegame(std::string savename)
{
	std::string filename(filesystem::writedir() + "savegames/" + savename + ".ini");
	
	con_debug << "Deleteing " << filename << std::endl;
	::remove(filename.c_str());
}

void SaveGameMenu::savegame(std::string savename)
{
	if (!core::server() || !core::server()->module())
		return;
	
	if (!core::localcontrol()) {
		return;
	}

	// find an approriate savegame name if none was specified
	if (!savename.size()) {
		std::string filename;
		unsigned long n = 0;
		do {
			n++;
			std::ostringstream numberstr;
			numberstr << std::setfill('0') << std::setw(4) << n;
			
			savename.assign("save");
			savename.append(numberstr.str());
			
			filename.assign(filesystem::writedir() + "savegames/" + savename + ".ini");
		} while (sys::file_exists(filename));
	}
	
	// create savegame diretory if required
	std::string savegamepath(filesystem::writedir());
	savegamepath.append("savegames");
	if (!sys::directory_exists(savegamepath)) {
		sys::mkdir(savegamepath);
	}
	
	filesystem::OFileStream ofs("savegames/" + savename + ".ini");
	
	con_debug << "Saving " << ofs.filename() << std::endl;
	
	ofs << "; Project::OSiRiON" << std::endl;
	ofs << "; savegame data" << std::endl;
	ofs << std::endl;
	
	ofs << "[savegame]" << std::endl;	
	ofs << "description=";
	if (core::localcontrol()->state() == core::Entity::Docked) {
		// FIXME implement entity->dock()
		ofs << core::localplayer()->view()->name();
	} else {
		ofs << core::localplayer()->zone()->name();
	}
	ofs << std::endl;
	
	int year = 0; int month = 0; int day = 0; int hours = 0; int minutes = 0; int seconds = 0; int msec = 0;
	sys::get_localtime(year, month, day, hours, minutes, seconds, msec);
	
	std::ostringstream timestampstr;	
	timestampstr << std::setfill('0') << std::setw(4) << year << "-" << std::setw(2) << month << "-" << std::setw(2) << day << " ";
	timestampstr << std::setw(2) << hours << ":" << std::setw(2) << minutes << ":" << std::setw(2) << seconds;

	ofs << "timestamp=" << timestampstr.str() << std::endl;
	
	ofs << std::endl;
	
	core::server()->module()->game_save(core::localplayer(), ofs);
	
	ofs.close();
}

void SaveGameMenu::loadgame(std::string savename) {
	
	filesystem::IniFile inifile("savegames/" + savename);
	if (!inifile.is_open()) {
		con_warn << "Could not open file " << inifile.filename() << "!" << std::endl;
		return;
	}
	
	con_debug << "Loading " << inifile.filename() << std::endl;
	
	if (core::application()->connected())
		core::application()->disconnect();
	
	core::application()->connect("");
	
	if (core::application()->connected() && core::server() && core::server()->module()) {
		core::server()->module()->game_load(core::localplayer(), inifile);
	}
	
	inifile.close();
}

} // namespace client