OLD | NEW |
| (Empty) |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #ifndef CHROME_FRAME_INFOBARS_INTERNAL_SUBCLASSING_WINDOW_WITH_DELEGATE_H_ | |
6 #define CHROME_FRAME_INFOBARS_INTERNAL_SUBCLASSING_WINDOW_WITH_DELEGATE_H_ | |
7 | |
8 #include <atlbase.h> | |
9 #include <atlcrack.h> | |
10 #include <atlwin.h> | |
11 | |
12 #include "base/basictypes.h" | |
13 #include "base/logging.h" | |
14 #include "base/memory/scoped_ptr.h" | |
15 #include "chrome_frame/pin_module.h" | |
16 | |
17 // Implements behavior common to HostWindowManager and DisplacedWindowManager. | |
18 template<typename T> class SubclassingWindowWithDelegate | |
19 : public CWindowImpl<T> { | |
20 public: | |
21 // Allows clients to modify the dimensions of the displaced window. | |
22 // Through its destructor, allows clients to know when the subclassed window | |
23 // is destroyed. | |
24 class Delegate { | |
25 public: | |
26 // The delegate will be deleted when the subclassed window is destroyed. | |
27 virtual ~Delegate() {} | |
28 | |
29 // Receives the natural dimensions of the displaced window. Upon return, | |
30 // rect should contain the adjusted dimensions (i.e., possibly reduced to | |
31 // accomodate an infobar). | |
32 virtual void AdjustDisplacedWindowDimensions(RECT* rect) = 0; | |
33 }; // class Delegate | |
34 | |
35 SubclassingWindowWithDelegate() {} | |
36 | |
37 // Returns true if the window is successfully subclassed, in which case this | |
38 // instance will take responsibility for its own destruction when the window | |
39 // is destroyed. If this method returns false, the caller should delete the | |
40 // instance immediately. | |
41 // | |
42 // Takes ownership of delegate in either case, deleting it when the window | |
43 // is destroyed (or immediately, in case of failure). | |
44 bool Initialize(HWND hwnd, Delegate* delegate) { | |
45 DCHECK(delegate != NULL); | |
46 DCHECK(delegate_ == NULL); | |
47 scoped_ptr<Delegate> new_delegate(delegate); | |
48 | |
49 if (!::IsWindow(hwnd) || !SubclassWindow(hwnd)) { | |
50 PLOG(ERROR) << "Failed to subclass an HWND"; | |
51 return false; | |
52 } | |
53 | |
54 // Ensure we won't be unloaded while our window proc is attached to the tab | |
55 // window. | |
56 chrome_frame::PinModule(); | |
57 | |
58 delegate_.swap(new_delegate); | |
59 | |
60 return true; | |
61 } | |
62 | |
63 // Returns the delegate associated with the specified window, if any. | |
64 static Delegate* GetDelegateForHwnd(HWND hwnd) { | |
65 return reinterpret_cast<Delegate*>( | |
66 ::SendMessage(hwnd, RegisterGetDelegateMessage(), NULL, NULL)); | |
67 } | |
68 | |
69 BEGIN_MSG_MAP_EX(SubclassingWindowWithDelegate) | |
70 MESSAGE_HANDLER(RegisterGetDelegateMessage(), OnGetDelegate) | |
71 MSG_WM_DESTROY(OnDestroy) | |
72 END_MSG_MAP() | |
73 | |
74 // This instance is now free to be deleted. | |
75 virtual void OnFinalMessage(HWND hwnd) { | |
76 delete this; | |
77 } | |
78 | |
79 protected: | |
80 scoped_ptr<Delegate>& delegate() { return delegate_; } | |
81 | |
82 private: | |
83 // Registers a unique ID for our custom event. | |
84 static UINT RegisterGetDelegateMessage() { | |
85 static UINT message_id( | |
86 RegisterWindowMessage(L"SubclassingWindowWithDelegate::OnGetDelegate")); | |
87 return message_id; | |
88 } | |
89 | |
90 // The subclassed window has been destroyed. Delete the delegate. We will | |
91 // delete ourselves in OnFinalMessage. | |
92 void OnDestroy() { | |
93 delegate_.reset(); | |
94 } | |
95 | |
96 LRESULT OnGetDelegate(UINT message, | |
97 WPARAM wparam, | |
98 LPARAM lparam, | |
99 BOOL& handled) { | |
100 return reinterpret_cast<LRESULT>(delegate_.get()); | |
101 } | |
102 | |
103 scoped_ptr<Delegate> delegate_; | |
104 DISALLOW_COPY_AND_ASSIGN(SubclassingWindowWithDelegate); | |
105 }; // class SubclassingWindowWithDelegate | |
106 | |
107 #endif // CHROME_FRAME_INFOBARS_INTERNAL_SUBCLASSING_WINDOW_WITH_DELEGATE_H_ | |
OLD | NEW |