//********************************** Banshee Engine (www.banshee3d.com) **************************************************//
//**************** Copyright (c) 2016 Marko Pintera (marko.pintera@gmail.com). All rights reserved. **********************//
#pragma once

#include "BsEditorPrerequisites.h"
#include "Reflection/BsRTTIType.h"
#include "Reflection/BsRTTIPlain.h"
#include "RTTI/BsPathRTTI.h"
#include "Library/BsProjectLibraryEntries.h"
#include "String/BsUnicode.h"

namespace bs
{
	/** @cond RTTI */
	/** @addtogroup RTTI-Impl-Editor
	 *  @{
	 */

	class ProjectLibraryEntriesRTTI : public RTTIType<ProjectLibraryEntries, IReflectable, ProjectLibraryEntriesRTTI>
	{
	private:
		ProjectLibrary::DirectoryEntry& getRootElement(ProjectLibraryEntries* obj) { return *obj->mRootEntry; }
		void setRootElement(ProjectLibraryEntries* obj, ProjectLibrary::DirectoryEntry& val)
		{
			obj->mRootEntry = bs_ushared_ptr_new<ProjectLibrary::DirectoryEntry>(val);

			for(auto& child : obj->mRootEntry->mChildren)
				child->parent = obj->mRootEntry.get();
		} 

	public:
		ProjectLibraryEntriesRTTI()
		{
			addPlainField("mRootElement", 0, &ProjectLibraryEntriesRTTI::getRootElement, &ProjectLibraryEntriesRTTI::setRootElement);
		}

		virtual const String& getRTTIName()
		{
			static String name = "ProjectLibraryEntries";
			return name;
		}

		virtual UINT32 getRTTIId()
		{
			return TID_ProjectLibraryEntries;
		}

		virtual SPtr<IReflectable> newRTTIObject()
		{
			return ProjectLibraryEntries::createEmpty();
		}
	};

	template<> struct RTTIPlainType<ProjectLibrary::FileEntry>
	{	
		enum { id = TID_ProjectLibraryResEntry }; enum { hasDynamicSize = 1 };

		static BitLength toMemory(const ProjectLibrary::FileEntry& data, Bitstream& stream, const RTTIFieldInfo& fieldInfo, bool compress)
		{ 
			return rtti_write_with_size_header(stream, data, compress, [&data, &stream]()
			{
				BitLength size = 0;

				// For compatibility, encoding the name as a wide string
				WString elemName = UTF8::toWide(data.elementName);

				auto type = (uint32_t)data.type;
				size += rtti_write(type, stream);
				size += rtti_write(data.path, stream);
				size += rtti_write(elemName, stream);
				size += rtti_write(data.lastUpdateTime, stream);

				return size;

			});
		}

		static BitLength fromMemory(ProjectLibrary::FileEntry& data, Bitstream& stream, const RTTIFieldInfo& fieldInfo, bool compress)
		{ 
			BitLength size;
			rtti_read_size_header(stream, compress, size);

			uint32_t type;
			rtti_read(type, stream);
			data.type = (ProjectLibrary::LibraryEntryType)type;

			rtti_read(data.path, stream);

			WString elemName;
			rtti_read(elemName, stream);
			data.elementName = UTF8::fromWide(elemName);
			data.elementNameHash = bs_hash(UTF8::toLower(data.elementName));

			rtti_read(data.lastUpdateTime, stream);

			return size;
		}

		static BitLength getSize(const ProjectLibrary::FileEntry& data, const RTTIFieldInfo& fieldInfo, bool compress)	
		{ 
			WString elemName = UTF8::toWide(data.elementName);

			BitLength dataSize = rtti_size(data.type) + rtti_size(data.path) +
				rtti_size(elemName) + rtti_size(data.lastUpdateTime);

			rtti_add_header_size(dataSize, compress);
			return dataSize;
		}	
	}; 

	template<> struct RTTIPlainType<ProjectLibrary::DirectoryEntry>
	{	
		enum { id = TID_ProjectLibraryDirEntry }; enum { hasDynamicSize = 1 };

		static BitLength toMemory(const ProjectLibrary::DirectoryEntry& data, Bitstream& stream, const RTTIFieldInfo& fieldInfo, bool compress)
		{ 
			return rtti_write_with_size_header(stream, data, compress, [&data, &stream]()
			{
				BitLength size = 0;

				// For compatibility, encoding the name as a wide string
				WString elemName = UTF8::toWide(data.elementName);

				size += rtti_write(data.type, stream);
				size += rtti_write(data.path, stream);
				size += rtti_write(elemName, stream);

				uint32_t numChildren = (UINT32)data.mChildren.size();
				size += rtti_write(numChildren, stream);

				for(auto& child : data.mChildren)
				{
					if(child->type == ProjectLibrary::LibraryEntryType::File)
					{
						auto* childResEntry = static_cast<ProjectLibrary::FileEntry*>(child.get());
						size += rtti_write(*childResEntry, stream);
					}
					else if(child->type == ProjectLibrary::LibraryEntryType::Directory)
					{
						auto* childDirEntry = static_cast<ProjectLibrary::DirectoryEntry*>(child.get());
						size += rtti_write(*childDirEntry, stream);
					}
				}

				return size;
			});
		}

		static BitLength fromMemory(ProjectLibrary::DirectoryEntry& data, Bitstream& stream, const RTTIFieldInfo& fieldInfo, bool compress)
		{
			BitLength size;
			rtti_read_size_header(stream, compress, size);

			rtti_read(data.type, stream);
			rtti_read(data.path, stream);

			WString elemName;
			rtti_read(elemName, stream);
			data.elementName = UTF8::fromWide(elemName);
			data.elementNameHash = bs_hash(UTF8::toLower(data.elementName));

			UINT32 numChildren = 0;
			rtti_read(numChildren, stream);

			for (UINT32 i = 0; i < numChildren; i++)
			{
				ProjectLibrary::LibraryEntryType childType = ProjectLibrary::LibraryEntryType::File;

				uint64_t prevLoc = stream.tell();
				stream.skipBytes(sizeof(uint32_t)); // Skip ahead to get the type
				rtti_read(childType, stream);
				stream.seek(prevLoc);

				if (childType == ProjectLibrary::LibraryEntryType::File)
				{
					USPtr<ProjectLibrary::FileEntry> childResEntry = bs_ushared_ptr_new<ProjectLibrary::FileEntry>();
					// Note: Assumes that ProjectLibrary takes care of the cleanup
					rtti_read(*childResEntry, stream);

					childResEntry->parent = &data;
					data.mChildren.push_back(childResEntry);
				}
				else if (childType == ProjectLibrary::LibraryEntryType::Directory)
				{
					USPtr<ProjectLibrary::DirectoryEntry> childDirEntry = bs_ushared_ptr_new<ProjectLibrary::DirectoryEntry>();
					// Note: Assumes that ProjectLibrary takes care of the cleanup
					rtti_read(*childDirEntry, stream);

					childDirEntry->parent = &data;
					data.mChildren.push_back(childDirEntry);
				}
			}

			return size;
		}

		static BitLength getSize(const ProjectLibrary::DirectoryEntry& data, const RTTIFieldInfo& fieldInfo, bool compress)
		{ 
			WString elemName = UTF8::toWide(data.elementName);
			BitLength dataSize = rtti_size(data.type) + rtti_size(data.path) + rtti_size(elemName) + sizeof(uint32_t);

			for(auto& child : data.mChildren)
			{
				if(child->type == ProjectLibrary::LibraryEntryType::File)
				{
					ProjectLibrary::FileEntry* childResEntry = static_cast<ProjectLibrary::FileEntry*>(child.get());
					dataSize += rtti_size(*childResEntry);
				}
				else if(child->type == ProjectLibrary::LibraryEntryType::Directory)
				{
					ProjectLibrary::DirectoryEntry* childDirEntry = static_cast<ProjectLibrary::DirectoryEntry*>(child.get());
					dataSize += rtti_size(*childDirEntry);
				}
			}

			rtti_add_header_size(dataSize, compress);
			return dataSize;
		}	
	};

	/** @} */
	/** @endcond */
}