LibGUI: Start working on a file picker dialog (GFilePicker).

Have LibGUI adopt GDirectoryModel from FileManager since it fits perfectly
for the needs of a file picker.

1 files changed, 326 insertions, 0 deletions

diff --git a/LibGUI/GDirectoryModel.cpp b/LibGUI/GDirectoryModel.cpp
new file mode 100644
index 0000000000..9c0c32570f
--- /dev/null
+++ b/LibGUI/GDirectoryModel.cpp
@@ -0,0 +1,326 @@
+#include "GDirectoryModel.h"
+#include <dirent.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <grp.h>
+#include <pwd.h>
+#include <AK/FileSystemPath.h>
+#include <AK/StringBuilder.h>
+#include <SharedGraphics/GraphicsBitmap.h>
+#include <LibGUI/GPainter.h>
+#include <LibCore/CLock.h>
+
+static CLockable<HashMap<String, RetainPtr<GraphicsBitmap>>>& thumbnail_cache()
+{
+    static CLockable<HashMap<String, RetainPtr<GraphicsBitmap>>>* s_map;
+    if (!s_map)
+        s_map = new CLockable<HashMap<String, RetainPtr<GraphicsBitmap>>>();
+    return *s_map;
+}
+
+int thumbnail_thread(void* model_ptr)
+{
+    auto& model = *(GDirectoryModel*)model_ptr;
+    for (;;) {
+        sleep(1);
+        Vector<String> to_generate;
+        {
+            LOCKER(thumbnail_cache().lock());
+            to_generate.ensure_capacity(thumbnail_cache().resource().size());
+            for (auto& it : thumbnail_cache().resource()) {
+                if (it.value)
+                    continue;
+                to_generate.append(it.key);
+            }
+        }
+        if (to_generate.is_empty())
+            continue;
+        for (int i = 0; i < to_generate.size(); ++i) {
+            auto& path = to_generate[i];
+            auto png_bitmap = GraphicsBitmap::load_from_file(path);
+            if (!png_bitmap)
+                continue;
+            auto thumbnail = GraphicsBitmap::create(png_bitmap->format(), { 32, 32 });
+            Painter painter(*thumbnail);
+            painter.draw_scaled_bitmap(thumbnail->rect(), *png_bitmap, png_bitmap->rect());
+            {
+                LOCKER(thumbnail_cache().lock());
+                thumbnail_cache().resource().set(path, move(thumbnail));
+            }
+            if (model.on_thumbnail_progress)
+                model.on_thumbnail_progress(i + 1, to_generate.size());
+            model.did_update();
+        }
+    }
+}
+
+GDirectoryModel::GDirectoryModel()
+{
+    create_thread(thumbnail_thread, this);
+
+    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");
+
+    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();
+}
+
+GDirectoryModel::~GDirectoryModel()
+{
+}
+
+int GDirectoryModel::row_count(const GModelIndex&) const
+{
+    return m_directories.size() + m_files.size();
+}
+
+int GDirectoryModel::column_count(const GModelIndex&) const
+{
+    return Column::__Count;
+}
+
+String GDirectoryModel::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::Inode: return "Inode";
+    }
+    ASSERT_NOT_REACHED();
+}
+
+GModel::ColumnMetadata GDirectoryModel::column_metadata(int column) const
+{
+    switch (column) {
+    case Column::Icon: return { 16, TextAlignment::Center };
+    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::Permissions: return { 80, TextAlignment::CenterLeft };
+    case Column::Inode: return { 80, TextAlignment::CenterRight };
+    }
+    ASSERT_NOT_REACHED();
+}
+
+GIcon GDirectoryModel::icon_for(const Entry& entry) const
+{
+    if (S_ISDIR(entry.mode))
+        return m_directory_icon;
+    if (S_ISLNK(entry.mode))
+        return m_symlink_icon;
+    if (S_ISSOCK(entry.mode))
+        return m_socket_icon;
+    if (entry.mode & S_IXUSR)
+        return m_executable_icon;
+    if (entry.name.to_lowercase().ends_with(".png")) {
+        if (!entry.thumbnail) {
+            auto path = entry.full_path(*this);
+            LOCKER(thumbnail_cache().lock());
+            auto it = thumbnail_cache().resource().find(path);
+            if (it != thumbnail_cache().resource().end()) {
+                entry.thumbnail = (*it).value.copy_ref();
+            } else {
+                thumbnail_cache().resource().set(path, nullptr);
+            }
+        }
+        if (!entry.thumbnail)
+            return m_filetype_image_icon;
+        return GIcon(m_filetype_image_icon.bitmap_for_size(16), *entry.thumbnail);
+    }
+    return m_file_icon;
+}
+
+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();
+}
+
+String GDirectoryModel::name_for_uid(uid_t uid) const
+{
+    auto it = m_user_names.find(uid);
+    if (it == m_user_names.end())
+        return String::format("%u", uid);
+    return (*it).value;
+}
+
+String GDirectoryModel::name_for_gid(uid_t gid) const
+{
+    auto it = m_user_names.find(gid);
+    if (it == m_user_names.end())
+        return String::format("%u", gid);
+    return (*it).value;
+}
+
+GVariant GDirectoryModel::data(const GModelIndex& index, Role role) const
+{
+    ASSERT(is_valid(index));
+    auto& entry = this->entry(index.row());
+    if (role == Role::Sort) {
+        switch (index.column()) {
+        case Column::Icon: return entry.is_directory() ? 0 : 1;
+        case Column::Name: return entry.name;
+        case Column::Size: return (int)entry.size;
+        case Column::Owner: return name_for_uid(entry.uid);
+        case Column::Group: return name_for_gid(entry.gid);
+        case Column::Permissions: return permission_string(entry.mode);
+        case Column::Inode: return (int)entry.inode;
+        }
+        ASSERT_NOT_REACHED();
+    }
+    if (role == Role::Display) {
+        switch (index.column()) {
+        case Column::Icon: return icon_for(entry);
+        case Column::Name: return entry.name;
+        case Column::Size: return (int)entry.size;
+        case Column::Owner: return name_for_uid(entry.uid);
+        case Column::Group: return name_for_gid(entry.gid);
+        case Column::Permissions: return permission_string(entry.mode);
+        case Column::Inode: return (int)entry.inode;
+        }
+    }
+    if (role == Role::Icon) {
+        return icon_for(entry);
+    }
+    return { };
+}
+
+void GDirectoryModel::update()
+{
+    DIR* dirp = opendir(m_path.characters());
+    if (!dirp) {
+        perror("opendir");
+        exit(1);
+    }
+    m_directories.clear();
+    m_files.clear();
+
+    m_bytes_in_files = 0;
+    while (auto* de = readdir(dirp)) {
+        Entry entry;
+        entry.name = de->d_name;
+        if (entry.name == "." || entry.name == "..")
+            continue;
+        struct stat st;
+        int rc = lstat(String::format("%s/%s", m_path.characters(), de->d_name).characters(), &st);
+        if (rc < 0) {
+            perror("lstat");
+            continue;
+        }
+        entry.size = st.st_size;
+        entry.mode = st.st_mode;
+        entry.uid = st.st_uid;
+        entry.gid = st.st_gid;
+        entry.inode = st.st_ino;
+        auto& entries = S_ISDIR(st.st_mode) ? m_directories : m_files;
+        entries.append(move(entry));
+
+        if (S_ISREG(entry.mode))
+            m_bytes_in_files += st.st_size;
+    }
+    closedir(dirp);
+
+    did_update();
+}
+
+void GDirectoryModel::open(const String& a_path)
+{
+    FileSystemPath canonical_path(a_path);
+    auto path = canonical_path.string();
+    if (m_path == path)
+        return;
+    DIR* dirp = opendir(path.characters());
+    if (!dirp)
+        return;
+    closedir(dirp);
+    m_path = path;
+    update();
+    set_selected_index(index(0, 0));
+}
+
+void GDirectoryModel::activate(const GModelIndex& index)
+{
+    if (!index.is_valid())
+        return;
+    auto& entry = this->entry(index.row());
+    FileSystemPath path(String::format("%s/%s", m_path.characters(), entry.name.characters()));
+    if (entry.is_directory()) {
+        open(path.string());
+        return;
+    }
+    if (entry.is_executable()) {
+        if (fork() == 0) {
+            int rc = execl(path.string().characters(), path.string().characters(), nullptr);
+            if (rc < 0)
+                perror("exec");
+            ASSERT_NOT_REACHED();
+        }
+        return;
+    }
+
+    if (path.string().to_lowercase().ends_with(".png")) {
+        if (fork() == 0) {
+            int rc = execl("/bin/qs", "/bin/qs", path.string().characters(), nullptr);
+            if (rc < 0)
+                perror("exec");
+            ASSERT_NOT_REACHED();
+        }
+        return;
+    }
+
+    if (fork() == 0) {
+        int rc = execl("/bin/TextEditor", "/bin/TextEditor", path.string().characters(), nullptr);
+        if (rc < 0)
+            perror("exec");
+        ASSERT_NOT_REACHED();
+    }
+    return;
+}