#include "stdafx.h"
#include "SiaCommon.h"
#include <ttmath/ttmath.h>
#include <Wincrypt.h>

NS_BEGIN(Sia)
NS_BEGIN(Api)

String AFX_EXT_API GenerateSha256(const String& str)
{
	String ret;
	HCRYPTPROV hCryptProv = 0;
	HCRYPTHASH hHash = 0;
	BOOL ok = CryptAcquireContext(&hCryptProv, NULL, NULL, PROV_RSA_AES, 0);
	ok = ok && CryptCreateHash(hCryptProv, CALG_SHA_256, 0, 0, &hHash);
	ok = ok && CryptHashData(hHash, reinterpret_cast<const BYTE*>(&str[0]), str.length() * sizeof(TCHAR), 0);
	if (ok)
	{
		DWORD dwHashLen;
		DWORD dwCount = sizeof(DWORD);
		if (CryptGetHashParam(hHash, HP_HASHSIZE, reinterpret_cast<BYTE *>(&dwHashLen), &dwCount, 0))
		{
			std::vector<unsigned char> hash(dwHashLen);
			if (CryptGetHashParam(hHash, HP_HASHVAL, reinterpret_cast<BYTE *>(&hash[0]), &dwHashLen, 0))
			{
				std::ostringstream ss;
				ss << std::hex << std::uppercase << std::setfill('0');
				for (int c : hash)
				{
					ss << std::setw(2) << c;
				}
				std::string s = ss.str();
				ret = CA2W(s.c_str()).m_psz;
			}
		}
	}
	
	if (hHash) CryptDestroyHash(hHash);
	if (hCryptProv) CryptReleaseContext(hCryptProv, 0);

	return ret;
}

BOOL RecurDeleteFilesByExtentsion(const String& folder, const String& extensionWithDot)
{
	BOOL ret = FALSE;
	WIN32_FIND_DATA fd = { 0 };	
	
	String search;
	search.resize(MAX_PATH + 1);
	::PathCombine(&search[0], folder.c_str(), L"*.*");

	HANDLE find = FindFirstFile(search.c_str(), &fd);
	if (find != INVALID_HANDLE_VALUE)
	{
		ret = TRUE;
		do
		{
			if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
			{
				if ((_tcscmp(fd.cFileName, L".") != 0) && (_tcscmp(fd.cFileName, L"..") != 0))
				{
					String nextFolder;
					nextFolder.resize(MAX_PATH + 1);
					::PathCombine(&nextFolder[0], folder.c_str(), fd.cFileName);
					ret = RecurDeleteFilesByExtentsion(nextFolder, extensionWithDot);
				}
			}
			else
			{
				String ext = ::PathFindExtension(fd.cFileName);
				if (_tcscmp(ext.c_str(), extensionWithDot.c_str()) == 0)
				{
					String filePath;
					filePath.resize(MAX_PATH + 1);
					::PathCombine(&filePath[0], folder.c_str(), fd.cFileName);

					ret = RetryDeleteFileIfExists(filePath.c_str());
				}
			}
		} while (ret && (FindNextFile(find, &fd) != 0));
	}

	return ret;
}

NS_END(2)