git.oblomov.eu Git - wine/blob - dlls/mshtml/persist.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 #define NONAMELESSUNION
  26 #define NONAMELESSSTRUCT
  27
  28 #include "windef.h"
  29 #include "winbase.h"
  30 #include "winuser.h"
  31 #include "ole2.h"
  32 #include "shlguid.h"
  33 #include "idispids.h"
  34
  35 #include "wine/debug.h"
  36 #include "wine/unicode.h"
  37
  38 #include "mshtml_private.h"
  39 #include "htmlevent.h"
  40 #include "resource.h"
  41
  42 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
  43
  44 typedef struct {
  45     task_t header;
  46     HTMLDocumentObj *doc;
  47     BOOL set_download;
  48     LPOLESTR url;
  49 } download_proc_task_t;
  50
  51 static BOOL use_gecko_script(LPCWSTR url)
  52 {
  53     static const WCHAR fileW[] = {'f','i','l','e',':'};
  54     static const WCHAR aboutW[] = {'a','b','o','u','t',':'};
  55     static const WCHAR resW[] = {'r','e','s',':'};
  56
  57     return strncmpiW(fileW, url, sizeof(fileW)/sizeof(WCHAR))
  58         && strncmpiW(aboutW, url, sizeof(aboutW)/sizeof(WCHAR))
  59         && strncmpiW(resW, url, sizeof(resW)/sizeof(WCHAR));
  60 }
  61
  62 void set_current_mon(HTMLWindow *This, IMoniker *mon)
  63 {
  64     HRESULT hres;
  65
  66     if(This->mon) {
  67         IMoniker_Release(This->mon);
  68         This->mon = NULL;
  69     }
  70
  71     if(This->url) {
  72         CoTaskMemFree(This->url);
  73         This->url = NULL;
  74     }
  75
  76     if(!mon)
  77         return;
  78
  79     IMoniker_AddRef(mon);
  80     This->mon = mon;
  81
  82     hres = IMoniker_GetDisplayName(mon, NULL, NULL, &This->url);
  83     if(FAILED(hres))
  84         WARN("GetDisplayName failed: %08x\n", hres);
  85
  86     set_script_mode(This, use_gecko_script(This->url) ? SCRIPTMODE_GECKO : SCRIPTMODE_ACTIVESCRIPT);
  87 }
  88
  89 static void set_progress_proc(task_t *_task)
  90 {
  91     docobj_task_t *task = (docobj_task_t*)_task;
  92     IOleCommandTarget *olecmd = NULL;
  93     HTMLDocumentObj *doc = task->doc;
  94     HRESULT hres;
  95
  96     TRACE("(%p)\n", doc);
  97
  98     if(doc->client)
  99         IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&olecmd);
 100
 101     if(olecmd) {
 102         VARIANT progress_max, progress;
 103
 104         V_VT(&progress_max) = VT_I4;
 105         V_I4(&progress_max) = 0; /* FIXME */
 106         IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_SETPROGRESSMAX, OLECMDEXECOPT_DONTPROMPTUSER,
 107                                &progress_max, NULL);
 108
 109         V_VT(&progress) = VT_I4;
 110         V_I4(&progress) = 0; /* FIXME */
 111         IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_SETPROGRESSPOS, OLECMDEXECOPT_DONTPROMPTUSER,
 112                                &progress, NULL);
 113         IOleCommandTarget_Release(olecmd);
 114     }
 115
 116     if(doc->usermode == EDITMODE && doc->hostui) {
 117         DOCHOSTUIINFO hostinfo;
 118
 119         memset(&hostinfo, 0, sizeof(DOCHOSTUIINFO));
 120         hostinfo.cbSize = sizeof(DOCHOSTUIINFO);
 121         hres = IDocHostUIHandler_GetHostInfo(doc->hostui, &hostinfo);
 122         if(SUCCEEDED(hres))
 123             /* FIXME: use hostinfo */
 124             TRACE("hostinfo = {%u %08x %08x %s %s}\n",
 125                     hostinfo.cbSize, hostinfo.dwFlags, hostinfo.dwDoubleClick,
 126                     debugstr_w(hostinfo.pchHostCss), debugstr_w(hostinfo.pchHostNS));
 127     }
 128 }
 129
 130 static void set_downloading_proc(task_t *_task)
 131 {
 132     download_proc_task_t *task = (download_proc_task_t*)_task;
 133     HTMLDocumentObj *doc = task->doc;
 134     IOleCommandTarget *olecmd;
 135     HRESULT hres;
 136
 137     TRACE("(%p)\n", doc);
 138
 139     set_statustext(doc, IDS_STATUS_DOWNLOADINGFROM, task->url);
 140     CoTaskMemFree(task->url);
 141
 142     if(!doc->client)
 143         return;
 144
 145     if(task->set_download) {
 146         hres = IOleClientSite_QueryInterface(doc->client, &IID_IOleCommandTarget, (void**)&olecmd);
 147         if(SUCCEEDED(hres)) {
 148             VARIANT var;
 149
 150             V_VT(&var) = VT_I4;
 151             V_I4(&var) = 1;
 152
 153             IOleCommandTarget_Exec(olecmd, NULL, OLECMDID_SETDOWNLOADSTATE,
 154                     OLECMDEXECOPT_DONTPROMPTUSER, &var, NULL);
 155             IOleCommandTarget_Release(olecmd);
 156         }
 157
 158         doc->download_state = 1;
 159     }
 160
 161     if(doc->view_sink)
 162         IAdviseSink_OnViewChange(doc->view_sink, DVASPECT_CONTENT, -1);
 163
 164     if(doc->hostui) {
 165         IDropTarget *drop_target = NULL;
 166
 167         hres = IDocHostUIHandler_GetDropTarget(doc->hostui, NULL /* FIXME */, &drop_target);
 168         if(drop_target) {
 169             FIXME("Use IDropTarget\n");
 170             IDropTarget_Release(drop_target);
 171         }
 172     }
 173 }
 174
 175 HRESULT set_moniker(HTMLDocument *This, IMoniker *mon, IBindCtx *pibc, nsChannelBSC *async_bsc, BOOL set_download)
 176 {
 177     nsChannelBSC *bscallback;
 178     docobj_task_t *task;
 179     download_proc_task_t *download_task;
 180     nsWineURI *nsuri;
 181     LPOLESTR url;
 182     HRESULT hres;
 183
 184     hres = IMoniker_GetDisplayName(mon, pibc, NULL, &url);
 185     if(FAILED(hres)) {
 186         WARN("GetDiaplayName failed: %08x\n", hres);
 187         return hres;
 188     }
 189
 190     TRACE("got url: %s\n", debugstr_w(url));
 191
 192     if(This->doc_obj->client) {
 193         VARIANT silent, offline;
 194
 195         hres = get_client_disp_property(This->doc_obj->client, DISPID_AMBIENT_SILENT, &silent);
 196         if(SUCCEEDED(hres)) {
 197             if(V_VT(&silent) != VT_BOOL)
 198                 WARN("V_VT(silent) = %d\n", V_VT(&silent));
 199             else if(V_BOOL(&silent))
 200                 FIXME("silent == true\n");
 201         }
 202
 203         hres = get_client_disp_property(This->doc_obj->client,
 204                 DISPID_AMBIENT_OFFLINEIFNOTCONNECTED, &offline);
 205         if(SUCCEEDED(hres)) {
 206             if(V_VT(&silent) != VT_BOOL)
 207                 WARN("V_VT(offline) = %d\n", V_VT(&silent));
 208             else if(V_BOOL(&silent))
 209                 FIXME("offline == true\n");
 210         }
 211     }
 212
 213     if(This->window->mon) {
 214         update_doc(This, UPDATE_TITLE|UPDATE_UI);
 215     }else {
 216         update_doc(This, UPDATE_TITLE);
 217         set_current_mon(This->window, mon);
 218     }
 219
 220     set_ready_state(This->window, READYSTATE_LOADING);
 221
 222     if(This->doc_obj->client) {
 223         IOleCommandTarget *cmdtrg = NULL;
 224
 225         hres = IOleClientSite_QueryInterface(This->doc_obj->client, &IID_IOleCommandTarget,
 226                 (void**)&cmdtrg);
 227         if(SUCCEEDED(hres)) {
 228             VARIANT var, out;
 229
 230             if(!async_bsc) {
 231                 V_VT(&var) = VT_I4;
 232                 V_I4(&var) = 0;
 233                 IOleCommandTarget_Exec(cmdtrg, &CGID_ShellDocView, 37, 0, &var, NULL);
 234             }else {
 235                 V_VT(&var) = VT_UNKNOWN;
 236                 V_UNKNOWN(&var) = (IUnknown*)HTMLWINDOW2(This->window);
 237                 V_VT(&out) = VT_EMPTY;
 238                 hres = IOleCommandTarget_Exec(cmdtrg, &CGID_ShellDocView, 63, 0, &var, &out);
 239                 if(SUCCEEDED(hres))
 240                     VariantClear(&out);
 241             }
 242
 243             IOleCommandTarget_Release(cmdtrg);
 244         }
 245     }
 246
 247     hres = create_doc_uri(This->window, url, &nsuri);
 248
 249     if(SUCCEEDED(hres))
 250     {
 251         if(async_bsc) {
 252             bscallback = async_bsc;
 253         }else {
 254             hres = create_channelbsc(mon, NULL, NULL, 0, &bscallback);
 255         }
 256     }
 257
 258     if(SUCCEEDED(hres))
 259     {
 260         hres = load_nsuri(This->window, nsuri, bscallback, LOAD_INITIAL_DOCUMENT_URI);
 261         nsISupports_Release((nsISupports*)nsuri); /* FIXME */
 262         if(SUCCEEDED(hres))
 263             set_window_bscallback(This->window, bscallback);
 264         if(bscallback != async_bsc)
 265             IUnknown_Release((IUnknown*)bscallback);
 266     }
 267
 268     if(FAILED(hres))
 269     {
 270         CoTaskMemFree(url);
 271         return hres;
 272     }
 273
 274     HTMLDocument_LockContainer(This->doc_obj, TRUE);
 275
 276     if(This->doc_obj->frame) {
 277         task = heap_alloc(sizeof(docobj_task_t));
 278         task->doc = This->doc_obj;
 279         push_task(&task->header, set_progress_proc, This->doc_obj->basedoc.task_magic);
 280     }
 281
 282     download_task = heap_alloc(sizeof(download_proc_task_t));
 283     download_task->doc = This->doc_obj;
 284     download_task->set_download = set_download;
 285     download_task->url = url;
 286     push_task(&download_task->header, set_downloading_proc, This->doc_obj->basedoc.task_magic);
 287
 288     return S_OK;
 289 }
 290
 291 void set_ready_state(HTMLWindow *window, READYSTATE readystate)
 292 {
 293     window->readystate = readystate;
 294
 295     if(window->doc_obj && window->doc_obj->basedoc.window == window)
 296         call_property_onchanged(&window->doc_obj->basedoc.cp_propnotif, DISPID_READYSTATE);
 297
 298     fire_event(window->doc, EVENTID_READYSTATECHANGE, FALSE, window->doc->node.nsnode, NULL);
 299
 300     if(window->frame_element)
 301         fire_event(window->frame_element->element.node.doc, EVENTID_READYSTATECHANGE,
 302                    TRUE, window->frame_element->element.node.nsnode, NULL);
 303 }
 304
 305 static HRESULT get_doc_string(HTMLDocumentNode *This, char **str)
 306 {
 307     nsIDOMNode *nsnode;
 308     LPCWSTR strw;
 309     nsAString nsstr;
 310     nsresult nsres;
 311
 312     if(!This->nsdoc) {
 313         WARN("NULL nsdoc\n");
 314         return E_UNEXPECTED;
 315     }
 316
 317     nsres = nsIDOMHTMLDocument_QueryInterface(This->nsdoc, &IID_nsIDOMNode, (void**)&nsnode);
 318     if(NS_FAILED(nsres)) {
 319         ERR("Could not get nsIDOMNode failed: %08x\n", nsres);
 320         return E_FAIL;
 321     }
 322
 323     nsAString_Init(&nsstr, NULL);
 324     nsnode_to_nsstring(nsnode, &nsstr);
 325     nsIDOMNode_Release(nsnode);
 326
 327     nsAString_GetData(&nsstr, &strw);
 328     TRACE("%s\n", debugstr_w(strw));
 329
 330     *str = heap_strdupWtoA(strw);
 331
 332     nsAString_Finish(&nsstr);
 333
 334     return S_OK;
 335 }
 336
 337
 338 /**********************************************************
 339  * IPersistMoniker implementation
 340  */
 341
 342 #define PERSISTMON_THIS(iface) DEFINE_THIS(HTMLDocument, PersistMoniker, iface)
 343
 344 static HRESULT WINAPI PersistMoniker_QueryInterface(IPersistMoniker *iface, REFIID riid, void **ppv)
 345 {
 346     HTMLDocument *This = PERSISTMON_THIS(iface);
 347     return htmldoc_query_interface(This, riid, ppv);
 348 }
 349
 350 static ULONG WINAPI PersistMoniker_AddRef(IPersistMoniker *iface)
 351 {
 352     HTMLDocument *This = PERSISTMON_THIS(iface);
 353     return htmldoc_addref(This);
 354 }
 355
 356 static ULONG WINAPI PersistMoniker_Release(IPersistMoniker *iface)
 357 {
 358     HTMLDocument *This = PERSISTMON_THIS(iface);
 359     return htmldoc_release(This);
 360 }
 361
 362 static HRESULT WINAPI PersistMoniker_GetClassID(IPersistMoniker *iface, CLSID *pClassID)
 363 {
 364     HTMLDocument *This = PERSISTMON_THIS(iface);
 365     return IPersist_GetClassID(PERSIST(This), pClassID);
 366 }
 367
 368 static HRESULT WINAPI PersistMoniker_IsDirty(IPersistMoniker *iface)
 369 {
 370     HTMLDocument *This = PERSISTMON_THIS(iface);
 371
 372     TRACE("(%p)\n", This);
 373
 374     return IPersistStreamInit_IsDirty(PERSTRINIT(This));
 375 }
 376
 377 static HRESULT WINAPI PersistMoniker_Load(IPersistMoniker *iface, BOOL fFullyAvailable,
 378         IMoniker *pimkName, LPBC pibc, DWORD grfMode)
 379 {
 380     HTMLDocument *This = PERSISTMON_THIS(iface);
 381     HRESULT hres;
 382
 383     TRACE("(%p)->(%x %p %p %08x)\n", This, fFullyAvailable, pimkName, pibc, grfMode);
 384
 385     if(pibc) {
 386         IUnknown *unk = NULL;
 387
 388         /* FIXME:
 389          * Use params:
 390          * "__PrecreatedObject"
 391          * "BIND_CONTEXT_PARAM"
 392          * "__HTMLLOADOPTIONS"
 393          * "__DWNBINDINFO"
 394          * "URL Context"
 395          * "_ITransData_Object_"
 396          * "_EnumFORMATETC_"
 397          */
 398
 399         IBindCtx_GetObjectParam(pibc, (LPOLESTR)SZ_HTML_CLIENTSITE_OBJECTPARAM, &unk);
 400         if(unk) {
 401             IOleClientSite *client = NULL;
 402
 403             hres = IUnknown_QueryInterface(unk, &IID_IOleClientSite, (void**)&client);
 404             if(SUCCEEDED(hres)) {
 405                 TRACE("Got client site %p\n", client);
 406                 IOleObject_SetClientSite(OLEOBJ(This), client);
 407                 IOleClientSite_Release(client);
 408             }
 409
 410             IUnknown_Release(unk);
 411         }
 412     }
 413
 414     hres = set_moniker(This, pimkName, pibc, NULL, TRUE);
 415     if(FAILED(hres))
 416         return hres;
 417
 418     return start_binding(This->window, NULL, (BSCallback*)This->window->bscallback, pibc);
 419 }
 420
 421 static HRESULT WINAPI PersistMoniker_Save(IPersistMoniker *iface, IMoniker *pimkName,
 422         LPBC pbc, BOOL fRemember)
 423 {
 424     HTMLDocument *This = PERSISTMON_THIS(iface);
 425     FIXME("(%p)->(%p %p %x)\n", This, pimkName, pbc, fRemember);
 426     return E_NOTIMPL;
 427 }
 428
 429 static HRESULT WINAPI PersistMoniker_SaveCompleted(IPersistMoniker *iface, IMoniker *pimkName, LPBC pibc)
 430 {
 431     HTMLDocument *This = PERSISTMON_THIS(iface);
 432     FIXME("(%p)->(%p %p)\n", This, pimkName, pibc);
 433     return E_NOTIMPL;
 434 }
 435
 436 static HRESULT WINAPI PersistMoniker_GetCurMoniker(IPersistMoniker *iface, IMoniker **ppimkName)
 437 {
 438     HTMLDocument *This = PERSISTMON_THIS(iface);
 439
 440     TRACE("(%p)->(%p)\n", This, ppimkName);
 441
 442     if(!This->window || !This->window->mon)
 443         return E_UNEXPECTED;
 444
 445     IMoniker_AddRef(This->window->mon);
 446     *ppimkName = This->window->mon;
 447     return S_OK;
 448 }
 449
 450 static const IPersistMonikerVtbl PersistMonikerVtbl = {
 451     PersistMoniker_QueryInterface,
 452     PersistMoniker_AddRef,
 453     PersistMoniker_Release,
 454     PersistMoniker_GetClassID,
 455     PersistMoniker_IsDirty,
 456     PersistMoniker_Load,
 457     PersistMoniker_Save,
 458     PersistMoniker_SaveCompleted,
 459     PersistMoniker_GetCurMoniker
 460 };
 461
 462 /**********************************************************
 463  * IMonikerProp implementation
 464  */
 465
 466 #define MONPROP_THIS(iface) DEFINE_THIS(HTMLDocument, MonikerProp, iface)
 467
 468 static HRESULT WINAPI MonikerProp_QueryInterface(IMonikerProp *iface, REFIID riid, void **ppv)
 469 {
 470     HTMLDocument *This = MONPROP_THIS(iface);
 471     return htmldoc_query_interface(This, riid, ppv);
 472 }
 473
 474 static ULONG WINAPI MonikerProp_AddRef(IMonikerProp *iface)
 475 {
 476     HTMLDocument *This = MONPROP_THIS(iface);
 477     return htmldoc_addref(This);
 478 }
 479
 480 static ULONG WINAPI MonikerProp_Release(IMonikerProp *iface)
 481 {
 482     HTMLDocument *This = MONPROP_THIS(iface);
 483     return htmldoc_release(This);
 484 }
 485
 486 static HRESULT WINAPI MonikerProp_PutProperty(IMonikerProp *iface, MONIKERPROPERTY mkp, LPCWSTR val)
 487 {
 488     HTMLDocument *This = MONPROP_THIS(iface);
 489
 490     TRACE("(%p)->(%d %s)\n", This, mkp, debugstr_w(val));
 491
 492     switch(mkp) {
 493     case MIMETYPEPROP:
 494         heap_free(This->doc_obj->mime);
 495         This->doc_obj->mime = heap_strdupW(val);
 496         break;
 497
 498     case CLASSIDPROP:
 499         break;
 500
 501     default:
 502         FIXME("mkp %d\n", mkp);
 503         return E_NOTIMPL;
 504     }
 505
 506     return S_OK;
 507 }
 508
 509 static const IMonikerPropVtbl MonikerPropVtbl = {
 510     MonikerProp_QueryInterface,
 511     MonikerProp_AddRef,
 512     MonikerProp_Release,
 513     MonikerProp_PutProperty
 514 };
 515
 516 /**********************************************************
 517  * IPersistFile implementation
 518  */
 519
 520 #define PERSISTFILE_THIS(iface) DEFINE_THIS(HTMLDocument, PersistFile, iface)
 521
 522 static HRESULT WINAPI PersistFile_QueryInterface(IPersistFile *iface, REFIID riid, void **ppv)
 523 {
 524     HTMLDocument *This = PERSISTFILE_THIS(iface);
 525     return htmldoc_query_interface(This, riid, ppv);
 526 }
 527
 528 static ULONG WINAPI PersistFile_AddRef(IPersistFile *iface)
 529 {
 530     HTMLDocument *This = PERSISTFILE_THIS(iface);
 531     return htmldoc_addref(This);
 532 }
 533
 534 static ULONG WINAPI PersistFile_Release(IPersistFile *iface)
 535 {
 536     HTMLDocument *This = PERSISTFILE_THIS(iface);
 537     return htmldoc_release(This);
 538 }
 539
 540 static HRESULT WINAPI PersistFile_GetClassID(IPersistFile *iface, CLSID *pClassID)
 541 {
 542     HTMLDocument *This = PERSISTFILE_THIS(iface);
 543
 544     TRACE("(%p)->(%p)\n", This, pClassID);
 545
 546     if(!pClassID)
 547         return E_INVALIDARG;
 548
 549     *pClassID = CLSID_HTMLDocument;
 550     return S_OK;
 551 }
 552
 553 static HRESULT WINAPI PersistFile_IsDirty(IPersistFile *iface)
 554 {
 555     HTMLDocument *This = PERSISTFILE_THIS(iface);
 556
 557     TRACE("(%p)\n", This);
 558
 559     return IPersistStreamInit_IsDirty(PERSTRINIT(This));
 560 }
 561
 562 static HRESULT WINAPI PersistFile_Load(IPersistFile *iface, LPCOLESTR pszFileName, DWORD dwMode)
 563 {
 564     HTMLDocument *This = PERSISTFILE_THIS(iface);
 565     FIXME("(%p)->(%s %08x)\n", This, debugstr_w(pszFileName), dwMode);
 566     return E_NOTIMPL;
 567 }
 568
 569 static HRESULT WINAPI PersistFile_Save(IPersistFile *iface, LPCOLESTR pszFileName, BOOL fRemember)
 570 {
 571     HTMLDocument *This = PERSISTFILE_THIS(iface);
 572     char *str;
 573     DWORD written=0;
 574     HANDLE file;
 575     HRESULT hres;
 576
 577     TRACE("(%p)->(%s %x)\n", This, debugstr_w(pszFileName), fRemember);
 578
 579     file = CreateFileW(pszFileName, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
 580                        FILE_ATTRIBUTE_NORMAL, NULL);
 581     if(file == INVALID_HANDLE_VALUE) {
 582         WARN("Could not create file: %u\n", GetLastError());
 583         return E_FAIL;
 584     }
 585
 586     hres = get_doc_string(This->doc_node, &str);
 587     if(SUCCEEDED(hres))
 588         WriteFile(file, str, strlen(str), &written, NULL);
 589
 590     CloseHandle(file);
 591     return hres;
 592 }
 593
 594 static HRESULT WINAPI PersistFile_SaveCompleted(IPersistFile *iface, LPCOLESTR pszFileName)
 595 {
 596     HTMLDocument *This = PERSISTFILE_THIS(iface);
 597     FIXME("(%p)->(%s)\n", This, debugstr_w(pszFileName));
 598     return E_NOTIMPL;
 599 }
 600
 601 static HRESULT WINAPI PersistFile_GetCurFile(IPersistFile *iface, LPOLESTR *pszFileName)
 602 {
 603     HTMLDocument *This = PERSISTFILE_THIS(iface);
 604     FIXME("(%p)->(%p)\n", This, pszFileName);
 605     return E_NOTIMPL;
 606 }
 607
 608 static const IPersistFileVtbl PersistFileVtbl = {
 609     PersistFile_QueryInterface,
 610     PersistFile_AddRef,
 611     PersistFile_Release,
 612     PersistFile_GetClassID,
 613     PersistFile_IsDirty,
 614     PersistFile_Load,
 615     PersistFile_Save,
 616     PersistFile_SaveCompleted,
 617     PersistFile_GetCurFile
 618 };
 619
 620 #define PERSTRINIT_THIS(iface) DEFINE_THIS(HTMLDocument, PersistStreamInit, iface)
 621
 622 static HRESULT WINAPI PersistStreamInit_QueryInterface(IPersistStreamInit *iface,
 623                                                        REFIID riid, void **ppv)
 624 {
 625     HTMLDocument *This = PERSTRINIT_THIS(iface);
 626     return htmldoc_query_interface(This, riid, ppv);
 627 }
 628
 629 static ULONG WINAPI PersistStreamInit_AddRef(IPersistStreamInit *iface)
 630 {
 631     HTMLDocument *This = PERSTRINIT_THIS(iface);
 632     return htmldoc_addref(This);
 633 }
 634
 635 static ULONG WINAPI PersistStreamInit_Release(IPersistStreamInit *iface)
 636 {
 637     HTMLDocument *This = PERSTRINIT_THIS(iface);
 638     return htmldoc_release(This);
 639 }
 640
 641 static HRESULT WINAPI PersistStreamInit_GetClassID(IPersistStreamInit *iface, CLSID *pClassID)
 642 {
 643     HTMLDocument *This = PERSTRINIT_THIS(iface);
 644     return IPersist_GetClassID(PERSIST(This), pClassID);
 645 }
 646
 647 static HRESULT WINAPI PersistStreamInit_IsDirty(IPersistStreamInit *iface)
 648 {
 649     HTMLDocument *This = PERSTRINIT_THIS(iface);
 650
 651     TRACE("(%p)\n", This);
 652
 653     if(This->doc_obj->usermode == EDITMODE)
 654         return editor_is_dirty(This);
 655
 656     return S_FALSE;
 657 }
 658
 659 static HRESULT WINAPI PersistStreamInit_Load(IPersistStreamInit *iface, LPSTREAM pStm)
 660 {
 661     HTMLDocument *This = PERSTRINIT_THIS(iface);
 662     IMoniker *mon;
 663     HRESULT hres;
 664
 665     static const WCHAR about_blankW[] = {'a','b','o','u','t',':','b','l','a','n','k',0};
 666
 667     TRACE("(%p)->(%p)\n", This, pStm);
 668
 669     hres = CreateURLMoniker(NULL, about_blankW, &mon);
 670     if(FAILED(hres)) {
 671         WARN("CreateURLMoniker failed: %08x\n", hres);
 672         return hres;
 673     }
 674
 675     hres = set_moniker(This, mon, NULL, NULL, TRUE);
 676     IMoniker_Release(mon);
 677     if(FAILED(hres))
 678         return hres;
 679
 680     return channelbsc_load_stream(This->window->bscallback, pStm);
 681 }
 682
 683 static HRESULT WINAPI PersistStreamInit_Save(IPersistStreamInit *iface, LPSTREAM pStm,
 684                                              BOOL fClearDirty)
 685 {
 686     HTMLDocument *This = PERSTRINIT_THIS(iface);
 687     char *str;
 688     DWORD written=0;
 689     HRESULT hres;
 690
 691     TRACE("(%p)->(%p %x)\n", This, pStm, fClearDirty);
 692
 693     hres = get_doc_string(This->doc_node, &str);
 694     if(FAILED(hres))
 695         return hres;
 696
 697     hres = IStream_Write(pStm, str, strlen(str), &written);
 698     if(FAILED(hres))
 699         FIXME("Write failed: %08x\n", hres);
 700
 701     heap_free(str);
 702
 703     if(fClearDirty)
 704         set_dirty(This, VARIANT_FALSE);
 705
 706     return S_OK;
 707 }
 708
 709 static HRESULT WINAPI PersistStreamInit_GetSizeMax(IPersistStreamInit *iface,
 710                                                    ULARGE_INTEGER *pcbSize)
 711 {
 712     HTMLDocument *This = PERSTRINIT_THIS(iface);
 713     FIXME("(%p)->(%p)\n", This, pcbSize);
 714     return E_NOTIMPL;
 715 }
 716
 717 static HRESULT WINAPI PersistStreamInit_InitNew(IPersistStreamInit *iface)
 718 {
 719     HTMLDocument *This = PERSTRINIT_THIS(iface);
 720     IMoniker *mon;
 721     HRESULT hres;
 722
 723     static const WCHAR about_blankW[] = {'a','b','o','u','t',':','b','l','a','n','k',0};
 724
 725     TRACE("(%p)\n", This);
 726
 727     hres = CreateURLMoniker(NULL, about_blankW, &mon);
 728     if(FAILED(hres)) {
 729         WARN("CreateURLMoniker failed: %08x\n", hres);
 730         return hres;
 731     }
 732
 733     hres = set_moniker(This, mon, NULL, NULL, FALSE);
 734     IMoniker_Release(mon);
 735     if(FAILED(hres))
 736         return hres;
 737
 738     return channelbsc_load_stream(This->window->bscallback, NULL);
 739 }
 740
 741 #undef PERSTRINIT_THIS
 742
 743 static const IPersistStreamInitVtbl PersistStreamInitVtbl = {
 744     PersistStreamInit_QueryInterface,
 745     PersistStreamInit_AddRef,
 746     PersistStreamInit_Release,
 747     PersistStreamInit_GetClassID,
 748     PersistStreamInit_IsDirty,
 749     PersistStreamInit_Load,
 750     PersistStreamInit_Save,
 751     PersistStreamInit_GetSizeMax,
 752     PersistStreamInit_InitNew
 753 };
 754
 755 /**********************************************************
 756  * IPersistHistory implementation
 757  */
 758
 759 #define PERSISTHIST_THIS(iface) DEFINE_THIS(HTMLDocument, PersistHistory, iface)
 760
 761 static HRESULT WINAPI PersistHistory_QueryInterface(IPersistHistory *iface, REFIID riid, void **ppv)
 762 {
 763     HTMLDocument *This = PERSISTHIST_THIS(iface);
 764     return htmldoc_query_interface(This, riid, ppv);
 765 }
 766
 767 static ULONG WINAPI PersistHistory_AddRef(IPersistHistory *iface)
 768 {
 769     HTMLDocument *This = PERSISTHIST_THIS(iface);
 770     return htmldoc_addref(This);
 771 }
 772
 773 static ULONG WINAPI PersistHistory_Release(IPersistHistory *iface)
 774 {
 775     HTMLDocument *This = PERSISTHIST_THIS(iface);
 776     return htmldoc_release(This);
 777 }
 778
 779 static HRESULT WINAPI PersistHistory_GetClassID(IPersistHistory *iface, CLSID *pClassID)
 780 {
 781     HTMLDocument *This = PERSISTHIST_THIS(iface);
 782     return IPersist_GetClassID(PERSIST(This), pClassID);
 783 }
 784
 785 static HRESULT WINAPI PersistHistory_LoadHistory(IPersistHistory *iface, IStream *pStream, IBindCtx *pbc)
 786 {
 787     HTMLDocument *This = PERSISTHIST_THIS(iface);
 788     FIXME("(%p)->(%p %p)\n", This, pStream, pbc);
 789     return E_NOTIMPL;
 790 }
 791
 792 static HRESULT WINAPI PersistHistory_SaveHistory(IPersistHistory *iface, IStream *pStream)
 793 {
 794     HTMLDocument *This = PERSISTHIST_THIS(iface);
 795     FIXME("(%p)->(%p)\n", This, pStream);
 796     return E_NOTIMPL;
 797 }
 798
 799 static HRESULT WINAPI PersistHistory_SetPositionCookie(IPersistHistory *iface, DWORD dwPositioncookie)
 800 {
 801     HTMLDocument *This = PERSISTHIST_THIS(iface);
 802     FIXME("(%p)->(%x)\n", This, dwPositioncookie);
 803     return E_NOTIMPL;
 804 }
 805
 806 static HRESULT WINAPI PersistHistory_GetPositionCookie(IPersistHistory *iface, DWORD *pdwPositioncookie)
 807 {
 808     HTMLDocument *This = PERSISTHIST_THIS(iface);
 809     FIXME("(%p)->(%p)\n", This, pdwPositioncookie);
 810     return E_NOTIMPL;
 811 }
 812
 813 #undef PERSISTHIST_THIS
 814
 815 static const IPersistHistoryVtbl PersistHistoryVtbl = {
 816     PersistHistory_QueryInterface,
 817     PersistHistory_AddRef,
 818     PersistHistory_Release,
 819     PersistHistory_GetClassID,
 820     PersistHistory_LoadHistory,
 821     PersistHistory_SaveHistory,
 822     PersistHistory_SetPositionCookie,
 823     PersistHistory_GetPositionCookie
 824 };
 825
 826 void HTMLDocument_Persist_Init(HTMLDocument *This)
 827 {
 828     This->lpPersistMonikerVtbl = &PersistMonikerVtbl;
 829     This->lpPersistFileVtbl = &PersistFileVtbl;
 830     This->lpMonikerPropVtbl = &MonikerPropVtbl;
 831     This->lpPersistStreamInitVtbl = &PersistStreamInitVtbl;
 832     This->lpPersistHistoryVtbl = &PersistHistoryVtbl;
 833 }