/* Copyright 2014 ROBLOX Corporation, All Rights Reserved */
#include "stdafx.h"

#include "V8DataModel/NonReplicatedCSGDictionaryService.h"

#include "V8DataModel/CSGDictionaryService.h"
#include "../App/include/v8datamodel/PartOperation.h"
#include "v8datamodel/Value.h"
#include "Util/RobloxGoogleAnalytics.h"

FASTFLAGVARIABLE(IgnoreBlankDataOnStore, true)

FASTFLAG(StudioCSGAssets)

namespace RBX
{
	const char *const sNonReplicatedCSGDictionaryService = "NonReplicatedCSGDictionaryService";

	NonReplicatedCSGDictionaryService::NonReplicatedCSGDictionaryService()
	{
		setName("NonReplicatedCSGDictionaryService");
	}
	
	void NonReplicatedCSGDictionaryService::reparentChildData(shared_ptr<RBX::Instance> childInstance)
	{
		if (!isChildData(childInstance))
			return;

		CSGDictionaryService* dictionaryService = ServiceProvider::create<CSGDictionaryService>(DataModel::get(this));
		childInstance->setParent(dictionaryService);

		if (shared_ptr<RBX::BinaryStringValue> bStrValue = RBX::Instance::fastSharedDynamicCast<RBX::BinaryStringValue>(childInstance))
		{
			std::string key = createHashKey(bStrValue->getValue().value());
			instanceMap.erase(key);
		}
	}

	void NonReplicatedCSGDictionaryService::storeData(PartOperation& partOperation, bool forceIncrement)
	{
		if (FFlag::StudioCSGAssets)
		{
			BinaryString tmpString = partOperation.getChildData();
			if ( FFlag::IgnoreBlankDataOnStore && tmpString.value().size() > 0 )
			{
				storeStringData(tmpString, forceIncrement, "ChildData");
				partOperation.setChildData(tmpString);
			}
		}
		else
		{
			BinaryString tmpString = partOperation.getChildData();
			storeStringData(tmpString, forceIncrement, "ChildData");
			partOperation.setChildData(tmpString);
		}
	}

	void NonReplicatedCSGDictionaryService::retrieveData(PartOperation& partOperation)
	{
		if (FFlag::StudioCSGAssets)
		{
			BinaryString tmpString = partOperation.getChildData();
			retrieveStringData(tmpString);
			partOperation.setChildData(tmpString);
		}
		else
		{
			BinaryString tmpString = partOperation.getChildData();
			retrieveStringData(tmpString);
			partOperation.setChildData(tmpString);
		}
	}

	void NonReplicatedCSGDictionaryService::storeAllDescendants(shared_ptr<Instance> instance)
	{
		if (instance->getChildren())
			for (RBX::Instances::const_iterator iter = instance->getChildren()->begin(); iter != instance->getChildren()->end(); ++iter)
				storeAllDescendants(*iter);

		if (shared_ptr<PartOperation> childOperation = RBX::Instance::fastSharedDynamicCast<PartOperation>(instance))
			storeData(*childOperation);
	}

	void NonReplicatedCSGDictionaryService::retrieveAllDescendants(shared_ptr<Instance> instance)
	{
		if (instance->getChildren())
			for (RBX::Instances::const_iterator iter = instance->getChildren()->begin(); iter != instance->getChildren()->end(); ++iter)
				retrieveAllDescendants(*iter);

		if (shared_ptr<PartOperation> childOperation = RBX::Instance::fastSharedDynamicCast<PartOperation>(instance))
			retrieveData(*childOperation);
	}

	void NonReplicatedCSGDictionaryService::refreshRefCountUnderInstance(RBX::Instance* instance)
	{
		if (RBX::PartOperation* partOperation = RBX::Instance::fastDynamicCast<RBX::PartOperation>(instance))
			storeData(*partOperation, true);

		if (instance->getChildren())
			for (RBX::Instances::const_iterator iter = instance->getChildren()->begin(); iter != instance->getChildren()->end(); ++iter)
				refreshRefCountUnderInstance(iter->get());
	}
}