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     TRACE("(%p)->(%p)\n", This, p);
 214
 215     if(This->nsselection)
 216         nsISelection_GetIsCollapsed(This->nsselection, &collapsed);
 217
 218     *p = SysAllocString(collapsed ? L"None" : L"Text"); /* FIXME: control */
 219     TRACE("ret %s\n", debugstr_w(*p));
 220     return S_OK;
 221 }
 222
 223 static const IHTMLSelectionObjectVtbl HTMLSelectionObjectVtbl = {
 224     HTMLSelectionObject_QueryInterface,
 225     HTMLSelectionObject_AddRef,
 226     HTMLSelectionObject_Release,
 227     HTMLSelectionObject_GetTypeInfoCount,
 228     HTMLSelectionObject_GetTypeInfo,
 229     HTMLSelectionObject_GetIDsOfNames,
 230     HTMLSelectionObject_Invoke,
 231     HTMLSelectionObject_createRange,
 232     HTMLSelectionObject_empty,
 233     HTMLSelectionObject_clear,
 234     HTMLSelectionObject_get_type
 235 };
 236
 237 static inline HTMLSelectionObject *impl_from_IHTMLSelectionObject2(IHTMLSelectionObject2 *iface)
 238 {
 239     return CONTAINING_RECORD(iface, HTMLSelectionObject, IHTMLSelectionObject2_iface);
 240 }
 241
 242 static HRESULT WINAPI HTMLSelectionObject2_QueryInterface(IHTMLSelectionObject2 *iface, REFIID riid, void **ppv)
 243 {
 244     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject2(iface);
 245
 246     return IHTMLSelectionObject_QueryInterface(&This->IHTMLSelectionObject_iface, riid, ppv);
 247 }
 248
 249 static ULONG WINAPI HTMLSelectionObject2_AddRef(IHTMLSelectionObject2 *iface)
 250 {
 251     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject2(iface);
 252
 253     return IHTMLSelectionObject_AddRef(&This->IHTMLSelectionObject_iface);
 254 }
 255
 256 static ULONG WINAPI HTMLSelectionObject2_Release(IHTMLSelectionObject2 *iface)
 257 {
 258     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject2(iface);
 259
 260     return IHTMLSelectionObject_Release(&This->IHTMLSelectionObject_iface);
 261 }
 262
 263 static HRESULT WINAPI HTMLSelectionObject2_GetTypeInfoCount(IHTMLSelectionObject2 *iface, UINT *pctinfo)
 264 {
 265     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject2(iface);
 266
 267     return IDispatchEx_GetTypeInfoCount(&This->dispex.IDispatchEx_iface, pctinfo);
 268 }
 269
 270 static HRESULT WINAPI HTMLSelectionObject2_GetTypeInfo(IHTMLSelectionObject2 *iface, UINT iTInfo,
 271         LCID lcid, ITypeInfo **ppTInfo)
 272 {
 273     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject2(iface);
 274
 275     return IDispatchEx_GetTypeInfo(&This->dispex.IDispatchEx_iface, iTInfo, lcid, ppTInfo);
 276 }
 277
 278 static HRESULT WINAPI HTMLSelectionObject2_GetIDsOfNames(IHTMLSelectionObject2 *iface, REFIID riid,
 279         LPOLESTR *rgszNames, UINT cNames, LCID lcid, DISPID *rgDispId)
 280 {
 281     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject2(iface);
 282
 283     return IDispatchEx_GetIDsOfNames(&This->dispex.IDispatchEx_iface, riid, rgszNames,
 284             cNames, lcid, rgDispId);
 285 }
 286
 287 static HRESULT WINAPI HTMLSelectionObject2_Invoke(IHTMLSelectionObject2 *iface, DISPID dispIdMember,
 288         REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams, VARIANT *pVarResult,
 289         EXCEPINFO *pExcepInfo, UINT *puArgErr)
 290 {
 291     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject2(iface);
 292
 293     return IDispatchEx_Invoke(&This->dispex.IDispatchEx_iface, dispIdMember, riid,
 294             lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
 295 }
 296
 297 static HRESULT WINAPI HTMLSelectionObject2_createRangeCollection(IHTMLSelectionObject2 *iface, IDispatch **rangeCollection)
 298 {
 299     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject2(iface);
 300     FIXME("(%p)->(%p)\n", This, rangeCollection);
 301     return E_NOTIMPL;
 302 }
 303
 304 static HRESULT WINAPI HTMLSelectionObject2_get_typeDetail(IHTMLSelectionObject2 *iface, BSTR *p)
 305 {
 306     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject2(iface);
 307
 308     FIXME("(%p)->(%p) semi-stub\n", This, p);
 309
 310     /* FIXME: We should try to use ISelectionServicesListener::GetTypeDetail here. */
 311     *p = SysAllocString(L"undefined");
 312     return *p ? S_OK : E_OUTOFMEMORY;
 313 }
 314
 315 static const IHTMLSelectionObject2Vtbl HTMLSelectionObject2Vtbl = {
 316     HTMLSelectionObject2_QueryInterface,
 317     HTMLSelectionObject2_AddRef,
 318     HTMLSelectionObject2_Release,
 319     HTMLSelectionObject2_GetTypeInfoCount,
 320     HTMLSelectionObject2_GetTypeInfo,
 321     HTMLSelectionObject2_GetIDsOfNames,
 322     HTMLSelectionObject2_Invoke,
 323     HTMLSelectionObject2_createRangeCollection,
 324     HTMLSelectionObject2_get_typeDetail
 325 };
 326
 327 static const tid_t HTMLSelectionObject_iface_tids[] = {
 328     IHTMLSelectionObject_tid,
 329     IHTMLSelectionObject2_tid,
 330     0
 331 };
 332 static dispex_static_data_t HTMLSelectionObject_dispex = {
 333     NULL,
 334     IHTMLSelectionObject_tid, /* FIXME: We have a test for that, but it doesn't expose IHTMLSelectionObject2 iface. */
 335     HTMLSelectionObject_iface_tids
 336 };
 337
 338 HRESULT HTMLSelectionObject_Create(HTMLDocumentNode *doc, nsISelection *nsselection, IHTMLSelectionObject **ret)
 339 {
 340     HTMLSelectionObject *selection;
 341
 342     selection = heap_alloc(sizeof(HTMLSelectionObject));
 343     if(!selection)
 344         return E_OUTOFMEMORY;
 345
 346     init_dispatch(&selection->dispex, (IUnknown*)&selection->IHTMLSelectionObject_iface,
 347                   &HTMLSelectionObject_dispex, dispex_compat_mode(&doc->node.event_target.dispex));
 348
 349     selection->IHTMLSelectionObject_iface.lpVtbl = &HTMLSelectionObjectVtbl;
 350     selection->IHTMLSelectionObject2_iface.lpVtbl = &HTMLSelectionObject2Vtbl;
 351     selection->ref = 1;
 352     selection->nsselection = nsselection; /* We shouldn't call AddRef here */
 353
 354     selection->doc = doc;
 355     list_add_head(&doc->selection_list, &selection->entry);
 356
 357     *ret = &selection->IHTMLSelectionObject_iface;
 358     return S_OK;
 359 }
 360
 361 void detach_selection(HTMLDocumentNode *This)
 362 {
 363     HTMLSelectionObject *iter, *next;
 364
 365     LIST_FOR_EACH_ENTRY_SAFE(iter, next, &This->selection_list, HTMLSelectionObject, entry) {
 366         iter->doc = NULL;
 367         list_remove(&iter->entry);
 368     }
 369 }