main_window.hpp

Go to the documentation of this file.

#pragma once
 
#include <gtkmm.h>
#include <giomm.h>
 
#include <chrono>
#include <cstdint>
#include <string>
#include <vector>
 
#include "device_enumerator.hpp"
#include "models.hpp"
#include "recovery_worker.hpp"
#include "scan_worker.hpp"
 
class MainWindow : public Gtk::Window {
public:
    MainWindow();
    ~MainWindow() override;
 
private:
    enum class ThemeMode { Dark, Light };
    enum class WizardStep { Source, Destination, Scan, Recover };
 
    void build_headerbar();
    void build_sidebar();
    void build_dashboard();
    void build_recovery_page();
    void build_wizard();
    void build_about_page();
    void update_action_controls();
 
    void apply_theme(ThemeMode mode);
    void apply_inline_css();
    void update_theme_buttons();
    void activate_sidebar_button(Gtk::Button* button);
 
    void refresh_devices();
    void populate_device_lists();
    void select_source(const std::string& path, const std::string& label);
    void update_source_state();
    void update_wizard_step();
    void append_log(const std::string& line);
 
    void on_sidebar_dashboard();
    void on_sidebar_recovery();
    void on_sidebar_about();
    void on_refresh_devices();
    void on_language_changed();
    void on_open_image();
    void on_choose_destination();
    void on_use_selected_device();
    void on_dashboard_device_activated(Gtk::ListBoxRow* row);
    void on_recovery_device_activated(Gtk::ListBoxRow* row);
    void on_theme_dark();
    void on_theme_light();
    void on_wizard_back();
    void on_wizard_next();
    void on_scan_now();
    void on_recover_now();
    void on_launch_elevated();
    void on_scan_progress(const ScanProgress& progress);
    void on_scan_completed(const ScanReport& report);
    void on_recovery_completed(const RecoveryResult& result);
    void populate_scan_results(const ScanReport& report);
    void update_selected_results_state();
    std::vector<size_t> selected_result_indices() const;
    void set_all_results_selected(bool selected);
    static std::string result_group_label(const ScanResult& result);
    static std::string result_subgroup_label(const ScanResult& result);
    void minimize_window();
    void hide_window();
 
    static std::string human_size(uint64_t bytes);
    static std::string read_language();
    static void save_language(const std::string& lang_code);
    static std::string read_theme();
    static void save_theme(const std::string& theme_code);
 
private:
    ThemeMode current_theme_ = ThemeMode::Dark;
    WizardStep current_step_ = WizardStep::Source;
 
    Glib::RefPtr<Gtk::CssProvider> css_provider_;
 
    Gtk::Box main_box_{Gtk::Orientation::HORIZONTAL, 0};
    Gtk::Stack page_stack_;
 
    Gtk::Box sidebar_{Gtk::Orientation::VERTICAL, 10};
    Gtk::Box dashboard_page_{Gtk::Orientation::VERTICAL, 16};
    Gtk::Box recovery_page_{Gtk::Orientation::VERTICAL, 16};
    Gtk::Box about_page_{Gtk::Orientation::VERTICAL, 24};
 
 
    Gtk::Grid cards_grid_;
    Gtk::Frame card_status_;
    Gtk::Frame card_source_;
    Gtk::Frame card_update_;
    Gtk::Label status_card_label_;
    Gtk::Label source_card_label_;
    Gtk::Label update_card_label_;
 
    Gtk::Frame devices_frame_;
    Gtk::ScrolledWindow devices_scroller_;
    Gtk::ListBox devices_list_;
 
    Gtk::Label brand_label_;
    Gtk::Button btn_dashboard_;
    Gtk::Button btn_recovery_;
    Gtk::Button btn_about_;
    Gtk::Button btn_refresh_;
    Gtk::Button btn_theme_dark_;
    Gtk::Button btn_theme_light_;
    Gtk::Button* dashboard_button_ = nullptr;
    Gtk::Button* recovery_button_ = nullptr;
    Gtk::Button* about_button_ = nullptr;
    std::vector<Gtk::Button*> sidebar_buttons_;
 
    Gtk::DropDown lang_combo_;
    bool lang_changing_ = false;
 
    Gtk::Box recovery_header_{Gtk::Orientation::VERTICAL, 8};
    Gtk::Label recovery_title_;
    Gtk::Label recovery_subtitle_;
 
    Gtk::Frame wizard_frame_;
    Gtk::Box wizard_root_{Gtk::Orientation::VERTICAL, 12};
    Gtk::Stack wizard_stack_;
    Gtk::Box wizard_nav_{Gtk::Orientation::HORIZONTAL, 8};
    Gtk::Button btn_wizard_back_;
    Gtk::Button btn_wizard_next_;
    Gtk::Label wizard_notice_label_;
 
    Gtk::Box source_step_box_{Gtk::Orientation::VERTICAL, 8};
    Gtk::Label source_step_label_;
    Gtk::Box source_button_row_{Gtk::Orientation::HORIZONTAL, 8};
    Gtk::Button btn_open_image_;
    Gtk::Button btn_use_selected_device_;
    Gtk::Button btn_launch_elevated_;
    Gtk::ScrolledWindow recovery_devices_scroller_;
    Gtk::ListBox recovery_devices_list_;
 
    Gtk::Box destination_step_box_{Gtk::Orientation::VERTICAL, 8};
    Gtk::Label destination_step_label_;
    Gtk::Box destination_button_row_{Gtk::Orientation::HORIZONTAL, 8};
    Gtk::Entry entry_output_;
    Gtk::Button btn_choose_destination_;
 
    Gtk::Box scan_step_box_{Gtk::Orientation::VERTICAL, 8};
    Gtk::Label scan_step_label_;
    Gtk::ProgressBar scan_progress_bar_;
    Gtk::Label scan_status_label_;
    Gtk::Label scan_timer_label_;
    Gtk::Button btn_scan_now_;
 
    Gtk::Box recover_step_box_{Gtk::Orientation::VERTICAL, 8};
    Gtk::Label recover_step_label_;
    Gtk::Frame results_frame_;
    Gtk::ScrolledWindow results_scroller_;
    Gtk::Box results_tree_box_{Gtk::Orientation::VERTICAL, 8};
    Gtk::Label results_count_label_;
    Gtk::Label results_selected_label_;
    Gtk::Box results_actions_row_{Gtk::Orientation::HORIZONTAL, 8};
    Gtk::Button btn_results_select_all_;
    Gtk::Button btn_results_clear_;
    Gtk::Button btn_recover_now_;
 
    Gtk::ScrolledWindow log_scroller_;
    Gtk::TextView log_view_;
    Glib::RefPtr<Gtk::TextBuffer> log_buffer_;
 
    Glib::RefPtr<Gtk::FileDialog> file_dialog_;
    Glib::RefPtr<Gtk::FileDialog> folder_dialog_;
 
    std::vector<DeviceInfo> devices_;
    std::string selected_source_path_;
    std::string selected_source_label_;
    std::string output_dir_;
    ScanReport last_report_;
    std::vector<bool> result_selection_;
    std::vector<Gtk::CheckButton*> result_check_buttons_;
 
    ScanWorker scan_worker_;
    RecoveryWorker recovery_worker_;
 
    ScanProgress last_scan_progress_;
    sigc::connection scan_timer_connection_;
    std::chrono::steady_clock::time_point scan_started_at_{};
};