mirror of
https://github.com/RGBCube/serenity
synced 2025-05-14 23:14:59 +00:00
ca2c81251a
Most of the models were just calling did_update anyway, which is pointless since it can be unified to the base Model class. Instead, code calling update() will now call invalidate(), which functions identically and is more obvious in what it does. Additionally, a default implementation is provided, which removes the need to add empty implementations of update() for each model subclass. Co-Authored-By: Ali Mohammad Pur <ali.mpfard@gmail.com>
115 lines
3.6 KiB
C++
115 lines
3.6 KiB
C++
/*
|
|
* Copyright (c) 2021, Luke Wilde <lukew@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include "MailboxTreeModel.h"
|
|
#include "AccountHolder.h"
|
|
|
|
MailboxTreeModel::MailboxTreeModel(AccountHolder const& account_holder)
|
|
: m_account_holder(account_holder)
|
|
{
|
|
m_mail_icon.set_bitmap_for_size(16, Gfx::Bitmap::try_load_from_file("/res/icons/16x16/app-mail.png"));
|
|
m_folder_icon.set_bitmap_for_size(16, Gfx::Bitmap::try_load_from_file("/res/icons/16x16/filetype-folder.png"));
|
|
m_account_icon.set_bitmap_for_size(16, Gfx::Bitmap::try_load_from_file("/res/icons/16x16/home-directory.png"));
|
|
}
|
|
|
|
MailboxTreeModel::~MailboxTreeModel()
|
|
{
|
|
}
|
|
|
|
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.display_name();
|
|
}
|
|
|
|
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 {};
|
|
}
|