ladybird/Userland/Applications/Mail/MailboxTreeModel.cpp
Luke b716e902ba Mail: Add an e-mail application called Mail
This utilises LibIMAP and LibWeb to provide an e-mail client.

The only way currently to connect to a server and login is with a
config file. This config file should be stored in ~/.config/Mail.ini
Here is an example config file:
```
[Connection]
Server=email.example.com
Port=993
TLS=true

[User]
Username=test@example.com
Password=Example!1
```

Since this is stored in plaintext and uses a less secure login method,
I'd recommend not using this on your main accounts :^)

This has been tested on Gmail and Outlook. For Gmail, you either have
to generate an app password if you have 2FA enabled, or enable access
from less secure apps in your account settings.
2021-07-24 20:11:28 +04:30

120 lines
3.7 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 {};
}
void MailboxTreeModel::update()
{
did_update();
}