dlls/mshtml/selection.c

   1 /*
   2  * Copyright 2006 Jacek Caban for CodeWeavers
   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 <stdarg.h>
  20
  21 #define COBJMACROS
  22
  23 #include "windef.h"
  24 #include "winbase.h"
  25 #include "winuser.h"
  26 #include "ole2.h"
  27
  28 #include "wine/debug.h"
  29
  30 #include "mshtml_private.h"
  31
  32 WINE_DEFAULT_DEBUG_CHANNEL(mshtml);
  33
  34 typedef struct {
  35     DispatchEx dispex;
  36     IHTMLSelectionObject IHTMLSelectionObject_iface;
  37     IHTMLSelectionObject2 IHTMLSelectionObject2_iface;
  38
  39     LONG ref;
  40
  41     nsISelection *nsselection;
  42     HTMLDocumentNode *doc;
  43
  44     struct list entry;
  45 } HTMLSelectionObject;
  46
  47 static inline HTMLSelectionObject *impl_from_IHTMLSelectionObject(IHTMLSelectionObject *iface)
  48 {
  49     return CONTAINING_RECORD(iface, HTMLSelectionObject, IHTMLSelectionObject_iface);
  50 }
  51
  52 static HRESULT WINAPI HTMLSelectionObject_QueryInterface(IHTMLSelectionObject *iface,
  53                                                          REFIID riid, void **ppv)
  54 {
  55     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
  56
  57     TRACE("(%p)->(%s %p)\n", This, debugstr_mshtml_guid(riid), ppv);
  58
  59     if(IsEqualGUID(&IID_IUnknown, riid)) {
  60         *ppv = &This->IHTMLSelectionObject_iface;
  61     }else if(IsEqualGUID(&IID_IDispatch, riid)) {
  62         *ppv = &This->IHTMLSelectionObject_iface;
  63     }else if(IsEqualGUID(&IID_IHTMLSelectionObject, riid)) {
  64         *ppv = &This->IHTMLSelectionObject_iface;
  65     }else if(IsEqualGUID(&IID_IHTMLSelectionObject2, riid)) {
  66         *ppv = &This->IHTMLSelectionObject2_iface;
  67     }else if(dispex_query_interface(&This->dispex, riid, ppv)) {
  68         return *ppv ? S_OK : E_NOINTERFACE;
  69     }else {
  70         *ppv = NULL;
  71         WARN("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
  72         return E_NOINTERFACE;
  73     }
  74
  75     IUnknown_AddRef((IUnknown*)*ppv);
  76     return S_OK;
  77 }
  78
  79 static ULONG WINAPI HTMLSelectionObject_AddRef(IHTMLSelectionObject *iface)
  80 {
  81     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
  82     LONG ref = InterlockedIncrement(&This->ref);
  83
  84     TRACE("(%p) ref=%d\n", This, ref);
  85
  86     return ref;
  87 }
  88
  89 static ULONG WINAPI HTMLSelectionObject_Release(IHTMLSelectionObject *iface)
  90 {
  91     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
  92     LONG ref = InterlockedDecrement(&This->ref);
  93
  94     TRACE("(%p) ref=%d\n", This, ref);
  95
  96     if(!ref) {
  97         if(This->nsselection)
  98             nsISelection_Release(This->nsselection);
  99         if(This->doc)
 100             list_remove(&This->entry);
 101         release_dispex(&This->dispex);
 102         heap_free(This);
 103     }
 104
 105     return ref;
 106 }
 107
 108 static HRESULT WINAPI HTMLSelectionObject_GetTypeInfoCount(IHTMLSelectionObject *iface, UINT *pctinfo)
 109 {
 110     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
 111
 112     return IDispatchEx_GetTypeInfoCount(&This->dispex.IDispatchEx_iface, pctinfo);
 113 }
 114
 115 static HRESULT WINAPI HTMLSelectionObject_GetTypeInfo(IHTMLSelectionObject *iface, UINT iTInfo,
 116                                               LCID lcid, ITypeInfo **ppTInfo)
 117 {
 118     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
 119
 120     return IDispatchEx_GetTypeInfo(&This->dispex.IDispatchEx_iface, iTInfo, lcid, ppTInfo);
 121 }
 122
 123 static HRESULT WINAPI HTMLSelectionObject_GetIDsOfNames(IHTMLSelectionObject *iface, REFIID riid,
 124                                                 LPOLESTR *rgszNames, UINT cNames,
 125                                                 LCID lcid, DISPID *rgDispId)
 126 {
 127     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
 128
 129     return IDispatchEx_GetIDsOfNames(&This->dispex.IDispatchEx_iface, riid, rgszNames,
 130             cNames, lcid, rgDispId);
 131 }
 132
 133 static HRESULT WINAPI HTMLSelectionObject_Invoke(IHTMLSelectionObject *iface, DISPID dispIdMember,
 134                             REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
 135                             VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
 136 {
 137     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
 138
 139
 140     return IDispatchEx_Invoke(&This->dispex.IDispatchEx_iface, dispIdMember, riid,
 141             lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
 142 }
 143
 144 static HRESULT WINAPI HTMLSelectionObject_createRange(IHTMLSelectionObject *iface, IDispatch **range)
 145 {
 146     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
 147     IHTMLTxtRange *range_obj = NULL;
 148     nsIDOMRange *nsrange = NULL;
 149     HRESULT hres;
 150
 151     TRACE("(%p)->(%p)\n", This, range);
 152
 153     if(This->nsselection) {
 154         LONG nsrange_cnt = 0;
 155         nsresult nsres;
 156
 157         nsISelection_GetRangeCount(This->nsselection, &nsrange_cnt);
 158         if(!nsrange_cnt) {
 159             nsIDOMHTMLElement *nsbody = NULL;
 160
 161             TRACE("nsrange_cnt = 0\n");
 162
 163             if(!This->doc->nsdoc) {
 164                 WARN("nsdoc is NULL\n");
 165                 return E_UNEXPECTED;
 166             }
 167
 168             nsres = nsIDOMHTMLDocument_GetBody(This->doc->nsdoc, &nsbody);
 169             if(NS_FAILED(nsres) || !nsbody) {
 170                 ERR("Could not get body: %08x\n", nsres);
 171                 return E_FAIL;
 172             }
 173
 174             nsres = nsISelection_Collapse(This->nsselection, (nsIDOMNode*)nsbody, 0);
 175             nsIDOMHTMLElement_Release(nsbody);
 176             if(NS_FAILED(nsres))
 177                 ERR("Collapse failed: %08x\n", nsres);
 178         }else if(nsrange_cnt > 1) {
 179             FIXME("range_cnt = %d\n", nsrange_cnt);
 180         }
 181
 182         nsres = nsISelection_GetRangeAt(This->nsselection, 0, &nsrange);
 183         if(NS_FAILED(nsres))
 184             ERR("GetRangeAt failed: %08x\n", nsres);
 185     }
 186
 187     hres = HTMLTxtRange_Create(This->doc, nsrange, &range_obj);
 188
 189     if (nsrange) nsIDOMRange_Release(nsrange);
 190     *range = (IDispatch*)range_obj;
 191     return hres;
 192 }
 193
 194 static HRESULT WINAPI HTMLSelectionObject_empty(IHTMLSelectionObject *iface)
 195 {
 196     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
 197     FIXME("(%p)\n", This);
 198     return E_NOTIMPL;
 199 }
 200
 201 static HRESULT WINAPI HTMLSelectionObject_clear(IHTMLSelectionObject *iface)
 202 {
 203     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
 204     FIXME("(%p)\n", This);
 205     return E_NOTIMPL;
 206 }
 207
 208 static HRESULT WINAPI HTMLSelectionObject_get_type(IHTMLSelectionObject *iface, BSTR *p)
 209 {
 210     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
 211     cpp_bool collapsed = TRUE;
 212
 213     static const WCHAR wszNone[] = {'N','o','n','e',0};
 214     static const WCHAR wszText[] = {'T','e','x','t',0};
 215
 216     TRACE("(%p)->(%p)\n", This, p);
 217
 218     if(This->nsselection)
 219         nsISelection_GetIsCollapsed(This->nsselection, &collapsed);
 220
 221     *p = SysAllocString(collapsed ? wszNone : wszText); /* FIXME: control */
 222     TRACE("ret %s\n", debugstr_w(*p));
 223     return S_OK;
 224 }
 225
 226 static const IHTMLSelectionObjectVtbl HTMLSelectionObjectVtbl = {
 227     HTMLSelectionObject_QueryInterface,
 228     HTMLSelectionObject_AddRef,
 229     HTMLSelectionObject_Release,
 230     HTMLSelectionObject_GetTypeInfoCount,
 231     HTMLSelectionObject_GetTypeInfo,
 232     HTMLSelectionObject_GetIDsOfNames,
 233     HTMLSelectionObject_Invoke,
 234     HTMLSelectionObject_createRange,
 235     HTMLSelectionObject_empty,
 236     HTMLSelectionObject_clear,
 237     HTMLSelectionObject_get_type
 238 };
 239
 240 static inline HTMLSelectionObject *impl_from_IHTMLSelectionObject2(IHTMLSelectionObject2 *iface)
 241 {
 242     return CONTAINING_RECORD(iface, HTMLSelectionObject, IHTMLSelectionObject2_iface);
 243 }
 244
 245 static HRESULT WINAPI HTMLSelectionObject2_QueryInterface(IHTMLSelectionObject2 *iface, REFIID riid, void **ppv)
 246 {
 247     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject2(iface);
 248
 249     return IHTMLSelectionObject_QueryInterface(&This->IHTMLSelectionObject_iface, riid, ppv);
 250 }
 251
 252 static ULONG WINAPI HTMLSelectionObject2_AddRef(IHTMLSelectionObject2 *iface)
 253 {
 254     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject2(iface);
 255
 256     return IHTMLSelectionObject_AddRef(&This->IHTMLSelectionObject_iface);
 257 }
 258
 259 static ULONG WINAPI HTMLSelectionObject2_Release(IHTMLSelectionObject2 *iface)
 260 {
 261     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject2(iface);
 262
 263     return IHTMLSelectionObject_Release(&This->IHTMLSelectionObject_iface);
 264 }
 265
 266 static HRESULT WINAPI HTMLSelectionObject2_GetTypeInfoCount(IHTMLSelectionObject2 *iface, UINT *pctinfo)
 267 {
 268     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject2(iface);
 269
 270     return IDispatchEx_GetTypeInfoCount(&This->dispex.IDispatchEx_iface, pctinfo);
 271 }
 272
 273 static HRESULT WINAPI HTMLSelectionObject2_GetTypeInfo(IHTMLSelectionObject2 *iface, UINT iTInfo,
 274         LCID lcid, ITypeInfo **ppTInfo)
 275 {
 276     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject2(iface);
 277
 278     return IDispatchEx_GetTypeInfo(&This->dispex.IDispatchEx_iface, iTInfo, lcid, ppTInfo);
 279 }
 280
 281 static HRESULT WINAPI HTMLSelectionObject2_GetIDsOfNames(IHTMLSelectionObject2 *iface, REFIID riid,
 282         LPOLESTR *rgszNames, UINT cNames, LCID lcid, DISPID *rgDispId)
 283 {
 284     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject2(iface);
 285
 286     return IDispatchEx_GetIDsOfNames(&This->dispex.IDispatchEx_iface, riid, rgszNames,
 287             cNames, lcid, rgDispId);
 288 }
 289
 290 static HRESULT WINAPI HTMLSelectionObject2_Invoke(IHTMLSelectionObject2 *iface, DISPID dispIdMember,
 291         REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams, VARIANT *pVarResult,
 292         EXCEPINFO *pExcepInfo, UINT *puArgErr)
 293 {
 294     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject2(iface);
 295
 296     return IDispatchEx_Invoke(&This->dispex.IDispatchEx_iface, dispIdMember, riid,
 297             lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
 298 }
 299
 300 static HRESULT WINAPI HTMLSelectionObject2_createRangeCollection(IHTMLSelectionObject2 *iface, IDispatch **rangeCollection)
 301 {
 302     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject2(iface);
 303     FIXME("(%p)->(%p)\n", This, rangeCollection);
 304     return E_NOTIMPL;
 305 }
 306
 307 static HRESULT WINAPI HTMLSelectionObject2_get_typeDetail(IHTMLSelectionObject2 *iface, BSTR *p)
 308 {
 309     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject2(iface);
 310
 311     static const WCHAR undefinedW[] = {'u','n','d','e','f','i','n','e','d',0};
 312
 313     FIXME("(%p)->(%p) semi-stub\n", This, p);
 314
 315     /* FIXME: We should try to use ISelectionServicesListener::GetTypeDetail here. */
 316     *p = SysAllocString(undefinedW);
 317     return *p ? S_OK : E_OUTOFMEMORY;
 318 }
 319
 320 static const IHTMLSelectionObject2Vtbl HTMLSelectionObject2Vtbl = {
 321     HTMLSelectionObject2_QueryInterface,
 322     HTMLSelectionObject2_AddRef,
 323     HTMLSelectionObject2_Release,
 324     HTMLSelectionObject2_GetTypeInfoCount,
 325     HTMLSelectionObject2_GetTypeInfo,
 326     HTMLSelectionObject2_GetIDsOfNames,
 327     HTMLSelectionObject2_Invoke,
 328     HTMLSelectionObject2_createRangeCollection,
 329     HTMLSelectionObject2_get_typeDetail
 330 };
 331
 332 static const tid_t HTMLSelectionObject_iface_tids[] = {
 333     IHTMLSelectionObject_tid,
 334     IHTMLSelectionObject2_tid,
 335     0
 336 };
 337 static dispex_static_data_t HTMLSelectionObject_dispex = {
 338     NULL,
 339     IHTMLSelectionObject_tid, /* FIXME: We have a test for that, but it doesn't expose IHTMLSelectionObject2 iface. */
 340     HTMLSelectionObject_iface_tids
 341 };
 342
 343 HRESULT HTMLSelectionObject_Create(HTMLDocumentNode *doc, nsISelection *nsselection, IHTMLSelectionObject **ret)
 344 {
 345     HTMLSelectionObject *selection;
 346
 347     selection = heap_alloc(sizeof(HTMLSelectionObject));
 348     if(!selection)
 349         return E_OUTOFMEMORY;
 350
 351     init_dispex(&selection->dispex, (IUnknown*)&selection->IHTMLSelectionObject_iface, &HTMLSelectionObject_dispex);
 352
 353     selection->IHTMLSelectionObject_iface.lpVtbl = &HTMLSelectionObjectVtbl;
 354     selection->IHTMLSelectionObject2_iface.lpVtbl = &HTMLSelectionObject2Vtbl;
 355     selection->ref = 1;
 356     selection->nsselection = nsselection; /* We shouldn't call AddRef here */
 357
 358     selection->doc = doc;
 359     list_add_head(&doc->selection_list, &selection->entry);
 360
 361     *ret = &selection->IHTMLSelectionObject_iface;
 362     return S_OK;
 363 }
 364
 365 void detach_selection(HTMLDocumentNode *This)
 366 {
 367     HTMLSelectionObject *iter, *next;
 368
 369     LIST_FOR_EACH_ENTRY_SAFE(iter, next, &This->selection_list, HTMLSelectionObject, entry) {
 370         iter->doc = NULL;
 371         list_remove(&iter->entry);
 372     }
 373 }