AudioSourceSPIFFS.h

#pragma once
 
#include <SPIFFS.h>
#include "AudioLogger.h"
#include "AudioTools/Disk/AudioSource.h"
#include "AudioTools/Disk/SDDirect.h"
 
namespace audio_tools {
 
class AudioSourceSPIFFS : public AudioSource {
public:
  AudioSourceSPIFFS(const char *startFilePath = "/", const char *ext = ".mp3") {
    start_path = startFilePath;
    exension = ext;
  }
  
  virtual ~AudioSourceSPIFFS() {
    end();
  }
 
  virtual bool begin() override {
    TRACED();
    if (!is_sd_setup) {
      int retry = 10;
      while (!SPIFFS.begin()) {
        LOGE("SPIFFS.begin failed");
        delay(500);
        if (--retry <= 0) {
          return false;
        }
      }
      is_sd_setup = true;
    }
    idx.begin(start_path, exension, file_name_pattern);
    idx_pos = 0;
    return is_sd_setup;
  }
 
  void end() {
    file.close();
    SPIFFS.end();
    is_sd_setup = false;
  }
 
  virtual Stream *nextStream(int offset = 1) override {
    LOGI("nextStream: %d", offset);
    return selectStream(idx_pos + offset);
  }
 
  virtual Stream *selectStream(int index) override {
    LOGI("selectStream: %d", index);
    file.close();
    idx_pos = index;
    file_name = idx[index];
    if (file_name==nullptr) return nullptr;
    LOGI("Using file %s", file_name);
    file = SPIFFS.open(file_name);
    return file ? &file : nullptr;
  }
 
  virtual Stream *selectStream(const char *path) override {
    file.close();
    file = SPIFFS.open(path);
    file_name = file.name();
    LOGI("-> selectStream: %s", path);
    return file ? &file : nullptr;
  }
 
  void setFileFilter(const char *filter) { file_name_pattern = filter; }
 
  int index() { return idx_pos; }
 
  const char *toStr() { return file_name; }
 
  // provides default setting go to the next
  virtual bool isAutoNext() { return true; }
 
  virtual void setPath(const char *p) { start_path = p; }
 
  long size() { return idx.size();}
 
protected:
  SDDirect<fs::SPIFFSFS,fs::File> idx{SPIFFS};
  File file;
  size_t idx_pos = 0;
  const char *file_name;
  const char *exension = nullptr;
  const char *start_path = nullptr;
  const char *file_name_pattern = "*";
  bool is_sd_setup = false;
};
 
} // namespace audio_tools