// Copyright Epic Games, Inc. All Rights Reserved.

#include "ConversationGraphNode_Task.h"
#include "ConversationTaskNode.h"
#include "ConversationGraphTypes.h"
#include "SGraphEditorActionMenuAI.h"
#include "ToolMenu.h"

#define LOCTEXT_NAMESPACE "ConversationGraph"

UConversationGraphNode_Task::UConversationGraphNode_Task(const FObjectInitializer& ObjectInitializer) : Super(ObjectInitializer)
{
}

void UConversationGraphNode_Task::AllocateDefaultPins()
{
	CreatePin(EGPD_Input, UConversationGraphTypes::PinCategory_MultipleNodes, TEXT("In"));
	CreatePin(EGPD_Output, UConversationGraphTypes::PinCategory_MultipleNodes, TEXT("Out"));
}

FText UConversationGraphNode_Task::GetNodeTitle(ENodeTitleType::Type TitleType) const
{
	if (const UConversationNode* MyNode = Cast<UConversationNode>(NodeInstance))
	{
		return MyNode->GetDisplayNameText();
	}

	return Super::GetNodeTitle(TitleType);
}

FLinearColor UConversationGraphNode_Task::GetNodeBodyTintColor() const
{
	FLinearColor Result = FLinearColor::Red;
	if (NodeInstance)
	{
		if (!GetRuntimeNode<UConversationTaskNode>()->GetNodeBodyColor(/*out*/ Result))
		{
			Result = Super::GetNodeBodyTintColor();
		}
	}
	return Result;
}

void UConversationGraphNode_Task::GetNodeContextMenuActions(UToolMenu* Menu, UGraphNodeContextMenuContext* Context) const
{
	TWeakObjectPtr<UConversationGraphNode> WeakThis(const_cast<UConversationGraphNode_Task*>(this));

	auto CreateAddRequirementSubMenu = [WeakThis](FMenuBuilder& MenuBuilder, UEdGraph* Graph)
	{
		TSharedRef<SGraphEditorActionMenuAI> Menu =
			SNew(SGraphEditorActionMenuAI)
			.GraphObj(Graph)
			.GraphNode(WeakThis.Get())
			.SubNodeFlags((int32)EConversationGraphSubNodeType::Requirement)
			.AutoExpandActionMenu(true);

		MenuBuilder.AddWidget(Menu, FText(), true);
	};

	FToolMenuSection& Section = Menu->FindOrAddSection("ConversationGraphNode_Task");

	Section.AddSubMenu(
		"AddRequirement",
		LOCTEXT("AddRequirement", "Add Requirement..."),
		LOCTEXT("AddRequirementTooltip", "Adds a new requirement as a subnode"),
		FNewMenuDelegate::CreateLambda(CreateAddRequirementSubMenu, (UEdGraph*)Context->Graph));


	auto CreateAddSideEffectSubMenu = [WeakThis](FMenuBuilder& MenuBuilder, UEdGraph* Graph)
	{
		TSharedRef<SGraphEditorActionMenuAI> Menu =
			SNew(SGraphEditorActionMenuAI)
			.GraphObj(Graph)
			.GraphNode(WeakThis.Get())
			.SubNodeFlags((int32)EConversationGraphSubNodeType::SideEffect)
			.AutoExpandActionMenu(true);

		MenuBuilder.AddWidget(Menu, FText(), true);
	};

	Section.AddSubMenu(
		"AddSideEffect",
		LOCTEXT("AddSideEffect", "Add Side Effect..."),
		LOCTEXT("AddSideEffectTooltip", "Adds a new side effect as a subnode"),
		FNewMenuDelegate::CreateLambda(CreateAddSideEffectSubMenu, (UEdGraph*)Context->Graph));


	auto CreateAddChoiceSubMenu = [WeakThis](FMenuBuilder& MenuBuilder, UEdGraph* Graph)
	{
		TSharedRef<SGraphEditorActionMenuAI> Menu =
			SNew(SGraphEditorActionMenuAI)
			.GraphObj(Graph)
			.GraphNode(WeakThis.Get())
			.SubNodeFlags((int32)EConversationGraphSubNodeType::Choice)
			.AutoExpandActionMenu(true);

		MenuBuilder.AddWidget(Menu, FText(), true);
	};

	Section.AddSubMenu(
		"AddChoice",
		LOCTEXT("AddChoice", "Add Choice..."),
		LOCTEXT("AddChoiceTooltip", "Adds a new choice as a subnode"),
		FNewMenuDelegate::CreateLambda(CreateAddChoiceSubMenu, (UEdGraph*)Context->Graph));
}

#undef LOCTEXT_NAMESPACE