mirror of
https://github.com/SerenityOS/serenity
synced 2026-04-25 17:15:42 +02:00
d73bad14ea
This PR implements the standard behavior of displaying the mailbox name and parenthesized unseen message count in bold when the unseen message count is greater than zero.
120 lines
4.1 KiB
C++
120 lines
4.1 KiB
C++
/*
|
|
* Copyright (c) 2021, Luke Wilde <lukew@serenityos.org>
|
|
* Copyright (c) 2022, the SerenityOS developers.
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include "MailboxTreeModel.h"
|
|
#include "AccountHolder.h"
|
|
#include <LibGfx/Font/FontDatabase.h>
|
|
|
|
MailboxTreeModel::MailboxTreeModel(AccountHolder const& account_holder)
|
|
: m_account_holder(account_holder)
|
|
{
|
|
m_mail_icon.set_bitmap_for_size(16, Gfx::Bitmap::load_from_file("/res/icons/16x16/app-mail.png"sv).release_value_but_fixme_should_propagate_errors());
|
|
m_folder_icon.set_bitmap_for_size(16, Gfx::Bitmap::load_from_file("/res/icons/16x16/filetype-folder.png"sv).release_value_but_fixme_should_propagate_errors());
|
|
m_account_icon.set_bitmap_for_size(16, Gfx::Bitmap::load_from_file("/res/icons/16x16/home-directory.png"sv).release_value_but_fixme_should_propagate_errors());
|
|
}
|
|
|
|
GUI::ModelIndex MailboxTreeModel::index(int row, int column, GUI::ModelIndex const& parent) const
|
|
{
|
|
if (!parent.is_valid()) {
|
|
if (m_account_holder.accounts().is_empty())
|
|
return {};
|
|
return create_index(row, column, m_account_holder.accounts().at(row));
|
|
}
|
|
auto& base_node = *static_cast<BaseNode*>(parent.internal_data());
|
|
|
|
if (is<MailboxNode>(base_node)) {
|
|
auto& remote_mailbox = verify_cast<MailboxNode>(base_node);
|
|
return create_index(row, column, remote_mailbox.children().at(row));
|
|
}
|
|
|
|
auto& remote_parent = verify_cast<AccountNode>(base_node);
|
|
return create_index(row, column, remote_parent.mailboxes().at(row));
|
|
}
|
|
|
|
GUI::ModelIndex MailboxTreeModel::parent_index(GUI::ModelIndex const& index) const
|
|
{
|
|
if (!index.is_valid())
|
|
return {};
|
|
|
|
auto& base_node = *static_cast<BaseNode*>(index.internal_data());
|
|
|
|
if (is<AccountNode>(base_node))
|
|
return {};
|
|
|
|
auto& mailbox_node = verify_cast<MailboxNode>(base_node);
|
|
|
|
if (!mailbox_node.has_parent()) {
|
|
for (size_t row = 0; row < mailbox_node.associated_account().mailboxes().size(); ++row) {
|
|
if (mailbox_node.associated_account().mailboxes()[row] == &mailbox_node) {
|
|
return create_index(row, index.column(), &mailbox_node.associated_account());
|
|
}
|
|
}
|
|
} else {
|
|
VERIFY(mailbox_node.parent()->has_children());
|
|
for (size_t row = 0; row < mailbox_node.parent()->children().size(); ++row) {
|
|
if (mailbox_node.parent()->children()[row] == &mailbox_node) {
|
|
return create_index(row, index.column(), mailbox_node.parent());
|
|
}
|
|
}
|
|
}
|
|
|
|
VERIFY_NOT_REACHED();
|
|
return {};
|
|
}
|
|
|
|
int MailboxTreeModel::row_count(GUI::ModelIndex const& index) const
|
|
{
|
|
if (!index.is_valid())
|
|
return m_account_holder.accounts().size();
|
|
|
|
auto& base_node = *static_cast<BaseNode*>(index.internal_data());
|
|
|
|
if (is<MailboxNode>(base_node))
|
|
return verify_cast<MailboxNode>(base_node).children().size();
|
|
|
|
auto& node = verify_cast<AccountNode>(base_node);
|
|
return node.mailboxes().size();
|
|
}
|
|
|
|
int MailboxTreeModel::column_count(GUI::ModelIndex const&) const
|
|
{
|
|
return 1;
|
|
}
|
|
|
|
GUI::Variant MailboxTreeModel::data(GUI::ModelIndex const& index, GUI::ModelRole role) const
|
|
{
|
|
auto& base_node = *static_cast<BaseNode*>(index.internal_data());
|
|
|
|
if (role == GUI::ModelRole::Display) {
|
|
if (is<AccountNode>(base_node)) {
|
|
auto& account_node = verify_cast<AccountNode>(base_node);
|
|
return account_node.name();
|
|
}
|
|
|
|
auto& mailbox_node = verify_cast<MailboxNode>(base_node);
|
|
return mailbox_node.unseen_count() ? mailbox_node.display_name_with_unseen_count() : mailbox_node.display_name();
|
|
}
|
|
|
|
if (role == GUI::ModelRole::Font && is<MailboxNode>(base_node)) {
|
|
auto& mailbox_node = verify_cast<MailboxNode>(base_node);
|
|
if (mailbox_node.unseen_count())
|
|
return Gfx::FontDatabase::default_font().bold_variant();
|
|
}
|
|
|
|
if (role == GUI::ModelRole::Icon) {
|
|
if (is<AccountNode>(base_node))
|
|
return m_account_icon;
|
|
|
|
auto& mailbox_node = verify_cast<MailboxNode>(base_node);
|
|
if (!mailbox_node.children().is_empty())
|
|
return m_folder_icon;
|
|
return m_mail_icon;
|
|
}
|
|
|
|
return {};
|
|
}
|