WechatExporter/vcproject/ShellImpl.h

#pragma once

#include "stdafx.h"
#include <string>
#include <shlobj_core.h>
#include "Core.h"

class ShellImpl : public Shell
{
public:
	~ShellImpl() {}

	bool existsDirectory(const std::string& path) const
	{
		CW2T pszT(CA2W(path.c_str(), CP_UTF8));

		DWORD dwAttrib = ::GetFileAttributes((LPCTSTR)pszT);

		return (dwAttrib != INVALID_FILE_ATTRIBUTES &&
			(dwAttrib & FILE_ATTRIBUTE_DIRECTORY) == FILE_ATTRIBUTE_DIRECTORY);
	}

	bool makeDirectory(const std::string& path) const
	{
		CW2T pszT(CA2W(path.c_str(), CP_UTF8));

		return ::SHCreateDirectoryEx(NULL, (LPCTSTR)pszT, NULL) == ERROR_SUCCESS;
	}

	bool listSubDirectories(const std::string& path, std::vector<std::string>& subDirectories) const
	{
		WIN32_FIND_DATA FindFileData;
		HANDLE hFind = INVALID_HANDLE_VALUE;

		std::string formatedPath = combinePath(path, "*.*");
		std::replace(formatedPath.begin(), formatedPath.end(), DIR_SEP_R, DIR_SEP);

		CW2T localPath(CA2W(formatedPath.c_str(), CP_UTF8));

		hFind = FindFirstFile((LPTSTR)localPath, &FindFileData);
		if (hFind == INVALID_HANDLE_VALUE)
		{
			return false;
		}
		do
		{
			if (_tcscmp(FindFileData.cFileName, TEXT(".")) == 0 || _tcscmp(FindFileData.cFileName, TEXT("..")) == 0)
			{
				continue;
			}
			if (FindFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
			{
				CW2A pszU8(CT2W(FindFileData.cFileName), CP_UTF8);
				subDirectories.push_back((LPCSTR)pszU8);
			}
		} while (::FindNextFile(hFind, &FindFileData));
		FindClose(hFind);

		return true;
	}
	
	bool copyFile(const std::string& src, const std::string& dest, bool overwrite) const
	{
		CW2T pszSrc(CA2W(src.c_str(), CP_UTF8));
		CW2T pszDest(CA2W(dest.c_str(), CP_UTF8));

		BOOL bRet = ::CopyFile((LPCTSTR)pszSrc, (LPCTSTR)pszDest, (overwrite ? FALSE : TRUE));
		return (bRet == TRUE);
	}

	bool openOutputFile(std::ofstream& ofs, const std::string& fileName, std::ios_base::openmode mode/* = std::ios::out*/) const
	{
		if (ofs.is_open())
		{
			return false;
		}

		CW2A psz(CA2W(fileName.c_str(), CP_UTF8));
		ofs.open((LPCSTR)psz, mode);

		return ofs.is_open();
	}

protected:
	
};