#ifndef SPIFFS_TASKS_H
#define SPIFFS_TASKS_H

#include "FS.h"
#include "SPIFFS.h"

void listDir(fs::FS &fs, const char *dirname, uint8_t levels) {
    Serial.printf("Listing directory: %s\r\n", dirname);
    File root = fs.open(dirname);
    if (!root) {
        Serial.println("- failed to open directory");
        return;
    }
    if (!root.isDirectory()) {
        Serial.println(" - not a directory");
        return;
    }
    
    File file = root.openNextFile();
    while (file) {
        if (file.isDirectory()) {
            Serial.print("  DIR : ");
            Serial.println(file.name());
            if (levels) {
                listDir(fs, file.name(), levels - 1);
            }
        } else {
            Serial.print("  FILE: ");
            Serial.print(file.name());
            Serial.print("\tSIZE: ");
            Serial.println(file.size());
        }
        file = root.openNextFile();
    }
}

void readFile(fs::FS &fs, const char *path) {
    Serial.printf("Reading file: %s\r\n", path);
    File file = fs.open(path);
    if (!file || file.isDirectory()) {
        Serial.println("- failed to open file for reading");
        return;
    }
    
    Serial.println("- read from file:");
    while (file.available()) {
        Serial.write(file.read());
    }
    file.close();
}

void writeFile(fs::FS &fs, const char *path, const char *message) {
    Serial.printf("Writing file: %s\r\n", path);
    File file = fs.open(path, FILE_WRITE);
    if (!file) {
        Serial.println("- failed to open file for writing");
        return;
    }
    
    if (file.print(message)) {
        Serial.println("- file written");
    } else {
        Serial.println("- write failed");
    }
    file.close();
}

void appendFile(fs::FS &fs, const char *path, const char *message) {
    Serial.printf("Appending to file: %s\r\n", path);
    File file = fs.open(path, FILE_APPEND);
    if (!file) {
        Serial.println("- failed to open file for appending");
        return;
    }
    
    if (file.print(message)) {
        Serial.println("- message appended");
    } else {
        Serial.println("- append failed");
    }
    file.close();
}

void renameFile(fs::FS &fs, const char *path1, const char *path2) {
    Serial.printf("Renaming file %s to %s\r\n", path1, path2);
    if (fs.rename(path1, path2)) {
        Serial.println("- file renamed");
    } else {
        Serial.println("- rename failed");
    }
}

void deleteFile(fs::FS &fs, const char *path) {
    Serial.printf("Deleting file: %s\r\n", path);
    if (fs.remove(path)) {
        Serial.println("- file deleted");
    } else {
        Serial.println("- delete failed");
    }
}

void testFileIO(fs::FS &fs, const char *path) {
    Serial.printf("Testing file I/O with %s\r\n", path);
    static uint8_t buf[512];
    size_t len = 0;
    File file = fs.open(path, FILE_WRITE);
    if (!file) {
        Serial.println("- failed to open file for writing");
        return;
    }

    Serial.print("- writing");
    uint32_t start = millis();
    for (size_t i = 0; i < 2048; i++) {
        if ((i & 0x001F) == 0x001F) {
            Serial.print(".");
        }
        file.write(buf, 512);
    }
    Serial.println("");
    uint32_t end = millis() - start;
    Serial.printf(" - %u bytes written in %u ms\r\n", 2048 * 512, end);
    file.close();

    file = fs.open(path);
    start = millis();
    end = start;
    size_t i = 0;
    if (file && !file.isDirectory()) {
        len = file.size();
        size_t flen = len;
        start = millis();
        Serial.print("- reading");
        while (len) {
            size_t toRead = len > 512 ? 512 : len;
            file.read(buf, toRead);
            if ((i++ & 0x001F) == 0x001F) {
                Serial.print(".");
            }
            len -= toRead;
        }
        Serial.println("");
        end = millis() - start;
        Serial.printf("- %u bytes read in %u ms\r\n", flen, end);
        file.close();
    } else {
        Serial.println("- failed to open file for reading");
    }
}

void readFile(const char *path) {
    File file = SPIFFS.open(path, "r");
    if (file) {
        Serial.print("File: " + String(path) + ", Content: ");

        if (String(path).endsWith(".bin")) {
            while (file.available()) {
                Serial.print(file.read(), HEX);
                Serial.print(" ");
            }
        } else {
            Serial.print(file.readString());
        }

        Serial.println();
        file.close();
    } else {
        Serial.println("Failed to open file for reading: " + String(path));
    }
}

void writeBinaryDataToFile(const char *path) {
    Serial.print("Creating file with binary data: ");
    Serial.println(String(path));

    File file = SPIFFS.open(path, "w");
    if (file) {
        uint8_t binaryData[] = {0x01, 0x02, 0x03, 0x04};
        file.write(binaryData, sizeof(binaryData));
        file.close();
        Serial.println("Binary file created and written successfully");
    } else {
        Serial.println("Failed to create binary file");
    }
}

void writeFileWithPrint(const char *path, const char *content) {
    Serial.print("Creating file with SPIFFS.print(): ");
    Serial.println(String(path));

    File file = SPIFFS.open(path, "w");
    if (file) {
        file.print(content);
        file.close();
        Serial.println("Text file created and written successfully");
    } else {
        Serial.println("Failed to create text file");
    }
}

void readAndPrintFiles() {
    Serial.println("Reading and printing files:");
    readFile("/binaryFile.bin");
    readFile("/textFile.txt");
}

void listFiles(const char *dir) {
    Serial.print("Listing files in directory: ");
    Serial.println(String(dir));

    File root = SPIFFS.open(dir);
    if (!root) {
        Serial.println("Failed to open directory");
        return;
    }

    File file = root.openNextFile();
    while (file) {
        Serial.println("File: " + String(file.name()) + ", Size: " + String(file.size()));
        file = root.openNextFile();
    }
}

#endif // SPIFFS_TASKS_H