Include msxml2.h rather than msxml.h and xmldom.h.
[wine] / dlls / msxml3 / nodelist.c
1 /*
2  *    Node list implementation
3  *
4  * Copyright 2005 Mike McCormack
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  */
20
21 #define COBJMACROS
22
23 #include "config.h"
24
25 #include <stdarg.h>
26 #include "windef.h"
27 #include "winbase.h"
28 #include "winuser.h"
29 #include "ole2.h"
30 #include "msxml2.h"
31
32 #include "msxml_private.h"
33
34 #include "wine/debug.h"
35
36 WINE_DEFAULT_DEBUG_CHANNEL(msxml);
37
38 #ifdef HAVE_LIBXML2
39
40 #ifdef HAVE_LIBXSLT
41
42 #ifdef HAVE_LIBXSLT_PATTERN_H
43 #include <libxslt/pattern.h>
44 #endif
45 #ifdef HAVE_LIBXSLT_TRANSFORM_H
46 #include <libxslt/transform.h>
47 #endif
48
49 struct xslt_info {
50     xsltTransformContextPtr ctxt;
51     xsltCompMatchPtr pattern;
52     xsltStylesheetPtr sheet;
53 };
54
55 static void xlst_info_init( struct xslt_info *info )
56 {
57     info->ctxt = NULL;
58     info->pattern = NULL;
59     info->sheet = NULL;
60 }
61
62 static int create_xslt_parser( struct xslt_info *info, xmlNodePtr node, const xmlChar *str )
63 {
64     info->sheet = xsltNewStylesheet();
65     if (!info->sheet)
66         return 0;
67
68     info->ctxt = xsltNewTransformContext( info->sheet, node->doc );
69     if (!info->ctxt)
70         return 0;
71
72     info->pattern = xsltCompilePattern( str, node->doc,
73                                         node, info->sheet, info->ctxt );
74     if (!info->pattern)
75         return 0;
76     return 1;
77 }
78  
79 void free_xslt_info( struct xslt_info *info )
80 {
81     if (info->pattern)
82         xsltFreeCompMatchList( info->pattern );
83     if (info->sheet)
84         xsltFreeStylesheet( info->sheet );
85     if (info->ctxt)
86         xsltFreeTransformContext( info->ctxt );
87 }
88
89 static HRESULT xslt_next_match( struct xslt_info *info, xmlNodePtr *node )
90 {
91     if (!info->ctxt)
92         return S_FALSE;
93  
94     /* make sure that the current element matches the pattern */
95     while ( *node )
96     {
97         int r;
98
99         r = xsltTestCompMatchList( info->ctxt, *node, info->pattern );
100         if ( 1 == r )
101         {
102             TRACE("Matched %p (%s)\n", *node, (*node)->name );
103             return S_OK;
104         }
105         if (r != 0)
106         {
107             ERR("Pattern match failed\n");
108             return E_FAIL;
109         }
110         *node = (*node)->next;
111     }
112     return S_OK;
113 }
114
115 #else
116
117 struct xslt_info {
118     /* empty */
119 };
120
121 static void xlst_info_init( struct xslt_info *info )
122 {
123 }
124
125 void free_xslt_info( struct xslt_info *info )
126 {
127 }
128
129 static int create_xslt_parser( struct xslt_info *info, xmlNodePtr node, const xmlChar *str )
130 {
131     MESSAGE("libxslt was missing at compile time\n");
132     return 0;
133 }
134
135 static HRESULT xslt_next_match( struct xslt_info *info, xmlNodePtr *node )
136 {
137     return S_FALSE;
138 }
139
140 #endif
141
142 typedef struct _xmlnodelist
143 {
144     const struct IXMLDOMNodeListVtbl *lpVtbl;
145     LONG ref;
146     xmlNodePtr node;
147     xmlNodePtr current;
148     struct xslt_info xinfo;
149 } xmlnodelist;
150
151 static inline xmlnodelist *impl_from_IXMLDOMNodeList( IXMLDOMNodeList *iface )
152 {
153     return (xmlnodelist *)((char*)iface - FIELD_OFFSET(xmlnodelist, lpVtbl));
154 }
155
156 static HRESULT WINAPI xmlnodelist_QueryInterface(
157     IXMLDOMNodeList *iface,
158     REFIID riid,
159     void** ppvObject )
160 {
161     TRACE("%p %s %p\n", iface, debugstr_guid(riid), ppvObject);
162
163     if ( IsEqualGUID( riid, &IID_IUnknown ) ||
164          IsEqualGUID( riid, &IID_IDispatch ) ||
165          IsEqualGUID( riid, &IID_IXMLDOMNodeList ) )
166     {
167         *ppvObject = iface;
168     }
169     else
170         return E_NOINTERFACE;
171
172     IXMLDOMNodeList_AddRef( iface );
173
174     return S_OK;
175 }
176
177 static ULONG WINAPI xmlnodelist_AddRef(
178     IXMLDOMNodeList *iface )
179 {
180     xmlnodelist *This = impl_from_IXMLDOMNodeList( iface );
181     return InterlockedIncrement( &This->ref );
182 }
183
184 static ULONG WINAPI xmlnodelist_Release(
185     IXMLDOMNodeList *iface )
186 {
187     xmlnodelist *This = impl_from_IXMLDOMNodeList( iface );
188     ULONG ref;
189
190     ref = InterlockedDecrement( &This->ref );
191     if ( ref == 0 )
192     {
193         free_xslt_info( &This->xinfo );
194         HeapFree( GetProcessHeap(), 0, This );
195     }
196
197     return ref;
198 }
199
200 static HRESULT WINAPI xmlnodelist_GetTypeInfoCount(
201     IXMLDOMNodeList *iface,
202     UINT* pctinfo )
203 {
204     FIXME("\n");
205     return E_NOTIMPL;
206 }
207
208 static HRESULT WINAPI xmlnodelist_GetTypeInfo(
209     IXMLDOMNodeList *iface,
210     UINT iTInfo,
211     LCID lcid,
212     ITypeInfo** ppTInfo )
213 {
214     FIXME("\n");
215     return E_NOTIMPL;
216 }
217
218 static HRESULT WINAPI xmlnodelist_GetIDsOfNames(
219     IXMLDOMNodeList *iface,
220     REFIID riid,
221     LPOLESTR* rgszNames,
222     UINT cNames,
223     LCID lcid,
224     DISPID* rgDispId )
225 {
226     FIXME("\n");
227     return E_NOTIMPL;
228 }
229
230 static HRESULT WINAPI xmlnodelist_Invoke(
231     IXMLDOMNodeList *iface,
232     DISPID dispIdMember,
233     REFIID riid,
234     LCID lcid,
235     WORD wFlags,
236     DISPPARAMS* pDispParams,
237     VARIANT* pVarResult,
238     EXCEPINFO* pExcepInfo,
239     UINT* puArgErr )
240 {
241     FIXME("\n");
242     return E_NOTIMPL;
243 }
244
245 static HRESULT WINAPI xmlnodelist_get_item(
246         IXMLDOMNodeList* iface,
247         long index,
248         IXMLDOMNode** listItem)
249 {
250     xmlnodelist *This = impl_from_IXMLDOMNodeList( iface );
251     xmlNodePtr curr;
252     long nodeIndex = 0;
253     HRESULT r;
254
255     TRACE("%p %ld\n", This, index);
256  
257     *listItem = NULL;
258
259     if (index < 0)
260         return S_FALSE;
261
262     curr = This->node;
263
264     while(curr)
265     {
266         r = xslt_next_match( &This->xinfo, &curr );
267         if(FAILED(r) || !curr) return S_FALSE;
268         if(nodeIndex++ == index) break;
269         curr = curr->next;
270     }
271     if(!curr) return S_FALSE;
272
273     *listItem = create_node( curr );
274
275     return S_OK;
276 }
277
278 static HRESULT WINAPI xmlnodelist_get_length(
279         IXMLDOMNodeList* iface,
280         long* listLength)
281 {
282
283     xmlNodePtr curr;
284     long nodeCount = 0;
285     HRESULT r;
286
287     xmlnodelist *This = impl_from_IXMLDOMNodeList( iface );
288
289     TRACE("%p\n", This);
290
291     if (This->node == NULL) {
292         *listLength = 0;
293         return S_OK;
294     }
295         
296     for(curr = This->node; curr; curr = curr->next)
297     {
298         r = xslt_next_match( &This->xinfo, &curr );
299         if(FAILED(r) || !curr) break;
300         nodeCount++;
301     }
302
303     *listLength = nodeCount;
304     return S_OK;
305 }
306
307 static HRESULT WINAPI xmlnodelist_nextNode(
308         IXMLDOMNodeList* iface,
309         IXMLDOMNode** nextItem)
310 {
311     xmlnodelist *This = impl_from_IXMLDOMNodeList( iface );
312     HRESULT r;
313
314     TRACE("%p %p\n", This, nextItem );
315
316     r = xslt_next_match( &This->xinfo, &This->current );
317     if (FAILED(r) )
318         return r;
319
320     if (!This->current)
321         return S_FALSE;
322
323     *nextItem = create_node( This->current );
324     This->current = This->current->next;
325     return S_OK;
326 }
327
328 static HRESULT WINAPI xmlnodelist_reset(
329         IXMLDOMNodeList* iface)
330 {
331     xmlnodelist *This = impl_from_IXMLDOMNodeList( iface );
332
333     TRACE("%p\n", This);
334     This->current = This->node;
335     return S_OK;
336 }
337
338 static HRESULT WINAPI xmlnodelist__newEnum(
339         IXMLDOMNodeList* iface,
340         IUnknown** ppUnk)
341 {
342     FIXME("\n");
343     return E_NOTIMPL;
344 }
345
346
347 static const struct IXMLDOMNodeListVtbl xmlnodelist_vtbl =
348 {
349     xmlnodelist_QueryInterface,
350     xmlnodelist_AddRef,
351     xmlnodelist_Release,
352     xmlnodelist_GetTypeInfoCount,
353     xmlnodelist_GetTypeInfo,
354     xmlnodelist_GetIDsOfNames,
355     xmlnodelist_Invoke,
356     xmlnodelist_get_item,
357     xmlnodelist_get_length,
358     xmlnodelist_nextNode,
359     xmlnodelist_reset,
360     xmlnodelist__newEnum,
361 };
362
363 static xmlnodelist *new_nodelist( xmlNodePtr node )
364 {
365     xmlnodelist *nodelist;
366
367     nodelist = HeapAlloc( GetProcessHeap(), 0, sizeof *nodelist );
368     if ( !nodelist )
369         return NULL;
370
371     nodelist->lpVtbl = &xmlnodelist_vtbl;
372     nodelist->ref = 1;
373     nodelist->node = node;
374     nodelist->current = node;
375     xlst_info_init( &nodelist->xinfo );
376
377     return nodelist;
378 }
379
380 IXMLDOMNodeList* create_nodelist( xmlNodePtr node )
381 {
382     xmlnodelist *nodelist = new_nodelist( node );
383     if (!node)
384         return NULL;
385     return (IXMLDOMNodeList*) &nodelist->lpVtbl;
386 }
387
388 IXMLDOMNodeList* create_filtered_nodelist( xmlNodePtr node, const xmlChar *str )
389 {
390     xmlnodelist *This = new_nodelist( node );
391
392     if (create_xslt_parser( &This->xinfo, node, str ))
393         return (IXMLDOMNodeList*) &This->lpVtbl;
394
395     IXMLDOMNodeList_Release( (IXMLDOMNodeList*) &This->lpVtbl );
396     return NULL;
397 }
398
399 #endif