1 /* -*- Mode: C++; tab-width: 3; indent-tabs-mode: nil; c-basic-offset: 2 -*-
2  *
3  * This Source Code Form is subject to the terms of the Mozilla Public
4  * License, v. 2.0. If a copy of the MPL was not distributed with this
5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
6 
7 // Local Includes
8 #include "nsChromeTreeOwner.h"
9 #include "nsXULWindow.h"
10 
11 // Helper Classes
12 #include "nsString.h"
13 #include "nsIEmbeddingSiteWindow.h"
14 #include "nsIServiceManager.h"
15 #include "nsIDocShellTreeItem.h"
16 
17 // Interfaces needed to include
18 #include "nsIPrompt.h"
19 #include "nsIAuthPrompt.h"
20 #include "nsIBrowserDOMWindow.h"
21 #include "nsIWebProgress.h"
22 #include "nsIWidget.h"
23 #include "nsIWindowMediator.h"
24 #include "nsIDOMChromeWindow.h"
25 #include "nsIDOMNode.h"
26 #include "nsIDOMElement.h"
27 #include "nsIDOMNodeList.h"
28 #include "nsIDOMXULElement.h"
29 #include "nsIXULBrowserWindow.h"
30 #include "mozilla/dom/Element.h"
31 
32 using namespace mozilla;
33 
34 //*****************************************************************************
35 // nsChromeTreeOwner string literals
36 //*****************************************************************************
37 
38 const nsLiteralString kPersist(u"persist");
39 const nsLiteralString kScreenX(u"screenX");
40 const nsLiteralString kScreenY(u"screenY");
41 const nsLiteralString kWidth(u"width");
42 const nsLiteralString kHeight(u"height");
43 const nsLiteralString kSizemode(u"sizemode");
44 const nsLiteralString kSpace(u" ");
45 
46 //*****************************************************************************
47 //***    nsChromeTreeOwner: Object Management
48 //*****************************************************************************
49 
nsChromeTreeOwner()50 nsChromeTreeOwner::nsChromeTreeOwner() : mXULWindow(nullptr) {}
51 
~nsChromeTreeOwner()52 nsChromeTreeOwner::~nsChromeTreeOwner() {}
53 
54 //*****************************************************************************
55 // nsChromeTreeOwner::nsISupports
56 //*****************************************************************************
57 
58 NS_IMPL_ADDREF(nsChromeTreeOwner)
NS_IMPL_RELEASE(nsChromeTreeOwner)59 NS_IMPL_RELEASE(nsChromeTreeOwner)
60 
61 NS_INTERFACE_MAP_BEGIN(nsChromeTreeOwner)
62   NS_INTERFACE_MAP_ENTRY_AMBIGUOUS(nsISupports, nsIDocShellTreeOwner)
63   NS_INTERFACE_MAP_ENTRY(nsIDocShellTreeOwner)
64   NS_INTERFACE_MAP_ENTRY(nsIBaseWindow)
65   NS_INTERFACE_MAP_ENTRY(nsIWebProgressListener)
66   NS_INTERFACE_MAP_ENTRY(nsIInterfaceRequestor)
67   NS_INTERFACE_MAP_ENTRY(nsISupportsWeakReference)
68 NS_INTERFACE_MAP_END
69 
70 //*****************************************************************************
71 // nsChromeTreeOwner::nsIInterfaceRequestor
72 //*****************************************************************************
73 
74 NS_IMETHODIMP nsChromeTreeOwner::GetInterface(const nsIID& aIID, void** aSink) {
75   NS_ENSURE_ARG_POINTER(aSink);
76 
77   if (aIID.Equals(NS_GET_IID(nsIPrompt))) {
78     NS_ENSURE_STATE(mXULWindow);
79     return mXULWindow->GetInterface(aIID, aSink);
80   }
81   if (aIID.Equals(NS_GET_IID(nsIAuthPrompt))) {
82     NS_ENSURE_STATE(mXULWindow);
83     return mXULWindow->GetInterface(aIID, aSink);
84   }
85   if (aIID.Equals(NS_GET_IID(nsIWebBrowserChrome))) {
86     NS_ENSURE_STATE(mXULWindow);
87     return mXULWindow->GetInterface(aIID, aSink);
88   }
89   if (aIID.Equals(NS_GET_IID(nsIEmbeddingSiteWindow))) {
90     NS_ENSURE_STATE(mXULWindow);
91     return mXULWindow->GetInterface(aIID, aSink);
92   }
93   if (aIID.Equals(NS_GET_IID(nsIXULWindow))) {
94     NS_ENSURE_STATE(mXULWindow);
95     return mXULWindow->QueryInterface(aIID, aSink);
96   }
97 
98   return QueryInterface(aIID, aSink);
99 }
100 
101 //*****************************************************************************
102 // nsChromeTreeOwner::nsIDocShellTreeOwner
103 //*****************************************************************************
104 
105 NS_IMETHODIMP
ContentShellAdded(nsIDocShellTreeItem * aContentShell,bool aPrimary)106 nsChromeTreeOwner::ContentShellAdded(nsIDocShellTreeItem* aContentShell,
107                                      bool aPrimary) {
108   NS_ENSURE_STATE(mXULWindow);
109   return mXULWindow->ContentShellAdded(aContentShell, aPrimary);
110 }
111 
112 NS_IMETHODIMP
ContentShellRemoved(nsIDocShellTreeItem * aContentShell)113 nsChromeTreeOwner::ContentShellRemoved(nsIDocShellTreeItem* aContentShell) {
114   NS_ENSURE_STATE(mXULWindow);
115   return mXULWindow->ContentShellRemoved(aContentShell);
116 }
117 
GetPrimaryContentShell(nsIDocShellTreeItem ** aShell)118 NS_IMETHODIMP nsChromeTreeOwner::GetPrimaryContentShell(
119     nsIDocShellTreeItem** aShell) {
120   NS_ENSURE_STATE(mXULWindow);
121   return mXULWindow->GetPrimaryContentShell(aShell);
122 }
123 
124 NS_IMETHODIMP
TabParentAdded(nsITabParent * aTab,bool aPrimary)125 nsChromeTreeOwner::TabParentAdded(nsITabParent* aTab, bool aPrimary) {
126   NS_ENSURE_STATE(mXULWindow);
127   return mXULWindow->TabParentAdded(aTab, aPrimary);
128 }
129 
130 NS_IMETHODIMP
TabParentRemoved(nsITabParent * aTab)131 nsChromeTreeOwner::TabParentRemoved(nsITabParent* aTab) {
132   NS_ENSURE_STATE(mXULWindow);
133   return mXULWindow->TabParentRemoved(aTab);
134 }
135 
136 NS_IMETHODIMP
GetPrimaryTabParent(nsITabParent ** aTab)137 nsChromeTreeOwner::GetPrimaryTabParent(nsITabParent** aTab) {
138   NS_ENSURE_STATE(mXULWindow);
139   return mXULWindow->GetPrimaryTabParent(aTab);
140 }
141 
142 NS_IMETHODIMP
GetPrimaryContentSize(int32_t * aWidth,int32_t * aHeight)143 nsChromeTreeOwner::GetPrimaryContentSize(int32_t* aWidth, int32_t* aHeight) {
144   NS_ENSURE_STATE(mXULWindow);
145   return mXULWindow->GetPrimaryContentSize(aWidth, aHeight);
146 }
147 
148 NS_IMETHODIMP
SetPrimaryContentSize(int32_t aWidth,int32_t aHeight)149 nsChromeTreeOwner::SetPrimaryContentSize(int32_t aWidth, int32_t aHeight) {
150   NS_ENSURE_STATE(mXULWindow);
151   return mXULWindow->SetPrimaryContentSize(aWidth, aHeight);
152 }
153 
154 NS_IMETHODIMP
GetRootShellSize(int32_t * aWidth,int32_t * aHeight)155 nsChromeTreeOwner::GetRootShellSize(int32_t* aWidth, int32_t* aHeight) {
156   NS_ENSURE_STATE(mXULWindow);
157   return mXULWindow->GetRootShellSize(aWidth, aHeight);
158 }
159 
160 NS_IMETHODIMP
SetRootShellSize(int32_t aWidth,int32_t aHeight)161 nsChromeTreeOwner::SetRootShellSize(int32_t aWidth, int32_t aHeight) {
162   NS_ENSURE_STATE(mXULWindow);
163   return mXULWindow->SetRootShellSize(aWidth, aHeight);
164 }
165 
SizeShellTo(nsIDocShellTreeItem * aShellItem,int32_t aCX,int32_t aCY)166 NS_IMETHODIMP nsChromeTreeOwner::SizeShellTo(nsIDocShellTreeItem* aShellItem,
167                                              int32_t aCX, int32_t aCY) {
168   NS_ENSURE_STATE(mXULWindow);
169   return mXULWindow->SizeShellTo(aShellItem, aCX, aCY);
170 }
171 
172 NS_IMETHODIMP
SetPersistence(bool aPersistPosition,bool aPersistSize,bool aPersistSizeMode)173 nsChromeTreeOwner::SetPersistence(bool aPersistPosition, bool aPersistSize,
174                                   bool aPersistSizeMode) {
175   NS_ENSURE_STATE(mXULWindow);
176   nsCOMPtr<dom::Element> docShellElement = mXULWindow->GetWindowDOMElement();
177   if (!docShellElement) return NS_ERROR_FAILURE;
178 
179   nsAutoString persistString;
180   docShellElement->GetAttribute(kPersist, persistString);
181 
182   bool saveString = false;
183   int32_t index;
184 
185 #define FIND_PERSIST_STRING(aString, aCond)     \
186   index = persistString.Find(aString);          \
187   if (!aCond && index > kNotFound) {            \
188     persistString.Cut(index, aString.Length()); \
189     saveString = true;                          \
190   } else if (aCond && index == kNotFound) {     \
191     persistString.Append(kSpace + aString);     \
192     saveString = true;                          \
193   }
194   FIND_PERSIST_STRING(kScreenX, aPersistPosition);
195   FIND_PERSIST_STRING(kScreenY, aPersistPosition);
196   FIND_PERSIST_STRING(kWidth, aPersistSize);
197   FIND_PERSIST_STRING(kHeight, aPersistSize);
198   FIND_PERSIST_STRING(kSizemode, aPersistSizeMode);
199 
200   ErrorResult rv;
201   if (saveString) {
202     docShellElement->SetAttribute(kPersist, persistString, rv);
203   }
204 
205   return NS_OK;
206 }
207 
208 NS_IMETHODIMP
GetPersistence(bool * aPersistPosition,bool * aPersistSize,bool * aPersistSizeMode)209 nsChromeTreeOwner::GetPersistence(bool* aPersistPosition, bool* aPersistSize,
210                                   bool* aPersistSizeMode) {
211   NS_ENSURE_STATE(mXULWindow);
212   nsCOMPtr<dom::Element> docShellElement = mXULWindow->GetWindowDOMElement();
213   if (!docShellElement) return NS_ERROR_FAILURE;
214 
215   nsAutoString persistString;
216   docShellElement->GetAttribute(kPersist, persistString);
217 
218   // data structure doesn't quite match the question, but it's close enough
219   // for what we want (since this method is never actually called...)
220   if (aPersistPosition)
221     *aPersistPosition = persistString.Find(kScreenX) > kNotFound ||
222                         persistString.Find(kScreenY) > kNotFound;
223   if (aPersistSize)
224     *aPersistSize = persistString.Find(kWidth) > kNotFound ||
225                     persistString.Find(kHeight) > kNotFound;
226   if (aPersistSizeMode)
227     *aPersistSizeMode = persistString.Find(kSizemode) > kNotFound;
228 
229   return NS_OK;
230 }
231 
232 NS_IMETHODIMP
GetTabCount(uint32_t * aResult)233 nsChromeTreeOwner::GetTabCount(uint32_t* aResult) {
234   if (mXULWindow) {
235     return mXULWindow->GetTabCount(aResult);
236   }
237 
238   *aResult = 0;
239   return NS_OK;
240 }
241 
242 NS_IMETHODIMP
GetHasPrimaryContent(bool * aResult)243 nsChromeTreeOwner::GetHasPrimaryContent(bool* aResult) {
244   NS_ENSURE_STATE(mXULWindow);
245   return mXULWindow->GetHasPrimaryContent(aResult);
246 }
247 
248 //*****************************************************************************
249 // nsChromeTreeOwner::nsIBaseWindow
250 //*****************************************************************************
251 
InitWindow(nativeWindow aParentNativeWindow,nsIWidget * parentWidget,int32_t x,int32_t y,int32_t cx,int32_t cy)252 NS_IMETHODIMP nsChromeTreeOwner::InitWindow(nativeWindow aParentNativeWindow,
253                                             nsIWidget* parentWidget, int32_t x,
254                                             int32_t y, int32_t cx, int32_t cy) {
255   // Ignore widget parents for now.  Don't think those are a vaild thing to
256   // call.
257   NS_ENSURE_SUCCESS(SetPositionAndSize(x, y, cx, cy, 0), NS_ERROR_FAILURE);
258 
259   return NS_OK;
260 }
261 
Create()262 NS_IMETHODIMP nsChromeTreeOwner::Create() {
263   NS_ASSERTION(false, "You can't call this");
264   return NS_ERROR_UNEXPECTED;
265 }
266 
Destroy()267 NS_IMETHODIMP nsChromeTreeOwner::Destroy() {
268   NS_ENSURE_STATE(mXULWindow);
269   return mXULWindow->Destroy();
270 }
271 
GetUnscaledDevicePixelsPerCSSPixel(double * aScale)272 NS_IMETHODIMP nsChromeTreeOwner::GetUnscaledDevicePixelsPerCSSPixel(
273     double* aScale) {
274   NS_ENSURE_STATE(mXULWindow);
275   return mXULWindow->GetUnscaledDevicePixelsPerCSSPixel(aScale);
276 }
277 
GetDevicePixelsPerDesktopPixel(double * aScale)278 NS_IMETHODIMP nsChromeTreeOwner::GetDevicePixelsPerDesktopPixel(
279     double* aScale) {
280   NS_ENSURE_STATE(mXULWindow);
281   return mXULWindow->GetDevicePixelsPerDesktopPixel(aScale);
282 }
283 
SetPositionDesktopPix(int32_t x,int32_t y)284 NS_IMETHODIMP nsChromeTreeOwner::SetPositionDesktopPix(int32_t x, int32_t y) {
285   NS_ENSURE_STATE(mXULWindow);
286   return mXULWindow->SetPositionDesktopPix(x, y);
287 }
288 
SetPosition(int32_t x,int32_t y)289 NS_IMETHODIMP nsChromeTreeOwner::SetPosition(int32_t x, int32_t y) {
290   NS_ENSURE_STATE(mXULWindow);
291   return mXULWindow->SetPosition(x, y);
292 }
293 
GetPosition(int32_t * x,int32_t * y)294 NS_IMETHODIMP nsChromeTreeOwner::GetPosition(int32_t* x, int32_t* y) {
295   NS_ENSURE_STATE(mXULWindow);
296   return mXULWindow->GetPosition(x, y);
297 }
298 
SetSize(int32_t cx,int32_t cy,bool fRepaint)299 NS_IMETHODIMP nsChromeTreeOwner::SetSize(int32_t cx, int32_t cy,
300                                          bool fRepaint) {
301   NS_ENSURE_STATE(mXULWindow);
302   return mXULWindow->SetSize(cx, cy, fRepaint);
303 }
304 
GetSize(int32_t * cx,int32_t * cy)305 NS_IMETHODIMP nsChromeTreeOwner::GetSize(int32_t* cx, int32_t* cy) {
306   NS_ENSURE_STATE(mXULWindow);
307   return mXULWindow->GetSize(cx, cy);
308 }
309 
SetPositionAndSize(int32_t x,int32_t y,int32_t cx,int32_t cy,uint32_t aFlags)310 NS_IMETHODIMP nsChromeTreeOwner::SetPositionAndSize(int32_t x, int32_t y,
311                                                     int32_t cx, int32_t cy,
312                                                     uint32_t aFlags) {
313   NS_ENSURE_STATE(mXULWindow);
314   return mXULWindow->SetPositionAndSize(x, y, cx, cy, aFlags);
315 }
316 
GetPositionAndSize(int32_t * x,int32_t * y,int32_t * cx,int32_t * cy)317 NS_IMETHODIMP nsChromeTreeOwner::GetPositionAndSize(int32_t* x, int32_t* y,
318                                                     int32_t* cx, int32_t* cy) {
319   NS_ENSURE_STATE(mXULWindow);
320   return mXULWindow->GetPositionAndSize(x, y, cx, cy);
321 }
322 
Repaint(bool aForce)323 NS_IMETHODIMP nsChromeTreeOwner::Repaint(bool aForce) {
324   NS_ENSURE_STATE(mXULWindow);
325   return mXULWindow->Repaint(aForce);
326 }
327 
GetParentWidget(nsIWidget ** aParentWidget)328 NS_IMETHODIMP nsChromeTreeOwner::GetParentWidget(nsIWidget** aParentWidget) {
329   NS_ENSURE_STATE(mXULWindow);
330   return mXULWindow->GetParentWidget(aParentWidget);
331 }
332 
SetParentWidget(nsIWidget * aParentWidget)333 NS_IMETHODIMP nsChromeTreeOwner::SetParentWidget(nsIWidget* aParentWidget) {
334   NS_ASSERTION(false, "You can't call this");
335   return NS_ERROR_NOT_IMPLEMENTED;
336 }
337 
GetParentNativeWindow(nativeWindow * aParentNativeWindow)338 NS_IMETHODIMP nsChromeTreeOwner::GetParentNativeWindow(
339     nativeWindow* aParentNativeWindow) {
340   NS_ENSURE_STATE(mXULWindow);
341   return mXULWindow->GetParentNativeWindow(aParentNativeWindow);
342 }
343 
SetParentNativeWindow(nativeWindow aParentNativeWindow)344 NS_IMETHODIMP nsChromeTreeOwner::SetParentNativeWindow(
345     nativeWindow aParentNativeWindow) {
346   NS_ASSERTION(false, "You can't call this");
347   return NS_ERROR_NOT_IMPLEMENTED;
348 }
349 
GetNativeHandle(nsAString & aNativeHandle)350 NS_IMETHODIMP nsChromeTreeOwner::GetNativeHandle(nsAString& aNativeHandle) {
351   NS_ENSURE_STATE(mXULWindow);
352   return mXULWindow->GetNativeHandle(aNativeHandle);
353 }
354 
GetVisibility(bool * aVisibility)355 NS_IMETHODIMP nsChromeTreeOwner::GetVisibility(bool* aVisibility) {
356   NS_ENSURE_STATE(mXULWindow);
357   return mXULWindow->GetVisibility(aVisibility);
358 }
359 
SetVisibility(bool aVisibility)360 NS_IMETHODIMP nsChromeTreeOwner::SetVisibility(bool aVisibility) {
361   NS_ENSURE_STATE(mXULWindow);
362   return mXULWindow->SetVisibility(aVisibility);
363 }
364 
GetEnabled(bool * aEnabled)365 NS_IMETHODIMP nsChromeTreeOwner::GetEnabled(bool* aEnabled) {
366   NS_ENSURE_STATE(mXULWindow);
367   return mXULWindow->GetEnabled(aEnabled);
368 }
369 
SetEnabled(bool aEnable)370 NS_IMETHODIMP nsChromeTreeOwner::SetEnabled(bool aEnable) {
371   NS_ENSURE_STATE(mXULWindow);
372   return mXULWindow->SetEnabled(aEnable);
373 }
374 
GetMainWidget(nsIWidget ** aMainWidget)375 NS_IMETHODIMP nsChromeTreeOwner::GetMainWidget(nsIWidget** aMainWidget) {
376   NS_ENSURE_ARG_POINTER(aMainWidget);
377   NS_ENSURE_STATE(mXULWindow);
378 
379   *aMainWidget = mXULWindow->mWindow;
380   NS_IF_ADDREF(*aMainWidget);
381 
382   return NS_OK;
383 }
384 
SetFocus()385 NS_IMETHODIMP nsChromeTreeOwner::SetFocus() {
386   NS_ENSURE_STATE(mXULWindow);
387   return mXULWindow->SetFocus();
388 }
389 
GetTitle(nsAString & aTitle)390 NS_IMETHODIMP nsChromeTreeOwner::GetTitle(nsAString& aTitle) {
391   NS_ENSURE_STATE(mXULWindow);
392   return mXULWindow->GetTitle(aTitle);
393 }
394 
SetTitle(const nsAString & aTitle)395 NS_IMETHODIMP nsChromeTreeOwner::SetTitle(const nsAString& aTitle) {
396   NS_ENSURE_STATE(mXULWindow);
397   return mXULWindow->SetTitle(aTitle);
398 }
399 
400 //*****************************************************************************
401 // nsChromeTreeOwner::nsIWebProgressListener
402 //*****************************************************************************
403 
404 NS_IMETHODIMP
OnProgressChange(nsIWebProgress * aWebProgress,nsIRequest * aRequest,int32_t aCurSelfProgress,int32_t aMaxSelfProgress,int32_t aCurTotalProgress,int32_t aMaxTotalProgress)405 nsChromeTreeOwner::OnProgressChange(nsIWebProgress* aWebProgress,
406                                     nsIRequest* aRequest,
407                                     int32_t aCurSelfProgress,
408                                     int32_t aMaxSelfProgress,
409                                     int32_t aCurTotalProgress,
410                                     int32_t aMaxTotalProgress) {
411   return NS_OK;
412 }
413 
414 NS_IMETHODIMP
OnStateChange(nsIWebProgress * aWebProgress,nsIRequest * aRequest,uint32_t aProgressStateFlags,nsresult aStatus)415 nsChromeTreeOwner::OnStateChange(nsIWebProgress* aWebProgress,
416                                  nsIRequest* aRequest,
417                                  uint32_t aProgressStateFlags,
418                                  nsresult aStatus) {
419   return NS_OK;
420 }
421 
422 NS_IMETHODIMP
OnLocationChange(nsIWebProgress * aWebProgress,nsIRequest * aRequest,nsIURI * aLocation,uint32_t aFlags)423 nsChromeTreeOwner::OnLocationChange(nsIWebProgress* aWebProgress,
424                                     nsIRequest* aRequest, nsIURI* aLocation,
425                                     uint32_t aFlags) {
426   NS_ENSURE_STATE(mXULWindow);
427 
428   // If loading a new root .xul document, then redo chrome.
429   if (aWebProgress) {
430     nsCOMPtr<nsIDocShell> docshell;
431     mXULWindow->GetDocShell(getter_AddRefs(docshell));
432 
433     nsCOMPtr<nsIWebProgress> webProgress(do_QueryInterface(docshell));
434     if (webProgress != aWebProgress) {
435       return NS_OK;
436     }
437   }
438 
439   mXULWindow->mChromeLoaded = false;
440   return NS_OK;
441 }
442 
443 NS_IMETHODIMP
OnStatusChange(nsIWebProgress * aWebProgress,nsIRequest * aRequest,nsresult aStatus,const char16_t * aMessage)444 nsChromeTreeOwner::OnStatusChange(nsIWebProgress* aWebProgress,
445                                   nsIRequest* aRequest, nsresult aStatus,
446                                   const char16_t* aMessage) {
447   return NS_OK;
448 }
449 
450 NS_IMETHODIMP
OnSecurityChange(nsIWebProgress * aWebProgress,nsIRequest * aRequest,uint32_t state)451 nsChromeTreeOwner::OnSecurityChange(nsIWebProgress* aWebProgress,
452                                     nsIRequest* aRequest, uint32_t state) {
453   return NS_OK;
454 }
455 
456 //*****************************************************************************
457 // nsChromeTreeOwner: Helpers
458 //*****************************************************************************
459 
460 //*****************************************************************************
461 // nsChromeTreeOwner: Accessors
462 //*****************************************************************************
463 
XULWindow(nsXULWindow * aXULWindow)464 void nsChromeTreeOwner::XULWindow(nsXULWindow* aXULWindow) {
465   mXULWindow = aXULWindow;
466 }
467 
XULWindow()468 nsXULWindow* nsChromeTreeOwner::XULWindow() { return mXULWindow; }
469