mirror of
https://github.com/BlueMatthew/WechatExporter.git
synced 2026-08-30 09:00:33 +08:00
Matthew
98 lines
2.3 KiB
C++
98 lines
2.3 KiB
C++
#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 existsFile(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) == 0);
|
|
}
|
|
|
|
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:
|
|
|
|
};
|
|
|