blob: 87c31dbf0898d71143855e45c4488752149e9cc4 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
/*************************************************************************
** Directory.cpp **
** **
** This file is part of dvisvgm -- a fast DVI to SVG converter **
** Copyright (C) 2005-2022 Martin Gieseking <martin.gieseking@uos.de> **
** **
** This program is free software; you can redistribute it and/or **
** modify it under the terms of the GNU General Public License as **
** published by the Free Software Foundation; either version 3 of **
** the License, or (at your option) any later version. **
** **
** This program is distributed in the hope that it will be useful, but **
** WITHOUT ANY WARRANTY; without even the implied warranty of **
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the **
** GNU General Public License for more details. **
** **
** You should have received a copy of the GNU General Public License **
** along with this program; if not, see <http://www.gnu.org/licenses/>. **
*************************************************************************/
#include "Directory.hpp"
using namespace std;
#ifndef _WIN32
#include <cerrno>
#include <sys/stat.h>
#endif
Directory::Directory () {
#ifdef _WIN32
memset(&_fileData, 0, sizeof(WIN32_FIND_DATA));
#endif
}
Directory::Directory (const string &dirname) : Directory() {
open(dirname);
}
Directory::~Directory () {
close();
}
bool Directory::open (string dirname) {
_dirname = dirname;
#ifdef _WIN32
_firstread = true;
if (dirname[dirname.length()-1] == '/' || dirname[dirname.length()-1] == '\\')
dirname = dirname.substr(0, dirname.length()-1);
dirname += "\\*";
_handle = FindFirstFile(dirname.c_str(), &_fileData);
return _handle != INVALID_HANDLE_VALUE;
#else
_dir = opendir(_dirname.c_str());
return bool(_dir);
#endif
}
void Directory::close () {
#ifdef _WIN32
FindClose(_handle);
#else
if (_dir) {
closedir(_dir);
_dir = nullptr;
}
#endif
}
/** Reads first/next directory entry.
* @param[in] type type of entry to return (a: file or dir, f: file, d: dir)
* @return name of entry */
const char* Directory::read (EntryType type) {
#ifdef _WIN32
if (_handle == INVALID_HANDLE_VALUE)
return nullptr;
while (_firstread || FindNextFile(_handle, &_fileData)) {
_firstread = false;
if (_fileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
if (type == ET_FILE_OR_DIR || type == ET_DIR)
return _fileData.cFileName;
}
else if (type == ET_FILE_OR_DIR || type == ET_FILE)
return _fileData.cFileName;
}
FindClose(_handle);
_handle = INVALID_HANDLE_VALUE;
return nullptr;
#else
if (_dir) {
while ((_dirent = readdir(_dir))) {
string path = _dirname + "/" + _dirent->d_name;
struct stat stats;
if (stat(path.c_str(), &stats) == 0) {
if (S_ISDIR(stats.st_mode)) {
if (type == ET_FILE_OR_DIR || type == ET_DIR)
return _dirent->d_name;
}
else if (type == ET_FILE_OR_DIR || type == ET_FILE)
return _dirent->d_name;
}
}
closedir(_dir);
_dir = nullptr;
}
return nullptr;
#endif
}
|
