// Copyright 2019 The Chromium Authors // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. #ifndef CHROME_UPDATER_WIN_UI_PROGRESS_WND_H_ #define CHROME_UPDATER_WIN_UI_PROGRESS_WND_H_ #include #include #include #include #include #include #include #include "base/gtest_prod_util.h" #include "base/memory/raw_ptr.h" #include "base/sequence_checker.h" #include "base/win/scoped_gdi_object.h" #include "chrome/updater/app/app_install_progress.h" #include "chrome/updater/win/ui/complete_wnd.h" #include "chrome/updater/win/ui/message_loop.h" #include "chrome/updater/win/ui/owner_draw_controls.h" #include "chrome/updater/win/ui/resources/resources.grh" #include "ui/gfx/win/msg_util.h" #include "url/gurl.h" namespace base { class TimeDelta; class Version; } // namespace base namespace updater::ui { class ProgressWndEvents : public CompleteWndEvents { public: // Restarts the running browsers. // If |restart_all_browsers| is true, all known browsers will be restarted. virtual bool DoRestartBrowser(bool restart_all_browsers, const std::vector& urls) = 0; // Initiates a reboot and returns whether it was initiated successfully. virtual bool DoReboot() = 0; // Indicates that current operation is cancelled. virtual void DoCancel() = 0; }; inline constexpr UINT WM_SET_APP_LOGO = WM_APP + 10; // Implements the UI progress window. class ProgressWnd : public CompleteWnd, public AppInstallProgress { public: ProgressWnd(MessageLoop* message_loop, HWND parent); ProgressWnd(const ProgressWnd&) = delete; ProgressWnd& operator=(const ProgressWnd&) = delete; ~ProgressWnd() override; void SetEventSink(ProgressWndEvents* ev); CR_BEGIN_MSG_MAP_EX(ProgressWnd) CR_MESSAGE_HANDLER_EX(WM_SET_APP_LOGO, OnSetAppLogo) CR_MESSAGE_HANDLER_EX(WM_INITDIALOG, OnInitDialog) CR_MESSAGE_HANDLER_EX(WM_SIZE, OnSize) CR_MESSAGE_HANDLER_EX(WM_ERASEBKGND, OnEraseBkgnd) CR_MESSAGE_HANDLER_EX(WM_SYSCOLORCHANGE, OnThemeChanged) CR_MESSAGE_HANDLER_EX(WM_SETTINGCHANGE, OnSettingChange) CR_MESSAGE_HANDLER_EX(WM_THEMECHANGED, OnThemeChanged) CR_MSG_WM_CTLCOLORSTATIC(OnCtlColorStatic) CR_COMMAND_HANDLER_EX(IDC_BUTTON1, BN_CLICKED, OnClickedButton) CR_COMMAND_HANDLER_EX(IDC_BUTTON2, BN_CLICKED, OnClickedButton) CR_COMMAND_HANDLER_EX(IDC_CLOSE, BN_CLICKED, OnClickedButton) CR_CHAIN_MSG_MAP(CompleteWnd) CR_END_MSG_MAP() private: FRIEND_TEST_ALL_PREFIXES(ProgressWndTest, ClickedButton); FRIEND_TEST_ALL_PREFIXES(ProgressWndTest, OnInstallStopped); FRIEND_TEST_ALL_PREFIXES(ProgressWndTest, MaybeCloseWindow); FRIEND_TEST_ALL_PREFIXES(ProgressWndTest, GetBundleCompletionCode); FRIEND_TEST_ALL_PREFIXES(ProgressWndTest, DeterminePostInstallUrls); FRIEND_TEST_ALL_PREFIXES(ProgressWndTest, OnCheckingForUpdate); FRIEND_TEST_ALL_PREFIXES(ProgressWndTest, OnWaitingToDownload); FRIEND_TEST_ALL_PREFIXES(ProgressWndTest, OnWaitingRetryDownload); FRIEND_TEST_ALL_PREFIXES(ProgressWndTest, OnPause); FRIEND_TEST_ALL_PREFIXES(ProgressWndTest, OnDownloading); FRIEND_TEST_ALL_PREFIXES(ProgressWndTest, OnComplete); FRIEND_TEST_ALL_PREFIXES(ProgressWndTest, LaunchCmdLine); FRIEND_TEST_ALL_PREFIXES(ProgressWndTest, FlatButtonSubclass); FRIEND_TEST_ALL_PREFIXES(ProgressWndTest, SetAppLogoDynamicSizing); FRIEND_TEST_ALL_PREFIXES(ProgressWndTest, SetAppLogoThemeSwitching); enum class States { STATE_INIT = 0, STATE_CHECKING_FOR_UPDATE, STATE_WAITING_TO_DOWNLOAD, STATE_DOWNLOADING, STATE_WAITING_TO_INSTALL, STATE_INSTALLING, STATE_PAUSED, STATE_COMPLETE_SUCCESS, STATE_COMPLETE_ERROR, STATE_COMPLETE_RESTART_BROWSER, STATE_COMPLETE_RESTART_ALL_BROWSERS, STATE_COMPLETE_REBOOT, STATE_END, }; static CompletionCodes GetBundleCompletionCode( const ObserverCompletionInfo& info); static std::wstring GetBundleCompletionErrorMessages( const ObserverCompletionInfo& info); // Overrides for AppInstallProgress. // These functions are called on the thread which owns this window. void OnCheckingForUpdate() override; void OnUpdateAvailable(const std::string& app_id, const std::u16string& app_name, const base::Version& version) override; void OnWaitingToDownload(const std::string& app_id, const std::u16string& app_name) override; void OnDownloading(const std::string& app_id, const std::u16string& app_name, const std::optional time_remaining, int pos) override; void OnWaitingRetryDownload(const std::string& app_id, const std::u16string& app_name, base::Time next_retry_time) override; void OnWaitingToInstall(const std::string& app_id, const std::u16string& app_name) override; void OnInstalling(const std::string& app_id, const std::u16string& app_name, const std::optional time_remaining, int pos) override; void OnPause() override; void OnComplete(const ObserverCompletionInfo& observer_info) override; LRESULT OnSetAppLogo(UINT msg, WPARAM wparam, LPARAM lparam); LRESULT OnInitDialog(UINT msg, WPARAM wparam, LPARAM lparam); LRESULT OnSize(UINT msg, WPARAM wparam, LPARAM lparam); void OnClickedButton(UINT notify_code, int id, HWND wnd_ctl); LRESULT OnEraseBkgnd(UINT msg, WPARAM wparam, LPARAM lparam); LRESULT OnSettingChange(UINT msg, WPARAM wparam, LPARAM lparam); LRESULT OnThemeChanged(UINT msg, WPARAM wparam, LPARAM lparam); HBRUSH OnCtlColorStatic(HDC dc, HWND ctl_hwnd); void SetControlText(int id, const std::wstring& text); void SetAppLogo(HBITMAP light_bitmap, HBITMAP dark_bitmap); void UpdateAppLogo(); HBITMAP GetCurrentAppLogoBitmap() const; // Returns true if this window is closed. bool MaybeCloseWindow() override; HRESULT ChangeControlState(); HRESULT SetMarqueeMode(bool is_marquee); void HandleCancelRequest(); void UpdateWindowRgn(); void ApplyDpiScaling(int dpi); int GetScaledCornerRadius() const; void DeterminePostInstallUrls(const ObserverCompletionInfo& info); SEQUENCE_CHECKER(sequence_checker_); States cur_state_ = States::STATE_INIT; raw_ptr events_sink_ = nullptr; std::vector post_install_urls_; bool is_canceled_ = false; struct ControlState { private: static constexpr size_t kNumControlAttributes = 1 + std::to_underlying(States::STATE_END); public: const int id; const std::array attr; }; static const ControlState ctls_[]; // Background image cache for both light and dark themes. base::win::ScopedGDIObject light_bg_bmp_; base::win::ScopedGDIObject dark_bg_bmp_; // Cached original app logo bitmaps for light and dark themes received via // WM_SET_APP_LOGO. base::win::ScopedGDIObject light_app_logo_bmp_; base::win::ScopedGDIObject dark_app_logo_bmp_; // Scaled app logo bitmap dynamically sized for the window's current DPI. base::win::ScopedGDIObject scaled_app_logo_bmp_; // The design-time bottom Y-coordinate of `IDC_APP_BITMAP` normalized to // standard 96-DPI space. The bottom edge is locked to preserve the design // margin above the dialog buttons, allowing taller square logos (e.g. 48x48) // to expand upward into the empty area below the progress bar. int initial_app_logo_base_bottom_y_ = -1; // Returns the bounding rectangle of `control` in parent client coordinates, // normalized for right-to-left (RTL) mirrored layouts. RECT GetControlClientRect(HWND control) const; HBITMAP GetBackgroundBitmap(bool is_dark_mode); // The speed by which the progress bar moves in marquee mode. static constexpr int kMarqueeModeUpdatesMs = 15; // Subclassed buttons representing the standard dialog actions. // btn1_ (Primary): Used for "Restart Now" actions on reboot/restart screens. // btn2_ (Secondary): Used for "Restart Later" actions on reboot/restart // screens. close_btn_ (Primary): Used for "Close" or "Cancel" actions. // get_help_btn_ (Secondary): Used for the "Get Help" link action. FlatButton btn1_; FlatButton btn2_; FlatButton close_btn_; FlatButton get_help_btn_; FRIEND_TEST_ALL_PREFIXES(ProgressWndTest, SetAppLogoDynamicSizing); CR_MSG_MAP_CLASS_DECLARATIONS(ProgressWnd) }; } // namespace updater::ui #endif // CHROME_UPDATER_WIN_UI_PROGRESS_WND_H_