feat: add cachecleaner
This commit is contained in:
@@ -0,0 +1,128 @@
|
|||||||
|
#include "cachecleaner.h"
|
||||||
|
|
||||||
|
#include <nerutils/log.h>
|
||||||
|
|
||||||
|
#include <filesystem>
|
||||||
|
#include <unordered_set>
|
||||||
|
|
||||||
|
#include <ShlObj_core.h>
|
||||||
|
#include <TlHelp32.h>
|
||||||
|
|
||||||
|
CacheCleaner::CacheCleaner() : _running(false), _monitorThread(nullptr) {}
|
||||||
|
|
||||||
|
CacheCleaner::~CacheCleaner()
|
||||||
|
{
|
||||||
|
shutdown();
|
||||||
|
}
|
||||||
|
|
||||||
|
void CacheCleaner::init()
|
||||||
|
{
|
||||||
|
std::string localAppData = getLocalAppDataPath();
|
||||||
|
if (localAppData == "") return Log::error("Failed to get LocalAppData?");
|
||||||
|
|
||||||
|
_cacheDir = localAppData + "\\DeadByDaylight\\Saved\\PersistentDownloadDir";
|
||||||
|
|
||||||
|
// this is pointless but just in case
|
||||||
|
std::filesystem::create_directories(_cacheDir);
|
||||||
|
|
||||||
|
clearFolder(_cacheDir);
|
||||||
|
|
||||||
|
_running = true;
|
||||||
|
//_monitorThread = std::make_unique<std::thread>(&CacheCleaner::monitorLoop, this);
|
||||||
|
}
|
||||||
|
|
||||||
|
void CacheCleaner::shutdown()
|
||||||
|
{
|
||||||
|
if (!_running) return;
|
||||||
|
_running = false;
|
||||||
|
|
||||||
|
if (_monitorThread && _monitorThread->joinable()) _monitorThread->join();
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string CacheCleaner::getLocalAppDataPath()
|
||||||
|
{
|
||||||
|
char buffer[MAX_PATH];
|
||||||
|
if (SHGetFolderPathA(NULL, CSIDL_LOCAL_APPDATA, NULL, 0, buffer) == S_OK)
|
||||||
|
{
|
||||||
|
return std::string(buffer);
|
||||||
|
}
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
|
void CacheCleaner::clearFolder(std::string& path)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
for (const auto& entry : std::filesystem::directory_iterator(path))
|
||||||
|
{
|
||||||
|
std::filesystem::remove_all(entry.path());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (...)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void CacheCleaner::monitorLoop()
|
||||||
|
{
|
||||||
|
HANDLE changeHandle = FindFirstChangeNotificationA(_cacheDir.c_str(), TRUE,
|
||||||
|
FILE_NOTIFY_CHANGE_FILE_NAME | FILE_NOTIFY_CHANGE_DIR_NAME |
|
||||||
|
FILE_NOTIFY_CHANGE_LAST_WRITE);
|
||||||
|
|
||||||
|
if (changeHandle == INVALID_HANDLE_VALUE)
|
||||||
|
{
|
||||||
|
Log::error("Failed to create change notification");
|
||||||
|
_running = false;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
Log::verbose("Cache cleaner monitoring started");
|
||||||
|
|
||||||
|
std::unordered_set<DWORD> previousPids;
|
||||||
|
|
||||||
|
while (_running)
|
||||||
|
{
|
||||||
|
// Wait 2 seconds between checks
|
||||||
|
for (int i = 0; i < 20 && _running; i++)
|
||||||
|
Sleep(100);
|
||||||
|
|
||||||
|
if (!_running) break;
|
||||||
|
|
||||||
|
// Take snapshot and check for new processes in one pass
|
||||||
|
HANDLE hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
|
||||||
|
if (hSnapshot == INVALID_HANDLE_VALUE) continue;
|
||||||
|
|
||||||
|
PROCESSENTRY32 pe32;
|
||||||
|
pe32.dwSize = sizeof(PROCESSENTRY32);
|
||||||
|
|
||||||
|
std::unordered_set<DWORD> currentPids;
|
||||||
|
|
||||||
|
if (Process32First(hSnapshot, &pe32))
|
||||||
|
{
|
||||||
|
do
|
||||||
|
{
|
||||||
|
DWORD pid = pe32.th32ProcessID;
|
||||||
|
currentPids.insert(pid);
|
||||||
|
|
||||||
|
if (previousPids.find(pid) == previousPids.end())
|
||||||
|
{
|
||||||
|
std::string processName(pe32.szExeFile);
|
||||||
|
|
||||||
|
for (char& c : processName)
|
||||||
|
c = static_cast<char>(tolower(c));
|
||||||
|
|
||||||
|
if (processName.find("deadbydaylight") != std::string::npos)
|
||||||
|
{
|
||||||
|
Log::verbose("New DeadByDaylight process detected (PID: {}), clearing cache", pid);
|
||||||
|
clearFolder(_cacheDir);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} while (Process32Next(hSnapshot, &pe32));
|
||||||
|
}
|
||||||
|
|
||||||
|
CloseHandle(hSnapshot);
|
||||||
|
previousPids = std::move(currentPids);
|
||||||
|
}
|
||||||
|
|
||||||
|
FindCloseChangeNotification(changeHandle);
|
||||||
|
}
|
||||||
@@ -0,0 +1,24 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
#include <atomic>
|
||||||
|
#include <thread>
|
||||||
|
|
||||||
|
class CacheCleaner
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
CacheCleaner();
|
||||||
|
~CacheCleaner();
|
||||||
|
|
||||||
|
void init();
|
||||||
|
void shutdown();
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::string getLocalAppDataPath();
|
||||||
|
void clearFolder(std::string& path);
|
||||||
|
void monitorLoop();
|
||||||
|
|
||||||
|
std::atomic<bool> _running;
|
||||||
|
std::unique_ptr<std::thread> _monitorThread;
|
||||||
|
std::string _cacheDir;
|
||||||
|
};
|
||||||
Reference in New Issue
Block a user