#include "StdAfx.h"
#include "CFilePathManager.h"

// 경고 막기.
#pragma warning ( disable : 4996 )


//-----------------------------------------------------------------------------
//		전역 번수 정의 
CFilePathManager		g_FilePathMng( "Resources" );

//-----------------------------------------------------------------------------
//		내부에서 사용하는 C함수 전방 선언
//-----------------------------------------------------------------------------

int		FindFolderUpTokenCount( LPCSTR str );
void	FilePathUp( LPSTR pPathBuffer, int count );
void	CharacterChange( LPSTR pPathBuffer, CHAR DestCh, CHAR SrcCh );


//-----------------------------------------------------------------------------
//
//			CFilePathManager
//
//-----------------------------------------------------------------------------

CFilePathManager::CFilePathManager(void)
: m_WorkingPath		()
, m_FileNames			()
, m_OverlapCount	( 0 )
{
}

CFilePathManager::CFilePathManager( LPCSTR p_WorkingFolder )
{
	Initialize( p_WorkingFolder );
}

CFilePathManager::~CFilePathManager(void)
{
	m_FileNames.clear();	
}


//	---------------------- Funtion Info -------------------------
//	목적 :	  작업폴더의 모든 파일이름을 얻어낸다.
//	매개변수:	p_WorkingFolder가 NULL 이면 응용프로그램 기본 경로로 지정된다.
//	반환값:		성공여부
//	설명:
//	----------------------------------------------------------
BOOL			CFilePathManager::Initialize( LPCSTR p_WorkingFolder )
{
	string	FullWorkingFolder;
	CHAR	CurrentDir[MAX_PATH];
	CHAR	WorkingFolderPath[MAX_PATH];
	GetCurrentDirectory( MAX_PATH, CurrentDir );

	strcpy( WorkingFolderPath, p_WorkingFolder );
	CharacterChange( WorkingFolderPath, '/', '\\' );

	// p_WorkingFolder가 NULL이 아니면
	// 경로를 검사한다.
	if ( p_WorkingFolder )
	{
		// "../"가 몇개 들어있는지 검사를 한다.
		int UpTokenCount = FindFolderUpTokenCount( WorkingFolderPath );

		// "../"의 개수 만큼 경로를 축소시킨다.
		FilePathUp( CurrentDir, UpTokenCount );

		CHAR	WorkingDirName[MAX_PATH];
		strcpy( WorkingDirName, WorkingFolderPath+UpTokenCount*3 );
		strcat( CurrentDir, "\\" );
		strcat( CurrentDir, WorkingDirName );
	}

	FullWorkingFolder = CurrentDir;

	if ( !FindAllFileNames( FullWorkingFolder.c_str() ) )
	{
		return FALSE;
	}

	// 작업경로를 지정한다.
	m_WorkingPath = FullWorkingFolder;
	return TRUE;
}




//	---------------------- Funtion Info -------------------------
//	목적 :	  파일이름의 전체 경로를 얻어낸다.
//	매개변수:
//	반환값:
//	설명:
//	----------------------------------------------------------
BOOL			CFilePathManager::GetFullPath( LPSTR p_FileName, LPSTR p_RetFullFileName )
{
	MapFileName::iterator		retIter;
	retIter = m_FileNames.find( p_FileName );
	if ( retIter == m_FileNames.end() )
	{
		return FALSE;
	}

	strcpy( p_RetFullFileName, retIter->second.c_str() );
	strcat( p_RetFullFileName, "\\" );
	strcat( p_RetFullFileName, retIter->first.c_str() );

	return TRUE;
}

//	---------------------- Funtion Info -------------------------
//	목적 :	  매개변수로 넘어온 폴더가 포함하는 모든 파일의 이름을 찾아서 저장한다.
//	매개변수:		검사할 폴더.
//	반환값:		성공 여부
//	설명:			재귀함수이다.
//	----------------------------------------------------------
BOOL			CFilePathManager::FindAllFileNames( LPCSTR p_FolderName )
{
	if ( NULL == p_FolderName )
		return FALSE;

	string		FolderName;
	HANDLE	hFind			= INVALID_HANDLE_VALUE;
	WIN32_FIND_DATA		FindFileData;

	FolderName = p_FolderName;
	FolderName += "\\*";

	hFind = FindFirstFile( FolderName.c_str(), &FindFileData );
	if ( hFind == INVALID_HANDLE_VALUE )
	{
		return FALSE;
	}

	list<string>		FolderList;		// 하위 폴더의 이름들.

	// 예외 이름이 아니고 폴더가 아닐때 파일이름목록에 추가한다.
	if ( !IsExceptionName(FindFileData.cFileName) &&
		( FindFileData.dwFileAttributes != FILE_ATTRIBUTE_DIRECTORY ) )
	{
		if ( !InsertFileName( FindFileData.cFileName, p_FolderName ) )
		{
			OutputDebugString("파일이름이 중복이라서 실패하였습니다.");
		}
	}

	// 예외 이름이 아니고 폴더일때 폴더 리스트에 추가한다.
	if ( !IsExceptionName(FindFileData.cFileName) &&
		( FindFileData.dwFileAttributes != FILE_ATTRIBUTE_DIRECTORY ) )
	{
		FolderList.push_back( FindFileData.cFileName );
	}

	// 루프를 돌면서 다음 파일이름을 찾는다.
	while ( FindNextFile( hFind, &FindFileData ) != 0 )
	{
		// 처리하면 안되는 파일이름일 경우 추가하지 않는다.
		if ( IsExceptionName(FindFileData.cFileName) )
		{
			continue;
		}

		// 속성이 폴더일 경우에 하위 폴더 리스트에 추가한다.
		if ( FindFileData.dwFileAttributes == FILE_ATTRIBUTE_DIRECTORY )
		{
			FolderList.push_back( FindFileData.cFileName );
			continue;
		}

		// 파일이름을 추가한다.
		if ( !InsertFileName( FindFileData.cFileName, p_FolderName ) )
		{
			OutputDebugString("파일이름이 중복이라서 실패하였습니다.");
		}
	}

	FindClose( hFind );

	// 하위 폴더를 검색한다.
	list<string>::iterator		iter;
	string		ChildFolder;
	for ( iter = FolderList.begin() ; iter != FolderList.end() ; ++iter )
	{
		ChildFolder = p_FolderName;
		ChildFolder += "\\";
		ChildFolder += (*iter);

		if ( !FindAllFileNames( ChildFolder.c_str() ) )
		{
			return FALSE;
		}
	}

	return TRUE;
}


//	---------------------- Funtion Info -------------------------
//	목적 :	  예외로 취급할 파일 이름을 검사한다.
//	매개변수:	검사할 파일이름.
//	반환값:		예외파일이름일경우 TRUE
//	설명:
//				폴더가 포함하는 파일이름을 검사할 경우
//				"..", "."가 나오기 때문에 이 이름인지를 검사해야한다.
//	----------------------------------------------------------
BOOL			CFilePathManager::IsExceptionName( LPCSTR p_Name )
{
	int ret = strcmp( p_Name, ".." );
	if ( ret == 0 )
	{
		return TRUE;
	}

	ret = strcmp( p_Name, "." );
	if ( ret == 0 )
	{
		return TRUE;
	}

	return FALSE;
}


//	---------------------- Funtion Info -------------------------
//	목적 :	파일 이름을 추가한다.
//	매개변수:	파일이름과 이 파일의 경로
//	반환값:		성공여부
//	설명:			파일이름이 중복일 경우에 실패한다.
//	----------------------------------------------------------
BOOL			CFilePathManager::InsertFileName( LPCSTR p_FileName, LPCSTR p_FilePath )
{
	pair<MapFileName::iterator,bool>		retVal;
	retVal = m_FileNames.insert( MapFileName::value_type( string(p_FileName), string(p_FilePath) ) );
	if ( retVal.second == false )
	{
		m_OverlapCount++;		// 중복횟수를 저장한다. 디버깅시에 편하라고.
		return FALSE;
	}
	return TRUE;
}



//-----------------------------------------------------------------------------
//		내부에서 사용하는 C함수 구현
//-----------------------------------------------------------------------------


//	---------------------- Funtion Info -------------------------
//	목적 :	  "../"가 몇개 들어있는지 검사하는 함수.
//	매개변수:
//	반환값:
//	설명:
//	----------------------------------------------------------
int		FindFolderUpTokenCount( LPCSTR str )
{
	int i = 0;
	int count = 0;
	while ( str[i] != 0 )
	{
		if ( str[i] != '.' )		break;
		i++;

		if ( str[i] != '.' )		break;
		i++;

		if ( str[i] != '\\' )		break;
		i++;

		count ++;
	}
	return count;
}

void	FilePathUp( LPSTR pPathBuffer, int count )
{
	int Length = lstrlen( pPathBuffer );
	while ( count > 0 )
	{
		if ( pPathBuffer[--Length] == '\\' )
		{
			--count;
		}
		pPathBuffer[Length] = NULL;
	}
}

void	CharacterChange( LPSTR pPathBuffer, CHAR DestCh, CHAR SrcCh )
{
	int Length = lstrlen( pPathBuffer );
	for ( int i = 0 ; i < Length ; ++i )
	{
		if ( pPathBuffer[i] == DestCh )
		{
			pPathBuffer[i] = SrcCh;
		}
	}
}