path:补充家目录相关函数。

This commit is contained in:
taynpg 2024-12-01 17:03:56 +08:00
parent 50f455bd23
commit 0b760f5363
3 changed files with 60 additions and 7 deletions

View File

@ -16,6 +16,10 @@ public:
static ofString replaceStr(const ofString& str, const ofString& from, const ofString& to);
static ofString getFullRunPath();
static ofString getHome();
static ofString getConfigDir(const ofString& sub_dir, bool create = false);
static ofString getFull(const ofString& path, const ofString& sub_file_path);
static bool isExist(const ofString& path);
static bool writeBin(const ofString& path, const char* data, int len);
};
}; // namespace ofen
#endif

View File

@ -1,5 +1,6 @@
#include "of_path.h"
#include <filesystem>
#include <fstream>
#ifdef _WIN32
#include <windows.h>
@ -64,7 +65,7 @@ ofString COfPath::getFullRunPath()
#elif defined(__clang__) && defined(__APPLE__)
uint32_t size = 0;
_NSGetExecutablePath(NULL, &size); // 获取路径缓冲区的大小
std::vector<ofChar> buffer(size); // 创建缓冲区
std::vector<ofChar> buffer(size); // 创建缓冲区
if (_NSGetExecutablePath(buffer.data(), &size) != 0) {
return ofT("");
}
@ -85,17 +86,20 @@ ofString COfPath::getHome()
#if defined(_WIN32)
ofChar* value = nullptr;
std::size_t len = 0;
#ifdef UNICODE_OFSTR
auto err = _wdupenv_s(&value, &len, ofT("USERPROFILE"));
#else
auto err = _dupenv_s(&value, &len, "USERPROFILE");
#endif
if (err == 0 && value != nullptr) {
ofString ret(value);
free(value);
return ret;
}
else {
} else {
return ofT("");
}
#else
ofChar *homedir = getenv("HOME");
ofChar* homedir = getenv("HOME");
if (homedir) {
return ofString(homedir);
}
@ -103,4 +107,49 @@ ofString COfPath::getHome()
#endif
}
ofString COfPath::getConfigDir(const ofString& sub_dir, bool create)
{
fs::path userHome = fs::path(getHome());
userHome.append(".config");
userHome.append(sub_dir);
if (create) {
if (!fs::exists(userHome)) {
fs::create_directories(userHome);
}
}
#ifdef UNICODE_OFSTR
return ofString(userHome.wstring());
#else
return ofString(userHome.string());
#endif
}
ofString COfPath::getFull(const ofString& path, const ofString& sub_file_path)
{
fs::path p(path);
p.append(sub_file_path);
#ifdef UNICODE_OFSTR
return ofString(p.wstring());
#else
return ofString(p.string());
#endif
}
bool COfPath::isExist(const ofString& path)
{
fs::path p(path);
return fs::exists(p);
}
bool COfPath::writeBin(const ofString& path, const char* data, int len)
{
std::ofstream file(path, std::ios::binary);
if (!file.is_open()) {
return false;
}
file.write(data, len);
file.close();
return true;
}
} // namespace ofen

View File

@ -24,18 +24,18 @@ OfStatus COfWin::appendPathVariable(const ofString& path)
return STA_CANOT_OPEN_REG;
}
DWORD bufferSize = 0;
if (RegQueryValueEx(hKey, "Path", nullptr, nullptr, nullptr, &bufferSize) != ERROR_SUCCESS) {
if (RegQueryValueEx(hKey, ofT("Path"), nullptr, nullptr, nullptr, &bufferSize) != ERROR_SUCCESS) {
RegCloseKey(hKey);
return STA_CANOT_QUERY_REG;
}
ofString currentPath(bufferSize, ofT('\0'));
if (RegQueryValueEx(hKey, "Path", nullptr, nullptr, reinterpret_cast<LPBYTE>(&currentPath[0]), &bufferSize) != ERROR_SUCCESS) {
if (RegQueryValueEx(hKey, ofT("Path"), nullptr, nullptr, reinterpret_cast<LPBYTE>(&currentPath[0]), &bufferSize) != ERROR_SUCCESS) {
RegCloseKey(hKey);
return STA_CANOT_QUERY_REG;
}
currentPath.resize(bufferSize - 1);
ofString updatedPath = currentPath + ofT(";") + path;
if (RegSetValueEx(hKey, "Path", 0, REG_EXPAND_SZ, reinterpret_cast<const BYTE*>(updatedPath.c_str()), updatedPath.size() + 1) != ERROR_SUCCESS) {
if (RegSetValueEx(hKey, ofT("Path"), 0, REG_EXPAND_SZ, reinterpret_cast<const BYTE*>(updatedPath.c_str()), updatedPath.size() + 1) != ERROR_SUCCESS) {
RegCloseKey(hKey);
return STA_CANOT_SET_REG;
}