/*
 * Copyright (c) Contributors to the Open 3D Engine Project
 * 
 * SPDX-License-Identifier: Apache-2.0 OR MIT
 *
 */
#include "AzToolsFramework_precompiled.h"
#include "PushToSliceCommand.h"

#include <AzCore/Component/TransformBus.h>
#include <AzToolsFramework/Entity/SliceEditorEntityOwnershipServiceBus.h>
#include <AzToolsFramework/Slice/SliceUtilities.h>
namespace AzToolsFramework
{
    PushToSliceCommand::PushToSliceCommand(const AZStd::string& friendlyName)
        : BaseSliceCommand(friendlyName)
    {
    }

    void PushToSliceCommand::Capture(const AZ::Data::Asset<AZ::SliceAsset>& sliceAsset, const AZStd::unordered_map<AZ::EntityId, AZ::EntityId>& addedEntityIdRemaps)
    {
        if (!sliceAsset.Get())
        {
            AZ_Error("PushToSliceCommand::Capture",
                false,
                "Invalid SliceAsset passed in. Unable to capture undo/redo state");

            return;
        }

        // Nothing to do if no remaps
        if (addedEntityIdRemaps.empty())
        {
            AZ_Warning("PushToSliceCommand::Capture",
                false,
                "Emtpty addedEntityIdRemaps passed in. Nothing to undo/redo");

            return;
        }

        AZ::SliceComponent* editorRootSlice = nullptr;
        SliceEditorEntityOwnershipServiceRequestBus::BroadcastResult(editorRootSlice,
            &SliceEditorEntityOwnershipServiceRequestBus::Events::GetEditorRootSlice);

        // Find the parent EntityID highest up in the slice hierarchy
        AZ::EntityId parentId;
        AZ::TransformBus::EventResult(parentId, addedEntityIdRemaps.begin()->first, &AZ::TransformBus::Events::GetParentId);

        while (parentId.IsValid() && !SliceUtilities::IsSliceOrSubsliceRootEntity(parentId))
        {
            AZ::EntityId currentParentId = parentId;
            AZ::TransformBus::EventResult(parentId, currentParentId, &AZ::TransformBus::Events::GetParentId);
        }

        if (!parentId.IsValid())
        {
            AZ_Error("SliceUtilities::SlicePostPushCallback", false, "Unable to find Slice Parent Entity cannot proceed with callback");
            return;
        }

        // Acquire the highest level owning slice of this hierarchy
        AZ::SliceComponent::SliceInstanceAddress targetSliceAddress = editorRootSlice->FindSlice(parentId);

        // Capture undo state for each entity
        for (auto addedIdRemapPair : addedEntityIdRemaps)
        {
            const AZ::EntityId& liveEntityId = addedIdRemapPair.first;
            const AZ::EntityId& assetEntityId = addedIdRemapPair.second;

            // Manufacture restoreInfo for the redo to add the entity back into targetSliceAddress
            AZ::SliceComponent::EntityRestoreInfo redoRestoreInfo(sliceAsset, targetSliceAddress.GetInstance()->GetId(), assetEntityId, AZ::DataPatch::FlagsMap());
            m_entityRedoRestoreInfoArray.emplace_back(liveEntityId, redoRestoreInfo);

            bool restoreCaptured = BaseSliceCommand::CaptureRestoreInfoForUndo(liveEntityId);

            if (!restoreCaptured)
            {
                m_entityRedoRestoreInfoArray.clear();
                m_entityUndoRestoreInfoArray.clear();

                return;
            }
        }
    }

    void PushToSliceCommand::Undo()
    {
        BaseSliceCommand::RestoreEntities(m_entityUndoRestoreInfoArray);
    }

    void PushToSliceCommand::Redo()
    {
        BaseSliceCommand::RestoreEntities(m_entityRedoRestoreInfoArray, false, AzToolsFramework::SliceEntityRestoreType::Added);
    }
}