jcs.org
Serenity Operating System
1/*
2 * Copyright (c) 2022, the SerenityOS developers.
3 * Copyright (c) 2022, Sam Atkins <atkinssj@serenityos.org>
4 *
5 * SPDX-License-Identifier: BSD-2-Clause
6 */
7
8#include "StorageWidget.h"
9#include "CookiesModel.h"
10#include "StorageModel.h"
11#include <Applications/Browser/StorageWidgetGML.h>
12#include <LibGUI/Menu.h>
13#include <LibGUI/TabWidget.h>
14#include <LibGUI/TableView.h>
15#include <LibWeb/Cookie/Cookie.h>
16
17namespace Browser {
18
19StorageWidget::StorageWidget()
20{
21 load_from_gml(storage_widget_gml).release_value_but_fixme_should_propagate_errors();
22 auto& tab_widget = *find_descendant_of_type_named<GUI::TabWidget>("tab_widget");
23
24 m_cookies_table_view = tab_widget.find_descendant_of_type_named<GUI::TableView>("cookies_tableview");
25 m_cookies_textbox = tab_widget.find_descendant_of_type_named<GUI::TextBox>("cookies_filter_textbox");
26 m_cookies_model = adopt_ref(*new CookiesModel());
27
28 m_cookies_filtering_model = MUST(GUI::FilteringProxyModel::create(*m_cookies_model));
29 m_cookies_filtering_model->set_filter_term(""sv);
30
31 m_cookies_textbox->on_change = [this] {
32 m_cookies_filtering_model->set_filter_term(m_cookies_textbox->text());
33 if (m_cookies_filtering_model->row_count() != 0)
34 m_cookies_table_view->set_cursor(m_cookies_filtering_model->index(0, 0), GUI::AbstractView::SelectionUpdate::Set);
35 };
36
37 m_cookies_table_view->set_model(m_cookies_filtering_model);
38 m_cookies_table_view->set_column_headers_visible(true);
39 m_cookies_table_view->set_alternating_row_colors(true);
40
41 auto delete_cookie_action = GUI::Action::create(
42 "&Delete Cookie", { Key_Delete }, Gfx::Bitmap::load_from_file("/res/icons/16x16/delete.png"sv).release_value_but_fixme_should_propagate_errors(), [&](auto const&) {
43 auto cookie_index = m_cookies_table_view->selection().first();
44 delete_cookie(m_cookies_model->take_cookie(cookie_index));
45 },
46 m_cookies_table_view);
47
48 auto delete_all_cookies_action = GUI::Action::create(
49 "Delete &All Cookies", [&](auto const&) {
50 auto cookies = m_cookies_model->take_all_cookies();
51
52 for (auto& cookie : cookies)
53 delete_cookie(move(cookie));
54 },
55 m_cookies_table_view);
56
57 m_cookies_context_menu = GUI::Menu::construct();
58 m_cookies_context_menu->add_action(delete_cookie_action);
59 m_cookies_context_menu->add_action(delete_all_cookies_action);
60 m_cookies_table_view->on_context_menu_request = [&](auto& index, auto& event) {
61 if (index.is_valid())
62 m_cookies_context_menu->popup(event.screen_position());
63 };
64
65 m_local_storage_table_view = tab_widget.find_descendant_of_type_named<GUI::TableView>("local_storage_tableview");
66 m_local_storage_textbox = tab_widget.find_descendant_of_type_named<GUI::TextBox>("local_storage_filter_textbox");
67 m_local_storage_model = adopt_ref(*new StorageModel());
68
69 m_local_storage_filtering_model = MUST(GUI::FilteringProxyModel::create(*m_local_storage_model));
70 m_local_storage_filtering_model->set_filter_term(""sv);
71
72 m_local_storage_textbox->on_change = [this] {
73 m_local_storage_filtering_model->set_filter_term(m_local_storage_textbox->text());
74 if (m_local_storage_filtering_model->row_count() != 0)
75 m_local_storage_table_view->set_cursor(m_local_storage_filtering_model->index(0, 0), GUI::AbstractView::SelectionUpdate::Set);
76 };
77
78 m_local_storage_table_view->set_model(m_local_storage_filtering_model);
79 m_local_storage_table_view->set_column_headers_visible(true);
80 m_local_storage_table_view->set_alternating_row_colors(true);
81
82 m_session_storage_table_view = tab_widget.find_descendant_of_type_named<GUI::TableView>("session_storage_tableview");
83 m_session_storage_textbox = tab_widget.find_descendant_of_type_named<GUI::TextBox>("session_storage_filter_textbox");
84 m_session_storage_model = adopt_ref(*new StorageModel());
85
86 m_session_storage_filtering_model = MUST(GUI::FilteringProxyModel::create(*m_session_storage_model));
87 m_session_storage_filtering_model->set_filter_term(""sv);
88
89 m_session_storage_textbox->on_change = [this] {
90 m_session_storage_filtering_model->set_filter_term(m_session_storage_textbox->text());
91 if (m_session_storage_filtering_model->row_count() != 0)
92 m_session_storage_table_view->set_cursor(m_session_storage_filtering_model->index(0, 0), GUI::AbstractView::SelectionUpdate::Set);
93 };
94
95 m_session_storage_table_view->set_model(m_session_storage_filtering_model);
96 m_session_storage_table_view->set_column_headers_visible(true);
97 m_session_storage_table_view->set_alternating_row_colors(true);
98}
99
100void StorageWidget::set_cookies_entries(Vector<Web::Cookie::Cookie> entries)
101{
102 m_cookies_model->set_items(entries);
103}
104
105void StorageWidget::clear_cookies()
106{
107 m_cookies_model->clear_items();
108}
109
110void StorageWidget::set_local_storage_entries(OrderedHashMap<DeprecatedString, DeprecatedString> entries)
111{
112 m_local_storage_model->set_items(entries);
113}
114
115void StorageWidget::clear_local_storage_entries()
116{
117 m_local_storage_model->clear_items();
118}
119
120void StorageWidget::set_session_storage_entries(OrderedHashMap<DeprecatedString, DeprecatedString> entries)
121{
122 m_session_storage_model->set_items(entries);
123}
124
125void StorageWidget::clear_session_storage_entries()
126{
127 m_session_storage_model->clear_items();
128}
129
130void StorageWidget::delete_cookie(Web::Cookie::Cookie cookie)
131{
132 // Delete cookie by making its expiry time in the past.
133 cookie.expiry_time = Time::from_seconds(0);
134 if (on_update_cookie)
135 on_update_cookie(move(cookie));
136}
137
138}