SdFatParser.h

Go to the documentation of this file.

#pragma once
#include <Print.h>
 
#include <string>
 
namespace tiny_dlna {
 
struct SdFatFileInfo {
  std::string name;
  bool is_directory;
  int level;
};
 
class SdFatParser : public Print {
 public:
  SdFatParser() { name.reserve(80); }
 
  size_t write(uint8_t c) override {
    switch (c) {
      case '\n':
        parse();
        break;
      case '\t':
        break;
      default:
        name += c;
        break;
    }
    return 1;
  }
 
  void setCallback(void (*cb)(SdFatFileInfo&, void* ref), void* ref = nullptr) {
    this->ref = ref;
    this->cb = cb;
  }
 
 protected:
  std::string name;
  void (*cb)(SdFatFileInfo&, void* ref) = nullptr;
  void* ref = nullptr;
  SdFatFileInfo info;
 
  int spaceCount() {
    for (int j = 0; j < static_cast<int>(name.size()); j++) {
      if (name[j] != ' ') return j;
    }
    return 0;
  }
 
  void parse() {
    int spaces = spaceCount();
    info.level = spaces / 2;
    trim(name);
    info.name = name;
    // directories have a trailing /
    info.is_directory =
        !info.name.empty() && info.name[info.name.size() - 1] == '/';
    // if it is a directory remove trailing /
    if (info.is_directory) {
        info.name.pop_back();
    }
    if (cb) cb(info, ref);
    name.clear();
  }
 
  inline void rtrim(std::string& s) {
    s.erase(std::find_if(s.rbegin(), s.rend(),
                         [](unsigned char ch) { return !std::isspace(ch); })
                .base(),
            s.end());
  }
 
  // Trim from the start (in place)
  inline void ltrim(std::string& s) {
    s.erase(s.begin(), std::find_if(s.begin(), s.end(), [](unsigned char ch) {
              return !std::isspace(ch);
            }));
  }
 
  // Trim from both ends (in place)
  inline void trim(std::string& s) {
    rtrim(s);
    ltrim(s);
  }
};
 
}  // namespace tiny_dlna