SD.h

/*
  SD.h 
  Copyright (c) 2025 Phil Schatzmann. All right reserved.
 
  This library is free software; you can redistribute it and/or
  modify it under the terms of the GNU Lesser General Public
  License as published by the Free Software Foundation; either
  version 2.1 of the License, or (at your option) any later version.
 
  This library 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
  Lesser General Public License for more details.
 
  You should have received a copy of the GNU Lesser General Public
  License along with this library; if not, write to the Free Software
  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*/
#pragma once
 
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
 
#include <cstdint>
#include <filesystem>
#include <fstream>
#include <iostream>
#include <string>
 
#include "Stream.h"
#ifdef USE_FILESYSTEM
#include <filesystem>
#endif
 
#define SD_SCK_MHZ(maxMhz) (1000000UL * (maxMhz))
#define SPI_FULL_SPEED SD_SCK_MHZ(50)
#define SPI_DIV3_SPEED SD_SCK_MHZ(16)
#define SPI_HALF_SPEED SD_SCK_MHZ(4)
#define SPI_DIV6_SPEED SD_SCK_MHZ(8)
#define SPI_QUARTER_SPEED SD_SCK_MHZ(2)
#define SPI_EIGHTH_SPEED SD_SCK_MHZ(1)
#define SPI_SIXTEENTH_SPEED SD_SCK_HZ(500000)
 
#define O_RDONLY ios::in           
#define O_WRONLY ios::out          
#define O_RDWR ios::in | ios::out  
#define O_AT_END ios::ate          
#define O_APPEND ios::ate          
#define O_CREAT ios::trunc         
#define O_TRUNC ios::trunc         
#define O_EXCL 0                   
#define O_SYNC 0                   
#define O_READ O_RDONLY
#define O_WRITE O_WRONLY
#define FILE_READ ios::in
#define FILE_WRITE ios::out
 
#ifndef SS
#define SS -1
#endif
 
using namespace std;
 
class SdSpiConfig {
  SdSpiConfig(int a = 0, int b = 0, int c = 0) {}
};
 
class File : public Stream {
 public:
  bool isOpen() {
    if (filename == "") return false;
    if (is_dir) return true;
    bool result = file.is_open();
    return result;
  }
 
  bool open(const char *name, int flags) {
    this->filename = name;
    int rc = stat(name, &info);
    if ((flags == O_RDONLY) && rc == -1) {
      // if we want to read but the file does not exist we fail
      return false;
    } else if (rc == 0 && info.st_mode & S_IFDIR) {
      // file exists and it is a directory
      is_dir = true;
      size_bytes = 0;
#ifdef USE_FILESYSTEM
      // prevent authorization exceptions
      try {
        dir_path = std::filesystem::path(filename);
        iterator = std::filesystem::directory_iterator({dir_path});
      } catch (...) {
      }
#endif
      return true;
    } else {
      is_dir = false;
      size_bytes = rc != -1 ? info.st_size : 0;
      file.open(filename.c_str(), toMode(flags));
      return isOpen();
    }
  }
 
  bool close() {
    file.close();
    return !isOpen();
  }
 
  size_t read(uint8_t *buffer, size_t length) {
    return readBytes((char *)buffer, length);
  }
 
  size_t readBytes(char *buffer, size_t length) {
    file.read(buffer, length);
    return static_cast<size_t>(file.gcount());
  }
 
  size_t write(const uint8_t *buffer, size_t size) override {
    file.write((const char *)buffer, size);
    return file.good() ? size : 0;
  }
  size_t write(uint8_t ch) override {
    file.put(ch);
    return file.good() ? 1 : 0;
  }
  int available() override { return size_bytes - file.tellg(); };
  int availableForWrite() override { return 1024; }
  int read() override { return file.get(); }
  int peek() override { return file.peek(); }
  void flush() override {}
  void getName(char *str, size_t len) { strncpy(str, filename.c_str(), len); }
  bool isDir() { return is_dir; }
  bool isHidden() { return false; }
  bool rewind() {
    pos = 0;
    return is_dir;
  }
  bool openNext(File &dir, int flags = O_RDONLY) {
    if (!*this) return false;
 
    // if (pos == 0) {
    //   dir_path = std::filesystem::path(filename);
    //   iterator = std::filesystem::directory_iterator({dir_path});
    // }
    // pos++;
#ifdef USE_FILESYSTEM
    if (iterator != end(iterator)) {
      std::filesystem::directory_entry entry = *iterator++;
      return dir.open(entry.path().c_str(), flags);
    }
#endif
    return false;
  }
  int dirIndex() { return pos; }
 
  File openNextFile() {
    File next_file;
    openNext(next_file, O_RDONLY);
    return next_file;
  }
 
  size_t size() {
    int rc = stat(filename.c_str(), &info);
    size_bytes = rc != -1 ? info.st_size : 0;
    return size_bytes;
  }
 
  size_t position() { return file.tellg(); }
 
  bool seek(size_t pos) {
    file.seekg(pos);
    return file ? true : false;
  }
 
  operator bool() { return isOpen(); }
 
  bool isDirectory() { return is_dir; }
 
  const char *name() { return filename.c_str(); }
 
  void rewindDirectory() {
    if (!is_dir) return;
#ifdef USE_FILESYSTEM
    try {
      iterator = std::filesystem::directory_iterator(dir_path);
    } catch (...) {
      // Handle errors silently
    }
#endif
  }
 
 protected:
  std::fstream file;
  size_t size_bytes = 0;
  bool is_dir = false;
  int pos = 0;
  std::string filename = "";
  struct stat info;
 
#ifdef USE_FILESYSTEM
  std::filesystem::directory_iterator iterator;
  std::filesystem::path dir_path;
#endif
 
  std::ios_base::openmode toMode(int flags) {
    return (std::ios_base::openmode)flags;
  }
};
 
class SdFat {
 public:
  bool begin(int cs = SS, int speed = 0) { return true; }
 
  bool begin(SdSpiConfig &cfg) { return true; }
 
  void end() {}
 
  void errorHalt(const char *msg) {
    puts(msg);
    exit(0);
  }
  void initErrorHalt() { exit(0); }
 
  bool exists(const char *name) {
    struct stat info;
    return stat(name, &info) == 0;
  }
 
  File open(const char *name, int flags = O_READ) {
    File file;
    file.open(name, flags);
    return file;
  }
 
  bool remove(const char *name) { return std::remove(name) == 0; }
  bool mkdir(const char *name) { return ::mkdir(name, 0777) == 0; }
  bool rmdir(const char *path) {
    int rc = ::rmdir(path);
    return rc == 0;
  }
  uint64_t totalBytes() {
    std::error_code ec;
    const std::filesystem::space_info si = std::filesystem::space("/home", ec);
    return si.capacity;
  }
  uint64_t usedBytes() {
    std::error_code ec;
    const std::filesystem::space_info si = std::filesystem::space("/home", ec);
    return si.capacity - si.available;
  }
};
 
static SdFat SD;
using SDClass = SdFat;