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
|
// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef UI_SHELL_DIALOGS_AUTO_CLOSE_DIALOG_EVENT_HANDLER_WIN_H_
#define UI_SHELL_DIALOGS_AUTO_CLOSE_DIALOG_EVENT_HANDLER_WIN_H_
#include <shobjidl_core.h>
#include <wrl.h>
#include <wrl/client.h>
#include "base/threading/thread_checker.h"
namespace ui {
// This IFileDialogEvents implementation ensures that the file dialog it is
// attached to will be closed if its owner is closed.
class AutoCloseDialogEventHandler
: public Microsoft::WRL::RuntimeClass<
Microsoft::WRL::RuntimeClassFlags<Microsoft::WRL::ClassicCom>,
IFileDialogEvents> {
public:
explicit AutoCloseDialogEventHandler(HWND owner_window);
private:
~AutoCloseDialogEventHandler() override;
// Initializes the event hook to watch for the owner window closing.
HRESULT Initialize(IFileDialog* file_dialog);
// Invoked by the event hook whenever a window is destroyed.
void OnWindowDestroyedNotification(HWND window);
static void CALLBACK EventHookCallback(HWINEVENTHOOK handle,
DWORD event,
HWND hwnd,
LONG id_object,
LONG id_child,
DWORD event_thread,
DWORD event_time);
// IFileDialogEvents:
IFACEMETHODIMP OnTypeChange(IFileDialog* pfd) override;
IFACEMETHODIMP OnFileOk(IFileDialog*) override;
IFACEMETHODIMP OnFolderChange(IFileDialog*) override;
IFACEMETHODIMP OnFolderChanging(IFileDialog*, IShellItem*) override;
IFACEMETHODIMP OnSelectionChange(IFileDialog*) override;
IFACEMETHODIMP OnShareViolation(IFileDialog*,
IShellItem*,
FDE_SHAREVIOLATION_RESPONSE*) override;
IFACEMETHODIMP OnOverwrite(IFileDialog*,
IShellItem*,
FDE_OVERWRITE_RESPONSE*) override;
// This is the owner window. When it closes, the dialog window also needs to
// be closed.
HWND owner_window_ = nullptr;
// Indicates if `Initialize()` has been invoked. Used to prevent multiple
// initializations.
bool initialize_called_ = false;
// The event hook handle. Used to uninstall the hook.
HWINEVENTHOOK event_hook_ = nullptr;
HWND dialog_window_ = nullptr;
THREAD_CHECKER(thread_checker_);
};
} // namespace ui
#endif // UI_SHELL_DIALOGS_AUTO_CLOSE_DIALOG_EVENT_HANDLER_WIN_H_
|
