//  
// Copyright (c) 2023 articy Software GmbH & Co. KG. All rights reserved.  
//

#pragma once

#include "CoreMinimal.h"
#include "ArticyObject.h"
#include "UObject/UObjectHash.h"
#include "ArticyPackage.generated.h"


UCLASS(BlueprintType)
class ARTICYRUNTIME_API UArticyPackage : public UDataAsset
{
	GENERATED_BODY()

protected:

	UPROPERTY(BlueprintReadOnly, VisibleAnywhere, Category = "Articy")
	TArray<UArticyObject*> Assets;

	UPROPERTY(BlueprintReadOnly, VisibleAnywhere, Category = "Articy")
	TMap<FName, TSoftObjectPtr<UArticyObject>> AssetsByTechnicalName;

	UPROPERTY(BlueprintReadOnly, VisibleAnywhere, Category = "Articy")
	TMap<FArticyId, TSoftObjectPtr<UArticyObject>> AssetsById;
public: 

	virtual FPrimaryAssetId GetPrimaryAssetId() const override { return FPrimaryAssetId(FName(TEXT("ArticyPackage")), GetFName()); }

	void AddAsset(UArticyObject* ArticyObject);

	UFUNCTION()
	const int AssetNum()const;

	UFUNCTION()
	void Clear();

	UFUNCTION()
	const TArray<UArticyObject*> GetAssets();

	UFUNCTION()
	const TMap<FName, TSoftObjectPtr<UArticyObject>> GetAssetsDict();

	UFUNCTION()
	UArticyObject* GetAssetById(const FArticyId& Id) const;

	UFUNCTION()
	UArticyObject* GetAssetByTechnicalName(const FName& TechnicalName) const;

	const bool IsAssetContained(FName TechnicalName) const;

	const bool IsAssetContained(const FArticyId& Id) const;

	UPROPERTY(VisibleAnywhere, BlueprintReadOnly, Category = "Package")
	FString Name;
	UPROPERTY(VisibleAnywhere, BlueprintReadOnly, Category = "Package")
	FString Description;
	UPROPERTY(VisibleAnywhere, BlueprintReadOnly, Category = "Package")
	bool bIsDefaultPackage = false;

private:
	// used to determine which objects are still parented to this package, which may include outdated articy objects that have to be deleted
	TArray<UObject*> GetInnerObjects() const;
};

inline const int UArticyPackage::AssetNum()const
{
	return Assets.Num();
}

inline void UArticyPackage::Clear()
{
	Assets.Empty();
	AssetsById.Empty();
	AssetsByTechnicalName.Empty();
}

inline const TArray<UArticyObject*> UArticyPackage::GetAssets()
{
	return Assets;
}

inline const TMap<FName, TSoftObjectPtr<UArticyObject>> UArticyPackage::GetAssetsDict()
{
	return AssetsByTechnicalName;
}

inline const bool UArticyPackage::IsAssetContained(FName TechnicalName) const
{	
	return AssetsByTechnicalName.Contains(TechnicalName);
}

inline const bool UArticyPackage::IsAssetContained(const FArticyId &Id) const
{
	return AssetsById.Contains(Id);
}

inline UArticyObject* UArticyPackage::GetAssetById(const FArticyId& Id) const
{
	if(AssetsById.Contains(Id))
	{
		return AssetsById[Id].Get();
	}

	return nullptr;
}

inline UArticyObject* UArticyPackage::GetAssetByTechnicalName(const FName& TechnicalName) const
{
	if (AssetsByTechnicalName.Contains(TechnicalName))
	{
		return AssetsByTechnicalName[TechnicalName].Get();
	}

	return nullptr;
}

inline TArray<UObject*> UArticyPackage::GetInnerObjects() const
{
	TArray<UObject*> OutObjects;
	GetObjectsWithOuter(this, OutObjects, false);

	return OutObjects;
}

inline void UArticyPackage::AddAsset(UArticyObject* ArticyObject)
{
	const FArticyId& ArticyId = ArticyObject->GetId();

	if (!IsAssetContained(ArticyId))
	{
		Assets.Add(ArticyObject);
		AssetsByTechnicalName.Add(ArticyObject->GetTechnicalName(), ArticyObject);
		AssetsById.Add(ArticyId, ArticyObject);
	}
}