flipperzero-firmware/lib/toolbox/dir_walk.c

#include "dir_walk.h"
#include <m-list.h>

LIST_DEF(DirIndexList, uint32_t);

struct DirWalk {
    File* file;
    string_t path;
    DirIndexList_t index_list;
    uint32_t current_index;
    bool recursive;
    DirWalkFilterCb filter_cb;
    void* filter_context;
};

DirWalk* dir_walk_alloc(Storage* storage) {
    DirWalk* dir_walk = malloc(sizeof(DirWalk));
    string_init(dir_walk->path);
    dir_walk->file = storage_file_alloc(storage);
    DirIndexList_init(dir_walk->index_list);
    dir_walk->recursive = true;
    dir_walk->filter_cb = NULL;
    return dir_walk;
}

void dir_walk_free(DirWalk* dir_walk) {
    storage_file_free(dir_walk->file);
    string_clear(dir_walk->path);
    DirIndexList_clear(dir_walk->index_list);
    free(dir_walk);
}

void dir_walk_set_recursive(DirWalk* dir_walk, bool recursive) {
    dir_walk->recursive = recursive;
}

void dir_walk_set_filter_cb(DirWalk* dir_walk, DirWalkFilterCb cb, void* context) {
    dir_walk->filter_cb = cb;
    dir_walk->filter_context = context;
}

bool dir_walk_open(DirWalk* dir_walk, const char* path) {
    string_set(dir_walk->path, path);
    dir_walk->current_index = 0;
    return storage_dir_open(dir_walk->file, path);
}

static bool dir_walk_filter(DirWalk* dir_walk, const char* name, FileInfo* fileinfo) {
    if(dir_walk->filter_cb) {
        return dir_walk->filter_cb(name, fileinfo, dir_walk->filter_context);
    } else {
        return true;
    }
}

static DirWalkResult dir_walk_iter(DirWalk* dir_walk, string_t return_path, FileInfo* fileinfo) {
    DirWalkResult result = DirWalkError;
    char* name = malloc(256); // FIXME: remove magic number
    FileInfo info;
    bool end = false;

    while(!end) {
        storage_dir_read(dir_walk->file, &info, name, 255);

        if(storage_file_get_error(dir_walk->file) == FSE_OK) {
            result = DirWalkOK;
            dir_walk->current_index++;

            if(dir_walk_filter(dir_walk, name, &info)) {
                if(return_path != NULL) {
                    string_printf(return_path, "%s/%s", string_get_cstr(dir_walk->path), name);
                }

                if(fileinfo != NULL) {
                    memcpy(fileinfo, &info, sizeof(FileInfo));
                }

                end = true;
            }

            if((info.flags & FSF_DIRECTORY) && dir_walk->recursive) {
                // step into
                DirIndexList_push_back(dir_walk->index_list, dir_walk->current_index);
                dir_walk->current_index = 0;
                storage_dir_close(dir_walk->file);

                string_cat_printf(dir_walk->path, "/%s", name);
                storage_dir_open(dir_walk->file, string_get_cstr(dir_walk->path));
            }
        } else if(storage_file_get_error(dir_walk->file) == FSE_NOT_EXIST) {
            if(DirIndexList_size(dir_walk->index_list) == 0) {
                // last
                result = DirWalkLast;
                end = true;
            } else {
                // step out
                uint32_t index;
                DirIndexList_pop_back(&index, dir_walk->index_list);
                dir_walk->current_index = 0;

                storage_dir_close(dir_walk->file);

                size_t last_char = string_search_rchar(dir_walk->path, '/');
                if(last_char != STRING_FAILURE) {
                    string_left(dir_walk->path, last_char);
                }

                storage_dir_open(dir_walk->file, string_get_cstr(dir_walk->path));

                // rewind
                while(true) {
                    if(index == dir_walk->current_index) {
                        result = DirWalkOK;
                        break;
                    }

                    if(!storage_dir_read(dir_walk->file, &info, name, 255)) {
                        result = DirWalkError;
                        end = true;
                        break;
                    }

                    dir_walk->current_index++;
                }
            }
        } else {
            result = DirWalkError;
            end = true;
        }
    }

    free(name);
    return result;
}

FS_Error dir_walk_get_error(DirWalk* dir_walk) {
    return storage_file_get_error(dir_walk->file);
}

DirWalkResult dir_walk_read(DirWalk* dir_walk, string_t return_path, FileInfo* fileinfo) {
    return dir_walk_iter(dir_walk, return_path, fileinfo);
}

void dir_walk_close(DirWalk* dir_walk) {
    if(storage_file_is_open(dir_walk->file)) {
        storage_dir_close(dir_walk->file);
    }

    DirIndexList_reset(dir_walk->index_list);
    string_reset(dir_walk->path);
    dir_walk->current_index = 0;
}