/*==LICENSE==*

CyanWorlds.com Engine - MMOG client, server and tools
Copyright (C) 2011  Cyan Worlds, Inc.

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.

You can contact Cyan Worlds, Inc. by email legal@cyan.com
 or by snail mail at:
      Cyan Worlds, Inc.
      14617 N Newport Hwy
      Mead, WA   99021

*==LICENSE==*/
//////////////////////////////////////////////////////////////////////////////
//
//	plRawResManager - Small public resManager thingy for reading/writing
//					  objects raw.
//
//////////////////////////////////////////////////////////////////////////////

#include "hsTypes.h"
#include "hsStream.h"
#include "plRawResManager.h"

#include "../plResMgr/plRegistry.h"
#include "../plResMgr/plPageInfo.h"
#include "../plResMgr/plRegistrySource.h"
#include "../plResMgr/plRegistryNode.h"
#include "../plResMgr/plRegistryHelpers.h"
#include "../pnKeyedObject/plKeyImp.h"


plRegistryPageNode	*plRawResManager::FindPage( const char *age, const char *chapter, const char *page )
{
	return fRegistry->FindPage( age, chapter, page );
}

plRegistryPageNode	*plRawResManager::CreatePage( const plPageInfo &info )
{
	plRegistryPageNode *page = fRegistry->CreatePage( info.GetLocation(), info.GetAge(), info.GetChapter(), info.GetPage() );

	if( page != nil )
	{
		page->SetLoaded( true );	// We're "loaded", i.e. constructing this at runtime
		fIOSources[ 0 ]->AddLocToSource( page );
	}

	return page;
}

UInt8	*plRawResManager::ReadObjectBuffer( plKeyImp *pKey, UInt32 &retBuffLength )
{
	UInt8	*buffer = nil;


	hsAssert( pKey, "Null Key" );
	hsAssert( pKey->GetStartPos() != (UInt32)-1, "Missing StartPos" );
	hsAssert( pKey->GetDataLen() != (UInt32)-1, "Missing Data Length" );

	if( pKey->GetStartPos() == (UInt32)-1 || pKey->GetDataLen() == (UInt32)-1 )
	{
		// Try to recover from this by just not reading an object
		retBuffLength = 0;
		return nil;
	}

	plRegistryDataStream *dataStream = fRegistry->OpenPageDataStream( pKey->GetUoid().GetLocation(), false );

	if( dataStream != nil && dataStream->GetStream() != nil )
	{
		hsStream *stream = dataStream->GetStream();

		UInt32 oldPos = stream->GetPosition();
		stream->SetPosition( pKey->GetStartPos() );

		buffer = new UInt8[ pKey->GetDataLen() ];
		if( buffer != nil )
		{
			*( (UInt32 *)buffer ) = pKey->GetDataLen();
			stream->Read( pKey->GetDataLen(), (UInt8 *)buffer );
			retBuffLength = pKey->GetDataLen();
		}
		else
			retBuffLength = 0;

		// Restore old position now
		stream->SetPosition( oldPos );
	}
	delete dataStream;

	return buffer;
}

plKey	plRawResManager::NewBlankKey( const plUoid &newUoid )
{
	plKeyImp	*newKey = new plKeyImp;


	newKey->SetUoid( newUoid );
	fRegistry->AddKey( newKey );

	plKey	keyPtr = plKey::Make( newKey );
	return keyPtr;
}