mshtml: Correctly handle document with already loaded moniker in set_moniker.
[wine] / dlls / mshtml / oleobj.c
1 /*
2  * Copyright 2005 Jacek Caban
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17  */
18
19 #include "config.h"
20
21 #include <stdarg.h>
22 #include <stdio.h>
23
24 #define COBJMACROS
25
26 #include "windef.h"
27 #include "winbase.h"
28 #include "winuser.h"
29 #include "ole2.h"
30 #include "shlguid.h"
31 #include "mshtmdid.h"
32 #include "idispids.h"
33
34 #include "wine/debug.h"
35
36 #include "mshtml_private.h"
37
38 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
39
40 /**********************************************************
41  * IOleObject implementation
42  */
43
44 #define OLEOBJ_THIS(iface) DEFINE_THIS(HTMLDocument, OleObject, iface)
45
46 static HRESULT WINAPI OleObject_QueryInterface(IOleObject *iface, REFIID riid, void **ppvObject)
47 {
48     HTMLDocument *This = OLEOBJ_THIS(iface);
49     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppvObject);
50 }
51
52 static ULONG WINAPI OleObject_AddRef(IOleObject *iface)
53 {
54     HTMLDocument *This = OLEOBJ_THIS(iface);
55     return IHTMLDocument2_AddRef(HTMLDOC(This));
56 }
57
58 static ULONG WINAPI OleObject_Release(IOleObject *iface)
59 {
60     HTMLDocument *This = OLEOBJ_THIS(iface);
61     return IHTMLDocument2_Release(HTMLDOC(This));
62 }
63
64 static void update_hostinfo(HTMLDocumentObj *This, DOCHOSTUIINFO *hostinfo)
65 {
66     nsIScrollable *scrollable;
67     nsresult nsres;
68
69     if(!This->nscontainer)
70         return;
71
72     nsres = nsIWebBrowser_QueryInterface(This->nscontainer->webbrowser, &IID_nsIScrollable, (void**)&scrollable);
73     if(NS_SUCCEEDED(nsres)) {
74         nsres = nsIScrollable_SetDefaultScrollbarPreferences(scrollable, ScrollOrientation_Y,
75                 (hostinfo->dwFlags & DOCHOSTUIFLAG_SCROLL_NO) ? Scrollbar_Never : Scrollbar_Always);
76         if(NS_FAILED(nsres))
77             ERR("Could not set default Y scrollbar prefs: %08x\n", nsres);
78
79         nsres = nsIScrollable_SetDefaultScrollbarPreferences(scrollable, ScrollOrientation_X,
80                 hostinfo->dwFlags & DOCHOSTUIFLAG_SCROLL_NO ? Scrollbar_Never : Scrollbar_Auto);
81         if(NS_FAILED(nsres))
82             ERR("Could not set default X scrollbar prefs: %08x\n", nsres);
83
84         nsIScrollable_Release(scrollable);
85     }else {
86         ERR("Could not get nsIScrollable: %08x\n", nsres);
87     }
88 }
89
90 static HRESULT WINAPI OleObject_SetClientSite(IOleObject *iface, IOleClientSite *pClientSite)
91 {
92     HTMLDocument *This = OLEOBJ_THIS(iface);
93     IDocHostUIHandler *pDocHostUIHandler = NULL;
94     IOleCommandTarget *cmdtrg = NULL;
95     VARIANT silent;
96     HRESULT hres;
97
98     TRACE("(%p)->(%p)\n", This, pClientSite);
99
100     if(pClientSite == This->doc_obj->client)
101         return S_OK;
102
103     if(This->doc_obj->client) {
104         IOleClientSite_Release(This->doc_obj->client);
105         This->doc_obj->client = NULL;
106         This->doc_obj->usermode = UNKNOWN_USERMODE;
107     }
108
109     if(This->doc_obj->hostui) {
110         IDocHostUIHandler_Release(This->doc_obj->hostui);
111         This->doc_obj->hostui = NULL;
112     }
113
114     memset(&This->doc_obj->hostinfo, 0, sizeof(DOCHOSTUIINFO));
115
116     if(!pClientSite)
117         return S_OK;
118
119     hres = IOleObject_QueryInterface(pClientSite, &IID_IDocHostUIHandler, (void**)&pDocHostUIHandler);
120     if(SUCCEEDED(hres)) {
121         DOCHOSTUIINFO hostinfo;
122         LPOLESTR key_path = NULL, override_key_path = NULL;
123         IDocHostUIHandler2 *pDocHostUIHandler2;
124
125         memset(&hostinfo, 0, sizeof(DOCHOSTUIINFO));
126         hostinfo.cbSize = sizeof(DOCHOSTUIINFO);
127         hres = IDocHostUIHandler_GetHostInfo(pDocHostUIHandler, &hostinfo);
128         if(SUCCEEDED(hres)) {
129             TRACE("hostinfo = {%u %08x %08x %s %s}\n",
130                     hostinfo.cbSize, hostinfo.dwFlags, hostinfo.dwDoubleClick,
131                     debugstr_w(hostinfo.pchHostCss), debugstr_w(hostinfo.pchHostNS));
132             update_hostinfo(This->doc_obj, &hostinfo);
133             This->doc_obj->hostinfo = hostinfo;
134         }
135
136         if(!This->doc_obj->has_key_path) {
137             hres = IDocHostUIHandler_GetOptionKeyPath(pDocHostUIHandler, &key_path, 0);
138             if(hres == S_OK && key_path) {
139                 if(key_path[0]) {
140                     /* FIXME: use key_path */
141                     TRACE("key_path = %s\n", debugstr_w(key_path));
142                 }
143                 CoTaskMemFree(key_path);
144             }
145
146             hres = IDocHostUIHandler_QueryInterface(pDocHostUIHandler, &IID_IDocHostUIHandler2,
147                     (void**)&pDocHostUIHandler2);
148             if(SUCCEEDED(hres)) {
149                 hres = IDocHostUIHandler2_GetOverrideKeyPath(pDocHostUIHandler2, &override_key_path, 0);
150                 if(hres == S_OK && override_key_path && override_key_path[0]) {
151                     if(override_key_path[0]) {
152                         /*FIXME: use override_key_path */
153                         TRACE("override_key_path = %s\n", debugstr_w(override_key_path));
154                     }
155                     CoTaskMemFree(override_key_path);
156                 }
157                 IDocHostUIHandler2_Release(pDocHostUIHandler2);
158             }
159
160             This->doc_obj->has_key_path = TRUE;
161         }
162     }
163
164     /* Native calls here GetWindow. What is it for?
165      * We don't have anything to do with it here (yet). */
166     if(pClientSite) {
167         IOleWindow *pOleWindow = NULL;
168         HWND hwnd;
169
170         hres = IOleClientSite_QueryInterface(pClientSite, &IID_IOleWindow, (void**)&pOleWindow);
171         if(SUCCEEDED(hres)) {
172             IOleWindow_GetWindow(pOleWindow, &hwnd);
173             IOleWindow_Release(pOleWindow);
174         }
175     }
176
177     hres = IOleClientSite_QueryInterface(pClientSite, &IID_IOleCommandTarget, (void**)&cmdtrg);
178     if(SUCCEEDED(hres)) {
179         VARIANT var;
180         OLECMD cmd = {OLECMDID_SETPROGRESSTEXT, 0};
181
182         IOleCommandTarget_QueryStatus(cmdtrg, NULL, 1, &cmd, NULL);
183
184         V_VT(&var) = VT_I4;
185         V_I4(&var) = 0;
186         IOleCommandTarget_Exec(cmdtrg, NULL, OLECMDID_SETPROGRESSMAX,
187                 OLECMDEXECOPT_DONTPROMPTUSER, &var, NULL);
188         IOleCommandTarget_Exec(cmdtrg, NULL, OLECMDID_SETPROGRESSPOS, 
189                 OLECMDEXECOPT_DONTPROMPTUSER, &var, NULL);
190
191         IOleCommandTarget_Release(cmdtrg);
192     }
193
194     IOleClientSite_AddRef(pClientSite);
195     This->doc_obj->client = pClientSite;
196     This->doc_obj->hostui = pDocHostUIHandler;
197
198     if(This->doc_obj->usermode == UNKNOWN_USERMODE)
199         IOleControl_OnAmbientPropertyChange(CONTROL(This), DISPID_AMBIENT_USERMODE);
200
201     IOleControl_OnAmbientPropertyChange(CONTROL(This), DISPID_AMBIENT_OFFLINEIFNOTCONNECTED); 
202
203     hres = get_client_disp_property(This->doc_obj->client, DISPID_AMBIENT_SILENT, &silent);
204     if(SUCCEEDED(hres)) {
205         if(V_VT(&silent) != VT_BOOL)
206             WARN("V_VT(silent) = %d\n", V_VT(&silent));
207         else if(V_BOOL(&silent))
208             FIXME("silent == true\n");
209     }
210
211     IOleControl_OnAmbientPropertyChange(CONTROL(This), DISPID_AMBIENT_USERAGENT);
212     IOleControl_OnAmbientPropertyChange(CONTROL(This), DISPID_AMBIENT_PALETTE);
213
214     return S_OK;
215 }
216
217 static HRESULT WINAPI OleObject_GetClientSite(IOleObject *iface, IOleClientSite **ppClientSite)
218 {
219     HTMLDocument *This = OLEOBJ_THIS(iface);
220
221     TRACE("(%p)->(%p)\n", This, ppClientSite);
222
223     if(!ppClientSite)
224         return E_INVALIDARG;
225
226     if(This->doc_obj->client)
227         IOleClientSite_AddRef(This->doc_obj->client);
228     *ppClientSite = This->doc_obj->client;
229
230     return S_OK;
231 }
232
233 static HRESULT WINAPI OleObject_SetHostNames(IOleObject *iface, LPCOLESTR szContainerApp, LPCOLESTR szContainerObj)
234 {
235     HTMLDocument *This = OLEOBJ_THIS(iface);
236     FIXME("(%p)->(%s %s)\n", This, debugstr_w(szContainerApp), debugstr_w(szContainerObj));
237     return E_NOTIMPL;
238 }
239
240 static HRESULT WINAPI OleObject_Close(IOleObject *iface, DWORD dwSaveOption)
241 {
242     HTMLDocument *This = OLEOBJ_THIS(iface);
243
244     TRACE("(%p)->(%08x)\n", This, dwSaveOption);
245
246     if(dwSaveOption == OLECLOSE_PROMPTSAVE)
247         FIXME("OLECLOSE_PROMPTSAVE not implemented\n");
248
249     if(This->doc_obj->in_place_active)
250         IOleInPlaceObjectWindowless_InPlaceDeactivate(INPLACEWIN(This));
251
252     HTMLDocument_LockContainer(This->doc_obj, FALSE);
253
254     if(This->advise_holder)
255         IOleAdviseHolder_SendOnClose(This->advise_holder);
256     
257     return S_OK;
258 }
259
260 static HRESULT WINAPI OleObject_SetMoniker(IOleObject *iface, DWORD dwWhichMoniker, IMoniker *pmk)
261 {
262     HTMLDocument *This = OLEOBJ_THIS(iface);
263     FIXME("(%p %d %p)->()\n", This, dwWhichMoniker, pmk);
264     return E_NOTIMPL;
265 }
266
267 static HRESULT WINAPI OleObject_GetMoniker(IOleObject *iface, DWORD dwAssign, DWORD dwWhichMoniker, IMoniker **ppmk)
268 {
269     HTMLDocument *This = OLEOBJ_THIS(iface);
270     FIXME("(%p)->(%d %d %p)\n", This, dwAssign, dwWhichMoniker, ppmk);
271     return E_NOTIMPL;
272 }
273
274 static HRESULT WINAPI OleObject_InitFromData(IOleObject *iface, IDataObject *pDataObject, BOOL fCreation,
275                                         DWORD dwReserved)
276 {
277     HTMLDocument *This = OLEOBJ_THIS(iface);
278     FIXME("(%p)->(%p %x %d)\n", This, pDataObject, fCreation, dwReserved);
279     return E_NOTIMPL;
280 }
281
282 static HRESULT WINAPI OleObject_GetClipboardData(IOleObject *iface, DWORD dwReserved, IDataObject **ppDataObject)
283 {
284     HTMLDocument *This = OLEOBJ_THIS(iface);
285     FIXME("(%p)->(%d %p)\n", This, dwReserved, ppDataObject);
286     return E_NOTIMPL;
287 }
288
289 static HRESULT WINAPI OleObject_DoVerb(IOleObject *iface, LONG iVerb, LPMSG lpmsg, IOleClientSite *pActiveSite,
290                                         LONG lindex, HWND hwndParent, LPCRECT lprcPosRect)
291 {
292     HTMLDocument *This = OLEOBJ_THIS(iface);
293     IOleDocumentSite *pDocSite;
294     HRESULT hres;
295
296     TRACE("(%p)->(%d %p %p %d %p %p)\n", This, iVerb, lpmsg, pActiveSite, lindex, hwndParent, lprcPosRect);
297
298     if(iVerb != OLEIVERB_SHOW && iVerb != OLEIVERB_UIACTIVATE && iVerb != OLEIVERB_INPLACEACTIVATE) { 
299         FIXME("iVerb = %d not supported\n", iVerb);
300         return E_NOTIMPL;
301     }
302
303     if(!pActiveSite)
304         pActiveSite = This->doc_obj->client;
305
306     hres = IOleClientSite_QueryInterface(pActiveSite, &IID_IOleDocumentSite, (void**)&pDocSite);
307     if(SUCCEEDED(hres)) {
308         HTMLDocument_LockContainer(This->doc_obj, TRUE);
309
310         /* FIXME: Create new IOleDocumentView. See CreateView for more info. */
311         hres = IOleDocumentSite_ActivateMe(pDocSite, DOCVIEW(This));
312         IOleDocumentSite_Release(pDocSite);
313     }else {
314         hres = IOleDocumentView_UIActivate(DOCVIEW(This), TRUE);
315         if(SUCCEEDED(hres)) {
316             if(lprcPosRect) {
317                 RECT rect; /* We need to pass rect as not const pointer */
318                 rect = *lprcPosRect;
319                 IOleDocumentView_SetRect(DOCVIEW(This), &rect);
320             }
321             IOleDocumentView_Show(DOCVIEW(This), TRUE);
322         }
323     }
324
325     return hres;
326 }
327
328 static HRESULT WINAPI OleObject_EnumVerbs(IOleObject *iface, IEnumOLEVERB **ppEnumOleVerb)
329 {
330     HTMLDocument *This = OLEOBJ_THIS(iface);
331     FIXME("(%p)->(%p)\n", This, ppEnumOleVerb);
332     return E_NOTIMPL;
333 }
334
335 static HRESULT WINAPI OleObject_Update(IOleObject *iface)
336 {
337     HTMLDocument *This = OLEOBJ_THIS(iface);
338     FIXME("(%p)\n", This);
339     return E_NOTIMPL;
340 }
341
342 static HRESULT WINAPI OleObject_IsUpToDate(IOleObject *iface)
343 {
344     HTMLDocument *This = OLEOBJ_THIS(iface);
345     FIXME("(%p)\n", This);
346     return E_NOTIMPL;
347 }
348
349 static HRESULT WINAPI OleObject_GetUserClassID(IOleObject *iface, CLSID *pClsid)
350 {
351     HTMLDocument *This = OLEOBJ_THIS(iface);
352
353     TRACE("(%p)->(%p)\n", This, pClsid);
354
355     if(!pClsid)
356         return E_INVALIDARG;
357
358     *pClsid = CLSID_HTMLDocument;
359     return S_OK;
360 }
361
362 static HRESULT WINAPI OleObject_GetUserType(IOleObject *iface, DWORD dwFormOfType, LPOLESTR *pszUserType)
363 {
364     HTMLDocument *This = OLEOBJ_THIS(iface);
365     FIXME("(%p)->(%d %p)\n", This, dwFormOfType, pszUserType);
366     return E_NOTIMPL;
367 }
368
369 static HRESULT WINAPI OleObject_SetExtent(IOleObject *iface, DWORD dwDrawAspect, SIZEL *psizel)
370 {
371     HTMLDocument *This = OLEOBJ_THIS(iface);
372     FIXME("(%p)->(%d %p)\n", This, dwDrawAspect, psizel);
373     return E_NOTIMPL;
374 }
375
376 static HRESULT WINAPI OleObject_GetExtent(IOleObject *iface, DWORD dwDrawAspect, SIZEL *psizel)
377 {
378     HTMLDocument *This = OLEOBJ_THIS(iface);
379     FIXME("(%p)->(%d %p)\n", This, dwDrawAspect, psizel);
380     return E_NOTIMPL;
381 }
382
383 static HRESULT WINAPI OleObject_Advise(IOleObject *iface, IAdviseSink *pAdvSink, DWORD *pdwConnection)
384 {
385     HTMLDocument *This = OLEOBJ_THIS(iface);
386     TRACE("(%p)->(%p %p)\n", This, pAdvSink, pdwConnection);
387
388     if(!pdwConnection)
389         return E_INVALIDARG;
390
391     if(!pAdvSink) {
392         *pdwConnection = 0;
393         return E_INVALIDARG;
394     }
395
396     if(!This->advise_holder) {
397         CreateOleAdviseHolder(&This->advise_holder);
398         if(!This->advise_holder)
399             return E_OUTOFMEMORY;
400     }
401
402     return IOleAdviseHolder_Advise(This->advise_holder, pAdvSink, pdwConnection);
403 }
404
405 static HRESULT WINAPI OleObject_Unadvise(IOleObject *iface, DWORD dwConnection)
406 {
407     HTMLDocument *This = OLEOBJ_THIS(iface);
408     TRACE("(%p)->(%d)\n", This, dwConnection);
409
410     if(!This->advise_holder)
411         return OLE_E_NOCONNECTION;
412
413     return IOleAdviseHolder_Unadvise(This->advise_holder, dwConnection);
414 }
415
416 static HRESULT WINAPI OleObject_EnumAdvise(IOleObject *iface, IEnumSTATDATA **ppenumAdvise)
417 {
418     HTMLDocument *This = OLEOBJ_THIS(iface);
419
420     if(!This->advise_holder) {
421         *ppenumAdvise = NULL;
422         return S_OK;
423     }
424
425     return IOleAdviseHolder_EnumAdvise(This->advise_holder, ppenumAdvise);
426 }
427
428 static HRESULT WINAPI OleObject_GetMiscStatus(IOleObject *iface, DWORD dwAspect, DWORD *pdwStatus)
429 {
430     HTMLDocument *This = OLEOBJ_THIS(iface);
431     FIXME("(%p)->(%d %p)\n", This, dwAspect, pdwStatus);
432     return E_NOTIMPL;
433 }
434
435 static HRESULT WINAPI OleObject_SetColorScheme(IOleObject *iface, LOGPALETTE *pLogpal)
436 {
437     HTMLDocument *This = OLEOBJ_THIS(iface);
438     FIXME("(%p)->(%p)\n", This, pLogpal);
439     return E_NOTIMPL;
440 }
441
442 #undef OLEPBJ_THIS
443
444 static const IOleObjectVtbl OleObjectVtbl = {
445     OleObject_QueryInterface,
446     OleObject_AddRef,
447     OleObject_Release,
448     OleObject_SetClientSite,
449     OleObject_GetClientSite,
450     OleObject_SetHostNames,
451     OleObject_Close,
452     OleObject_SetMoniker,
453     OleObject_GetMoniker,
454     OleObject_InitFromData,
455     OleObject_GetClipboardData,
456     OleObject_DoVerb,
457     OleObject_EnumVerbs,
458     OleObject_Update,
459     OleObject_IsUpToDate,
460     OleObject_GetUserClassID,
461     OleObject_GetUserType,
462     OleObject_SetExtent,
463     OleObject_GetExtent,
464     OleObject_Advise,
465     OleObject_Unadvise,
466     OleObject_EnumAdvise,
467     OleObject_GetMiscStatus,
468     OleObject_SetColorScheme
469 };
470
471 /**********************************************************
472  * IOleDocument implementation
473  */
474
475 #define OLEDOC_THIS(iface) DEFINE_THIS(HTMLDocument, OleDocument, iface)
476
477 static HRESULT WINAPI OleDocument_QueryInterface(IOleDocument *iface, REFIID riid, void **ppvObject)
478 {
479     HTMLDocument *This = OLEDOC_THIS(iface);
480     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppvObject);
481 }
482
483 static ULONG WINAPI OleDocument_AddRef(IOleDocument *iface)
484 {
485     HTMLDocument *This = OLEDOC_THIS(iface);
486     return IHTMLDocument2_AddRef(HTMLDOC(This));
487 }
488
489 static ULONG WINAPI OleDocument_Release(IOleDocument *iface)
490 {
491     HTMLDocument *This = OLEDOC_THIS(iface);
492     return IHTMLDocument2_Release(HTMLDOC(This));
493 }
494
495 static HRESULT WINAPI OleDocument_CreateView(IOleDocument *iface, IOleInPlaceSite *pIPSite, IStream *pstm,
496                                    DWORD dwReserved, IOleDocumentView **ppView)
497 {
498     HTMLDocument *This = OLEDOC_THIS(iface);
499     HRESULT hres;
500
501     TRACE("(%p)->(%p %p %d %p)\n", This, pIPSite, pstm, dwReserved, ppView);
502
503     if(!ppView)
504         return E_INVALIDARG;
505
506     /* FIXME:
507      * Windows implementation creates new IOleDocumentView when function is called for the
508      * first time and returns E_FAIL when it is called for the second time, but it doesn't matter
509      * if the application uses returned interfaces, passed to ActivateMe or returned by
510      * QueryInterface, so there is no reason to create new interface. This needs more testing.
511      */
512
513     if(pIPSite) {
514         hres = IOleDocumentView_SetInPlaceSite(DOCVIEW(This), pIPSite);
515         if(FAILED(hres))
516             return hres;
517     }
518
519     if(pstm)
520         FIXME("pstm is not supported\n");
521
522     IOleDocumentView_AddRef(DOCVIEW(This));
523     *ppView = DOCVIEW(This);
524     return S_OK;
525 }
526
527 static HRESULT WINAPI OleDocument_GetDocMiscStatus(IOleDocument *iface, DWORD *pdwStatus)
528 {
529     HTMLDocument *This = OLEDOC_THIS(iface);
530     FIXME("(%p)->(%p)\n", This, pdwStatus);
531     return E_NOTIMPL;
532 }
533
534 static HRESULT WINAPI OleDocument_EnumViews(IOleDocument *iface, IEnumOleDocumentViews **ppEnum,
535                                    IOleDocumentView **ppView)
536 {
537     HTMLDocument *This = OLEDOC_THIS(iface);
538     FIXME("(%p)->(%p %p)\n", This, ppEnum, ppView);
539     return E_NOTIMPL;
540 }
541
542 #undef OLEDOC_THIS
543
544 static const IOleDocumentVtbl OleDocumentVtbl = {
545     OleDocument_QueryInterface,
546     OleDocument_AddRef,
547     OleDocument_Release,
548     OleDocument_CreateView,
549     OleDocument_GetDocMiscStatus,
550     OleDocument_EnumViews
551 };
552
553 /**********************************************************
554  * IOleControl implementation
555  */
556
557 #define CONTROL_THIS(iface) DEFINE_THIS(HTMLDocument, OleControl, iface)
558
559 static HRESULT WINAPI OleControl_QueryInterface(IOleControl *iface, REFIID riid, void **ppv)
560 {
561     HTMLDocument *This = CONTROL_THIS(iface);
562     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppv);
563 }
564
565 static ULONG WINAPI OleControl_AddRef(IOleControl *iface)
566 {
567     HTMLDocument *This = CONTROL_THIS(iface);
568     return IHTMLDocument2_AddRef(HTMLDOC(This));
569 }
570
571 static ULONG WINAPI OleControl_Release(IOleControl *iface)
572 {
573     HTMLDocument *This = CONTROL_THIS(iface);
574     return IHTMLDocument_Release(HTMLDOC(This));
575 }
576
577 static HRESULT WINAPI OleControl_GetControlInfo(IOleControl *iface, CONTROLINFO *pCI)
578 {
579     HTMLDocument *This = CONTROL_THIS(iface);
580     FIXME("(%p)->(%p)\n", This, pCI);
581     return E_NOTIMPL;
582 }
583
584 static HRESULT WINAPI OleControl_OnMnemonic(IOleControl *iface, MSG *pMsg)
585 {
586     HTMLDocument *This = CONTROL_THIS(iface);
587     FIXME("(%p)->(%p)\n", This, pMsg);
588     return E_NOTIMPL;
589 }
590
591 HRESULT get_client_disp_property(IOleClientSite *client, DISPID dispid, VARIANT *res)
592 {
593     IDispatch *disp = NULL;
594     DISPPARAMS dispparams = {NULL, 0};
595     UINT err;
596     HRESULT hres;
597
598     hres = IOleClientSite_QueryInterface(client, &IID_IDispatch, (void**)&disp);
599     if(FAILED(hres)) {
600         TRACE("Could not get IDispatch\n");
601         return hres;
602     }
603
604     VariantInit(res);
605
606     hres = IDispatch_Invoke(disp, dispid, &IID_NULL, LOCALE_SYSTEM_DEFAULT,
607             DISPATCH_PROPERTYGET, &dispparams, res, NULL, &err);
608
609     IDispatch_Release(disp);
610
611     return hres;
612 }
613
614 static HRESULT on_change_dlcontrol(HTMLDocument *This)
615 {
616     VARIANT res;
617     HRESULT hres;
618     
619     hres = get_client_disp_property(This->doc_obj->client, DISPID_AMBIENT_DLCONTROL, &res);
620     if(SUCCEEDED(hres))
621         FIXME("unsupported dlcontrol %08x\n", V_I4(&res));
622
623     return S_OK;
624 }
625
626 static HRESULT WINAPI OleControl_OnAmbientPropertyChange(IOleControl *iface, DISPID dispID)
627 {
628     HTMLDocument *This = CONTROL_THIS(iface);
629     IOleClientSite *client;
630     VARIANT res;
631     HRESULT hres;
632
633     client = This->doc_obj->client;
634     if(!client) {
635         TRACE("client = NULL\n");
636         return S_OK;
637     }
638
639     switch(dispID) {
640     case DISPID_AMBIENT_USERMODE:
641         TRACE("(%p)->(DISPID_AMBIENT_USERMODE)\n", This);
642         hres = get_client_disp_property(client, DISPID_AMBIENT_USERMODE, &res);
643         if(FAILED(hres))
644             return S_OK;
645
646         if(V_VT(&res) == VT_BOOL) {
647             if(V_BOOL(&res)) {
648                 This->doc_obj->usermode = BROWSEMODE;
649             }else {
650                 FIXME("edit mode is not supported\n");
651                 This->doc_obj->usermode = EDITMODE;
652             }
653         }else {
654             FIXME("V_VT(res)=%d\n", V_VT(&res));
655         }
656         return S_OK;
657     case DISPID_AMBIENT_DLCONTROL:
658         TRACE("(%p)->(DISPID_AMBIENT_DLCONTROL)\n", This);
659         return on_change_dlcontrol(This);
660     case DISPID_AMBIENT_OFFLINEIFNOTCONNECTED:
661         TRACE("(%p)->(DISPID_AMBIENT_OFFLINEIFNOTCONNECTED)\n", This);
662         on_change_dlcontrol(This);
663         hres = get_client_disp_property(client, DISPID_AMBIENT_OFFLINEIFNOTCONNECTED, &res);
664         if(FAILED(hres))
665             return S_OK;
666
667         if(V_VT(&res) == VT_BOOL) {
668             if(V_BOOL(&res)) {
669                 FIXME("offline connection is not supported\n");
670                 hres = E_FAIL;
671             }
672         }else {
673             FIXME("V_VT(res)=%d\n", V_VT(&res));
674         }
675         return S_OK;
676     case DISPID_AMBIENT_SILENT:
677         TRACE("(%p)->(DISPID_AMBIENT_SILENT)\n", This);
678         on_change_dlcontrol(This);
679         hres = get_client_disp_property(client, DISPID_AMBIENT_SILENT, &res);
680         if(FAILED(hres))
681             return S_OK;
682
683         if(V_VT(&res) == VT_BOOL) {
684             if(V_BOOL(&res)) {
685                 FIXME("silent mode is not supported\n");
686                 hres = E_FAIL;
687             }
688         }else {
689             FIXME("V_VT(res)=%d\n", V_VT(&res));
690         }
691         return S_OK;
692     case DISPID_AMBIENT_USERAGENT:
693         TRACE("(%p)->(DISPID_AMBIENT_USERAGENT)\n", This);
694         hres = get_client_disp_property(client, DISPID_AMBIENT_USERAGENT, &res);
695         if(FAILED(hres))
696             return S_OK;
697
698         FIXME("not supported AMBIENT_USERAGENT\n");
699         hres = E_FAIL;
700         return S_OK;
701     case DISPID_AMBIENT_PALETTE:
702         TRACE("(%p)->(DISPID_AMBIENT_PALETTE)\n", This);
703         hres = get_client_disp_property(client, DISPID_AMBIENT_PALETTE, &res);
704         if(FAILED(hres))
705             return S_OK;
706
707         FIXME("not supported AMBIENT_PALETTE\n");
708         hres = E_FAIL;
709         return S_OK;
710     }
711
712     FIXME("(%p) unsupported dispID=%d\n", This, dispID);
713     return E_FAIL;
714 }
715
716 static HRESULT WINAPI OleControl_FreezeEvents(IOleControl *iface, BOOL bFreeze)
717 {
718     HTMLDocument *This = CONTROL_THIS(iface);
719     FIXME("(%p)->(%x)\n", This, bFreeze);
720     return E_NOTIMPL;
721 }
722
723 #undef CONTROL_THIS
724
725 static const IOleControlVtbl OleControlVtbl = {
726     OleControl_QueryInterface,
727     OleControl_AddRef,
728     OleControl_Release,
729     OleControl_GetControlInfo,
730     OleControl_OnMnemonic,
731     OleControl_OnAmbientPropertyChange,
732     OleControl_FreezeEvents
733 };
734
735 /**********************************************************
736  * IObjectWithSite implementation
737  */
738
739 #define OBJSITE_THIS(iface) DEFINE_THIS(HTMLDocument, ObjectWithSite, iface)
740
741 static HRESULT WINAPI ObjectWithSite_QueryInterface(IObjectWithSite *iface, REFIID riid, void **ppvObject)
742 {
743     HTMLDocument *This = OBJSITE_THIS(iface);
744     return IHTMLDocument2_QueryInterface(HTMLDOC(This), riid, ppvObject);
745 }
746
747 static ULONG WINAPI ObjectWithSite_AddRef(IObjectWithSite *iface)
748 {
749     HTMLDocument *This = OBJSITE_THIS(iface);
750     return IHTMLDocument2_AddRef(HTMLDOC(This));
751 }
752
753 static ULONG WINAPI ObjectWithSite_Release(IObjectWithSite *iface)
754 {
755     HTMLDocument *This = OBJSITE_THIS(iface);
756     return IHTMLDocument2_Release(HTMLDOC(This));
757 }
758
759 static HRESULT WINAPI ObjectWithSite_SetSite(IObjectWithSite *iface, IUnknown *pUnkSite)
760 {
761     HTMLDocument *This = OBJSITE_THIS(iface);
762     FIXME("(%p)->(%p)\n", This, pUnkSite);
763     return E_NOTIMPL;
764 }
765
766 static HRESULT WINAPI ObjectWithSite_GetSite(IObjectWithSite* iface, REFIID riid, PVOID *ppvSite)
767 {
768     HTMLDocument *This = OBJSITE_THIS(iface);
769     FIXME("(%p)->(%p)\n", This, ppvSite);
770     return E_NOTIMPL;
771 }
772
773 #undef OBJSITE_THIS
774
775 static const IObjectWithSiteVtbl ObjectWithSiteVtbl = {
776     ObjectWithSite_QueryInterface,
777     ObjectWithSite_AddRef,
778     ObjectWithSite_Release,
779     ObjectWithSite_SetSite,
780     ObjectWithSite_GetSite
781 };
782
783 void HTMLDocument_LockContainer(HTMLDocumentObj *This, BOOL fLock)
784 {
785     IOleContainer *container;
786     HRESULT hres;
787
788     if(!This->client || This->container_locked == fLock)
789         return;
790
791     hres = IOleClientSite_GetContainer(This->client, &container);
792     if(SUCCEEDED(hres)) {
793         IOleContainer_LockContainer(container, fLock);
794         This->container_locked = fLock;
795         IOleContainer_Release(container);
796     }
797 }
798
799 void HTMLDocument_OleObj_Init(HTMLDocument *This)
800 {
801     This->lpOleObjectVtbl = &OleObjectVtbl;
802     This->lpOleDocumentVtbl = &OleDocumentVtbl;
803     This->lpOleControlVtbl = &OleControlVtbl;
804     This->lpObjectWithSiteVtbl = &ObjectWithSiteVtbl;
805 }