/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/
#include "stdafx.h"

#include "EditorCommon.h"
#include "AssetTreeEntry.h"

namespace SliceMenuHelpers
{

    void CreateMenuActionsAndSubMenus(
        AssetTreeEntry* sliceAssetTree,
        HierarchyWidget* hierarchy,
        QTreeWidgetItemRawPtrQList& selectedItems,
        QMenu* parentMenu,
        bool addAtRoot,
        const AZ::Vector2& viewportPosition)
    {
        // Add the file entries at this level alphabetically
        for (auto fileEntry : sliceAssetTree->m_files)
        {
            QAction* action = new QAction(fileEntry.first.c_str(), parentMenu);
            parentMenu->addAction(action);

            // get the pathname relative to the asset root
            const AZ::Data::AssetId& assetId = fileEntry.second;

            // Connect the action to a lamba that instantiates the slice
            QObject::connect(action,
                &QAction::triggered,
                hierarchy,
                [assetId, hierarchy, addAtRoot, viewportPosition](bool checked)
                {
                    if (addAtRoot)
                    {
                        hierarchy->clearSelection();
                    }

                    hierarchy->GetEditorWindow()->GetSliceManager()->InstantiateSlice(assetId, viewportPosition);
                });
        }

        // Add the sub-folder entries at this level alphabetically
        for (auto folderEntry : sliceAssetTree->m_folders)
        {
            QMenu* subMenu = parentMenu->addMenu(folderEntry.first.c_str());
            CreateMenuActionsAndSubMenus(folderEntry.second,
                hierarchy, selectedItems, subMenu, addAtRoot, viewportPosition);
        }
    }

    void CreateInstantiateSliceMenu(HierarchyWidget* hierarchy,
        QTreeWidgetItemRawPtrQList& selectedItems,
        QMenu* parent,
        bool addAtRoot,
        const AZ::Vector2& viewportPosition)
    {
        AssetTreeEntry* sliceAssetTree = hierarchy->GetEditorWindow()->GetSliceLibraryTree();

        QMenu* sliceLibraryMenu = parent->addMenu(QString("Element%1 from Slice &Library").arg(!addAtRoot && selectedItems.size() > 1 ? "s" : ""));

        CreateMenuActionsAndSubMenus(sliceAssetTree,
            hierarchy, selectedItems, sliceLibraryMenu, addAtRoot, viewportPosition);
    }

}   // namespace SliceMenuHelpers