#include "global.h"
#include "MemoryCardManager.h"
#include "arch/MemoryCard/MemoryCardDriver.h"	// for UsbStorageDevice
#include "ScreenManager.h"
#include "ThemeManager.h"
#include "PrefsManager.h"
#include "RageLog.h"
#include "ScreenManager.h"
#include "ProfileManager.h"
#include "Foreach.h"
#include "GameState.h"

MemoryCardManager*	MEMCARDMAN = NULL;	// global and accessable from anywhere in our program


MemoryCardManager::MemoryCardManager()
{
	m_pDriver = MakeMemoryCardDriver();
	m_bCardsLocked = false;
	FOREACH_PlayerNumber( p )
	{
		m_bTooLate[p] = false;
	}

	m_soundReady.Load( THEME->GetPathToS("MemoryCardManager ready") );
	m_soundError.Load( THEME->GetPathToS("MemoryCardManager error") );
	m_soundTooLate.Load( THEME->GetPathToS("MemoryCardManager too late") );
	m_soundDisconnect.Load( THEME->GetPathToS("MemoryCardManager disconnect") );
}

MemoryCardManager::~MemoryCardManager()
{
	delete m_pDriver;
}

void MemoryCardManager::Update( float fDelta )
{
	if( m_pDriver->StorageDevicesChanged() )
	{
		vector<UsbStorageDevice> vOld = m_vStorageDevices;	// copy
		m_pDriver->GetStorageDevices( m_vStorageDevices );
		vector<UsbStorageDevice> &vNew = m_vStorageDevices;
		vector<UsbStorageDevice> vConnects;	// fill these in below
		vector<UsbStorageDevice> vDisconnects;	// fill these in below


		// check for disconnects
		FOREACH( UsbStorageDevice, vOld, old )
		  {
		    bool bFoundMatch = false;
		    FOREACH( UsbStorageDevice, vNew, newd )
		      {
			if( old->IdsMatch(*newd) )
			{
			  bFoundMatch = true;
			  break;
			}
		      }
		    if( !bFoundMatch )
		      {
			LOG->Trace( "Disconnected bus %d port %d device %d path %s", old->iBus, old->iPort, old->iLevel, old->sOsMountDir.c_str() );
			vDisconnects.push_back( *old );
		      }
		  }

		// check for connects
		FOREACH( UsbStorageDevice, vNew, newd )
                  {
                    bool bFoundMatch = false;
                    FOREACH( UsbStorageDevice, vOld, old )
                      {
                        if( old->IdsMatch(*newd) )
                          {
			    bFoundMatch = true;
			    break;
			  }
                      }
                    if( !bFoundMatch )
                      {
                        LOG->Trace( "Connected bus %d port %d device %d path %s", newd->iBus, newd->iPort, newd->iLevel, newd->sOsMountDir.c_str() );
                        vDisconnects.push_back( *newd );
                      }
                  }

		// unassign cards that were disconnected
		FOREACH_PlayerNumber( p )
		  {
		    UsbStorageDevice &assigned_device = m_Device[p];
		    if( assigned_device.IsBlank() )	// not assigned a card
		      continue;
		    
		    FOREACH( UsbStorageDevice, vDisconnects, disconnect )
                      {
			if( disconnect->IdsMatch(assigned_device) )
			  {
			    m_pDriver->Unmount(&assigned_device, MEM_CARD_MOUNT_POINT[p]);
			    
			    assigned_device.MakeBlank();
			    m_soundDisconnect.Play();
			    
			    if( PROFILEMAN->ProfileWasLoadedFromMemoryCard(p) )
			      PROFILEMAN->UnloadProfile( p );
			  }
		      }
		  }

		// Update the status of already-assigned cards.  It may contain updated info 
		// like WriteTest results and a new sName.
                FOREACH_PlayerNumber( p )
                  {
                    UsbStorageDevice &assigned_device = m_Device[p];
                    if( assigned_device.IsBlank() )     // no card assigned to this player
                      continue;

                    FOREACH( UsbStorageDevice, m_vStorageDevices, d )
		      {
			if( d->IdsMatch(assigned_device) )
			  {
			    // play write test error sound if write test failed since we last checked
			    if( assigned_device.bNeedsWriteTest && !d->bNeedsWriteTest && !d->bWriteTestSucceeded )
			      m_soundError.Play();
			    assigned_device = *d;
			  }
		      }
		  }

		// make a list of unassigned
		vector<UsbStorageDevice> vUnassignedDevices = m_vStorageDevices;        // copy
		
		// remove cards that are already assigned
		FOREACH_PlayerNumber( p )
		  {
		    UsbStorageDevice &assigned_device = m_Device[p];
		    if( assigned_device.IsBlank() )     // no card assigned to this player
		      continue;

                    FOREACH( UsbStorageDevice, vUnassignedDevices, d )
                      {
                        if( d->IdsMatch(assigned_device) )
			  {
			    vUnassignedDevices.erase( d );
			    break;
			  }
                      }
		  }
		
		// try to assign each device to a player
		FOREACH_PlayerNumber( p )
		  {
		    LOG->Trace( "Looking for a card for Player %d", p+1 );

                    UsbStorageDevice &assigned_device = m_Device[p];		    
		    if( !assigned_device.IsBlank() )    // they already have an assigned card
		      {
			LOG->Trace( "Player %d already has a card: '%s'", p+1, assigned_device.sOsMountDir.c_str() );
			continue;       // skip
		      }
		    
		    FOREACH( UsbStorageDevice, vUnassignedDevices, d )
		      {
			// search for card dir match
			if( !PREFSMAN->m_sMemoryCardOsMountPoint[p].empty() &&
			    d->sOsMountDir.CompareNoCase(PREFSMAN->m_sMemoryCardOsMountPoint[p]) )
			  continue;      // not a match

			// search for USB bus match
			if( PREFSMAN->m_iMemoryCardUsbBus[p] != -1 &&
			    PREFSMAN->m_iMemoryCardUsbBus[p] != d->iBus )
			  continue;       // not a match
			
			if( PREFSMAN->m_iMemoryCardUsbPort[p] != -1 &&
			    PREFSMAN->m_iMemoryCardUsbPort[p] != d->iPort )
			  continue;       // not a match
			
			if( PREFSMAN->m_iMemoryCardUsbLevel[p] != -1 &&
			    PREFSMAN->m_iMemoryCardUsbLevel[p] != d->iLevel )
			  continue;       // not a match
			
			LOG->Trace( "device match:  iScsiIndex: %d, iBus: %d, iLevel: %d, iPort: %d, sOsMountDir: %s",
				    d->iScsiIndex, d->iBus, d->iLevel, d->iPort, d->sOsMountDir.c_str() );

			assigned_device = *d;    // save a copy
			vUnassignedDevices.erase( d );       // remove the device so we don't match it for another player
			m_bTooLate[p] = GAMESTATE->m_bPlayersFinalized;    // the device is too late if inserted when cards were locked
			
			// play sound
			if( m_bTooLate[p] )
			  m_soundTooLate.Play();
			else if( !d->bNeedsWriteTest && !d->bWriteTestSucceeded )
			  m_soundError.Play();
			else
			  m_soundReady.Play();

			break;
		      }
		  }
		
		SCREENMAN->RefreshCreditsMessages();
	}
}

MemoryCardState MemoryCardManager::GetCardState( PlayerNumber pn )
{
  UsbStorageDevice &d = m_Device[pn];
	if( d.IsBlank() )
		return MEMORY_CARD_STATE_NO_CARD;
	else if( m_bTooLate[pn] )
		return MEMORY_CARD_STATE_TOO_LATE;
	else if( !d.bNeedsWriteTest && !d.bWriteTestSucceeded )
		return MEMORY_CARD_STATE_WRITE_ERROR;
	else
		return MEMORY_CARD_STATE_READY;
}

void MemoryCardManager::LockCards( bool bLock )
{
	bool bWasLocked = m_bCardsLocked;
	m_bCardsLocked = bLock;

	if( !bLock )
	{
		// clear too late flag
		FOREACH_PlayerNumber( p )
			m_bTooLate[p] = false;
	}

	if( !bWasLocked && bLock )
	{
		LOG->Trace( "do the final mount" );

		FOREACH_EnabledPlayer( p )
		{
			if( m_Device[p].IsBlank() )	// they don't have an assigned card
				continue;

			m_pDriver->MountAndTestWrite(&m_Device[p], MEM_CARD_MOUNT_POINT[p]);
		}
	}


	if( !bWasLocked && bLock )
		m_pDriver->SetMountThreadState( MemoryCardDriver::detect_and_dont_mount );
	else
		m_pDriver->SetMountThreadState( MemoryCardDriver::detect_and_mount );
}

void MemoryCardManager::FlushAndReset()
{
	FOREACH_PlayerNumber( p )
	{
	  UsbStorageDevice &d = m_Device[p];
		if( d.IsBlank() )	// no card assigned
			continue;	// skip
		if( (!d.bNeedsWriteTest && !d.bWriteTestSucceeded) || m_bTooLate[p] )
			continue;	// skip
		m_pDriver->Flush(&m_Device[p]);
	}

	m_pDriver->ResetUsbStorage();	// forces cards to be re-detected
}

bool MemoryCardManager::PathIsMemCard( CString sDir ) const
{
	FOREACH_PlayerNumber( p )
		if( !sDir.Left(MEM_CARD_MOUNT_POINT[p].size()).CompareNoCase( MEM_CARD_MOUNT_POINT[p] ) )
			return true;
	return false;
}

CString MemoryCardManager::GetName( PlayerNumber pn ) const
{
	return m_Device[pn].sName;
}

void MemoryCardManager::PauseMountingThread()
{
	m_pDriver->SetMountThreadState( MemoryCardDriver::paused );
}

void MemoryCardManager::UnPauseMountingThread()
{
	m_pDriver->SetMountThreadState( 
		m_bCardsLocked ? 
		MemoryCardDriver::detect_and_dont_mount : 
		MemoryCardDriver::detect_and_mount );
}

bool IsAnyPlayerUsingMemoryCard()
{
	FOREACH_HumanPlayer( pn )
		if( MEMCARDMAN->GetCardState(pn) == MEMORY_CARD_STATE_READY )
			return true;
	return false;
}

#include "LuaFunctions.h"
LuaFunction_NoArgs( IsAnyPlayerUsingMemoryCard,		IsAnyPlayerUsingMemoryCard() )

/*
 * (c) 2003-2004 Chris Danford
 * All rights reserved.
 * 
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the
 * "Software"), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, and/or sell copies of the Software, and to permit persons to
 * whom the Software is furnished to do so, provided that the above
 * copyright notice(s) and this permission notice appear in all copies of
 * the Software and that both the above copyright notice(s) and this
 * permission notice appear in supporting documentation.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT OF
 * THIRD PARTY RIGHTS. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR HOLDERS
 * INCLUDED IN THIS NOTICE BE LIABLE FOR ANY CLAIM, OR ANY SPECIAL INDIRECT
 * OR CONSEQUENTIAL DAMAGES, OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS
 * OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
 * OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
 * PERFORMANCE OF THIS SOFTWARE.
 */