PowerToys/src/modules/Workspaces/WorkspacesLib/LaunchingStatus.cpp

#include "pch.h"
#include "LaunchingStatus.h"

#include <common/logger/logger.h>

LaunchingStatus::LaunchingStatus(const WorkspacesData::WorkspacesProject& project, std::function<void(const WorkspacesData::LaunchingAppStateMap&)> updateCallback) :
    m_updateCallback(updateCallback)
{
    std::unique_lock lock(m_mutex);
    for (const auto& app : project.apps)
    {
        m_appsState.insert({ app, { app, nullptr, LaunchingState::Waiting } });
    }
}

const WorkspacesData::LaunchingAppStateMap& LaunchingStatus::Get() noexcept
{
    std::shared_lock lock(m_mutex);
    return m_appsState;
}

bool LaunchingStatus::AllLaunchedAndMoved() noexcept
{
    std::shared_lock lock(m_mutex);
    for (const auto& [app, data] : m_appsState)
    {
        if (data.state != LaunchingState::Failed && data.state != LaunchingState::LaunchedAndMoved)
        {
            return false;
        }
    }

    return true;
}

bool LaunchingStatus::AllLaunched() noexcept
{
    std::shared_lock lock(m_mutex);
    for (const auto& [app, data] : m_appsState)
    {
        if (data.state == LaunchingState::Waiting)
        {
            return false;
        }
    }

    return true;
}

void LaunchingStatus::Update(const WorkspacesData::WorkspacesProject::Application& app, LaunchingState state)
{
    std::unique_lock lock(m_mutex);
    if (!m_appsState.contains(app))
    {
        Logger::error(L"Error updating state: app {} is not tracked in the project", app.name);
        return;
    }

    m_appsState[app].state = state;

    if (m_updateCallback)
    {
        m_updateCallback(m_appsState);
    }
}