1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
|
// 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_UI_H_
#define CHROME_UPDATER_WIN_UI_UI_H_
#include <string>
#include <vector>
#include "base/memory/raw_ptr.h"
#include "base/sequence_checker.h"
#include "base/win/scoped_gdi_object.h"
#include "chrome/updater/updater_scope.h"
#include "chrome/updater/win/ui/owner_draw_controls.h"
#include "chrome/updater/win/ui/resources/resources.grh"
namespace updater::ui {
void EnableFlatButtons(HWND hwnd_parent);
void HideWindowChildren(HWND hwnd_parent);
class OmahaWndEvents {
public:
virtual ~OmahaWndEvents() = default;
virtual void DoClose() = 0;
virtual void DoExit() = 0;
};
// Implements the UI progress window.
class OmahaWnd : public CAxDialogImpl<OmahaWnd>,
public OwnerDrawTitleBar,
public CustomDlgColors,
public WTL::CMessageFilter {
using Base = CAxDialogImpl<OmahaWnd>;
public:
const int IDD;
OmahaWnd(const OmahaWnd&) = delete;
OmahaWnd& operator=(const OmahaWnd&) = delete;
~OmahaWnd() override;
virtual HRESULT Initialize();
// Overrides for CMessageFilter.
BOOL PreTranslateMessage(MSG* msg) override;
void SetEventSink(OmahaWndEvents* ev) { events_sink_ = ev; }
void set_scope(UpdaterScope scope) { scope_ = scope; }
void set_bundle_name(const std::u16string& bundle_name) {
bundle_name_ = bundle_name;
}
virtual void Show();
BEGIN_MSG_MAP(OmahaWnd)
MESSAGE_HANDLER(WM_CLOSE, OnClose)
MESSAGE_HANDLER(WM_NCDESTROY, OnNCDestroy)
COMMAND_ID_HANDLER(IDCANCEL, OnCancel)
CHAIN_MSG_MAP(Base)
CHAIN_MSG_MAP(OwnerDrawTitleBar)
CHAIN_MSG_MAP(CustomDlgColors)
END_MSG_MAP()
protected:
struct ControlAttributes {
const bool is_ignore_entry = false;
const bool is_visible = false;
const bool is_enabled = false;
const bool is_button = false;
const bool is_default = false;
};
OmahaWnd(int dialog_id,
WTL::CMessageLoop* message_loop,
HWND parent,
const std::wstring& lang);
// Message and command handlers.
LRESULT OnClose(UINT msg,
WPARAM wparam,
LPARAM lparam,
BOOL& handled); // NOLINT
LRESULT OnNCDestroy(UINT msg,
WPARAM wparam,
LPARAM lparam,
BOOL& handled); // NOLINT
LRESULT OnCancel(WORD notify_code,
WORD id,
HWND wnd_ctl,
BOOL& handled); // NOLINT
// Returns true if the window is closed.
virtual bool MaybeCloseWindow() = 0;
// Returns true to indicate that the client continues handling OnComplete.
bool OnComplete();
HRESULT CloseWindow();
void InitializeDialog();
HRESULT EnableClose(bool enable);
HRESULT EnableSystemCloseButton(bool enable);
void SetControlAttributes(int control_id,
const ControlAttributes& attributes);
void SetVisible(bool visible) {
ShowWindow(visible ? SW_SHOWNORMAL : SW_HIDE);
}
WTL::CMessageLoop* message_loop() { return message_loop_; }
std::wstring lang() const { return lang_; }
bool is_complete() { return is_complete_; }
bool is_close_enabled() { return is_close_enabled_; }
UpdaterScope scope() { return scope_; }
const std::u16string& bundle_name() { return bundle_name_; }
static const ControlAttributes kVisibleTextAttributes;
static const ControlAttributes kDefaultActiveButtonAttributes;
static const ControlAttributes kDisabledButtonAttributes;
static const ControlAttributes kNonDefaultActiveButtonAttributes;
static const ControlAttributes kVisibleImageAttributes;
static const ControlAttributes kDisabledNonButtonAttributes;
private:
HRESULT SetWindowIcon();
void MaybeRequestExitProcess();
void RequestExitProcess();
SEQUENCE_CHECKER(sequence_checker_);
raw_ptr<WTL::CMessageLoop> message_loop_;
HWND parent_;
const std::wstring lang_;
bool is_complete_;
bool is_close_enabled_;
raw_ptr<OmahaWndEvents> events_sink_;
UpdaterScope scope_;
std::u16string bundle_name_;
// Handle to large icon to show when ALT-TAB
base::win::ScopedGDIObject<HICON> hicon_;
WTL::CFont default_font_;
WTL::CFont font_;
WTL::CFont error_font_;
CustomProgressBarCtrl progress_bar_;
};
// Registers the specified common control classes from the common control DLL.
// Calls are cumulative, meaning control_classes are added to existing classes.
// UIs that use common controls should call this function to ensure that the UI
// supports visual styles.
HRESULT InitializeCommonControls(DWORD control_classes);
} // namespace updater::ui
#endif // CHROME_UPDATER_WIN_UI_UI_H_
|
