aboutsummaryrefslogtreecommitdiff
path: root/src/plugins/FileManager.cpp
blob: ccaf2c1748426afb4fd74d9a175d119b2f7bb342 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#include "../../plugins/FileManager.hpp"
#include "../../include/ImageUtils.hpp"
#include <filesystem>

namespace QuickMedia {
    FileManager::FileManager() : Plugin("file-manager"), current_dir("/") {
        
    }

    // Returns empty string if no extension
    static const char* get_ext(const std::filesystem::path &path) {
        const char *path_c = path.c_str();
        int len = strlen(path_c);
        for(int i = len - 1; i >= 0; --i) {
            if(path_c[i] == '.')
                return path_c + i;
        }
        return "";
    }

    static std::filesystem::file_time_type file_get_filetime_or(const std::filesystem::directory_entry &path, std::filesystem::file_time_type default_value) {
        try {
            return path.last_write_time();
        } catch(const std::filesystem::filesystem_error &err) {
            return default_value;
        }
    }

    PluginResult FileManager::get_files_in_directory(BodyItems &result_items) {
        std::vector<std::filesystem::directory_entry> paths;
        try {
            for(auto &p : std::filesystem::directory_iterator(current_dir)) {
                paths.push_back(p);
            }
        } catch(const std::filesystem::filesystem_error &err) {
            fprintf(stderr, "Failed to list files in directory %s, error: %s\n", current_dir.c_str(), err.what());
            return PluginResult::ERR;
        }

        std::sort(paths.begin(), paths.end(), [](const std::filesystem::directory_entry &path1, std::filesystem::directory_entry &path2) {
            return file_get_filetime_or(path1, std::filesystem::file_time_type::min()) > file_get_filetime_or(path2, std::filesystem::file_time_type::min());
        });

        if(current_dir != "/") {
            auto parent_item = BodyItem::create("..");
            result_items.push_back(std::move(parent_item));
        }

        for(auto &p : paths) {
            auto body_item = BodyItem::create(p.path().filename().string());
            // TODO: Check file magic number instead of extension?
            if(p.is_regular_file() && is_image_ext(get_ext(p.path()))) {
                body_item->thumbnail_is_local = true;
                body_item->thumbnail_url = p.path().string();
            }
            result_items.push_back(std::move(body_item));
        }

        return PluginResult::OK;
    }

    bool FileManager::set_current_directory(const std::string &path) {
        if(!std::filesystem::is_directory(path))
            return false;
        current_dir = path;
        return true;
    }

    bool FileManager::set_child_directory(const std::string &filename) {
        if(filename == "..") {
            std::filesystem::path new_path = current_dir.parent_path();
            if(std::filesystem::is_directory(new_path)) {
                current_dir = std::move(new_path);
                return true;
            }
            return false;
        } else {
            std::filesystem::path new_path = current_dir / filename;
            if(std::filesystem::is_directory(new_path)) {
                current_dir = std::move(new_path);
                return true;
            }
            return false;
        }
    }

    const std::filesystem::path& FileManager::get_current_dir() const {
        return current_dir;
    }
}