/**
 * @file
 *
 * @author  Mamadou Babaei <mamadou@senseglove.com>
 *
 * @section LICENSE
 *
 * (The MIT License)
 *
 * Copyright (c) 2020 - 2024 SenseGlove
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 *
 * @section DESCRIPTION
 *
 *
 */


#include "SGEditor/SGContentBrowserExtension_Skeleton.h"

#include "AssetRegistry/AssetData.h"
#include "ContentBrowserDelegates.h"
#include "ContentBrowserModule.h"
#include "Engine/SkeletalMeshSocket.h"
#include "Framework/MultiBox/MultiBoxBuilder.h"
#include "Framework/MultiBox/MultiBoxExtender.h"
#include "Math/Transform.h"
#include "ReferenceSkeleton.h"

#include "SGEditor/SGPluginStyle.h"
#include "SGLog/SGLog.h"

#define LOCTEXT_NAMESPACE "SGContentBrowserExtension_Skeleton"

void FSGContentBrowserExtension_Skeleton::Register()
{
    FContentBrowserModule& ContentBrowserModule{
        FModuleManager::LoadModuleChecked<FContentBrowserModule>(TEXT("ContentBrowser"))
    };

    TArray<FContentBrowserMenuExtender_SelectedAssets>& ContextMenuDelegates{
        ContentBrowserModule.GetAllAssetViewContextMenuExtenders()
    };

    ContextMenuDelegates.Add(FContentBrowserMenuExtender_SelectedAssets::CreateStatic(&OnExtendAssetSelectionMenu));
}

TSharedRef<FExtender> FSGContentBrowserExtension_Skeleton::OnExtendAssetSelectionMenu(
    const TArray<FAssetData>& SelectedAssets)
{
    TSharedRef<FExtender> Extender{MakeShared<FExtender>()};
    Extender->AddMenuExtension(
        "CommonAssetActions",
        EExtensionHook::After,
        nullptr,
        FMenuExtensionDelegate::CreateStatic(&AssetExtenderCallback, SelectedAssets)
        );
    return MoveTemp(Extender);
}

void FSGContentBrowserExtension_Skeleton::AssetExtenderCallback(
    FMenuBuilder& MenuBuilder, const TArray<FAssetData> SelectedAssets)
{
    MenuBuilder.BeginSection(FName{TEXT("SenseGlove")}, LOCTEXT("SenseGloveHeading", "SenseGlove"));
    {
        for (const FAssetData& Asset: SelectedAssets)
        {
            if (!Asset.IsRedirector()
                && FName{*Asset.AssetClassPath.ToString()} != NAME_Class
                && !(Asset.PackageFlags & PKG_FilterEditorOnly))
            {
                if (Asset.GetClass()->IsChildOf(USkeleton::StaticClass()))
                {
                    MenuBuilder.AddMenuEntry(
                        LOCTEXT("AddSenseGloveSocketsLabel", "Add SenseGlove Sockets"),
                        LOCTEXT(
                            "AddSenseGloveSocketsToolTip", "Adds and sets up the SenseGlove grab and touch sockets"),
                        FSGPluginStyle::GetStyleSetIcon(),
                        FUIAction{
                            FExecuteAction::CreateLambda([SelectedAssets]()-> void
                            {
                                for (const FAssetData& Asset: SelectedAssets)
                                {
                                    USkeleton* Skeleton{Cast<USkeleton>(Asset.GetAsset())};

                                    if (!IsValid(Skeleton))
                                    {
                                        continue;
                                    }

                                    AddSenseGloveSockets(Skeleton);
                                }
                            })
                        },
                        NAME_None,
                        EUserInterfaceActionType::Button,
                        NAME_None);
                }
            }
        }
    }
    MenuBuilder.EndSection();
}

void FSGContentBrowserExtension_Skeleton::AddSenseGloveSockets(USkeleton* Skeleton)
{
    if (!ensureAlwaysMsgf(IsValid(Skeleton), TEXT("Invalid Skeleton!")))
    {
        return;
    }

}

#undef LOCTEXT_NAMESPACE