mshtml: HTMLWindow_item code clean up.
[wine] / 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     IHTMLSelectionObject IHTMLSelectionObject_iface;
36
37     LONG ref;
38
39     nsISelection *nsselection;
40     HTMLDocumentNode *doc;
41
42     struct list entry;
43 } HTMLSelectionObject;
44
45 static inline HTMLSelectionObject *impl_from_IHTMLSelectionObject(IHTMLSelectionObject *iface)
46 {
47     return CONTAINING_RECORD(iface, HTMLSelectionObject, IHTMLSelectionObject_iface);
48 }
49
50 static HRESULT WINAPI HTMLSelectionObject_QueryInterface(IHTMLSelectionObject *iface,
51                                                          REFIID riid, void **ppv)
52 {
53     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
54
55     *ppv = NULL;
56
57     if(IsEqualGUID(&IID_IUnknown, riid)) {
58         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
59         *ppv = &This->IHTMLSelectionObject_iface;
60     }else if(IsEqualGUID(&IID_IDispatch, riid)) {
61         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
62         *ppv = &This->IHTMLSelectionObject_iface;
63     }else if(IsEqualGUID(&IID_IHTMLSelectionObject, riid)) {
64         TRACE("(%p)->(IID_IUnknown %p)\n", This, ppv);
65         *ppv = &This->IHTMLSelectionObject_iface;
66     }
67
68     if(*ppv) {
69         IUnknown_AddRef((IUnknown*)*ppv);
70         return S_OK;
71     }
72
73     WARN("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
74     return E_NOINTERFACE;
75 }
76
77 static ULONG WINAPI HTMLSelectionObject_AddRef(IHTMLSelectionObject *iface)
78 {
79     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
80     LONG ref = InterlockedIncrement(&This->ref);
81
82     TRACE("(%p) ref=%d\n", This, ref);
83
84     return ref;
85 }
86
87 static ULONG WINAPI HTMLSelectionObject_Release(IHTMLSelectionObject *iface)
88 {
89     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
90     LONG ref = InterlockedDecrement(&This->ref);
91
92     TRACE("(%p) ref=%d\n", This, ref);
93
94     if(!ref) {
95         if(This->nsselection)
96             nsISelection_Release(This->nsselection);
97         if(This->doc)
98             list_remove(&This->entry);
99         heap_free(This);
100     }
101
102     return ref;
103 }
104
105 static HRESULT WINAPI HTMLSelectionObject_GetTypeInfoCount(IHTMLSelectionObject *iface, UINT *pctinfo)
106 {
107     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
108     FIXME("(%p)->(%p)\n", This, pctinfo);
109     return E_NOTIMPL;
110 }
111
112 static HRESULT WINAPI HTMLSelectionObject_GetTypeInfo(IHTMLSelectionObject *iface, UINT iTInfo,
113                                               LCID lcid, ITypeInfo **ppTInfo)
114 {
115     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
116     FIXME("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
117     return E_NOTIMPL;
118 }
119
120 static HRESULT WINAPI HTMLSelectionObject_GetIDsOfNames(IHTMLSelectionObject *iface, REFIID riid,
121                                                 LPOLESTR *rgszNames, UINT cNames,
122                                                 LCID lcid, DISPID *rgDispId)
123 {
124     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
125     FIXME("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames,
126           lcid, rgDispId);
127     return E_NOTIMPL;
128 }
129
130 static HRESULT WINAPI HTMLSelectionObject_Invoke(IHTMLSelectionObject *iface, DISPID dispIdMember,
131                             REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS *pDispParams,
132                             VARIANT *pVarResult, EXCEPINFO *pExcepInfo, UINT *puArgErr)
133 {
134     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
135     FIXME("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
136           lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
137     return E_NOTIMPL;
138 }
139
140 static HRESULT WINAPI HTMLSelectionObject_createRange(IHTMLSelectionObject *iface, IDispatch **range)
141 {
142     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
143     IHTMLTxtRange *range_obj = NULL;
144     nsIDOMRange *nsrange = NULL;
145     HRESULT hres;
146
147     TRACE("(%p)->(%p)\n", This, range);
148
149     if(This->nsselection) {
150         PRInt32 nsrange_cnt = 0;
151         nsresult nsres;
152
153         nsISelection_GetRangeCount(This->nsselection, &nsrange_cnt);
154         if(!nsrange_cnt) {
155             nsIDOMHTMLElement *nsbody = NULL;
156
157             TRACE("nsrange_cnt = 0\n");
158
159             if(!This->doc->nsdoc) {
160                 WARN("nsdoc is NULL\n");
161                 return E_UNEXPECTED;
162             }
163
164             nsres = nsIDOMHTMLDocument_GetBody(This->doc->nsdoc, &nsbody);
165             if(NS_FAILED(nsres) || !nsbody) {
166                 ERR("Could not get body: %08x\n", nsres);
167                 return E_FAIL;
168             }
169
170             nsres = nsISelection_Collapse(This->nsselection, (nsIDOMNode*)nsbody, 0);
171             nsIDOMHTMLElement_Release(nsbody);
172             if(NS_FAILED(nsres))
173                 ERR("Collapse failed: %08x\n", nsres);
174         }else if(nsrange_cnt > 1) {
175             FIXME("range_cnt = %d\n", nsrange_cnt);
176         }
177
178         nsres = nsISelection_GetRangeAt(This->nsselection, 0, &nsrange);
179         if(NS_FAILED(nsres))
180             ERR("GetRangeAt failed: %08x\n", nsres);
181     }
182
183     hres = HTMLTxtRange_Create(This->doc, nsrange, &range_obj);
184
185     if (nsrange) nsIDOMRange_Release(nsrange);
186     *range = (IDispatch*)range_obj;
187     return hres;
188 }
189
190 static HRESULT WINAPI HTMLSelectionObject_empty(IHTMLSelectionObject *iface)
191 {
192     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
193     FIXME("(%p)\n", This);
194     return E_NOTIMPL;
195 }
196
197 static HRESULT WINAPI HTMLSelectionObject_clear(IHTMLSelectionObject *iface)
198 {
199     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
200     FIXME("(%p)\n", This);
201     return E_NOTIMPL;
202 }
203
204 static HRESULT WINAPI HTMLSelectionObject_get_type(IHTMLSelectionObject *iface, BSTR *p)
205 {
206     HTMLSelectionObject *This = impl_from_IHTMLSelectionObject(iface);
207     PRBool collapsed = TRUE;
208
209     static const WCHAR wszNone[] = {'N','o','n','e',0};
210     static const WCHAR wszText[] = {'T','e','x','t',0};
211
212     TRACE("(%p)->(%p)\n", This, p);
213
214     if(This->nsselection)
215         nsISelection_GetIsCollapsed(This->nsselection, &collapsed);
216
217     *p = SysAllocString(collapsed ? wszNone : wszText); /* FIXME: control */
218     TRACE("ret %s\n", debugstr_w(*p));
219     return S_OK;
220 }
221
222 static const IHTMLSelectionObjectVtbl HTMLSelectionObjectVtbl = {
223     HTMLSelectionObject_QueryInterface,
224     HTMLSelectionObject_AddRef,
225     HTMLSelectionObject_Release,
226     HTMLSelectionObject_GetTypeInfoCount,
227     HTMLSelectionObject_GetTypeInfo,
228     HTMLSelectionObject_GetIDsOfNames,
229     HTMLSelectionObject_Invoke,
230     HTMLSelectionObject_createRange,
231     HTMLSelectionObject_empty,
232     HTMLSelectionObject_clear,
233     HTMLSelectionObject_get_type
234 };
235
236 HRESULT HTMLSelectionObject_Create(HTMLDocumentNode *doc, nsISelection *nsselection, IHTMLSelectionObject **ret)
237 {
238     HTMLSelectionObject *selection;
239
240     selection = heap_alloc(sizeof(HTMLSelectionObject));
241     if(!selection)
242         return E_OUTOFMEMORY;
243
244     selection->IHTMLSelectionObject_iface.lpVtbl = &HTMLSelectionObjectVtbl;
245     selection->ref = 1;
246     selection->nsselection = nsselection; /* We shouldn't call AddRef here */
247
248     selection->doc = doc;
249     list_add_head(&doc->selection_list, &selection->entry);
250
251     *ret = &selection->IHTMLSelectionObject_iface;
252     return S_OK;
253 }
254
255 void detach_selection(HTMLDocumentNode *This)
256 {
257     HTMLSelectionObject *iter;
258
259     LIST_FOR_EACH_ENTRY(iter, &This->selection_list, HTMLSelectionObject, entry) {
260         iter->doc = NULL;
261     }
262 }