/***************************************************************************
 *   fheroes2: https://github.com/ihhub/fheroes2                           *
 *   Copyright (C) 2019 - 2025                                             *
 *                                                                         *
 *   Free Heroes2 Engine: http://sourceforge.net/projects/fheroes2         *
 *   Copyright (C) 2009 by Andrey Afletdinov <fheroes2@gmail.com>          *
 *                                                                         *
 *   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 2 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, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/

#include <algorithm>
#include <cassert>
#include <cstddef>
#include <cstdint>
#include <functional>
#include <string>
#include <utility>
#include <vector>

#include "agg_image.h"
#include "army_troop.h"
#include "bin_info.h"
#include "cursor.h"
#include "dialog.h" // IWYU pragma: associated
#include "game_hotkeys.h"
#include "icn.h"
#include "image.h"
#include "kingdom.h"
#include "localevent.h"
#include "math_base.h"
#include "monster.h"
#include "pal.h"
#include "resource.h"
#include "screen.h"
#include "settings.h"
#include "tools.h"
#include "translations.h"
#include "ui_button.h"
#include "ui_constants.h"
#include "ui_dialog.h"
#include "ui_keyboard.h"
#include "ui_text.h"
#include "ui_tool.h"
#include "ui_window.h"
#include "world.h"

namespace
{
    void drawButtonShadow( fheroes2::Image & output, const int buttonIcn, const uint32_t icnIndex, const fheroes2::Point & offset )
    {
        const fheroes2::Sprite & buttonSprite = fheroes2::AGG::GetICN( buttonIcn, icnIndex );

        fheroes2::addGradientShadow( buttonSprite, output, offset, { -5, 5 } );
    }

    void drawCostPerTroopFrame( fheroes2::Image & output, const fheroes2::Point & offset )
    {
        const fheroes2::Sprite & originalBackground = fheroes2::AGG::GetICN( ICN::RECRBKG, 0 );
        const fheroes2::Sprite & recruitWindowTitle = fheroes2::AGG::GetICN( ICN::BLDGXTRA, 0 );

        fheroes2::Image recruitWindow( 132, 67 );
        // Reset the transparent layer.
        recruitWindow.reset();

        // Copy recruit cost title background and borders to make shadows for it and to Blit it to the Recruit dialog.
        fheroes2::Copy( recruitWindowTitle, 3, 58, recruitWindow, 3, 0, 63, 14 );
        fheroes2::Copy( recruitWindowTitle, recruitWindowTitle.width() - 66, 58, recruitWindow, 66, 0, 63, 14 );
        fheroes2::Copy( originalBackground, 138, 54, recruitWindow, 0, 0, 3, 63 );
        fheroes2::Copy( originalBackground, 267, 54, recruitWindow, 129, 0, 3, 63 );
        fheroes2::Copy( originalBackground, 138, 117, recruitWindow, 0, 63, 132, 4 );

        const fheroes2::Text recruitWindowText( _( "Cost per troop:" ), fheroes2::FontType::smallWhite() );
        recruitWindowText.draw( ( recruitWindow.width() - recruitWindowText.width() ) / 2, 3, recruitWindow );

        if ( Settings::Get().isEvilInterfaceEnabled() ) {
            fheroes2::ApplyPalette( recruitWindow, PAL::GetPalette( PAL::PaletteType::GOOD_TO_EVIL_INTERFACE ) );
        }

        fheroes2::Blit( recruitWindow, output, offset.x, offset.y );
        fheroes2::addGradientShadow( recruitWindow, output, { offset.x, offset.y }, { -5, 5 } );
    }

    void RedrawCurrentInfo( const fheroes2::Point & pos, const uint32_t result, const Funds & paymentMonster, const Funds & paymentCosts, const Funds & funds,
                            const std::string & label, const fheroes2::Image & background = {} )
    {
        fheroes2::Display & display = fheroes2::Display::instance();

        fheroes2::Text text( std::to_string( result ), fheroes2::FontType::normalWhite() );

        // Restore the background of the text before rendering it.
        fheroes2::Copy( background, 118, 147, display, pos.x + 118, pos.y + 147, 68, text.height() );

        text.draw( pos.x + 151 - text.width() / 2, pos.y + 147, display );

        std::string sgold = std::to_string( paymentCosts.gold ) + " " + "(" + std::to_string( funds.gold - paymentCosts.gold ) + ")";
        const int rsext = paymentMonster.GetValidItems() & ~Resource::GOLD;

        text.set( std::move( sgold ), fheroes2::FontType::smallWhite() );

        // Restore the background of the text before rendering it.
        fheroes2::Copy( background, 0, 214, display, pos.x, pos.y + 214, background.width(), text.height() );

        if ( rsext ) {
            text.draw( pos.x + 117 - text.width() / 2, pos.y + 214, display );

            text.set( std::to_string( paymentCosts.Get( rsext ) ) + " " + "(" + std::to_string( funds.Get( rsext ) - paymentCosts.Get( rsext ) ) + ")",
                      fheroes2::FontType::smallWhite() );
            text.draw( pos.x + 179 - text.width() / 2, pos.y + 214, display );
        }
        else {
            text.draw( pos.x + 144 - text.width() / 2, pos.y + 214, display );
        }

        // Restore the background of the text before rendering it or to leave it blank if there is no text.
        fheroes2::Copy( background, 0, 166, display, pos.x, pos.y + 166, background.width(), text.height() );

        if ( !label.empty() ) {
            text.set( label, fheroes2::FontType::smallWhite() );
            text.draw( pos.x + 151 - text.width() / 2, pos.y + 166, display );
        }
    }

    void RedrawResourceInfo( const int resourceIcnIndex, const fheroes2::Point & pos, const int32_t value, const int32_t px1, const int32_t py1, const int32_t px2,
                             const int32_t py2, const bool showTotalSum )
    {
        fheroes2::Display & display = fheroes2::Display::instance();

        // In recruit dialog (where the total sum is also shown) the resource info is shifted by 10 pixels to the right.
        const int32_t offsetX = showTotalSum ? 10 : 0;

        const fheroes2::Sprite & sres = fheroes2::AGG::GetICN( ICN::RESOURCE, Resource::getIconIcnIndex( resourceIcnIndex ) );
        fheroes2::Blit( sres, fheroes2::Display::instance(), pos.x + px1 + offsetX, pos.y + py1 );

        const fheroes2::Text text( std::to_string( value ), fheroes2::FontType::smallWhite() );
        text.draw( pos.x + px2 - text.width() / 2 + offsetX, pos.y + py2, display );

        if ( showTotalSum ) {
            fheroes2::Blit( sres, display, pos.x + px1 - 45, pos.y + py1 + 125 );
        }
    }

    void RedrawMonsterInfo( const fheroes2::Rect & pos, const Monster & monster, const uint32_t available, const bool showTotalSum )
    {
        fheroes2::Display & display = fheroes2::Display::instance();
        const Funds paymentMonster = monster.GetCost();
        const bool needExtraResources = 2 == paymentMonster.GetValidItemsCount();

        // Recruit monster text.
        std::string str = _( "Recruit %{name}" );
        StringReplace( str, "%{name}", monster.GetMultiName() );
        fheroes2::Text text( std::move( str ), fheroes2::FontType::normalYellow() );
        fheroes2::Point dst_pt( pos.x + ( pos.width - text.width() ) / 2, pos.y + 11 );
        text.draw( dst_pt.x, dst_pt.y, display );

        // Monster sprite.
        const int monsterId = monster.GetID();
        const Bin_Info::MonsterAnimInfo & monsterInfo = Bin_Info::GetMonsterInfo( monsterId );
        assert( !monsterInfo.animationFrames[Bin_Info::MonsterAnimInfo::STATIC].empty() );

        const fheroes2::Sprite & smon = fheroes2::AGG::GetICN( monster.GetMonsterSprite(), monsterInfo.animationFrames[Bin_Info::MonsterAnimInfo::STATIC][0] );
        dst_pt.x = pos.x + 64 + smon.x() - ( monster.isWide() ? 22 : 0 );
        const int32_t monsterExtraOffsetY = std::max( 0, smon.height() - 96 );
        dst_pt.y = pos.y + 119 - smon.height() + monsterExtraOffsetY;

        if ( monsterId == Monster::CHAMPION ) {
            ++dst_pt.x;
        }

        fheroes2::Blit( smon, display, dst_pt.x, dst_pt.y );

        // Resources needed to buy monster.
        if ( needExtraResources ) {
            RedrawResourceInfo( Resource::GOLD, pos.getPosition(), paymentMonster.gold, 134, 59, 167, 89, showTotalSum );

            if ( paymentMonster.crystal > 0 ) {
                RedrawResourceInfo( Resource::CRYSTAL, pos.getPosition(), paymentMonster.crystal, 206, 53, 224, 89, showTotalSum );
            }
            else if ( paymentMonster.mercury > 0 ) {
                RedrawResourceInfo( Resource::MERCURY, pos.getPosition(), paymentMonster.mercury, 209, 56, 224, 89, showTotalSum );
            }
            else if ( paymentMonster.wood > 0 ) {
                RedrawResourceInfo( Resource::WOOD, pos.getPosition(), paymentMonster.wood, 209, 56, 224, 89, showTotalSum );
            }
            else if ( paymentMonster.ore > 0 ) {
                RedrawResourceInfo( Resource::ORE, pos.getPosition(), paymentMonster.ore, 209, 56, 224, 89, showTotalSum );
            }
            else if ( paymentMonster.sulfur > 0 ) {
                RedrawResourceInfo( Resource::SULFUR, pos.getPosition(), paymentMonster.sulfur, 209, 59, 224, 89, showTotalSum );
            }
            else if ( paymentMonster.gems > 0 ) {
                RedrawResourceInfo( Resource::GEMS, pos.getPosition(), paymentMonster.gems, 209, 59, 224, 89, showTotalSum );
            }
        }
        else {
            // Only gold is needed.
            RedrawResourceInfo( Resource::GOLD, pos.getPosition(), paymentMonster.gold, 159, 59, 189, 89, showTotalSum );
        }

        str = _( "Available: %{count}" );
        StringReplace( str, "%{count}", available );
        text.set( std::move( str ), fheroes2::FontType::smallWhite() );
        text.draw( pos.x + 64 - text.width() / 2, pos.y + 120 + std::max( monsterExtraOffsetY, 2 ), display );

        if ( showTotalSum ) {
            text.set( _( "Number to buy:" ), fheroes2::FontType::smallWhite() );
            text.draw( pos.x + 107 - text.width(), pos.y + 149, display );
        }
    }

    const char * SwitchMaxMinButtons( fheroes2::ButtonBase & btnMax, fheroes2::ButtonBase & btnMin, bool max )
    {
        if ( btnMax.isEnabled() || btnMin.isEnabled() ) {
            if ( max ) {
                btnMax.disable();
                btnMin.enable();

                return _( "Max" );
            }

            btnMin.disable();
            btnMax.enable();

            return _( "Min" );
        }

        return "";
    }

    uint32_t CalculateMax( const Monster & monster, const Kingdom & kingdom, const uint32_t available )
    {
        uint32_t max = 0;
        while ( kingdom.AllowPayment( monster.GetCost() * ( max + 1 ) ) && ( max + 1 ) <= available ) {
            ++max;
        }

        return max;
    }
}

Troop Dialog::RecruitMonster( const Monster & monster0, const uint32_t available, const bool allowDowngradedMonster, const int32_t windowOffsetY )
{
    const bool isEvilInterface = Settings::Get().isEvilInterfaceEnabled();

    fheroes2::Display & display = fheroes2::Display::instance();
    LocalEvent & le = LocalEvent::Get();

    // Set cursor.
    const CursorRestorer cursorRestorer( true, Cursor::POINTER );

    // Calculate max count.
    Monster monster = monster0;
    Funds paymentMonster = monster.GetCost();
    const Kingdom & kingdom = world.GetKingdom( Settings::Get().CurrentColor() );

    uint32_t max = CalculateMax( monster, kingdom, available );
    uint32_t result = max;

    Funds paymentCosts( paymentMonster * result );

    const fheroes2::Size windowSize{ 299, 272 };
    const fheroes2::Point dialogOffset( ( display.width() - windowSize.width ) / 2, ( display.height() - windowSize.height ) / 2 + windowOffsetY );

    const fheroes2::StandardWindow window( dialogOffset.x, dialogOffset.y, windowSize.width, windowSize.height, true, display );

    const fheroes2::Rect roi( window.totalArea() );

    const fheroes2::Rect windowActiveArea( window.activeArea() );

    const fheroes2::Sprite & originalBackground = fheroes2::AGG::GetICN( ICN::RECRBKG, 0 );

    // Render the recruit count background from original recruit dialog ICN.
    fheroes2::Image background( 68, 19 );
    fheroes2::Copy( originalBackground, 134, 159, background, 0, 0, background.width(), background.height() );

    const fheroes2::Rect recruitCountInputArea( dialogOffset.x + 118, dialogOffset.y + 147, background.width(), background.height() );

    if ( isEvilInterface ) {
        fheroes2::ApplyPalette( background, PAL::GetPalette( PAL::PaletteType::GOOD_TO_EVIL_INTERFACE ) );
    }

    fheroes2::Point dst_pt( dialogOffset.x + 118, dialogOffset.y + 143 );
    fheroes2::Copy( background, 0, 0, display, dst_pt.x, dst_pt.y, background.width(), background.height() );
    fheroes2::addGradientShadow( background, display, dst_pt, { -5, 5 } );

    dst_pt.x = dialogOffset.x + 132;
    dst_pt.y = dialogOffset.y + 38;
    drawCostPerTroopFrame( display, dst_pt );

    // Prepare buttons.
    const int32_t backgroundMargin = 18;
    dst_pt.x = dialogOffset.x + backgroundMargin;
    dst_pt.y = dialogOffset.y + 233;

    int buttonId = isEvilInterface ? ICN::BUTTON_SMALL_OKAY_EVIL : ICN::BUTTON_SMALL_OKAY_GOOD;
    fheroes2::Button buttonOk( dst_pt.x, dst_pt.y, buttonId, 0, 1 );
    drawButtonShadow( display, buttonId, 0, dst_pt );

    buttonId = isEvilInterface ? ICN::BUTTON_SMALL_CANCEL_EVIL : ICN::BUTTON_SMALL_CANCEL_GOOD;
    const int32_t buttonCancelWidth = fheroes2::AGG ::GetICN( buttonId, 0 ).width();
    dst_pt.x = dialogOffset.x + windowSize.width - backgroundMargin - buttonCancelWidth;
    fheroes2::Button buttonCancel( dst_pt.x, dst_pt.y, buttonId, 0, 1 );
    drawButtonShadow( display, buttonId, 0, dst_pt );

    buttonId = isEvilInterface ? ICN::BUTTON_SMALL_MAX_EVIL : ICN::BUTTON_SMALL_MAX_GOOD;
    const int32_t buttonMaxWidth = fheroes2::AGG ::GetICN( buttonId, 0 ).width();
    dst_pt.x = dialogOffset.x + 253 - buttonMaxWidth / 2;
    dst_pt.y = dialogOffset.y + 140;

    fheroes2::Button buttonMax( dst_pt.x, dst_pt.y, buttonId, 0, 1 );
    fheroes2::Button buttonMin( dst_pt.x, dst_pt.y, isEvilInterface ? ICN::BUTTON_SMALL_MIN_EVIL : ICN::BUTTON_SMALL_MIN_GOOD, 0, 1 );
    drawButtonShadow( display, buttonId, 0, dst_pt );

    dst_pt.x = dialogOffset.x + 189;
    dst_pt.y = dialogOffset.y + 138;
    fheroes2::Button buttonUp( dst_pt.x, dst_pt.y, ICN::RECRUIT, 0, 1 );
    drawButtonShadow( display, ICN::RECRUIT, 0, dst_pt );

    dst_pt.y = dialogOffset.y + 153;
    fheroes2::Button buttonDn( dst_pt.x, dst_pt.y, ICN::RECRUIT, 2, 3 );
    drawButtonShadow( display, ICN::RECRUIT, 2, dst_pt );

    fheroes2::TimedEventValidator timedButtonUp( [&buttonUp]() { return buttonUp.isPressed(); } );
    fheroes2::TimedEventValidator timedButtonDn( [&buttonDn]() { return buttonDn.isPressed(); } );

    buttonDn.subscribe( &timedButtonDn );
    buttonUp.subscribe( &timedButtonUp );

    const fheroes2::Rect rtWheel( dialogOffset.x + 114, dialogOffset.y + 139, 100, 30 );

    // Create monster switching arrows
    fheroes2::ButtonSprite monsterSwitchLeft;
    fheroes2::ButtonSprite monsterSwitchRight;

    const bool showDowngradedMonsterSwitchButtons = allowDowngradedMonster && ( monster0.GetDowngrade() != monster0 );

    if ( showDowngradedMonsterSwitchButtons ) {
        const fheroes2::Sprite & leftButtonSprite = fheroes2::AGG::GetICN( ICN::MONSTER_SWITCH_LEFT_ARROW, 0 );
        monsterSwitchLeft.setSprite( leftButtonSprite, fheroes2::AGG::GetICN( ICN::MONSTER_SWITCH_LEFT_ARROW, 1 ) );

        dst_pt.x = dialogOffset.x + 6;
        dst_pt.y = dialogOffset.y + 64;
        monsterSwitchLeft.setPosition( dst_pt.x, dst_pt.y );
        fheroes2::addGradientShadow( leftButtonSprite, display, dst_pt, { -5, 5 } );

        const fheroes2::Sprite & rightButtonSprite = fheroes2::AGG::GetICN( ICN::MONSTER_SWITCH_RIGHT_ARROW, 0 );
        monsterSwitchRight.setSprite( rightButtonSprite, fheroes2::AGG::GetICN( ICN::MONSTER_SWITCH_RIGHT_ARROW, 1 ) );
        dst_pt.x = dialogOffset.x + 105;
        monsterSwitchRight.setPosition( dst_pt.x, dst_pt.y );
        fheroes2::addGradientShadow( rightButtonSprite, display, dst_pt, { -5, 5 } );

        // Render Left and Right buttons to restore their initial state later.
        monsterSwitchLeft.draw();
        monsterSwitchRight.draw();
    }
    else {
        monsterSwitchLeft.hide();
        monsterSwitchRight.hide();

        monsterSwitchLeft.disable();
        monsterSwitchRight.disable();
    }

    // Render Up and Down buttons to restore their initial state later.
    buttonUp.draw();
    buttonDn.draw();

    // Make a copy of background dialog to restore its parts before updating some dialog elements.
    background.resize( windowSize.width, windowSize.height );
    fheroes2::Copy( display, dialogOffset.x, dialogOffset.y, background, 0, 0, windowSize.width, windowSize.height );

    RedrawMonsterInfo( windowActiveArea, monster, available, true );

    if ( 0 == result ) {
        buttonOk.disable();
        buttonMax.disable();
        buttonMin.disable();
        buttonMax.draw();
    }

    const Funds & funds = kingdom.GetFunds();
    std::string maxmin = SwitchMaxMinButtons( buttonMax, buttonMin, true );
    RedrawCurrentInfo( dialogOffset, result, paymentMonster, paymentCosts, funds, maxmin );

    buttonOk.draw();
    buttonCancel.draw();

    if ( buttonMax.isEnabled() ) {
        buttonMax.draw();
    }
    if ( buttonMin.isEnabled() ) {
        buttonMin.draw();
    }

    display.render( roi );

    const fheroes2::Rect monsterArea( dialogOffset.x + 24, dialogOffset.y + 19, 75, 95 );

    const auto buttonReleaseRestore = [&display, &background, &dialogOffset]( fheroes2::ButtonBase & button ) {
        if ( button.isReleased() ) {
            return;
        }

        // When the "Up"/"Down" button is pressed it is shifted 1 pixel down so we need to properly restore the background.
        button.release();

        const fheroes2::Rect & buttonRoi = button.area();

        fheroes2::Copy( background, buttonRoi.x - dialogOffset.x, buttonRoi.y - dialogOffset.y, display, buttonRoi.x, buttonRoi.y, buttonRoi.width, buttonRoi.height );
        // The non-pressed button is already on the "background" copy so we do not render the button.
        display.render( buttonRoi );
    };

    std::vector<Monster> upgrades = { monster0 };
    while ( upgrades.back().GetDowngrade() != upgrades.back() ) {
        upgrades.emplace_back( upgrades.back().GetDowngrade() );
    }

    const auto updateCurrentInfo = [&paymentMonster = std::as_const( paymentMonster ), &max = std::as_const( max ), &result = std::as_const( result ), &paymentCosts,
                                    &buttonMax, &buttonMin, &maxmin]() {
        paymentCosts = paymentMonster * result;

        if ( result == max ) {
            maxmin = SwitchMaxMinButtons( buttonMax, buttonMin, true );
        }
        else if ( result == 1 ) {
            maxmin = SwitchMaxMinButtons( buttonMax, buttonMin, false );
        }
        else {
            maxmin.clear();
        }
    };

    while ( le.HandleEvents() ) {
        bool redraw = false;

        if ( buttonOk.isEnabled() ) {
            buttonOk.drawOnState( le.isMouseLeftButtonPressedInArea( buttonOk.area() ) );
        }

        buttonCancel.drawOnState( le.isMouseLeftButtonPressedInArea( buttonCancel.area() ) );

        if ( le.isMouseLeftButtonPressedInArea( buttonUp.area() ) ) {
            buttonUp.drawOnPress();
        }
        else {
            buttonReleaseRestore( buttonUp );
        }

        if ( le.isMouseLeftButtonPressedInArea( buttonDn.area() ) ) {
            buttonDn.drawOnPress();
        }
        else {
            buttonReleaseRestore( buttonDn );
        }

        if ( buttonMax.isEnabled() ) {
            buttonMax.drawOnState( le.isMouseLeftButtonPressedInArea( buttonMax.area() ) );
        }
        if ( buttonMin.isEnabled() ) {
            buttonMin.drawOnState( le.isMouseLeftButtonPressedInArea( buttonMin.area() ) );
        }

        bool updateMonsterInfo = false;

        if ( showDowngradedMonsterSwitchButtons ) {
            if ( le.isMouseLeftButtonPressedInArea( monsterSwitchLeft.area() ) ) {
                monsterSwitchLeft.drawOnPress();
            }
            else {
                buttonReleaseRestore( monsterSwitchLeft );
            }

            if ( le.isMouseLeftButtonPressedInArea( monsterSwitchRight.area() ) ) {
                monsterSwitchRight.drawOnPress();
            }
            else {
                buttonReleaseRestore( monsterSwitchRight );
            }

            if ( le.MouseClickLeft( monsterSwitchLeft.area() ) || le.isKeyPressed( fheroes2::Key::KEY_LEFT ) ) {
                for ( size_t i = 0; i < upgrades.size(); ++i ) {
                    if ( upgrades[i] == monster ) {
                        if ( i < upgrades.size() - 1 ) {
                            monster = upgrades[i + 1];
                        }
                        else {
                            monster = upgrades[0];
                        }
                        break;
                    }
                }
                updateMonsterInfo = true;
            }
            else if ( le.MouseClickLeft( monsterSwitchRight.area() ) || le.isKeyPressed( fheroes2::Key::KEY_RIGHT ) ) {
                for ( size_t i = 0; i < upgrades.size(); ++i ) {
                    if ( upgrades[i] == monster ) {
                        if ( i > 0 ) {
                            monster = upgrades[i - 1];
                        }
                        else {
                            monster = upgrades.back();
                        }
                        break;
                    }
                }
                updateMonsterInfo = true;
            }
        }

        if ( updateMonsterInfo ) {
            // Restore the recruit dialog background.
            fheroes2::Copy( background, 0, 0, display, dialogOffset.x, dialogOffset.y, windowSize.width, windowSize.height );

            max = CalculateMax( monster, kingdom, available );

            if ( max == 0 ) {
                buttonMin.disable();
                buttonMax.disable();
            }
            else if ( !buttonMax.isEnabled() && !buttonMin.isEnabled() ) {
                buttonMin.enable();
            }

            result = max;
            paymentMonster = monster.GetCost();

            updateCurrentInfo();
            RedrawMonsterInfo( windowActiveArea, monster, available, true );

            redraw = true;
        }

        if ( le.isMouseRightButtonPressedInArea( monsterArea ) ) {
            ArmyInfo( Troop( monster, available ), ZERO );

            // Perform a full rendering to properly restore the parts of the screen outside of this dialog
            display.render();
            continue;
        }

        if ( le.MouseClickLeft( monsterArea ) ) {
            ArmyInfo( Troop( monster, available ), BUTTONS );

            // Perform a full rendering to properly restore the parts of the screen outside of this dialog
            display.render();
            continue;
        }

        if ( int32_t temp = static_cast<int32_t>( result ); fheroes2::processIntegerValueTyping( 0, static_cast<int32_t>( max ), temp ) ) {
            result = temp;

            updateCurrentInfo();

            redraw = true;
        }

        if ( le.MouseClickLeft( recruitCountInputArea ) ) {
            int32_t temp = static_cast<int32_t>( result );

            fheroes2::openVirtualNumpad( temp, 0, static_cast<int32_t>( max ) );
            assert( temp >= 0 && temp <= static_cast<int32_t>( max ) );

            result = temp;

            updateCurrentInfo();

            redraw = true;
        }
        else if ( ( le.isMouseWheelUpInArea( rtWheel ) || le.MouseClickLeft( buttonUp.area() ) || le.isKeyPressed( fheroes2::Key::KEY_UP )
                    || timedButtonUp.isDelayPassed() )
                  && result < max ) {
            ++result;

            updateCurrentInfo();

            redraw = true;
        }
        else if ( ( le.isMouseWheelDownInArea( rtWheel ) || le.MouseClickLeft( buttonDn.area() ) || le.isKeyPressed( fheroes2::Key::KEY_DOWN )
                    || timedButtonDn.isDelayPassed() )
                  && result ) {
            --result;

            updateCurrentInfo();

            redraw = true;
        }
        else if ( buttonMax.isEnabled() && le.MouseClickLeft( buttonMax.area() ) && result != max ) {
            result = max;

            updateCurrentInfo();

            redraw = true;
        }
        else if ( buttonMin.isEnabled() && le.MouseClickLeft( buttonMin.area() ) && result != 1 ) {
            result = 1;

            updateCurrentInfo();

            redraw = true;
        }
        else if ( le.isMouseRightButtonPressedInArea( buttonOk.area() ) ) {
            fheroes2::showStandardTextMessage( _( "Okay" ), _( "Recruit selected monsters." ), 0 );
        }
        else if ( le.isMouseRightButtonPressedInArea( buttonCancel.area() ) ) {
            fheroes2::showStandardTextMessage( _( "Cancel" ), _( "Exit this menu without doing anything." ), 0 );
        }
        else if ( buttonMax.isEnabled() && le.isMouseRightButtonPressedInArea( buttonMax.area() ) ) {
            fheroes2::showStandardTextMessage( _( "MAX" ), _( "Select maximum monsters to be recruited." ), 0 );
        }
        else if ( buttonMin.isEnabled() && le.isMouseRightButtonPressedInArea( buttonMin.area() ) ) {
            fheroes2::showStandardTextMessage( _( "MIN" ), _( "Select only 1 monster to be recruited." ), 0 );
        }

        if ( redraw ) {
            RedrawCurrentInfo( dialogOffset, result, paymentMonster, paymentCosts, funds, maxmin, background );

            if ( 0 == result ) {
                buttonOk.disable();
                buttonOk.draw();
            }
            else {
                buttonOk.enable();
                buttonOk.draw();
            }

            buttonCancel.draw();

            if ( buttonMax.isEnabled() || max == 0 ) {
                buttonMax.draw();
            }
            else if ( buttonMin.isEnabled() ) {
                buttonMin.draw();
            }

            monsterSwitchLeft.draw();
            monsterSwitchRight.draw();

            display.render( windowActiveArea );
        }

        if ( buttonOk.isEnabled() && ( le.MouseClickLeft( buttonOk.area() ) || ( Game::HotKeyPressEvent( Game::HotKeyEvent::DEFAULT_OKAY ) ) ) ) {
            break;
        }

        if ( le.MouseClickLeft( buttonCancel.area() ) || ( Game::HotKeyPressEvent( Game::HotKeyEvent::DEFAULT_CANCEL ) ) ) {
            result = 0;
            break;
        }
    }

    display.updateNextRenderRoi( roi );

    return { monster, result };
}

void Dialog::DwellingInfo( const Monster & monster, const uint32_t available )
{
    const fheroes2::Size windowSize{ 289, 141 };

    fheroes2::Display & display = fheroes2::Display::instance();

    // Set cursor.
    const CursorRestorer cursorRestorer( false );

    const fheroes2::Point dialogOffset( ( display.width() - windowSize.width ) / 2,
                                        display.height() / 2 - fheroes2::Display::DEFAULT_HEIGHT / 2 + fheroes2::borderWidthPx );

    const fheroes2::StandardWindow window( dialogOffset.x, dialogOffset.y, windowSize.width, windowSize.height, true, display );

    drawCostPerTroopFrame( display, { dialogOffset.x + 122, dialogOffset.y + 38 } );
    RedrawMonsterInfo( window.activeArea(), monster, available, false );

    display.render( window.totalArea() );

    LocalEvent & le = LocalEvent::Get();

    while ( le.HandleEvents() && le.isMouseRightButtonPressed() ) {
        // Do nothing.
    }
}
