sourcemod/tools/installer/CFileList.cpp

#include "CFileList.h"
#include "InstallerUtil.h"

using namespace std;

CFileList::CFileList(const TCHAR *name) : m_TotalSize(0), m_RecursiveSize(0), 
	m_bGotRecursiveSize(false)
{
	UTIL_Format(m_FolderName, sizeof(m_FolderName) / sizeof(TCHAR), _T("%s"), name);
}

CFileList::~CFileList()
{
	list<CFileList *>::iterator iter;

	for (iter = m_folder_list.begin();
		 iter != m_folder_list.end();
		 iter++)
	{
		delete (*iter);
	}
}

const TCHAR *CFileList::GetFolderName()
{
	return m_FolderName;
}

void CFileList::AddFolder(CFileList *pFileList)
{
	m_folder_list.push_back(pFileList);
}

void CFileList::AddFile(const TCHAR *name, unsigned __int64 size)
{
	CFileListEntry entry;
	
	UTIL_Format(entry.file, sizeof(entry.file) / sizeof(TCHAR), _T("%s"), name);
	entry.size = size;

	m_file_list.push_back(entry);
	m_TotalSize += size;
}

unsigned __int64 CFileList::GetRecursiveSize()
{
	if (m_bGotRecursiveSize)
	{
		return m_RecursiveSize;
	}

	m_RecursiveSize = m_TotalSize;

	list<CFileList *>::iterator iter;
	for (iter = m_folder_list.begin(); iter != m_folder_list.end(); iter++)
	{
		m_RecursiveSize += (*iter)->GetRecursiveSize();
	}

	m_bGotRecursiveSize = true;

	return m_RecursiveSize;
}

const TCHAR *CFileList::PeekCurrentFile()
{
	if (m_file_list.empty())
	{
		return NULL;
	}

	return m_file_list.begin()->file;
}

void CFileList::PopCurrentFile()
{
	m_file_list.erase(m_file_list.begin());
}

CFileList *CFileList::PeekCurrentFolder()
{
	if (m_folder_list.empty())
	{
		return NULL;
	}

	return *(m_folder_list.begin());
}

void CFileList::PopCurrentFolder()
{
	m_folder_list.erase(m_folder_list.begin());
}

void RecursiveBuildFileList(CFileList *file_list, const TCHAR *current_folder)
{
	HANDLE hFind;
	WIN32_FIND_DATA fd;
	TCHAR path[MAX_PATH];

	UTIL_PathFormat(path, sizeof(path) / sizeof(TCHAR), _T("%s\\*.*"), current_folder);

	if ((hFind = FindFirstFile(path, &fd)) == INVALID_HANDLE_VALUE)
	{
		return;
	}

	do
	{
		if (tstrcasecmp(fd.cFileName, _T(".")) == 0
			|| tstrcasecmp(fd.cFileName, _T("..")) == 0)
		{
			continue;
		}

		if ((fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == FILE_ATTRIBUTE_DIRECTORY)
		{
			CFileList *pSubList = new CFileList(fd.cFileName);

			UTIL_PathFormat(path,
				sizeof(path) / sizeof(TCHAR),
				_T("%s\\%s"), 
				current_folder,
				fd.cFileName);			

			RecursiveBuildFileList(pSubList, path);

			file_list->AddFolder(pSubList);
		}
		else
		{
			LARGE_INTEGER li;

			li.LowPart = fd.nFileSizeLow;
			li.HighPart = fd.nFileSizeHigh;

			file_list->AddFile(fd.cFileName, li.QuadPart);
		}
	} while (FindNextFile(hFind, &fd));

	FindClose(hFind);
}

CFileList *CFileList::BuildFileList(const TCHAR *name, const TCHAR *root_folder)
{
	CFileList *pFileList = new CFileList(name);

	RecursiveBuildFileList(pFileList, root_folder);

	return pFileList;
}