// Copyright (c) Samuel Kahn (samuel@kahncode.com). All Rights Reserved.

#include "K2Node_StructToBlob.h"

#include "Blob.h"
#include "BlobFunctionLibrary.h"
#include "BlueprintActionDatabaseRegistrar.h"
#include "BlueprintFieldNodeSpawner.h"
#include "EdGraphSchema_K2.h"
#include "EdGraphUtilities.h"
#include "EditorCategoryUtils.h"
#include "K2Node_CallFunction.h"
#include "KismetCompiler.h"

#define LOCTEXT_NAMESPACE "K2Node_StructToBlob"

//////////////////////////////////////////////////////////////////////////
// UK2Node_StructToBlob

void UK2Node_StructToBlob::AllocateDefaultPins()
{
	Super::AllocateDefaultPins();

	UEdGraphNode::FCreatePinParams PinParams;
	PinParams.bIsConst = true;
	PinParams.bIsReference = true;
	CreatePin(EGPD_Input, UEdGraphSchema_K2::PC_Wildcard, FName(TEXT("Struct")), PinParams);

	CreatePin(EGPD_Output, UEdGraphSchema_K2::PC_Struct, FBlob::StaticStruct(), FName(TEXT("Blob")));

	AllocateBehaviorPin();
	AllocateNetContextPin();
}

FText UK2Node_StructToBlob::GetNodeTitle(ENodeTitleType::Type TitleType) const
{
	if (CachedNodeTitle.IsOutOfDate(this))
	{
		CachedNodeTitle.SetCachedText(LOCTEXT("StructToBlob_Title", "Struct To Blob"), this);
	}
	return CachedNodeTitle;
}

FText UK2Node_StructToBlob::GetTooltipText() const
{
	if (CachedTooltip.IsOutOfDate(this))
	{
		CachedTooltip.SetCachedText(
			LOCTEXT("StructToBlob_Tooltip",
				"Node that serializes a struct into a binary blob, make sure to serialize it back to the same type of script at the other end"),
			this);
	}
	return CachedTooltip;
}

FSlateIcon UK2Node_StructToBlob::GetIconAndTint(FLinearColor& OutColor) const
{
	static FSlateIcon Icon("EditorStyle", "GraphEditor.MakeStruct_16x");
	return Icon;
}

void UK2Node_StructToBlob::ExpandNode(class FKismetCompilerContext& CompilerContext, UEdGraph* SourceGraph)
{
	// This is where the node is implemented, calls UBinaryBlobFunctionLibrary::StructToBlob

	Super::ExpandNode(CompilerContext, SourceGraph);

	if (StructType)
	{
		UEdGraphPin* OutBlobPin = GetBlobPin();
		UEdGraphPin* InStructPin = GetStructPin();

		UK2Node_CallFunction* PinCallFunction = CompilerContext.SpawnIntermediateNode<UK2Node_CallFunction>(this, SourceGraph);
		const UFunction* Function = GetStructToBlobFunction();
		PinCallFunction->SetFromFunction(Function);
		PinCallFunction->AllocateDefaultPins();

		UEdGraphPin* StructPin = PinCallFunction->FindPinChecked(TEXT("InStruct"));
		// Set the type of the OutRow pin on this expanded mode to match original
		StructPin->PinType = InStructPin->PinType;
		StructPin->PinType.PinSubCategoryObject = InStructPin->PinType.PinSubCategoryObject;

		CompilerContext.MovePinLinksToIntermediate(*InStructPin, *StructPin);

		UEdGraphPin* BlobPin = PinCallFunction->FindPinChecked(TEXT("OutBlob"));
		CompilerContext.MovePinLinksToIntermediate(*OutBlobPin, *BlobPin); // struct pin out is not recognized properly

		UEdGraphPin* BehaviorPin = PinCallFunction->FindPinChecked(TEXT("Behavior"));
		UEdGraphPin* NetContextPin = PinCallFunction->FindPinChecked(TEXT("NetContext"));
		CompilerContext.MovePinLinksToIntermediate(*GetBehaviorPin(), *BehaviorPin);
		CompilerContext.MovePinLinksToIntermediate(*GetNetContextPin(), *NetContextPin);

		// Move execution pin links
		CompilerContext.MovePinLinksToIntermediate(*GetThenPin(), *(PinCallFunction->GetThenPin()));
		CompilerContext.MovePinLinksToIntermediate(*GetExecPin(), *(PinCallFunction->GetExecPin()));
	}

	// Break any links to the expanded node
	BreakAllNodeLinks();
}

const UFunction* UK2Node_StructToBlob::GetStructToBlobFunction() const
{
	return UBlobFunctionLibrary::StaticClass()->FindFunctionByName(GET_FUNCTION_NAME_CHECKED(UBlobFunctionLibrary, StructToBlob));
}

#undef LOCTEXT_NAMESPACE