#include "build.h"
#include "function.h"
#include "viewerfactory.h"
#include "serializerbinary.h"

subExceptionDef( BuildException );

Build::Build() :
	pStrProc( NULL ),
	pView( NULL ),
	bCacheUpdated( false )
{
}

Build::~Build()
{
	if( sCacheName.size() > 0 && bCacheUpdated )
	{
		try
		{
			SerializerBinary ar( sCacheName.c_str(), Serializer::save );

			ar << cRequires;
		}
		catch( ExceptionBase &e )
		{
		}
	}
}

void Build::setView( const std::string &sView )
{
	pView = ViewerFactory::getInstance().instantiate( sView.c_str() );
}

void Build::setCache( const std::string &sFileName )
{
	sCacheName = sFileName;

	try
	{
		SerializerBinary ar( sCacheName.c_str(), Serializer::load );

		ar >> cRequires;
	}
	catch( ExceptionBase &e )
	{
	}
}

void Build::setStringProc( StringProc *pStrProc )
{
	delete this->pStrProc;
	this->pStrProc = pStrProc;
}

std::string Build::replVars( const std::string &sSrc, const StringList *pCont, VarMap *mExtra )
{
	if( pStrProc == NULL )
		throw BuildException(
			"No valid string processor was registered with the Build object."
			);

	return pStrProc->replVars( sSrc, pCont, mExtra );
}

void Build::execAction( const std::string &sWhat )
{
	if( mAction.find( sWhat ) == mAction.end() )
		throw BuildException(
			"No action matches %s, check your build.conf.",
			sWhat.c_str()
			);

	Action *pAct = mAction[sWhat];

	pView->beginAction( sWhat, pAct->size() );

	for( pAct->begin(); !pAct->isEnded(); pAct->next() )
	{
		if( mTarget.find( pAct->getWhat() ) == mTarget.end() )
			throw BuildException(
				"No target matches %s in action %s.",
				pAct->getWhat().c_str(),
				sWhat.c_str()
				);
		Target *pTarget = mTarget[pAct->getWhat()];
		pView->beginCommand( pAct->getAct(), pAct->getWhat() );
		switch( pAct->getAct() )
		{
			case Action::actCheck:
				pTarget->check( *this );
				break;

			case Action::actClean:
				pTarget->clean( *this );
				break;
		}
		pView->endCommand();
	}

	pView->endAction();

	return;
}

void Build::addTarget( Target *pTarget )
{
	TargetMap::iterator i = mTarget.find( pTarget->getName() );
	if( i == mTarget.end() )
	{
		mTarget[pTarget->getName()] = pTarget;
	}
	else
	{
		throw BuildException("Merging targets isn't working yet.");
	}
}

void Build::addRequires( const std::string &who, const std::string &what )
{
	mRequires[who].push_back( what );
}

void Build::addRule( Rule *pRule )
{
	mRule[pRule->getName()] = pRule;
}

Rule *Build::getRule( const std::string &name )
{
	if( mRule.find( name ) == mRule.end() )
		throw BuildException("No rule named %s found.", name.c_str() );

	return mRule[name];
}

void Build::addAction( Action *pAction )
{
	mAction[pAction->getName()] = pAction;
}

void Build::set( const std::string &cont, const std::string &var, const std::string &val )
{
	if( cont == "" )
	{
		mVars[var] = replVars( val, NULL, NULL );
	}
	else
	{
		StringList cl;
		cl.push_front( cont );
		mContVars[cont][var] = replVars( val, &cl, NULL );
	}
}

void Build::setAdd( const std::string &cont, const std::string &var, const std::string &val )
{
	if( cont == "" )
	{
		mVars[var] = getVar( NULL, var, NULL ) + " " + replVars( val, NULL, NULL );
	}
	else
	{
		StringList cl;
		cl.push_front( cont );
		mContVars[cont][var] = getVar( &cl, var, NULL ) + " " + replVars( val, &cl, NULL );
	}
}

std::string Build::getVar( const StringList *cont, const std::string &var, VarMap *mExtra )
{
	if( mExtra != NULL )
	{
		if( mExtra->find(var) == mExtra->end() )
		{
			return getVar( cont, var, NULL );
		}
		return (*mExtra)[var];
	}
	
	if( cont == NULL )
	{
		if( mVars.find(var) == mVars.end() )
		{
			if( getenv( var.c_str() ) == NULL )
			{
				mVars[var] = "";
			}
			else
			{
				mVars[var] = getenv( var.c_str() );
			}
		}
		return mVars[var];
	}
	else
	{
		if( cont->empty() )
		{
			return getVar( NULL, var, NULL );
		}
		std::string sTop = cont->front();
		if( mContVars[sTop].find(var) == mContVars[sTop].end() )
		{
			((StringList *)cont)->pop_front();
			mContVars[sTop][var] = getVar( cont, var, NULL );
			((StringList *)cont)->push_front( sTop );
		}
		return mContVars[sTop][var];
	}
}

void Build::debugDump()
{
	printf("Requires:\n");
	for( ReqMap::iterator i = mRequires.begin(); i != mRequires.end(); i++ )
	{
		printf("  %s: ", (*i).first.c_str() );

		for( StringList::iterator j = (*i).second.begin();
			 j != (*i).second.end(); j++ )
		{
			if( j != (*i).second.begin() )
				printf(", ");
			printf("%s", (*j).c_str() );
		}
		printf("\n");
	}

	printf("Targets:\n");
	for( TargetMap::iterator i = mTarget.begin(); i != mTarget.end(); i++ )
	{
		printf("  %s:\n", (*i).first.c_str() );
		printf("    Rule:  %s\n", (*i).second->getRule().c_str() );
		printf("    Input: ");
		for( StringList::iterator j = (*i).second->getInput().begin();
			 j != (*i).second->getInput().end(); j++ )
		{
			if( j != (*i).second->getInput().begin() )
				printf(", ");
			printf("%s", (*j).c_str() );
		}
		printf("\n");
	}

	printf("Global Variables:\n");
	for( VarMap::iterator i = mVars.begin(); i != mVars.end(); i++ )
	{
		printf("  \"%s\" = \"%s\"\n", (*i).first.c_str(), (*i).second.c_str() );
	}

	printf("Context Variables:\n");
	for( ContextMap::iterator i = mContVars.begin(); i != mContVars.end(); i++ )
	{
		printf("  %s:\n", (*i).first.c_str() );
		
		for( VarMap::iterator j = (*i).second.begin();
			 j != (*i).second.end(); j++ )
		{
			printf("    \"%s\" = \"%s\"\n",
				(*j).first.c_str(),
				(*j).second.c_str()
				);
		}
	}

	printf("Rules:\n");
	for( RuleMap::iterator i = mRule.begin(); i != mRule.end(); i++ )
	{
		printf("  %s:\n", (*i).first.c_str() );
		printf("    Matches:  func\n");
		printf("    Filters:  %d\n", (*i).second->getFilterList().size() );
		printf("    Performs: %d\n", (*i).second->getPerformList().size() );
		printf("    Produces:\n");
		printf("    Requires:\n");
	}

	printf("Actions:\n");
	for( ActionMap::iterator i = mAction.begin(); i != mAction.end(); i++ )
	{
		printf("  %s: ", (*i).first.c_str() );
		for( (*i).second->begin(); !(*i).second->isEnded(); (*i).second->next() )
		{
			printf("%d:%s ", (*i).second->getAct(), (*i).second->getWhat().c_str() );
		}
		printf("\n");
	}
}

RuleList Build::findChainRules( Rule *pHead )
{
	RuleList lOut;
	FunctionList &lMatches = pHead->getMatchesList();

	for( RuleMap::iterator i = mRule.begin(); i != mRule.end(); i++ )
	{
		if( (*i).second == pHead )
			continue;

		for( FunctionList::iterator j = lMatches.begin();
			 j != lMatches.end(); j++ )
		{
			StringList lTmp;
			(*j)->execute( NULL, (*i).second->getProducesList(), lTmp );
			if( !lTmp.empty() )
			{
				lOut.push_back( (*i).second );
				break;
			}
		}
	}

	return lOut;
}

StringList &Build::getRequires( std::string sName )
{
	return mRequires[sName];
}

bool Build::getCached( const std::string &sID, int nTime, StringList &lOut )
{
	Cache::Entry *pEnt = cRequires.get( sID );
	if( pEnt == NULL )
		return false;
	if( pEnt->tCreated < nTime )
		return false;

	lOut.insert( lOut.end(), pEnt->lData.begin(), pEnt->lData.end() );

	return true;
}

void Build::updateCache( const std::string &sID, FunctionList &lFunc, StringList &lOut )
{
	Cache::Entry *pEnt = new Cache::Entry;
	getView()->beginRequiresCheck( false, sID );
	for( FunctionList::iterator f = lFunc.begin(); f != lFunc.end(); f++ )
	{
		StringList lTmpIn;
		lTmpIn.push_back( sID );
		(*f)->execute( this, lTmpIn, pEnt->lData );
	}
	getView()->endRequiresCheck();

	lOut.insert( lOut.end(), pEnt->lData.begin(), pEnt->lData.end() );
	cRequires.put( sID, pEnt );

	pEnt->tCreated = time( NULL );

	bCacheUpdated = true;
}