/*****************************************************************************
 * Copyright (c) 2014-2025 OpenRCT2 developers
 *
 * For a complete list of all authors, please refer to contributors.md
 * Interested in contributing? Visit https://github.com/OpenRCT2/OpenRCT2
 *
 * OpenRCT2 is licensed under the GNU General Public License version 3.
 *****************************************************************************/

#include "ScenerySetRestrictedAction.h"

#include "../object/SceneryGroupObject.h"
#include "../world/Scenery.h"

namespace OpenRCT2::GameActions
{
    ScenerySetRestrictedAction::ScenerySetRestrictedAction(
        ObjectType objectType, ObjectEntryIndex entryIndex, bool isRestricted)
        : _objectType(objectType)
        , _objectIndex(entryIndex)
        , _isRestricted(isRestricted)
    {
    }

    void ScenerySetRestrictedAction::Serialise(DataSerialiser& stream)
    {
        GameAction::Serialise(stream);

        stream << DS_TAG(_objectType) << DS_TAG(_objectIndex) << DS_TAG(_isRestricted);
    }

    uint16_t ScenerySetRestrictedAction::GetActionFlags() const
    {
        return GameAction::GetActionFlags() | Flags::AllowWhilePaused;
    }

    Result ScenerySetRestrictedAction::Query(GameState_t& gameState) const
    {
        if (!ObjectTypeCanBeRestricted(_objectType))
        {
            return Result(Status::InvalidParameters, STR_CANT_RESTRICT_OBJECT, STR_OBJECT_TYPE_CANNOT_BE_RESTRICTED);
        }
        const auto* loadedObject = ObjectEntryGetObject(_objectType, _objectIndex);
        if (loadedObject == nullptr)
        {
            return Result(Status::InvalidParameters, STR_CANT_RESTRICT_OBJECT, STR_OBJECT_NOT_FOUND);
        }

        return Result();
    }

    Result ScenerySetRestrictedAction::Execute(GameState_t& gameState) const
    {
        auto sceneryType = GetSceneryTypeFromObjectType(_objectType);
        SetSceneryItemRestricted({ sceneryType, _objectIndex }, _isRestricted);

        return Result();
    }
} // namespace OpenRCT2::GameActions