dlls/mshtml/nsembed.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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17  */
  18
  19 #include "config.h"
  20
  21 #include <stdarg.h>
  22
  23 #define COBJMACROS
  24
  25 #include "windef.h"
  26 #include "winbase.h"
  27 #include "winuser.h"
  28 #include "winreg.h"
  29 #include "ole2.h"
  30
  31 #include "wine/debug.h"
  32 #include "wine/unicode.h"
  33
  34 #include "mshtml_private.h"
  35
  36 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
  37
  38 #define NS_APPSTARTUPNOTIFIER_CONTRACTID "@mozilla.org/embedcomp/appstartup-notifier;1"
  39 #define NS_WEBBROWSER_CONTRACTID "@mozilla.org/embedding/browser/nsWebBrowser;1"
  40 #define NS_IOSERVICE_CONTRACTID "@mozilla.org/network/io-service;1"
  41
  42 #define APPSTARTUP_TOPIC "app-startup"
  43
  44 #define PR_UINT32_MAX 0xffffffff
  45
  46 typedef struct nsACString {
  47     void *d1;
  48     PRUint32 d2;
  49     void *d3;
  50 } nsString;
  51
  52 static nsresult (*NS_InitXPCOM2)(nsIServiceManager**,void*,void*);
  53 static nsresult (*NS_ShutdownXPCOM)(nsIServiceManager*);
  54 static nsresult (*NS_StringContainerInit)(nsString*);
  55 static nsresult (*NS_CStringContainerInit)(nsACString*);
  56 static nsresult (*NS_StringContainerFinish)(nsString*);
  57 static nsresult (*NS_CStringContainerFinish)(nsACString*);
  58 static nsresult (*NS_StringSetData)(nsString*,const PRUnichar*,PRUint32);
  59 static nsresult (*NS_CStringSetData)(nsACString*,const char*,PRUint32);
  60 static nsresult (*NS_NewLocalFile)(const nsString*,PRBool,nsIFile**);
  61 static PRUint32 (*NS_CStringGetData)(nsACString*,const char**,PRBool*);
  62
  63 static HINSTANCE hXPCOM = NULL;
  64
  65 static nsIServiceManager *pServMgr = NULL;
  66 static nsIComponentManager *pCompMgr = NULL;
  67 static nsIIOService *pIOService = NULL;
  68
  69 static const WCHAR wszNsContainer[] = {'N','s','C','o','n','t','a','i','n','e','r',0};
  70
  71 static ATOM nscontainer_class;
  72
  73 static LRESULT WINAPI nsembed_proc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
  74 {
  75     HTMLDocument *This;
  76     nsresult nsres;
  77
  78     static const WCHAR wszTHIS[] = {'T','H','I','S',0};
  79
  80     if(msg == WM_CREATE) {
  81         This = *(HTMLDocument**)lParam;
  82         SetPropW(hwnd, wszTHIS, This);
  83     }else {
  84         This = (HTMLDocument*)GetPropW(hwnd, wszTHIS);
  85     }
  86
  87     switch(msg) {
  88         case WM_SIZE:
  89             TRACE("(%p)->(WM_SIZE)\n", This);
  90
  91             nsres = nsIBaseWindow_SetSize(This->nscontainer->window,
  92                     LOWORD(lParam), HIWORD(lParam), TRUE);
  93             if(NS_FAILED(nsres))
  94                 WARN("SetSize failed: %08lx\n", nsres);
  95     }
  96
  97     return DefWindowProcW(hwnd, msg, wParam, lParam);
  98 }
  99
 100
 101 static void register_nscontainer_class(void)
 102 {
 103     static WNDCLASSEXW wndclass = {
 104         sizeof(WNDCLASSEXW),
 105         CS_DBLCLKS,
 106         nsembed_proc,
 107         0, 0, NULL, NULL, NULL, NULL, NULL,
 108         wszNsContainer,
 109         NULL,
 110     };
 111     wndclass.hInstance = hInst;
 112     nscontainer_class = RegisterClassExW(&wndclass);
 113 }
 114
 115 static BOOL get_mozilla_path(PRUnichar *gre_path)
 116 {
 117     DWORD res, type, i, size = MAX_PATH;
 118     HKEY mozilla_key, hkey;
 119     WCHAR key_name[100];
 120     BOOL ret = FALSE;
 121
 122     static const WCHAR wszGreKey[] =
 123         {'S','o','f','t','w','a','r','e','\\',
 124             'm','o','z','i','l','l','a','.','o','r','g','\\',
 125                 'G','R','E',0};
 126
 127     static const WCHAR wszGreHome[] = {'G','r','e','H','o','m','e',0};
 128
 129     res = RegOpenKeyW(HKEY_LOCAL_MACHINE, wszGreKey, &mozilla_key);
 130     if(res != ERROR_SUCCESS) {
 131         TRACE("Could not open key %s\n", debugstr_w(wszGreKey));
 132         return FALSE;
 133     }
 134
 135     for(i=0; !ret && RegEnumKeyW(mozilla_key, i, key_name, sizeof(key_name)/sizeof(WCHAR)) == ERROR_SUCCESS; i++) {
 136         RegOpenKeyW(mozilla_key, key_name, &hkey);
 137         res = RegQueryValueExW(hkey, wszGreHome, NULL, &type, (LPBYTE)gre_path, &size);
 138         if(res == ERROR_SUCCESS)
 139             ret = TRUE;
 140         RegCloseKey(hkey);
 141     }
 142
 143     RegCloseKey(mozilla_key);
 144     return ret;
 145 }
 146
 147 static BOOL get_mozctl_path(PRUnichar *gre_path)
 148 {
 149     HKEY hkey;
 150     DWORD res, type, size = MAX_PATH;
 151
 152     static const WCHAR wszMozCtlKey[] =
 153         {'S','o','f','t','w','a','r','e','\\','M','o','z','i','l','l','a',0};
 154     static const WCHAR wszBinDirectoryPath[] =
 155         {'B','i','n','D','i','r','e','c','t','o','r','y','P','a','t','h',0};
 156
 157     res = RegOpenKeyW(HKEY_LOCAL_MACHINE, wszMozCtlKey, &hkey);
 158     if(res != ERROR_SUCCESS) {
 159         TRACE("Could not open key %s\n", debugstr_w(wszMozCtlKey));
 160         return FALSE;
 161     }
 162
 163     res = RegQueryValueExW(hkey, wszBinDirectoryPath, NULL, &type, (LPBYTE)gre_path, &size);
 164     if(res != ERROR_SUCCESS) {
 165         ERR("Could not get value %s\n", debugstr_w(wszBinDirectoryPath));
 166         return FALSE;
 167     }
 168
 169     return TRUE;
 170 }
 171
 172 static BOOL load_gecko()
 173 {
 174     nsresult nsres;
 175     nsIObserver *pStartNotif;
 176     nsString path;
 177     nsIFile *gre_dir;
 178     PRUnichar gre_path[MAX_PATH];
 179     WCHAR path_env[MAX_PATH];
 180     int len;
 181
 182     static BOOL tried_load = FALSE;
 183     static const WCHAR wszPATH[] = {'P','A','T','H',0};
 184     static const WCHAR strXPCOM[] = {'x','p','c','o','m','.','d','l','l',0};
 185
 186     TRACE("()\n");
 187
 188     if(tried_load)
 189         return pCompMgr != NULL;
 190     tried_load = TRUE;
 191
 192     if(!get_mozctl_path(gre_path) && !get_mozilla_path(gre_path)) {
 193         MESSAGE("Could not load Mozilla. HTML rendering will be disabled.\n");
 194         return FALSE;
 195     }
 196
 197     TRACE("found path %s\n", debugstr_w(gre_path));
 198
 199     /* We have to modify PATH as XPCOM loads other DLLs from this directory. */
 200     GetEnvironmentVariableW(wszPATH, path_env, sizeof(path_env)/sizeof(WCHAR));
 201     len = strlenW(path_env);
 202     path_env[len++] = ';';
 203     strcpyW(path_env+len, gre_path);
 204     SetEnvironmentVariableW(wszPATH, path_env);
 205
 206     hXPCOM = LoadLibraryW(strXPCOM);
 207     if(!hXPCOM) {
 208         ERR("Could not load XPCOM: %ld\n", GetLastError());
 209         return FALSE;
 210     }
 211
 212 #define NS_DLSYM(func) \
 213     func = (typeof(func))GetProcAddress(hXPCOM, #func); \
 214     if(!func) \
 215         ERR("Could not GetProcAddress(" #func ") failed\n")
 216
 217     NS_DLSYM(NS_InitXPCOM2);
 218     NS_DLSYM(NS_ShutdownXPCOM);
 219     NS_DLSYM(NS_StringContainerInit);
 220     NS_DLSYM(NS_CStringContainerInit);
 221     NS_DLSYM(NS_StringContainerFinish);
 222     NS_DLSYM(NS_CStringContainerFinish);
 223     NS_DLSYM(NS_StringSetData);
 224     NS_DLSYM(NS_CStringSetData);
 225     NS_DLSYM(NS_NewLocalFile);
 226     NS_DLSYM(NS_CStringGetData);
 227
 228 #undef NS_DLSYM
 229
 230     NS_StringContainerInit(&path);
 231     NS_StringSetData(&path, gre_path, PR_UINT32_MAX);
 232     nsres = NS_NewLocalFile(&path, FALSE, &gre_dir);
 233     NS_StringContainerFinish(&path);
 234     if(NS_FAILED(nsres)) {
 235         ERR("NS_NewLocalFile failed: %08lx\n", nsres);
 236         FreeLibrary(hXPCOM);
 237         return FALSE;
 238     }
 239
 240     nsres = NS_InitXPCOM2(&pServMgr, gre_dir, NULL);
 241     if(NS_FAILED(nsres)) {
 242         ERR("NS_InitXPCOM2 failed: %08lx\n", nsres);
 243         FreeLibrary(hXPCOM);
 244         return FALSE;
 245     }
 246
 247     nsres = nsIServiceManager_QueryInterface(pServMgr, &IID_nsIComponentManager, (void**)&pCompMgr);
 248     if(NS_FAILED(nsres))
 249         ERR("Could not get nsIComponentManager: %08lx\n", nsres);
 250
 251     nsres = nsIComponentManager_CreateInstanceByContractID(pCompMgr, NS_APPSTARTUPNOTIFIER_CONTRACTID,
 252             NULL, &IID_nsIObserver, (void**)&pStartNotif);
 253     if(NS_SUCCEEDED(nsres)) {
 254         nsres = nsIObserver_Observe(pStartNotif, NULL, APPSTARTUP_TOPIC, NULL);
 255         if(NS_FAILED(nsres))
 256             ERR("Observe failed: %08lx\n", nsres);
 257
 258         nsIObserver_Release(pStartNotif);
 259     }else {
 260         ERR("could not get appstartup-notifier: %08lx\n", nsres);
 261     }
 262
 263     return TRUE;
 264 }
 265
 266 nsACString *nsACString_Create(void)
 267 {
 268     nsACString *ret;
 269     ret = HeapAlloc(GetProcessHeap(), 0, sizeof(nsACString));
 270     NS_CStringContainerInit(ret);
 271     return ret;
 272 }
 273
 274 void nsACString_SetData(nsACString *str, const char *data)
 275 {
 276     NS_CStringSetData(str, data, PR_UINT32_MAX);
 277 }
 278
 279 static PRUint32 nsACString_GetData(nsACString *str, const char **data, PRBool *termited)
 280 {
 281     return NS_CStringGetData(str, data, termited);
 282 }
 283
 284 void nsACString_Destroy(nsACString *str)
 285 {
 286     NS_CStringContainerFinish(str);
 287     HeapFree(GetProcessHeap(), 0, str);
 288 }
 289
 290 void close_gecko()
 291 {
 292     TRACE("()\n");
 293
 294     if(pCompMgr)
 295         nsIComponentManager_Release(pCompMgr);
 296
 297     if(pServMgr)
 298         nsIServiceManager_Release(pServMgr);
 299
 300     if(hXPCOM)
 301         FreeLibrary(hXPCOM);
 302 }
 303
 304 nsIURI *get_nsIURI(LPCWSTR url)
 305 {
 306     nsIURI *ret;
 307     nsACString *acstr;
 308     nsresult nsres;
 309     char *urla;
 310     int len;
 311
 312     if(!pIOService) {
 313         nsres = nsIServiceManager_GetServiceByContactID(pServMgr, NS_IOSERVICE_CONTRACTID,
 314                 &IID_nsIIOService, (void**)&pIOService);
 315         if(NS_FAILED(nsres))
 316             ERR("Failed to create nsIOService: %08lx\n", nsres);
 317     }
 318
 319     len = WideCharToMultiByte(CP_ACP, 0, url, -1, NULL, -1, NULL, NULL);
 320     urla = HeapAlloc(GetProcessHeap(), 0, len);
 321     WideCharToMultiByte(CP_ACP, 0, url, -1, urla, -1, NULL, NULL);
 322
 323     acstr = nsACString_Create();
 324     nsACString_SetData(acstr, urla);
 325
 326     nsres = nsIIOService_NewURI(pIOService, acstr, NULL, NULL, &ret);
 327     if(NS_FAILED(nsres))
 328         FIXME("NewURI failed: %08lx\n", nsres);
 329
 330     nsACString_Destroy(acstr);
 331     HeapFree(GetProcessHeap(), 0, urla);
 332
 333     return ret;
 334 }
 335
 336 /**********************************************************
 337  *      nsIWebBrowserChrome interface
 338  */
 339
 340 #define NSWBCHROME_THIS(iface) DEFINE_THIS(NSContainer, WebBrowserChrome, iface)
 341
 342 static nsresult NSAPI nsWebBrowserChrome_QueryInterface(nsIWebBrowserChrome *iface,
 343         nsIIDRef riid, nsQIResult result)
 344 {
 345     NSContainer *This = NSWBCHROME_THIS(iface);
 346
 347     *result = NULL;
 348     if(IsEqualGUID(&IID_nsISupports, riid)) {
 349         TRACE("(%p)->(IID_nsISupports, %p)\n", This, result);
 350         *result = NSWBCHROME(This);
 351     }else if(IsEqualGUID(&IID_nsIWebBrowserChrome, riid)) {
 352         TRACE("(%p)->(IID_nsIWebBrowserChrome, %p)\n", This, result);
 353         *result = NSWBCHROME(This);
 354     }else if(IsEqualGUID(&IID_nsIContextMenuListener, riid)) {
 355         TRACE("(%p)->(IID_nsIContextMenuListener, %p)\n", This, result);
 356         *result = NSCML(This);
 357     }
 358
 359     if(*result)
 360         return NS_OK;
 361
 362     TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), result);
 363     return NS_NOINTERFACE;
 364 }
 365
 366 static nsrefcnt NSAPI nsWebBrowserChrome_AddRef(nsIWebBrowserChrome *iface)
 367 {
 368     NSContainer *This = NSWBCHROME_THIS(iface);
 369     TRACE("(%p)\n", This);
 370     return 2;  /* Should we implement ref conunting here? */
 371 }
 372
 373 static nsrefcnt NSAPI nsWebBrowserChrome_Release(nsIWebBrowserChrome *iface)
 374 {
 375     NSContainer *This = NSWBCHROME_THIS(iface);
 376     TRACE("(%p)\n", This);
 377     return 1;
 378 }
 379
 380 static nsresult NSAPI nsWebBrowserChrome_SetStatus(nsIWebBrowserChrome *iface,
 381         PRUint32 statusType, const PRUnichar *status)
 382 {
 383     NSContainer *This = NSWBCHROME_THIS(iface);
 384     TRACE("(%p)->(%ld %s)\n", This, statusType, debugstr_w(status));
 385     return NS_ERROR_NOT_IMPLEMENTED;
 386 }
 387
 388 static nsresult NSAPI nsWebBrowserChrome_GetWebBrowser(nsIWebBrowserChrome *iface,
 389         nsIWebBrowser **aWebBrowser)
 390 {
 391     NSContainer *This = NSWBCHROME_THIS(iface);
 392
 393     TRACE("(%p)->(%p)\n", This, aWebBrowser);
 394
 395     if(!aWebBrowser)
 396         return NS_ERROR_INVALID_ARG;
 397
 398     *aWebBrowser = This->webbrowser;
 399     return S_OK;
 400 }
 401
 402 static nsresult NSAPI nsWebBrowserChrome_SetWebBrowser(nsIWebBrowserChrome *iface,
 403         nsIWebBrowser *aWebBrowser)
 404 {
 405     NSContainer *This = NSWBCHROME_THIS(iface);
 406
 407     TRACE("(%p)->(%p)\n", This, aWebBrowser);
 408
 409     if(aWebBrowser != This->webbrowser)
 410         ERR("Wrong nsWebBrowser!\n");
 411
 412     return NS_OK;
 413 }
 414
 415 static nsresult NSAPI nsWebBrowserChrome_GetChromeFlags(nsIWebBrowserChrome *iface,
 416         PRUint32 *aChromeFlags)
 417 {
 418     NSContainer *This = NSWBCHROME_THIS(iface);
 419     WARN("(%p)->(%p)\n", This, aChromeFlags);
 420     return NS_ERROR_NOT_IMPLEMENTED;
 421 }
 422
 423 static nsresult NSAPI nsWebBrowserChrome_SetChromeFlags(nsIWebBrowserChrome *iface,
 424         PRUint32 aChromeFlags)
 425 {
 426     NSContainer *This = NSWBCHROME_THIS(iface);
 427     WARN("(%p)->(%08lx)\n", This, aChromeFlags);
 428     return NS_ERROR_NOT_IMPLEMENTED;
 429 }
 430
 431 static nsresult NSAPI nsWebBrowserChrome_DestroyBrowserWindow(nsIWebBrowserChrome *iface)
 432 {
 433     NSContainer *This = NSWBCHROME_THIS(iface);
 434     TRACE("(%p)\n", This);
 435     return NS_ERROR_NOT_IMPLEMENTED;
 436 }
 437
 438 static nsresult NSAPI nsWebBrowserChrome_SizeBrowserTo(nsIWebBrowserChrome *iface,
 439         PRInt32 aCX, PRInt32 aCY)
 440 {
 441     NSContainer *This = NSWBCHROME_THIS(iface);
 442     WARN("(%p)->(%ld %ld)\n", This, aCX, aCY);
 443     return NS_ERROR_NOT_IMPLEMENTED;
 444 }
 445
 446 static nsresult NSAPI nsWebBrowserChrome_ShowAsModal(nsIWebBrowserChrome *iface)
 447 {
 448     NSContainer *This = NSWBCHROME_THIS(iface);
 449     WARN("(%p)\n", This);
 450     return NS_ERROR_NOT_IMPLEMENTED;
 451 }
 452
 453 static nsresult NSAPI nsWebBrowserChrome_IsWindowModal(nsIWebBrowserChrome *iface, PRBool *_retval)
 454 {
 455     NSContainer *This = NSWBCHROME_THIS(iface);
 456     WARN("(%p)->(%p)\n", This, _retval);
 457     return NS_ERROR_NOT_IMPLEMENTED;
 458 }
 459
 460 static nsresult NSAPI nsWebBrowserChrome_ExitModalEventLoop(nsIWebBrowserChrome *iface,
 461         nsresult aStatus)
 462 {
 463     NSContainer *This = NSWBCHROME_THIS(iface);
 464     WARN("(%p)->(%08lx)\n", This, aStatus);
 465     return NS_ERROR_NOT_IMPLEMENTED;
 466 }
 467
 468 #undef NSWBCHROME_THIS
 469
 470 static const nsIWebBrowserChromeVtbl nsWebBrowserChromeVtbl = {
 471     nsWebBrowserChrome_QueryInterface,
 472     nsWebBrowserChrome_AddRef,
 473     nsWebBrowserChrome_Release,
 474     nsWebBrowserChrome_SetStatus,
 475     nsWebBrowserChrome_GetWebBrowser,
 476     nsWebBrowserChrome_SetWebBrowser,
 477     nsWebBrowserChrome_GetChromeFlags,
 478     nsWebBrowserChrome_SetChromeFlags,
 479     nsWebBrowserChrome_DestroyBrowserWindow,
 480     nsWebBrowserChrome_SizeBrowserTo,
 481     nsWebBrowserChrome_ShowAsModal,
 482     nsWebBrowserChrome_IsWindowModal,
 483     nsWebBrowserChrome_ExitModalEventLoop
 484 };
 485
 486 /**********************************************************
 487  *      nsIContextMenuListener interface
 488  */
 489
 490 #define NSCML_THIS(iface) DEFINE_THIS(NSContainer, ContextMenuListener, iface)
 491
 492 static nsresult NSAPI nsContextMenuListener_QueryInterface(nsIContextMenuListener *iface,
 493         nsIIDRef riid, nsQIResult result)
 494 {
 495     NSContainer *This = NSCML_THIS(iface);
 496     return nsIWebBrowserChrome_QueryInterface(NSWBCHROME(This), riid, result);
 497 }
 498
 499 static nsrefcnt NSAPI nsContextMenuListener_AddRef(nsIContextMenuListener *iface)
 500 {
 501     NSContainer *This = NSCML_THIS(iface);
 502     return nsIWebBrowserChrome_AddRef(NSWBCHROME(This));
 503 }
 504
 505 static nsrefcnt NSAPI nsContextMenuListener_Release(nsIContextMenuListener *iface)
 506 {
 507     NSContainer *This = NSCML_THIS(iface);
 508     return nsIWebBrowserChrome_Release(NSWBCHROME(This));
 509 }
 510
 511 static nsresult NSAPI nsContextMenuListener_OnShowContextMenu(nsIContextMenuListener *iface,
 512         PRUint32 aContextFlags, nsIDOMEvent *aEvent, nsIDOMNode *aNode)
 513 {
 514     NSContainer *This = NSCML_THIS(iface);
 515     nsIDOMMouseEvent *event;
 516     POINT pt;
 517     DWORD dwID = CONTEXT_MENU_DEFAULT;
 518     nsresult nsres;
 519
 520     TRACE("(%p)->(%08lx %p %p)\n", This, aContextFlags, aEvent, aNode);
 521
 522     nsres = nsIDOMEvent_QueryInterface(aEvent, &IID_nsIDOMMouseEvent, (void**)&event);
 523     if(NS_FAILED(nsres)) {
 524         ERR("Could not get nsIDOMMouseEvent interface: %08lx\n", nsres);
 525         return nsres;
 526     }
 527
 528     nsIDOMMouseEvent_GetScreenX(event, &pt.x);
 529     nsIDOMMouseEvent_GetScreenY(event, &pt.y);
 530     nsIDOMMouseEvent_Release(event);
 531
 532     switch(aContextFlags) {
 533     case CONTEXT_NONE:
 534     case CONTEXT_DOCUMENT:
 535     case CONTEXT_TEXT:
 536         dwID = CONTEXT_MENU_DEFAULT;
 537         break;
 538     case CONTEXT_IMAGE:
 539     case CONTEXT_IMAGE|CONTEXT_LINK:
 540         dwID = CONTEXT_MENU_IMAGE;
 541         break;
 542     case CONTEXT_LINK:
 543         dwID = CONTEXT_MENU_ANCHOR;
 544         break;
 545     case CONTEXT_INPUT:
 546         dwID = CONTEXT_MENU_CONTROL;
 547         break;
 548     default:
 549         FIXME("aContextFlags=%08lx\n", aContextFlags);
 550     };
 551
 552     HTMLDocument_ShowContextMenu(This->doc, dwID, &pt);
 553
 554     return NS_OK;
 555 }
 556
 557 #undef NSCML_THIS
 558
 559 static const nsIContextMenuListenerVtbl nsContextMenuListenerVtbl = {
 560     nsContextMenuListener_QueryInterface,
 561     nsContextMenuListener_AddRef,
 562     nsContextMenuListener_Release,
 563     nsContextMenuListener_OnShowContextMenu
 564 };
 565
 566 /**********************************************************
 567  *      nsIURIContentListener interface
 568  */
 569
 570 #define NSURICL_THIS(iface) DEFINE_THIS(NSContainer, URIContentListener, iface)
 571
 572 static nsresult NSAPI nsURIContentListener_QueryInterface(nsIURIContentListener *iface,
 573         nsIIDRef riid, nsQIResult result)
 574 {
 575     NSContainer *This = NSURICL_THIS(iface);
 576     return nsIWebBrowserChrome_QueryInterface(NSWBCHROME(This), riid, result);
 577 }
 578
 579 static nsrefcnt NSAPI nsURIContentListener_AddRef(nsIURIContentListener *iface)
 580 {
 581     NSContainer *This = NSURICL_THIS(iface);
 582     return nsIWebBrowserChrome_AddRef(NSWBCHROME(This));
 583 }
 584
 585 static nsrefcnt NSAPI nsURIContentListener_Release(nsIURIContentListener *iface)
 586 {
 587     NSContainer *This = NSURICL_THIS(iface);
 588     return nsIWebBrowserChrome_Release(NSWBCHROME(This));
 589 }
 590
 591 static nsresult NSAPI nsURIContentListener_OnStartURIOpen(nsIURIContentListener *iface, nsIURI *aURI,
 592         PRBool *_retval)
 593 {
 594     NSContainer *This = NSURICL_THIS(iface);
 595     BOOL do_load = TRUE;
 596     nsresult nsres;
 597
 598     TRACE("(%p)->(%p %p)\n", This, aURI, _retval);
 599
 600     nsACString *spec_str = nsACString_Create();
 601
 602     nsres = nsIURI_GetSpec(aURI, spec_str);
 603     if(NS_SUCCEEDED(nsres)) {
 604         const char *spec = NULL;
 605         LPWSTR specw;
 606         int len;
 607
 608         nsACString_GetData(spec_str, &spec, NULL);
 609
 610         len = MultiByteToWideChar(CP_ACP, 0, spec, -1, NULL, 0);
 611         specw = HeapAlloc(GetProcessHeap(), 0, len*sizeof(WCHAR));
 612         MultiByteToWideChar(CP_ACP, 0, spec, -1, specw, -1);
 613
 614         if(strcmpW(This->url, specw)) /* hack */
 615             do_load = HTMLDocument_OnLoad(This->doc, specw);
 616
 617         HeapFree(GetProcessHeap(), 0, specw);
 618     }else {
 619         ERR("GetSpec failed: %08lx\n", nsres);
 620     }
 621
 622     nsACString_Destroy(spec_str);
 623
 624     if(!do_load) {
 625         *_retval = TRUE;
 626         return NS_OK;
 627     }
 628
 629     return NS_ERROR_NOT_IMPLEMENTED;
 630 }
 631
 632 static nsresult NSAPI nsURIContentListener_DoContent(nsIURIContentListener *iface,
 633         const char *aContentType, PRBool aIsContentPreferred, nsIRequest *aRequest,
 634         nsIStreamListener **aContentHandler, PRBool *_retval)
 635 {
 636     NSContainer *This = NSURICL_THIS(iface);
 637     TRACE("(%p)->(%s %x %p %p %p)\n", This, debugstr_a(aContentType), aIsContentPreferred,
 638             aRequest, aContentHandler, _retval);
 639     return NS_ERROR_NOT_IMPLEMENTED;
 640 }
 641
 642 static nsresult NSAPI nsURIContentListener_IsPreferred(nsIURIContentListener *iface,
 643         const char *aContentType, char **aDesiredContentType, PRBool *_retval)
 644 {
 645     NSContainer *This = NSURICL_THIS(iface);
 646
 647     TRACE("(%p)->(%s %p %p)\n", This, debugstr_a(aContentType), aDesiredContentType, _retval);
 648
 649     /* FIXME: Should we do something here? */
 650     *_retval = TRUE;
 651     return NS_OK;
 652 }
 653
 654 static nsresult NSAPI nsURIContentListener_CanHandleContent(nsIURIContentListener *iface,
 655         const char *aContentType, PRBool aIsContentPreferred, char **aDesiredContentType,
 656         PRBool *_retval)
 657 {
 658     NSContainer *This = NSURICL_THIS(iface);
 659     TRACE("(%p)->(%s %x %p %p)\n", This, debugstr_a(aContentType), aIsContentPreferred,
 660             aDesiredContentType, _retval);
 661     return NS_ERROR_NOT_IMPLEMENTED;
 662 }
 663
 664 static nsresult NSAPI nsURIContentListener_GetLoadCookie(nsIURIContentListener *iface,
 665         nsISupports **aLoadCookie)
 666 {
 667     NSContainer *This = NSURICL_THIS(iface);
 668     WARN("(%p)->(%p)\n", This, aLoadCookie);
 669     return NS_ERROR_NOT_IMPLEMENTED;
 670 }
 671
 672 static nsresult NSAPI nsURIContentListener_SetLoadCookie(nsIURIContentListener *iface,
 673         nsISupports *aLoadCookie)
 674 {
 675     NSContainer *This = NSURICL_THIS(iface);
 676     WARN("(%p)->(%p)\n", This, aLoadCookie);
 677     return NS_ERROR_NOT_IMPLEMENTED;
 678 }
 679
 680 static nsresult NSAPI nsURIContentListener_GetParentContentListener(nsIURIContentListener *iface,
 681         nsIURIContentListener **aParentContentListener)
 682 {
 683     NSContainer *This = NSURICL_THIS(iface);
 684     WARN("(%p)->(%p)\n", This, aParentContentListener);
 685     return NS_ERROR_NOT_IMPLEMENTED;
 686 }
 687
 688 static nsresult NSAPI nsURIContentListener_SetParentContentListener(nsIURIContentListener *iface,
 689         nsIURIContentListener *aParentContentListener)
 690 {
 691     NSContainer *This = NSURICL_THIS(iface);
 692     WARN("(%p)->(%p)\n", This, aParentContentListener);
 693     return NS_ERROR_NOT_IMPLEMENTED;
 694 }
 695
 696 #undef NSURICL_THIS
 697
 698 static const nsIURIContentListenerVtbl nsURIContentListenerVtbl = {
 699     nsURIContentListener_QueryInterface,
 700     nsURIContentListener_AddRef,
 701     nsURIContentListener_Release,
 702     nsURIContentListener_OnStartURIOpen,
 703     nsURIContentListener_DoContent,
 704     nsURIContentListener_IsPreferred,
 705     nsURIContentListener_CanHandleContent,
 706     nsURIContentListener_GetLoadCookie,
 707     nsURIContentListener_SetLoadCookie,
 708     nsURIContentListener_GetParentContentListener,
 709     nsURIContentListener_SetParentContentListener
 710 };
 711
 712 void HTMLDocument_NSContainer_Init(HTMLDocument *This)
 713 {
 714     nsIWebBrowserSetup *wbsetup;
 715     nsresult nsres;
 716
 717     This->nscontainer = NULL;
 718
 719     if(!load_gecko())
 720         return;
 721
 722     This->nscontainer = HeapAlloc(GetProcessHeap(), 0, sizeof(NSContainer));
 723
 724     This->nscontainer->lpWebBrowserChromeVtbl    = &nsWebBrowserChromeVtbl;
 725     This->nscontainer->lpContextMenuListenerVtbl = &nsContextMenuListenerVtbl;
 726     This->nscontainer->lpURIContentListenerVtbl  = &nsURIContentListenerVtbl;
 727
 728     This->nscontainer->doc = This;
 729
 730     nsres = nsIComponentManager_CreateInstanceByContractID(pCompMgr, NS_WEBBROWSER_CONTRACTID,
 731             NULL, &IID_nsIWebBrowser, (void**)&This->nscontainer->webbrowser);
 732     if(NS_FAILED(nsres))
 733         ERR("Creating WebBrowser failed: %08lx\n", nsres);
 734
 735     nsres = nsIWebBrowser_SetContainerWindow(This->nscontainer->webbrowser,
 736             NSWBCHROME(This->nscontainer));
 737     if(NS_FAILED(nsres))
 738         ERR("SetContainerWindow failed: %08lx\n", nsres);
 739
 740     nsres = nsIWebBrowser_QueryInterface(This->nscontainer->webbrowser, &IID_nsIBaseWindow,
 741             (void**)&This->nscontainer->window);
 742     if(NS_FAILED(nsres))
 743         ERR("Could not get nsIBaseWindow interface: %08lx\n", nsres);
 744
 745     nsres = nsIWebBrowser_QueryInterface(This->nscontainer->webbrowser,
 746             &IID_nsIWebBrowserSetup, (void**)&wbsetup);
 747     if(NS_SUCCEEDED(nsres)) {
 748         nsres = nsIWebBrowserSetup_SetProperty(wbsetup, SETUP_IS_CHROME_WRAPPER, TRUE);
 749         if(NS_FAILED(nsres))
 750             ERR("SetProperty failed: %08lx\n", nsres);
 751         nsIWebBrowserSetup_Release(wbsetup);
 752     }else {
 753         ERR("Could not get nsIWebBrowserSetup interface\n");
 754     }
 755
 756     nsres = nsIWebBrowser_QueryInterface(This->nscontainer->webbrowser, &IID_nsIWebNavigation,
 757             (void**)&This->nscontainer->navigation);
 758     if(NS_FAILED(nsres))
 759         ERR("Could not get nsIWebNavigation interface: %08lx\n", nsres);
 760
 761     nsres = nsIWebBrowserFocus_QueryInterface(This->nscontainer->webbrowser, &IID_nsIWebBrowserFocus,
 762             (void**)&This->nscontainer->focus);
 763     if(NS_FAILED(nsres))
 764         ERR("Could not get nsIWebBrowserFocus interface: %08lx\n", nsres);
 765
 766 #if 0
 767     nsres = nsIWebBrowserStream_QueryInterface(This->nscontainer->webbrowser, &IID_nsIWebBrowserStream,
 768             (void**)&This->nscontainer->stream);
 769     if(NS_FAILED(nsres))
 770         ERR("Could not get nsIWebBrowserStream interface: %08lx\n", nsres);
 771 #else
 772     This->nscontainer->stream = NULL;
 773 #endif
 774
 775     if(!nscontainer_class)
 776         register_nscontainer_class();
 777
 778     This->nscontainer->hwnd = CreateWindowExW(0, wszNsContainer, NULL,
 779             WS_CHILD | WS_CLIPSIBLINGS | WS_CLIPCHILDREN, 0, 0, 100, 100,
 780             GetDesktopWindow(), NULL, hInst, This);
 781
 782     nsres = nsIBaseWindow_InitWindow(This->nscontainer->window, This->nscontainer->hwnd, NULL,
 783             0, 0, 100, 100);
 784     if(NS_SUCCEEDED(nsres)) {
 785         nsres = nsIBaseWindow_Create(This->nscontainer->window);
 786         if(NS_FAILED(nsres))
 787             WARN("Creating window failed: %08lx\n", nsres);
 788
 789         nsIBaseWindow_SetVisibility(This->nscontainer->window, FALSE);
 790         nsIBaseWindow_SetEnabled(This->nscontainer->window, FALSE);
 791     }else {
 792         ERR("InitWindow failed: %08lx\n", nsres);
 793     }
 794
 795     nsres = nsIWebBrowser_SetParentURIContentListener(This->nscontainer->webbrowser,
 796             NSURICL(This->nscontainer));
 797     if(NS_FAILED(nsres))
 798         ERR("SetParentURIContentListener failed: %08lx\n", nsres);
 799
 800     This->nscontainer->url = NULL;
 801 }
 802
 803 void HTMLDocument_NSContainer_Destroy(HTMLDocument *This)
 804 {
 805     TRACE("(%p)\n", This);
 806
 807     nsIWebBrowser_Release(This->nscontainer->webbrowser);
 808     nsIWebNavigation_Release(This->nscontainer->navigation);
 809     nsIBaseWindow_Release(This->nscontainer->window);
 810
 811     if(This->nscontainer->stream)
 812         nsIWebBrowserStream_Release(This->nscontainer->stream);
 813
 814     HeapFree(GetProcessHeap(), 0, This->nscontainer);
 815
 816     if(This->nscontainer->url)
 817         CoTaskMemFree(This->nscontainer->url);
 818 }