/**
 * @file mainmenu.cpp
 *
 * Implementation of functions for interacting with the main menu.
 */

#include "DiabloUI/diabloui.h"
#include "DiabloUI/extrasmenu.h"
#include "DiabloUI/selok.h"
#include "engine/demomode.h"
#include "hwcursor.hpp"
#include "init.h"
#include "movie.h"
#include "options.h"
#include "pfile.h"
#include "storm/storm.h"
#include "utils/language.h"

namespace devilution {

uint32_t gSaveNumber;

namespace {

/** The active music track id for the main menu. */
uint8_t menu_music_track_id = TMUSIC_INTRO;

void RefreshMusic()
{
	music_start(menu_music_track_id);

	if (gbIsSpawn && !gbIsHellfire) {
		return;
	}

	do {
		menu_music_track_id++;
		if (menu_music_track_id == NUM_MUSIC || (!gbIsHellfire && menu_music_track_id > TMUSIC_L4))
			menu_music_track_id = TMUSIC_L2;
		if (gbIsSpawn && menu_music_track_id > TMUSIC_L1)
			menu_music_track_id = TMUSIC_L5;
	} while (menu_music_track_id == TMUSIC_TOWN || menu_music_track_id == TMUSIC_L1);
}

bool InitMenu(_selhero_selections type)
{
	bool success;

	if (type == SELHERO_PREVIOUS)
		return true;

	music_stop();

	success = StartGame(type != SELHERO_CONTINUE, type != SELHERO_CONNECT);
	if (success)
		RefreshMusic();

	return success;
}

bool InitSinglePlayerMenu()
{
	gbIsMultiplayer = false;
	return InitMenu(SELHERO_NEW_DUNGEON);
}

bool InitMultiPlayerMenu()
{
	gbIsMultiplayer = true;
	return InitMenu(SELHERO_CONNECT);
}

void PlayIntro()
{
	music_stop();
	if (gbIsHellfire)
		play_movie("gendata\\Hellfire.smk", true);
	else
		play_movie("gendata\\diablo1.smk", true);
	RefreshMusic();
}

bool DummyGetHeroInfo(_uiheroinfo * /*pInfo*/)
{
	return true;
}

} // namespace

bool mainmenu_select_hero_dialog(GameData *gameData)
{
	uint32_t *pSaveNumberFromOptions = nullptr;
	_selhero_selections dlgresult = SELHERO_NEW_DUNGEON;
	if (demo::IsRunning()) {
		pfile_ui_set_hero_infos(DummyGetHeroInfo);
		gbLoadGame = true;
	} else if (!gbIsMultiplayer) {
		pSaveNumberFromOptions = gbIsHellfire ? &sgOptions.Hellfire.lastSinglePlayerHero : &sgOptions.Diablo.lastSinglePlayerHero;
		gSaveNumber = *pSaveNumberFromOptions;
		UiSelHeroSingDialog(
		    pfile_ui_set_hero_infos,
		    pfile_ui_save_create,
		    pfile_delete_save,
		    pfile_ui_set_class_stats,
		    &dlgresult,
		    &gSaveNumber,
		    &gameData->nDifficulty);

		gbLoadGame = (dlgresult == SELHERO_CONTINUE);
	} else {
		pSaveNumberFromOptions = gbIsHellfire ? &sgOptions.Hellfire.lastMultiplayerHero : &sgOptions.Diablo.lastMultiplayerHero;
		gSaveNumber = *pSaveNumberFromOptions;
		UiSelHeroMultDialog(
		    pfile_ui_set_hero_infos,
		    pfile_ui_save_create,
		    pfile_delete_save,
		    pfile_ui_set_class_stats,
		    &dlgresult,
		    &gSaveNumber);
	}
	if (dlgresult == SELHERO_PREVIOUS) {
		SErrSetLastError(1223);
		return false;
	}

	if (pSaveNumberFromOptions != nullptr)
		*pSaveNumberFromOptions = gSaveNumber;

	pfile_read_player_from_save(gSaveNumber, Players[MyPlayerId]);

	return true;
}

void mainmenu_wait_for_button_sound()
{
	SDL_FillRect(DiabloUiSurface(), nullptr, 0x000000);
	UiFadeIn();
	SDL_Delay(350); // delay to let button pressed sound finish playing
}

void mainmenu_loop()
{
	bool done;
	_mainmenu_selections menu = MAINMENU_NONE;

	RefreshMusic();
	done = false;

	do {
		if (menu == MAINMENU_NONE) {
			if (demo::IsRunning())
				menu = MAINMENU_SINGLE_PLAYER;
			else if (!UiMainMenuDialog(gszProductName, &menu, effects_play_sound, 30))
				app_fatal("%s", _("Unable to display mainmenu"));
		}

		switch (menu) {
		case MAINMENU_NONE:
			break;
		case MAINMENU_SINGLE_PLAYER:
			if (!InitSinglePlayerMenu())
				done = true;
			menu = MAINMENU_NONE;
			break;
		case MAINMENU_MULTIPLAYER:
			if (!InitMultiPlayerMenu())
				done = true;
			menu = MAINMENU_NONE;
			break;
		case MAINMENU_ATTRACT_MODE:
			if (gbIsSpawn && diabdat_mpq == nullptr)
				done = false;
			else if (gbActive)
				PlayIntro();
			menu = MAINMENU_NONE;
			break;
		case MAINMENU_REPLAY_INTRO:
			if (gbIsSpawn && diabdat_mpq == nullptr && hellfire_mpq == nullptr) {
				UiSelOkDialog(nullptr, _(/* TRANSLATORS:  Error Message when a Shareware User clicks on "Replay Intro" in the Main Menu */ "The Diablo introduction cinematic is only available in the full retail version of Diablo. Visit https://www.gog.com/game/diablo to purchase."), true);
			} else if (gbActive) {
				mainmenu_wait_for_button_sound();
				PlayIntro();
			}
			menu = MAINMENU_EXTRAS;
			break;
		case MAINMENU_SHOW_CREDITS:
			UiCreditsDialog();
			menu = MAINMENU_NONE;
			break;
		case MAINMENU_SHOW_SUPPORT:
			UiSupportDialog();
			menu = MAINMENU_EXTRAS;
			break;
		case MAINMENU_EXIT_DIABLO:
			mainmenu_wait_for_button_sound();
			done = true;
			break;
		case MAINMENU_EXTRAS:
			menu = UiExtrasMenu();
			break;
		case MAINMENU_SWITCHGAME:
			gbIsHellfire = !gbIsHellfire;
			sgOptions.Hellfire.startUpGameOption = gbIsHellfire ? StartUpGameOption::Hellfire : StartUpGameOption::Diablo;
			UiInitialize();
			FreeItemGFX();
			InitItemGFX();
			if (IsHardwareCursor())
				SetHardwareCursor(CursorInfo::UnknownCursor());
			RefreshMusic();
			menu = MAINMENU_NONE;
			break;
		case MAINMENU_TOGGLESPAWN:
			gbIsSpawn = !gbIsSpawn;
			UiSetSpawned(gbIsSpawn);
			RefreshMusic();
			menu = MAINMENU_NONE;
			break;
		}
	} while (!done);

	music_stop();
}

} // namespace devilution