mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-12-29 14:14:45 +03:00
bd853de716
When we moved from NonnullRefPtrVector<T> to Vector<NonnullRefPtr<T>>
in commit 8a48246ed1, the `at()` function
started returning a NonnullRefPtr<T>& instead of T&.
The code calling create_index() was not then updated and ended up taking
a pointer to a temporary NonnullRefPtr<>, instead of an actual object,
leading to a crash after logging in.
113 lines
3.8 KiB
C++
113 lines
3.8 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"
|
|
|
|
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.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 {};
|
|
}
|