#pragma once

#include "Object.h"
#include "NameTypes.h"

#include "reboot.h"
#include "DataTable.h"

enum class ECurveTableMode : unsigned char
{
	Empty,
	SimpleCurves,
	RichCurves
};

struct FSimpleCurveKey
{
	float                                              Time;                                                     // 0x0000(0x0004) (Edit, ZeroConstructor, IsPlainOldData, NoDestructor, HasGetValueTypeHash, NativeAccessSpecifierPublic)
	float                                              Value;                                                    // 0x0004(0x0004) (Edit, ZeroConstructor, IsPlainOldData, NoDestructor, HasGetValueTypeHash, NativeAccessSpecifierPublic)
};

struct FIndexedCurve
{

};

struct FRealCurve : public FIndexedCurve
{
};

struct FSimpleCurve : public FRealCurve
{
	TArray<FSimpleCurveKey>& GetKeys()
	{
		static auto KeysOffset = FindOffsetStruct("/Script/Engine.SimpleCurve", "Keys");
		return *(TArray<FSimpleCurveKey>*)(__int64(this) + KeysOffset);
	}
};

class UCurveTable : public UObject
{
public:
	static int GetCurveTableSize()
	{
		static auto CurveTableClass = FindObject<UClass>(L"/Script/Engine.CurveTable");
		return CurveTableClass->GetPropertiesSize();
	}

	ECurveTableMode GetCurveTableMode() const
	{
		static auto CurveTableModeOffset = GetCurveTableSize() - 8;
		return *(ECurveTableMode*)(__int64(this) + CurveTableModeOffset);
	}

	void* GetKey(const FName& RowName, int Index)
	{
		auto CurveTableMode = GetCurveTableMode();

		// LOG_INFO(LogDev, "RowName {} CurveTableMode {} Size {}", RowName.ComparisonIndex.Value ? RowName.ToString() : "InvalidComparision", (int)CurveTableMode, GetCurveTableSize());

		if (CurveTableMode == ECurveTableMode::SimpleCurves)
		{
			auto& RowMap = ((UDataTable*)this)->GetRowMap<FSimpleCurve>(); // its the same offset so
			auto CurvePtr = RowMap.Find(RowName);

			if (!CurvePtr)
			{
				LOG_WARN(LogDev, "Failed to get curve!");
				return nullptr;
			}

			auto Curve = *CurvePtr;
			auto& Keys = Curve->GetKeys();
			return Keys.Num() > Index ? &Curve->GetKeys().at(Index) : nullptr;
		}
		else if (CurveTableMode == ECurveTableMode::RichCurves)
		{
			LOG_INFO(LogDev, "RICHCURVE UNIMPLEMENTED!");
		}

		return nullptr;
	}

	float GetValueOfKey(void* Key)
	{
		if (!Key)
			return 0.f;

		auto CurveTableMode = GetCurveTableMode();

		if (CurveTableMode == ECurveTableMode::SimpleCurves) return ((FSimpleCurveKey*)Key)->Value;
		else if (CurveTableMode == ECurveTableMode::RichCurves) return 0.f;

		return 0.f;
	}
};

struct FCurveTableRowHandle
{
	UCurveTable* CurveTable;
	FName RowName; 
};