#ifndef NALL_DIRECTORY_HPP #define NALL_DIRECTORY_HPP #include #include #include #include #include #if defined(PLATFORM_WINDOWS) #include #else #include #include #include #endif namespace nall { struct directory { static bool create(const string& pathname, unsigned permissions = 0755); //recursive static bool remove(const string& pathname); //recursive static bool exists(const string& pathname); static lstring folders(const string& pathname, const string& pattern = "*") { lstring folders = directory::ufolders(pathname, pattern); folders.sort(); return folders; } static lstring files(const string& pathname, const string& pattern = "*") { lstring files = directory::ufiles(pathname, pattern); files.sort(); return files; } static lstring contents(const string& pathname, const string& pattern = "*") { lstring folders = directory::ufolders(pathname); //pattern search of contents should only filter files lstring files = directory::ufiles(pathname, pattern); folders.sort(); files.sort(); for(auto& file : files) folders.append(file); return folders; } static lstring ifolders(const string& pathname, const string& pattern = "*") { lstring folders = ufolders(pathname, pattern); folders.isort(); return folders; } static lstring ifiles(const string& pathname, const string& pattern = "*") { lstring files = ufiles(pathname, pattern); files.isort(); return files; } static lstring icontents(const string& pathname, const string& pattern = "*") { lstring folders = directory::ufolders(pathname); //pattern search of contents should only filter files lstring files = directory::ufiles(pathname, pattern); folders.isort(); files.isort(); for(auto& file : files) folders.append(file); return folders; } private: //internal functions; these return unsorted lists static lstring ufolders(const string& pathname, const string& pattern = "*"); static lstring ufiles(const string& pathname, const string& pattern = "*"); }; #if defined(PLATFORM_WINDOWS) inline bool directory::create(const string& pathname, unsigned permissions) { string path; lstring list = string{pathname}.transform("\\", "/").rtrim<1>("/").split("/"); bool result = true; for(auto& part : list) { path.append(part, "/"); result &= (_wmkdir(utf16_t(path)) == 0); } return result; } inline bool directory::remove(const string& pathname) { lstring list = directory::contents(pathname); for(auto& name : list) { if(name.endsWith("/")) directory::remove({pathname, name}); else file::remove({pathname, name}); } return _wrmdir(utf16_t(pathname)) == 0; } inline bool directory::exists(const string& pathname) { string name = pathname; name.trim<1>("\""); DWORD result = GetFileAttributes(utf16_t(name)); if(result == INVALID_FILE_ATTRIBUTES) return false; return (result & FILE_ATTRIBUTE_DIRECTORY); } inline lstring directory::ufolders(const string& pathname, const string& pattern) { lstring list; string path = pathname; path.transform("/", "\\"); if(!strend(path, "\\")) path.append("\\"); path.append("*"); HANDLE handle; WIN32_FIND_DATA data; handle = FindFirstFile(utf16_t(path), &data); if(handle != INVALID_HANDLE_VALUE) { if(wcscmp(data.cFileName, L".") && wcscmp(data.cFileName, L"..")) { if(data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) { string name = (const char*)utf8_t(data.cFileName); if(name.match(pattern)) list.append(name); } } while(FindNextFile(handle, &data) != false) { if(wcscmp(data.cFileName, L".") && wcscmp(data.cFileName, L"..")) { if(data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) { string name = (const char*)utf8_t(data.cFileName); if(name.match(pattern)) list.append(name); } } } FindClose(handle); } for(auto& name : list) name.append("/"); //must append after sorting return list; } inline lstring directory::ufiles(const string& pathname, const string& pattern) { lstring list; string path = pathname; path.transform("/", "\\"); if(!strend(path, "\\")) path.append("\\"); path.append("*"); HANDLE handle; WIN32_FIND_DATA data; handle = FindFirstFile(utf16_t(path), &data); if(handle != INVALID_HANDLE_VALUE) { if((data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == 0) { string name = (const char*)utf8_t(data.cFileName); if(name.match(pattern)) list.append(name); } while(FindNextFile(handle, &data) != false) { if((data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) == 0) { string name = (const char*)utf8_t(data.cFileName); if(name.match(pattern)) list.append(name); } } FindClose(handle); } return list; } #else inline bool directory::create(const string& pathname, unsigned permissions) { string path; lstring list = string{pathname}.rtrim<1>("/").split("/"); bool result = true; for(auto& part : list) { path.append(part, "/"); result &= (mkdir(path, permissions) == 0); } return result; } inline bool directory::remove(const string& pathname) { lstring list = directory::contents(pathname); for(auto& name : list) { if(name.endsWith("/")) directory::remove({pathname, name}); else file::remove({pathname, name}); } return rmdir(pathname) == 0; } inline bool directory::exists(const string& pathname) { DIR *dp = opendir(pathname); if(!dp) return false; closedir(dp); return true; } inline lstring directory::ufolders(const string& pathname, const string& pattern) { lstring list; DIR* dp; struct dirent* ep; dp = opendir(pathname); if(dp) { while(ep = readdir(dp)) { if(!strcmp(ep->d_name, ".")) continue; if(!strcmp(ep->d_name, "..")) continue; bool is_directory = ep->d_type & DT_DIR; if(ep->d_type & DT_UNKNOWN) { struct stat sp = {0}; stat(string{pathname, ep->d_name}, &sp); is_directory = S_ISDIR(sp.st_mode); } if(is_directory) { if(strmatch(ep->d_name, pattern)) list.append(ep->d_name); } } closedir(dp); } for(auto& name : list) name.append("/"); //must append after sorting return list; } inline lstring directory::ufiles(const string& pathname, const string& pattern) { lstring list; DIR* dp; struct dirent* ep; dp = opendir(pathname); if(dp) { while(ep = readdir(dp)) { if(!strcmp(ep->d_name, ".")) continue; if(!strcmp(ep->d_name, "..")) continue; if((ep->d_type & DT_DIR) == 0) { if(strmatch(ep->d_name, pattern)) list.append(ep->d_name); } } closedir(dp); } return list; } #endif } #endif