Common C++ classes and routines used by my applications such as argument parser, IO and conversion utilities
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

219 lines
7.0 KiB

7 years ago
#include "./path.h"
7 years ago
7 years ago
#include "../application/global.h"
#include "../conversion/widen.h"
7 years ago
#include <string>
#include <sstream>
#include <fstream>
#include <cstdlib>
#ifdef PLATFORM_UNIX
# include <unistd.h>
# include <sys/types.h>
# include <sys/stat.h>
# include <pwd.h>
# include <dirent.h>
7 years ago
#else
# ifdef PLATFORM_WINDOWS
# ifdef UNICODE
# undef UNICODE
# endif
# ifdef _UNICODE
# undef _UNICODE
# endif
# include <windows.h>
# endif
7 years ago
#endif
using namespace std;
using namespace ConversionUtilities;
namespace IoUtilities {
7 years ago
/*!
* \brief Returns the file name and extension of the specified \a path string.
7 years ago
*/
string fileName(const string &path)
7 years ago
{
size_t lastSlash = path.rfind('/');
size_t lastBackSlash = path.rfind('\\');
size_t lastSeparator;
if(lastSlash == string::npos && lastBackSlash == string::npos) {
7 years ago
return path;
} else if(lastSlash == string::npos) {
7 years ago
lastSeparator = lastBackSlash;
} else if(lastBackSlash == string::npos) {
7 years ago
lastSeparator = lastSlash;
} else {
7 years ago
lastSeparator = lastSlash > lastBackSlash ? lastSlash : lastBackSlash;
}
7 years ago
return path.substr(lastSeparator + 1);
}
/*!
* \brief Returns the directory of the specified \a path string (including trailing slash).
*/
string directory(const string &path)
{
size_t lastSlash = path.rfind('/');
size_t lastBackSlash = path.rfind('\\');
size_t lastSeparator;
if(lastSlash == string::npos && lastBackSlash == string::npos) {
return string();
} else if(lastSlash == string::npos) {
lastSeparator = lastBackSlash;
} else if(lastBackSlash == string::npos) {
lastSeparator = lastSlash;
} else {
lastSeparator = lastSlash > lastBackSlash ? lastSlash : lastBackSlash;
}
return path.substr(0, lastSeparator + 1);
}
7 years ago
/*!
* \brief Removes invalid characters from the specified \a fileName.
7 years ago
*
* The characters <, >, ?, !, *, |, /, :, \ and new lines are considered as invalid.
*/
void removeInvalidChars(std::string &fileName)
7 years ago
{
size_t startPos = 0;
static const char invalidPathChars[] = {'\"', '<', '>', '?', '!', '*', '|', '/', ':', '\\', '\n'};
for(const char *i = invalidPathChars, *end = invalidPathChars + sizeof(invalidPathChars); i != end; ++i) {
startPos = fileName.find(*i);
7 years ago
while(startPos != string::npos) {
fileName.replace(startPos, 1, string());
startPos = fileName.find(*i, startPos);
7 years ago
}
}
}
/*!
* \brief Locates a directory meant to store application settings.
7 years ago
* \param result Specifies a string to store the path in.
* \param applicationDirectoryName Specifies the name for the application subdirectory.
* \param createApplicationDirectory Indicates wheter the application subdirectory should be created if not present.
* \returns Returns if a settings directory could be located.
*/
bool settingsDirectory(std::string &result, std::string applicationDirectoryName, bool createApplicationDirectory)
7 years ago
{
result.clear();
fstream pathConfigFile("path.config", ios_base::in);
if(pathConfigFile.good()) {
for(string line; getline(pathConfigFile, line); ) {
string::size_type p = line.find('=');
if((p != string::npos) && (p + 1 < line.length())) {
string fieldName = line.substr(0, p);
if(fieldName == "settings") {
result.assign(line.substr(p + 1));
}
}
}
}
if(!result.empty()) {
#ifdef PLATFORM_UNIX
struct stat sb;
return (stat(result.c_str(), &sb) == 0 && S_ISDIR(sb.st_mode));
#else
# ifdef PLATFORM_WINDOWS
7 years ago
DWORD ftyp = GetFileAttributesA(result.c_str());
return (ftyp != INVALID_FILE_ATTRIBUTES) && (ftyp & FILE_ATTRIBUTE_DIRECTORY);
# else
# error Platform not supported.
# endif
7 years ago
#endif
} else {
if(!applicationDirectoryName.empty()) {
removeInvalidChars(applicationDirectoryName);
}
#ifdef PLATFORM_UNIX
if(char *homeDir = getenv("HOME")) {
result = string(homeDir);
} else {
struct passwd *pw = getpwuid(getuid());
result = string(pw->pw_dir);
}
struct stat sb;
result += "/.config";
if(createApplicationDirectory && !(stat(result.c_str(), &sb) == 0 && S_ISDIR(sb.st_mode))) {
if(mkdir(result.c_str(), S_IRUSR | S_IWUSR | S_IXUSR) != 0) {
return false;
}
}
if(!applicationDirectoryName.empty()) {
result += "/" + applicationDirectoryName;
if(createApplicationDirectory && !(stat(result.c_str(), &sb) == 0 && S_ISDIR(sb.st_mode))) {
if(mkdir(result.c_str(), S_IRUSR | S_IWUSR | S_IXUSR) != 0) {
return false;
}
}
}
#else
# ifdef PLATFORM_WINDOWS
7 years ago
if(char *appData = getenv("appdata")) {
result = appData;
if(!applicationDirectoryName.empty()) {
result += "\\" + applicationDirectoryName;
if(createApplicationDirectory) {
DWORD ftyp = GetFileAttributesA(result.c_str());
if(ftyp == INVALID_FILE_ATTRIBUTES) {
return false;
} else if(ftyp & FILE_ATTRIBUTE_DIRECTORY) {
return true;
} else {
if(CreateDirectory(result.c_str(), NULL) == 0) {
return false;
} else {
return true;
}
}
}
}
} else {
return false;
}
# else
# error Platform not supported.
# endif
#endif
7 years ago
}
return true;
}
/*!
* \brief Returns the names of the directory entries in the specified \a path with the specified \a types.
*/
std::list<std::string> directoryEntries(const char *path, DirectoryEntryType types)
{
#ifdef PLATFORM_UNIX
list<string> entries;
if(auto dir = opendir(path)) {
while(auto dirEntry = readdir(dir)) {
bool filter = false;
switch(dirEntry->d_type) {
case DT_REG:
filter = (types & DirectoryEntryType::File) != DirectoryEntryType::None;
break;
case DT_DIR:
filter = (types & DirectoryEntryType::Directory) != DirectoryEntryType::None;
break;
case DT_LNK:
filter = (types & DirectoryEntryType::Symlink) != DirectoryEntryType::None;
break;
default:
filter = (types & DirectoryEntryType::All) != DirectoryEntryType::None;
}
if(filter) {
entries.emplace_back(dirEntry->d_name);
}
}
closedir(dir);
}
return entries;
#else
return list<string>(); // TODO
#endif
}
}