summaryrefslogtreecommitdiff
path: root/src/tangara/lua/file_iterator.cpp
blob: d0eb0bae67d6d0da8a45020487487bae4efdcc03 (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
/*
 * 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 "ff.h"
#include "drivers/spi.hpp"

namespace lua {

[[maybe_unused]] static const char* kTag = "FileIterator";

FileIterator::FileIterator(std::string filepath) 
: original_path_(filepath),
  current_(),
  offset_(-1)
 {
  auto lock = drivers::acquire_spi();

  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() {
  auto lock = drivers::acquire_spi();
  f_closedir(&dir_);
}

auto FileIterator::value() const -> const std::optional<FileEntry>& {
  return current_;
}

auto FileIterator::next() -> void {
  iterate(false);
}

auto FileIterator::prev() -> void {
  if (offset_ == 0) {
    current_.reset();
    return;
  }
  f_rewinddir(&dir_);
  auto new_offset = offset_-1;
  offset_ = -1;
  for (int i = 0; i <= new_offset; i++) {
    iterate(false);
  }
}

auto FileIterator::iterate(bool reverse) -> bool {
    FILINFO info;
    {
      auto lock = drivers::acquire_spi();
      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();
    } else {
      // Update current value
      offset_++;
      current_ = FileEntry{
        .index = offset_,
        .isHidden = (info.fattrib & AM_HID) > 0,
        .isDirectory = (info.fattrib & AM_DIR) > 0,
        .isTrack = false, // TODO
        .filepath = original_path_ + (original_path_.size()>0?"/":"") + info.fname,

      };
    }
    return true;
}

}  // namespace lua