diff options
Diffstat (limited to 'Libraries/LibGUI/FileSystemModel.h')
| -rw-r--r-- | Libraries/LibGUI/FileSystemModel.h | 168 |
1 files changed, 168 insertions, 0 deletions
diff --git a/Libraries/LibGUI/FileSystemModel.h b/Libraries/LibGUI/FileSystemModel.h new file mode 100644 index 0000000000..9b54c4b0a3 --- /dev/null +++ b/Libraries/LibGUI/FileSystemModel.h @@ -0,0 +1,168 @@ +/* + * Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org> + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * 1. Redistributions of source code must retain the above copyright notice, this + * list of conditions and the following disclaimer. + * + * 2. Redistributions in binary form must reproduce the above copyright notice, + * this list of conditions and the following disclaimer in the documentation + * and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR + * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER + * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, + * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#pragma once + +#include <AK/HashMap.h> +#include <AK/NonnullOwnPtrVector.h> +#include <LibCore/Notifier.h> +#include <LibGUI/Model.h> +#include <sys/stat.h> +#include <time.h> + +namespace GUI { + +class FileSystemModel + : public Model + , public Weakable<FileSystemModel> { + friend struct Node; + +public: + enum Mode { + Invalid, + DirectoriesOnly, + FilesAndDirectories + }; + + enum Column { + Icon = 0, + Name, + Size, + Owner, + Group, + Permissions, + ModificationTime, + Inode, + SymlinkTarget, + __Count, + }; + + struct Node { + ~Node() { close(m_watch_fd); } + + String name; + String symlink_target; + size_t size { 0 }; + mode_t mode { 0 }; + uid_t uid { 0 }; + gid_t gid { 0 }; + ino_t inode { 0 }; + time_t mtime { 0 }; + + size_t total_size { 0 }; + + mutable RefPtr<Gfx::Bitmap> thumbnail; + bool is_directory() const { return S_ISDIR(mode); } + bool is_executable() const { return mode & (S_IXUSR | S_IXGRP | S_IXOTH); } + + String full_path(const FileSystemModel&) const; + + private: + friend class FileSystemModel; + + Node* parent { nullptr }; + NonnullOwnPtrVector<Node> children; + bool has_traversed { false }; + + int m_watch_fd { -1 }; + RefPtr<Core::Notifier> m_notifier; + + ModelIndex index(const FileSystemModel&, int column) const; + void traverse_if_needed(const FileSystemModel&); + void reify_if_needed(const FileSystemModel&); + bool fetch_data(const String& full_path, bool is_root); + }; + + static NonnullRefPtr<FileSystemModel> create(const StringView& root_path = "/", Mode mode = Mode::FilesAndDirectories) + { + return adopt(*new FileSystemModel(root_path, mode)); + } + virtual ~FileSystemModel() override; + + String root_path() const { return m_root_path; } + void set_root_path(const StringView&); + String full_path(const ModelIndex&) const; + ModelIndex index(const StringView& path, int column) const; + + const Node& node(const ModelIndex& index) const; + GIcon icon_for_file(const mode_t mode, const String& name) const; + + Function<void(int done, int total)> on_thumbnail_progress; + Function<void()> on_root_path_change; + + virtual int tree_column() const override { return Column::Name; } + virtual int row_count(const ModelIndex& = ModelIndex()) const override; + virtual int column_count(const ModelIndex& = ModelIndex()) const override; + virtual String column_name(int column) const override; + virtual ColumnMetadata column_metadata(int column) const override; + virtual Variant data(const ModelIndex&, Role = Role::Display) const override; + virtual void update() override; + virtual ModelIndex parent_index(const ModelIndex&) const override; + virtual ModelIndex index(int row, int column = 0, const ModelIndex& parent = ModelIndex()) const override; + virtual StringView drag_data_type() const override { return "url-list"; } + + static String timestamp_string(time_t timestamp) + { + auto* tm = localtime(×tamp); + return String::format("%4u-%02u-%02u %02u:%02u:%02u", + tm->tm_year + 1900, + tm->tm_mon + 1, + tm->tm_mday, + tm->tm_hour, + tm->tm_min, + tm->tm_sec); + } + +private: + FileSystemModel(const StringView& root_path, Mode); + + String name_for_uid(uid_t) const; + String name_for_gid(gid_t) const; + + HashMap<uid_t, String> m_user_names; + HashMap<gid_t, String> m_group_names; + + bool fetch_thumbnail_for(const Node& node); + GIcon icon_for(const Node& node) const; + + String m_root_path; + Mode m_mode { Invalid }; + OwnPtr<Node> m_root { nullptr }; + + GIcon m_directory_icon; + GIcon m_file_icon; + GIcon m_symlink_icon; + GIcon m_socket_icon; + GIcon m_executable_icon; + GIcon m_filetype_image_icon; + GIcon m_filetype_sound_icon; + GIcon m_filetype_html_icon; + + unsigned m_thumbnail_progress { 0 }; + unsigned m_thumbnail_progress_total { 0 }; +}; + +} |