/* * Copyright (c) 2020, Andreas Kling * Copyright (c) 2022, the SerenityOS developers. * * SPDX-License-Identifier: BSD-2-Clause */ #include #include #include #include #include #include #include namespace GUI { ProcessChooser::ProcessChooser(StringView window_title, StringView button_label, const Gfx::Bitmap* window_icon, GUI::Window* parent_window) : Dialog(parent_window) , m_window_title(window_title) , m_button_label(button_label) , m_window_icon(window_icon) { set_title(m_window_title); if (m_window_icon) set_icon(m_window_icon); else if (parent_window) set_icon(parent_window->icon()); resize(300, 340); center_on_screen(); auto& widget = set_main_widget(); widget.set_fill_with_background_color(true); widget.set_layout(); m_table_view = widget.add(); auto process_model = RunningProcessesModel::create(); auto sorting_model = MUST(GUI::SortingProxyModel::create(process_model)); sorting_model->set_sort_role(GUI::ModelRole::Display); m_table_view->set_model(sorting_model); m_table_view->set_key_column_and_sort_order(RunningProcessesModel::Column::PID, GUI::SortOrder::Descending); m_process_model = process_model; m_table_view->on_activation = [this](const ModelIndex& index) { set_pid_from_index_and_close(index); }; auto& button_container = widget.add(); button_container.set_fixed_height(30); button_container.set_layout(); button_container.layout()->set_margins({ 0, 4, 0 }); button_container.layout()->add_spacer(); auto& select_button = button_container.add(m_button_label); select_button.set_fixed_width(80); select_button.on_click = [this](auto) { if (m_table_view->selection().is_empty()) { GUI::MessageBox::show(this, "No process selected!", m_window_title, GUI::MessageBox::Type::Error); return; } auto index = m_table_view->selection().first(); set_pid_from_index_and_close(index); }; auto& cancel_button = button_container.add("Cancel"); cancel_button.set_fixed_width(80); cancel_button.on_click = [this](auto) { done(ExecCancel); }; m_process_model->update(); m_refresh_timer = add(); m_refresh_timer->start(m_refresh_interval); // Start the timer to update the processes m_refresh_timer->on_timeout = [this] { auto previous_selected_pid = -1; // Store the selection index to not to clear the selection upon update. if (!m_table_view->selection().is_empty()) { auto pid_as_variant = m_table_view->selection().first().data(GUI::ModelRole::Custom); previous_selected_pid = pid_as_variant.as_i32(); } m_process_model->update(); if (previous_selected_pid == -1) { return; } auto model = m_table_view->model(); auto row_count = model->row_count(); auto column_index = 1; // Corresponds to PID column in the m_table_view. for (int row_index = 0; row_index < row_count; ++row_index) { auto cell_index = model->index(row_index, column_index); auto pid_as_variant = cell_index.data(GUI::ModelRole::Custom); if (previous_selected_pid == pid_as_variant.as_i32()) { m_table_view->selection().set(cell_index); // Set only if PIDs are matched. } } }; } void ProcessChooser::set_pid_from_index_and_close(const ModelIndex& index) { m_pid = index.data(GUI::ModelRole::Custom).as_i32(); done(ExecOK); } }