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
|
/*
* Copyright 2023 ailurux <ailuruxx@gmail.com>
*
* SPDX-License-Identifier: GPL-3.0-only
*/
#include "lua/file_iterator.hpp"
#include "esp_log.h"
#include <string>
#include "drivers/spi.hpp"
#include "ff.h"
namespace lua {
[[maybe_unused]] static const char* kTag = "FileIterator";
FileIterator::FileIterator(std::string filepath, bool showHidden)
: original_path_(filepath), show_hidden_(showHidden), current_(), offset_(-1) {
const TCHAR* path = static_cast<const TCHAR*>(filepath.c_str());
FRESULT res = f_opendir(&dir_, path);
if (res != FR_OK) {
ESP_LOGE(kTag, "Error opening directory: %s", filepath.c_str());
}
}
FileIterator::~FileIterator() {
f_closedir(&dir_);
}
auto FileIterator::value() const -> const std::optional<FileEntry>& {
return current_;
}
auto FileIterator::next() -> void {
size_t prev_index = -1;
if (current_) {
prev_index = current_->index;
}
do {
bool res = iterate(show_hidden_);
if (!res) {
break;
}
} while (!current_ || current_->index == prev_index);
}
auto FileIterator::prev() -> void {
f_rewinddir(&dir_);
if (offset_ <= 0) {
offset_ = -1;
current_.reset();
return;
}
auto new_offset = offset_ - 1;
offset_ = -1;
while (offset_ < new_offset) {
if (!iterate(show_hidden_)) {
break;
}
}
}
auto FileIterator::iterate(bool show_hidden) -> bool {
FILINFO info;
auto res = f_readdir(&dir_, &info);
if (res != FR_OK) {
ESP_LOGE(kTag, "Error reading directory. Error: %d", res);
return false;
}
if (info.fname[0] == 0) {
// End of directory
// Set value to nil
current_.reset();
return false;
} else {
// Update current value
bool hidden = (info.fattrib & AM_HID) > 0 || info.fname[0] == '.';
if (!hidden || show_hidden) {
offset_++;
current_ = FileEntry{
.index = offset_,
.isHidden = hidden,
.isDirectory = (info.fattrib & AM_DIR) > 0,
.filepath = original_path_ + (original_path_.size() > 0 ? "/" : "") +
info.fname,
.name = info.fname,
};
}
}
return true;
}
} // namespace lua
|
