mirror of
https://github.com/SerenityOS/serenity.git
synced 2025-01-26 19:32:06 -05:00
3f3f45580a
Each of these strings would previously rely on StringView's char const* constructor overload, which would call __builtin_strlen on the string. Since we now have operator ""sv, we can replace these with much simpler versions. This opens the door to being able to remove StringView(char const*). No functional changes.
112 lines
3.8 KiB
C++
112 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::try_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::try_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::try_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 {};
|
|
}
|