ES-DE/es-app/src/guis/GuiScraperMulti.cpp

//  SPDX-License-Identifier: MIT
//
//  EmulationStation Desktop Edition
//  GuiScraperMulti.cpp
//
//  Multiple game scraping user interface.
//  Shows the progress for the scraping as it's running.
//  This interface is triggered from GuiScraperMenu.
//  GuiScraperSearch is called from here.
//

#include "guis/GuiScraperMulti.h"

#include "components/ButtonComponent.h"
#include "components/MenuComponent.h"
#include "components/TextComponent.h"
#include "guis/GuiMsgBox.h"
#include "guis/GuiScraperSearch.h"
#include "views/ViewController.h"
#include "Gamelist.h"
#include "MameNames.h"
#include "SystemData.h"
#include "Window.h"

GuiScraperMulti::GuiScraperMulti(
        Window* window,
        const std::queue<ScraperSearchParams>& searches,
        bool approveResults)
        : GuiComponent(window),
        mBackground(window, ":/graphics/frame.svg"),
        mGrid(window, Vector2i(1, 5)),
        mSearchQueue(searches),
        mApproveResults(approveResults)
{
    assert(mSearchQueue.size());

    addChild(&mBackground);
    addChild(&mGrid);

    mIsProcessing = true;

    mTotalGames = static_cast<int>(mSearchQueue.size());
    mCurrentGame = 0;
    mTotalSuccessful = 0;
    mTotalSkipped = 0;

    // Set up grid.
    mTitle = std::make_shared<TextComponent>(mWindow, "SCRAPING IN PROGRESS",
            Font::get(FONT_SIZE_LARGE), 0x555555FF, ALIGN_CENTER);
    mGrid.setEntry(mTitle, Vector2i(0, 0), false, true);

    mSystem = std::make_shared<TextComponent>(mWindow, "SYSTEM",
            Font::get(FONT_SIZE_MEDIUM), 0x777777FF, ALIGN_CENTER);
    mGrid.setEntry(mSystem, Vector2i(0, 1), false, true);

    mSubtitle = std::make_shared<TextComponent>(mWindow, "subtitle text",
            Font::get(FONT_SIZE_SMALL), 0x888888FF, ALIGN_CENTER);
    mGrid.setEntry(mSubtitle, Vector2i(0, 2), false, true);

    if (mApproveResults && !Settings::getInstance()->getBool("ScraperSemiautomatic"))
        mSearchComp = std::make_shared<GuiScraperSearch>(mWindow,
                GuiScraperSearch::NEVER_AUTO_ACCEPT, mTotalGames);
    else if (mApproveResults && Settings::getInstance()->getBool("ScraperSemiautomatic"))
        mSearchComp = std::make_shared<GuiScraperSearch>(mWindow,
                GuiScraperSearch::ACCEPT_SINGLE_MATCHES, mTotalGames);
    else if (!mApproveResults)
        mSearchComp = std::make_shared<GuiScraperSearch>(mWindow,
                GuiScraperSearch::ALWAYS_ACCEPT_FIRST_RESULT, mTotalGames);
    mSearchComp->setAcceptCallback(std::bind(&GuiScraperMulti::acceptResult,
            this, std::placeholders::_1));
    mSearchComp->setSkipCallback(std::bind(&GuiScraperMulti::skip, this));
    mSearchComp->setCancelCallback(std::bind(&GuiScraperMulti::finish, this));
    mGrid.setEntry(mSearchComp, Vector2i(0, 3), mSearchComp->getSearchType() !=
            GuiScraperSearch::ALWAYS_ACCEPT_FIRST_RESULT, true);

    std::vector< std::shared_ptr<ButtonComponent> > buttons;

    if (mApproveResults) {
        buttons.push_back(std::make_shared<ButtonComponent>(mWindow, "REFINE SEARCH",
                "refine search", [&] {
            mSearchComp->openInputScreen(mSearchQueue.front());
            mGrid.resetCursor();
        }));

        buttons.push_back(std::make_shared<ButtonComponent>(mWindow, "SKIP", "skip game", [&] {
            skip();
            mGrid.resetCursor();
        }));
    }

    buttons.push_back(std::make_shared<ButtonComponent>(mWindow, "STOP",
            "stop (progress saved)", std::bind(&GuiScraperMulti::finish, this)));

    mButtonGrid = makeButtonGrid(mWindow, buttons);
    mGrid.setEntry(mButtonGrid, Vector2i(0, 4), true, false);

    setSize(Renderer::getScreenWidth() * 0.95f, Renderer::getScreenHeight() * 0.849f);
    setPosition((Renderer::getScreenWidth() - mSize.x()) / 2, (Renderer::getScreenHeight() -
            mSize.y()) / 2);

    doNextSearch();
}

GuiScraperMulti::~GuiScraperMulti()
{
    if (mTotalSuccessful > 0) {
        // Sort all systems to possibly update their view style from Basic to Detailed or Video.
        for (auto it = SystemData::sSystemVector.cbegin();
                it !=SystemData::sSystemVector.cend(); it++) {
            (*it)->sortSystem();
        }
    }
    ViewController::get()->onPauseVideo();
}

void GuiScraperMulti::onSizeChanged()
{
    mBackground.fitTo(mSize, Vector3f::Zero(), Vector2f(-32, -32));

    mGrid.setRowHeightPerc(0, mTitle->getFont()->getLetterHeight() * 1.9725f / mSize.y(), false);
    mGrid.setRowHeightPerc(1, (mSystem->getFont()->getLetterHeight() + 2) / mSize.y(), false);
    mGrid.setRowHeightPerc(2, mSubtitle->getFont()->getHeight() * 1.75f / mSize.y(), false);
    mGrid.setRowHeightPerc(4, mButtonGrid->getSize().y() / mSize.y(), false);
    mGrid.setSize(mSize);
}

void GuiScraperMulti::doNextSearch()
{
    if (mSearchQueue.empty()) {
        finish();
        return;
    }

    // Update title.
    std::stringstream ss;
    mSystem->setText(Utils::String::toUpper(mSearchQueue.front().system->getFullName()));

    std::string scrapeName;

    if (Settings::getInstance()->getBool("ScraperSearchMetadataName")) {
        scrapeName = mSearchQueue.front().game->getName();
    }
    else {
        if (mSearchQueue.front().game->isArcadeGame() &&
                Settings::getInstance()->getString("Scraper") == "thegamesdb")
            scrapeName = Utils::FileSystem::getFileName(mSearchQueue.front().game->getPath()) +
                    " (" + MameNames::getInstance()->getCleanName(mSearchQueue.front().game->
                    getCleanName()) + ")";
        else
            scrapeName = Utils::FileSystem::getFileName(mSearchQueue.front().game->getPath());
    }

    // Extract possible subfolders from the path.
    std::string folderPath = Utils::String::replace(
            Utils::FileSystem::getParent(mSearchQueue.front().game->getPath()),
            mSearchQueue.front().system->getSystemEnvData()->mStartPath, "");

    if (folderPath.size() >= 2) {
        folderPath.erase(0, 1);
        #if defined(_WIN64)
        folderPath.push_back('\\');
        folderPath = Utils::String::replace(folderPath, "/", "\\");
        #else
        folderPath.push_back('/');
        #endif
    }

    // Update subtitle.
    ss.str("");
    ss << "GAME " << (mCurrentGame + 1) << " OF " << mTotalGames << " - " << folderPath <<
            scrapeName << ((mSearchQueue.front().game->getType() == FOLDER) ? "  " +
            ViewController::FOLDER_CHAR : "");
    mSubtitle->setText(ss.str());

    mSearchComp->search(mSearchQueue.front());
}

void GuiScraperMulti::acceptResult(const ScraperSearchResult& result)
{
    ScraperSearchParams& search = mSearchQueue.front();

    GuiScraperSearch::saveMetadata(result, search.game->metadata, search.game);

    updateGamelist(search.system);

    mSearchQueue.pop();
    mCurrentGame++;
    mTotalSuccessful++;
    doNextSearch();
}

void GuiScraperMulti::skip()
{
    mSearchQueue.pop();
    mCurrentGame++;
    mTotalSkipped++;
    mSearchComp->unsetRefinedSearch();
    doNextSearch();
}

void GuiScraperMulti::finish()
{
    std::stringstream ss;
    if (mTotalSuccessful == 0) {
        ss << "NO GAMES WERE SCRAPED";
    }
    else {
        ss << mTotalSuccessful << " GAME" <<
                ((mTotalSuccessful > 1) ? "S" : "") << " SUCCESSFULLY SCRAPED";

        if (mTotalSkipped > 0)
            ss << "\n" << mTotalSkipped << " GAME"
                    << ((mTotalSkipped > 1) ? "S" : "") << " SKIPPED";
    }

    mWindow->pushGui(new GuiMsgBox(mWindow, getHelpStyle(), ss.str(), "OK", [&] {
        mIsProcessing = false;
        delete this;
    }));
}

std::vector<HelpPrompt> GuiScraperMulti::getHelpPrompts()
{
    std::vector<HelpPrompt> prompts = mGrid.getHelpPrompts();
    // Remove the 'Choose' entry if in fully automatic mode.
    if (!mApproveResults)
        prompts.pop_back();
    return prompts;
}

HelpStyle GuiScraperMulti::getHelpStyle()
{
    HelpStyle style = HelpStyle();
    style.applyTheme(ViewController::get()->getState().getSystem()->getTheme(), "system");
    return style;
}