mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2025-01-23 09:46:04 -05:00
94ca55cefd
As suggested by Joshua, this commit adds the 2-clause BSD license as a comment block to the top of every source file. For the first pass, I've just added myself for simplicity. I encourage everyone to add themselves as copyright holders of any file they've added or modified in some significant way. If I've added myself in error somewhere, feel free to replace it with the appropriate copyright holder instead. Going forward, all new source files should include a license header.
531 lines
16 KiB
C++
531 lines
16 KiB
C++
/*
|
|
* 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.
|
|
*/
|
|
|
|
#include <AK/FileSystemPath.h>
|
|
#include <AK/StringBuilder.h>
|
|
#include <LibCore/CDirIterator.h>
|
|
#include <LibDraw/GraphicsBitmap.h>
|
|
#include <LibGUI/GFileSystemModel.h>
|
|
#include <LibGUI/GPainter.h>
|
|
#include <LibThread/BackgroundAction.h>
|
|
#include <dirent.h>
|
|
#include <grp.h>
|
|
#include <pwd.h>
|
|
#include <stdio.h>
|
|
#include <sys/stat.h>
|
|
#include <unistd.h>
|
|
|
|
GModelIndex GFileSystemModel::Node::index(const GFileSystemModel& model, int column) const
|
|
{
|
|
if (!parent)
|
|
return {};
|
|
for (int row = 0; row < parent->children.size(); ++row) {
|
|
if (&parent->children[row] == this)
|
|
return model.create_index(row, column, const_cast<Node*>(this));
|
|
}
|
|
ASSERT_NOT_REACHED();
|
|
}
|
|
|
|
bool GFileSystemModel::Node::fetch_data_using_lstat(const String& full_path)
|
|
{
|
|
struct stat st;
|
|
int rc = lstat(full_path.characters(), &st);
|
|
if (rc < 0) {
|
|
perror("lstat");
|
|
return false;
|
|
}
|
|
|
|
size = st.st_size;
|
|
mode = st.st_mode;
|
|
uid = st.st_uid;
|
|
gid = st.st_gid;
|
|
inode = st.st_ino;
|
|
mtime = st.st_mtime;
|
|
|
|
return true;
|
|
}
|
|
|
|
void GFileSystemModel::Node::traverse_if_needed(const GFileSystemModel& model)
|
|
{
|
|
if (!is_directory() || has_traversed)
|
|
return;
|
|
has_traversed = true;
|
|
total_size = 0;
|
|
|
|
auto full_path = this->full_path(model);
|
|
CDirIterator di(full_path, CDirIterator::SkipDots);
|
|
if (di.has_error()) {
|
|
fprintf(stderr, "CDirIterator: %s\n", di.error_string());
|
|
return;
|
|
}
|
|
|
|
while (di.has_next()) {
|
|
String name = di.next_path();
|
|
String child_path = String::format("%s/%s", full_path.characters(), name.characters());
|
|
NonnullOwnPtr<Node> child = make<Node>();
|
|
bool ok = child->fetch_data_using_lstat(child_path);
|
|
if (!ok)
|
|
continue;
|
|
if (model.m_mode == DirectoriesOnly && !S_ISDIR(child->mode))
|
|
continue;
|
|
child->name = name;
|
|
child->parent = this;
|
|
total_size += child->size;
|
|
children.append(move(child));
|
|
}
|
|
|
|
if (m_watch_fd >= 0)
|
|
return;
|
|
|
|
m_watch_fd = watch_file(full_path.characters(), full_path.length());
|
|
if (m_watch_fd < 0) {
|
|
perror("watch_file");
|
|
return;
|
|
}
|
|
fcntl(m_watch_fd, F_SETFD, FD_CLOEXEC);
|
|
dbg() << "Watching " << full_path << " for changes, m_watch_fd = " << m_watch_fd;
|
|
m_notifier = CNotifier::construct(m_watch_fd, CNotifier::Event::Read);
|
|
m_notifier->on_ready_to_read = [this, &model] {
|
|
char buffer[32];
|
|
int rc = read(m_notifier->fd(), buffer, sizeof(buffer));
|
|
ASSERT(rc >= 0);
|
|
|
|
has_traversed = false;
|
|
mode = 0;
|
|
children.clear();
|
|
reify_if_needed(model);
|
|
const_cast<GFileSystemModel&>(model).did_update();
|
|
};
|
|
}
|
|
|
|
void GFileSystemModel::Node::reify_if_needed(const GFileSystemModel& model)
|
|
{
|
|
traverse_if_needed(model);
|
|
if (mode != 0)
|
|
return;
|
|
fetch_data_using_lstat(full_path(model));
|
|
}
|
|
|
|
String GFileSystemModel::Node::full_path(const GFileSystemModel& model) const
|
|
{
|
|
Vector<String, 32> lineage;
|
|
for (auto* ancestor = parent; ancestor; ancestor = ancestor->parent) {
|
|
lineage.append(ancestor->name);
|
|
}
|
|
StringBuilder builder;
|
|
builder.append(model.root_path());
|
|
for (int i = lineage.size() - 1; i >= 0; --i) {
|
|
builder.append('/');
|
|
builder.append(lineage[i]);
|
|
}
|
|
builder.append('/');
|
|
builder.append(name);
|
|
return canonicalized_path(builder.to_string());
|
|
}
|
|
|
|
GModelIndex GFileSystemModel::index(const StringView& path, int column) const
|
|
{
|
|
FileSystemPath canonical_path(path);
|
|
const Node* node = m_root;
|
|
if (canonical_path.string() == "/")
|
|
return m_root->index(*this, column);
|
|
for (int i = 0; i < canonical_path.parts().size(); ++i) {
|
|
auto& part = canonical_path.parts()[i];
|
|
bool found = false;
|
|
for (auto& child : node->children) {
|
|
if (child.name == part) {
|
|
const_cast<Node&>(child).reify_if_needed(*this);
|
|
node = &child;
|
|
found = true;
|
|
if (i == canonical_path.parts().size() - 1)
|
|
return child.index(*this, column);
|
|
break;
|
|
}
|
|
}
|
|
if (!found)
|
|
return {};
|
|
}
|
|
return {};
|
|
}
|
|
|
|
String GFileSystemModel::full_path(const GModelIndex& index) const
|
|
{
|
|
auto& node = this->node(index);
|
|
const_cast<Node&>(node).reify_if_needed(*this);
|
|
return node.full_path(*this);
|
|
}
|
|
|
|
GFileSystemModel::GFileSystemModel(const StringView& root_path, Mode mode)
|
|
: m_root_path(canonicalized_path(root_path))
|
|
, m_mode(mode)
|
|
{
|
|
m_directory_icon = GIcon::default_icon("filetype-folder");
|
|
m_file_icon = GIcon::default_icon("filetype-unknown");
|
|
m_symlink_icon = GIcon::default_icon("filetype-symlink");
|
|
m_socket_icon = GIcon::default_icon("filetype-socket");
|
|
m_executable_icon = GIcon::default_icon("filetype-executable");
|
|
m_filetype_image_icon = GIcon::default_icon("filetype-image");
|
|
m_filetype_sound_icon = GIcon::default_icon("filetype-sound");
|
|
m_filetype_html_icon = GIcon::default_icon("filetype-html");
|
|
|
|
setpwent();
|
|
while (auto* passwd = getpwent())
|
|
m_user_names.set(passwd->pw_uid, passwd->pw_name);
|
|
endpwent();
|
|
|
|
setgrent();
|
|
while (auto* group = getgrent())
|
|
m_group_names.set(group->gr_gid, group->gr_name);
|
|
endgrent();
|
|
|
|
update();
|
|
}
|
|
|
|
GFileSystemModel::~GFileSystemModel()
|
|
{
|
|
}
|
|
|
|
String GFileSystemModel::name_for_uid(uid_t uid) const
|
|
{
|
|
auto it = m_user_names.find(uid);
|
|
if (it == m_user_names.end())
|
|
return String::number(uid);
|
|
return (*it).value;
|
|
}
|
|
|
|
String GFileSystemModel::name_for_gid(uid_t gid) const
|
|
{
|
|
auto it = m_user_names.find(gid);
|
|
if (it == m_user_names.end())
|
|
return String::number(gid);
|
|
return (*it).value;
|
|
}
|
|
|
|
static String permission_string(mode_t mode)
|
|
{
|
|
StringBuilder builder;
|
|
if (S_ISDIR(mode))
|
|
builder.append("d");
|
|
else if (S_ISLNK(mode))
|
|
builder.append("l");
|
|
else if (S_ISBLK(mode))
|
|
builder.append("b");
|
|
else if (S_ISCHR(mode))
|
|
builder.append("c");
|
|
else if (S_ISFIFO(mode))
|
|
builder.append("f");
|
|
else if (S_ISSOCK(mode))
|
|
builder.append("s");
|
|
else if (S_ISREG(mode))
|
|
builder.append("-");
|
|
else
|
|
builder.append("?");
|
|
|
|
builder.appendf("%c%c%c%c%c%c%c%c",
|
|
mode & S_IRUSR ? 'r' : '-',
|
|
mode & S_IWUSR ? 'w' : '-',
|
|
mode & S_ISUID ? 's' : (mode & S_IXUSR ? 'x' : '-'),
|
|
mode & S_IRGRP ? 'r' : '-',
|
|
mode & S_IWGRP ? 'w' : '-',
|
|
mode & S_ISGID ? 's' : (mode & S_IXGRP ? 'x' : '-'),
|
|
mode & S_IROTH ? 'r' : '-',
|
|
mode & S_IWOTH ? 'w' : '-');
|
|
|
|
if (mode & S_ISVTX)
|
|
builder.append("t");
|
|
else
|
|
builder.appendf("%c", mode & S_IXOTH ? 'x' : '-');
|
|
return builder.to_string();
|
|
}
|
|
|
|
void GFileSystemModel::set_root_path(const StringView& root_path)
|
|
{
|
|
m_root_path = canonicalized_path(root_path);
|
|
|
|
if (on_root_path_change)
|
|
on_root_path_change();
|
|
|
|
update();
|
|
}
|
|
|
|
void GFileSystemModel::update()
|
|
{
|
|
m_root = make<Node>();
|
|
m_root->reify_if_needed(*this);
|
|
|
|
did_update();
|
|
}
|
|
|
|
int GFileSystemModel::row_count(const GModelIndex& index) const
|
|
{
|
|
Node& node = const_cast<Node&>(this->node(index));
|
|
node.reify_if_needed(*this);
|
|
if (node.is_directory())
|
|
return node.children.size();
|
|
return 0;
|
|
}
|
|
|
|
const GFileSystemModel::Node& GFileSystemModel::node(const GModelIndex& index) const
|
|
{
|
|
if (!index.is_valid())
|
|
return *m_root;
|
|
return *(Node*)index.internal_data();
|
|
}
|
|
|
|
GModelIndex GFileSystemModel::index(int row, int column, const GModelIndex& parent) const
|
|
{
|
|
auto& node = this->node(parent);
|
|
const_cast<Node&>(node).reify_if_needed(*this);
|
|
if (row >= node.children.size())
|
|
return {};
|
|
return create_index(row, column, &node.children[row]);
|
|
}
|
|
|
|
GModelIndex GFileSystemModel::parent_index(const GModelIndex& index) const
|
|
{
|
|
if (!index.is_valid())
|
|
return {};
|
|
auto& node = this->node(index);
|
|
if (!node.parent) {
|
|
ASSERT(&node == m_root);
|
|
return {};
|
|
}
|
|
return node.parent->index(*this, index.column());
|
|
}
|
|
|
|
GVariant GFileSystemModel::data(const GModelIndex& index, Role role) const
|
|
{
|
|
ASSERT(index.is_valid());
|
|
auto& node = this->node(index);
|
|
|
|
if (role == Role::Custom) {
|
|
// For GFileSystemModel, custom role means the full path.
|
|
ASSERT(index.column() == Column::Name);
|
|
return node.full_path(*this);
|
|
}
|
|
|
|
if (role == Role::DragData) {
|
|
if (index.column() == Column::Name) {
|
|
StringBuilder builder;
|
|
builder.append("file://");
|
|
builder.append(node.full_path(*this));
|
|
return builder.to_string();
|
|
}
|
|
return {};
|
|
}
|
|
|
|
if (role == Role::Sort) {
|
|
switch (index.column()) {
|
|
case Column::Icon:
|
|
return node.is_directory() ? 0 : 1;
|
|
case Column::Name:
|
|
return node.name;
|
|
case Column::Size:
|
|
return (int)node.size;
|
|
case Column::Owner:
|
|
return name_for_uid(node.uid);
|
|
case Column::Group:
|
|
return name_for_gid(node.gid);
|
|
case Column::Permissions:
|
|
return permission_string(node.mode);
|
|
case Column::ModificationTime:
|
|
return node.mtime;
|
|
case Column::Inode:
|
|
return (int)node.inode;
|
|
}
|
|
ASSERT_NOT_REACHED();
|
|
}
|
|
|
|
if (role == Role::Display) {
|
|
switch (index.column()) {
|
|
case Column::Icon:
|
|
return icon_for(node);
|
|
case Column::Name:
|
|
return node.name;
|
|
case Column::Size:
|
|
return (int)node.size;
|
|
case Column::Owner:
|
|
return name_for_uid(node.uid);
|
|
case Column::Group:
|
|
return name_for_gid(node.gid);
|
|
case Column::Permissions:
|
|
return permission_string(node.mode);
|
|
case Column::ModificationTime:
|
|
return timestamp_string(node.mtime);
|
|
case Column::Inode:
|
|
return (int)node.inode;
|
|
}
|
|
}
|
|
|
|
if (role == Role::Icon) {
|
|
return icon_for(node);
|
|
}
|
|
return {};
|
|
}
|
|
|
|
GIcon GFileSystemModel::icon_for_file(const mode_t mode, const String& name) const
|
|
{
|
|
if (S_ISDIR(mode))
|
|
return m_directory_icon;
|
|
if (S_ISLNK(mode))
|
|
return m_symlink_icon;
|
|
if (S_ISSOCK(mode))
|
|
return m_socket_icon;
|
|
if (mode & S_IXUSR)
|
|
return m_executable_icon;
|
|
if (name.to_lowercase().ends_with(".wav"))
|
|
return m_filetype_sound_icon;
|
|
if (name.to_lowercase().ends_with(".html"))
|
|
return m_filetype_html_icon;
|
|
if (name.to_lowercase().ends_with(".png"))
|
|
return m_filetype_image_icon;
|
|
return m_file_icon;
|
|
}
|
|
|
|
GIcon GFileSystemModel::icon_for(const Node& node) const
|
|
{
|
|
if (node.name.to_lowercase().ends_with(".png")) {
|
|
if (!node.thumbnail) {
|
|
if (!const_cast<GFileSystemModel*>(this)->fetch_thumbnail_for(node))
|
|
return m_filetype_image_icon;
|
|
}
|
|
return GIcon(m_filetype_image_icon.bitmap_for_size(16), *node.thumbnail);
|
|
}
|
|
|
|
return icon_for_file(node.mode, node.name);
|
|
}
|
|
|
|
static HashMap<String, RefPtr<GraphicsBitmap>> s_thumbnail_cache;
|
|
|
|
static RefPtr<GraphicsBitmap> render_thumbnail(const StringView& path)
|
|
{
|
|
auto png_bitmap = GraphicsBitmap::load_from_file(path);
|
|
if (!png_bitmap)
|
|
return nullptr;
|
|
auto thumbnail = GraphicsBitmap::create(png_bitmap->format(), { 32, 32 });
|
|
Painter painter(*thumbnail);
|
|
painter.draw_scaled_bitmap(thumbnail->rect(), *png_bitmap, png_bitmap->rect());
|
|
return thumbnail;
|
|
}
|
|
|
|
bool GFileSystemModel::fetch_thumbnail_for(const Node& node)
|
|
{
|
|
// See if we already have the thumbnail
|
|
// we're looking for in the cache.
|
|
auto path = node.full_path(*this);
|
|
auto it = s_thumbnail_cache.find(path);
|
|
if (it != s_thumbnail_cache.end()) {
|
|
if (!(*it).value)
|
|
return false;
|
|
node.thumbnail = (*it).value;
|
|
return true;
|
|
}
|
|
|
|
// Otherwise, arrange to render the thumbnail
|
|
// in background and make it available later.
|
|
|
|
s_thumbnail_cache.set(path, nullptr);
|
|
m_thumbnail_progress_total++;
|
|
|
|
auto weak_this = make_weak_ptr();
|
|
|
|
LibThread::BackgroundAction<RefPtr<GraphicsBitmap>>::create(
|
|
[path] {
|
|
return render_thumbnail(path);
|
|
},
|
|
|
|
[this, path, weak_this](auto thumbnail) {
|
|
s_thumbnail_cache.set(path, move(thumbnail));
|
|
|
|
// The model was destroyed, no need to update
|
|
// progress or call any event handlers.
|
|
if (weak_this.is_null())
|
|
return;
|
|
|
|
m_thumbnail_progress++;
|
|
if (on_thumbnail_progress)
|
|
on_thumbnail_progress(m_thumbnail_progress, m_thumbnail_progress_total);
|
|
if (m_thumbnail_progress == m_thumbnail_progress_total) {
|
|
m_thumbnail_progress = 0;
|
|
m_thumbnail_progress_total = 0;
|
|
}
|
|
|
|
did_update();
|
|
});
|
|
|
|
return false;
|
|
}
|
|
|
|
int GFileSystemModel::column_count(const GModelIndex&) const
|
|
{
|
|
return Column::__Count;
|
|
}
|
|
|
|
String GFileSystemModel::column_name(int column) const
|
|
{
|
|
switch (column) {
|
|
case Column::Icon:
|
|
return "";
|
|
case Column::Name:
|
|
return "Name";
|
|
case Column::Size:
|
|
return "Size";
|
|
case Column::Owner:
|
|
return "Owner";
|
|
case Column::Group:
|
|
return "Group";
|
|
case Column::Permissions:
|
|
return "Mode";
|
|
case Column::ModificationTime:
|
|
return "Modified";
|
|
case Column::Inode:
|
|
return "Inode";
|
|
}
|
|
ASSERT_NOT_REACHED();
|
|
}
|
|
|
|
GModel::ColumnMetadata GFileSystemModel::column_metadata(int column) const
|
|
{
|
|
switch (column) {
|
|
case Column::Icon:
|
|
return { 16, TextAlignment::Center, nullptr, GModel::ColumnMetadata::Sortable::False };
|
|
case Column::Name:
|
|
return { 120, TextAlignment::CenterLeft };
|
|
case Column::Size:
|
|
return { 80, TextAlignment::CenterRight };
|
|
case Column::Owner:
|
|
return { 50, TextAlignment::CenterLeft };
|
|
case Column::Group:
|
|
return { 50, TextAlignment::CenterLeft };
|
|
case Column::ModificationTime:
|
|
return { 110, TextAlignment::CenterLeft };
|
|
case Column::Permissions:
|
|
return { 65, TextAlignment::CenterLeft };
|
|
case Column::Inode:
|
|
return { 60, TextAlignment::CenterRight };
|
|
}
|
|
ASSERT_NOT_REACHED();
|
|
}
|