avifil32: Reorder some code to avoid forward declarations for the IGetFrameVtbl methods.
[wine] / dlls / msctf / documentmgr.c
1 /*
2  *  ITfDocumentMgr implementation
3  *
4  *  Copyright 2009 Aric Stewart, CodeWeavers
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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 #include "config.h"
22
23 #include <stdarg.h>
24
25 #define COBJMACROS
26
27 #include "wine/debug.h"
28 #include "windef.h"
29 #include "winbase.h"
30 #include "winreg.h"
31 #include "winuser.h"
32 #include "shlwapi.h"
33 #include "winerror.h"
34 #include "objbase.h"
35
36 #include "wine/unicode.h"
37
38 #include "msctf.h"
39 #include "msctf_internal.h"
40
41 WINE_DEFAULT_DEBUG_CHANNEL(msctf);
42
43 typedef struct tagDocumentMgr {
44     ITfDocumentMgr ITfDocumentMgr_iface;
45     ITfSource ITfSource_iface;
46     LONG refCount;
47
48     /* Aggregation */
49     ITfCompartmentMgr  *CompartmentMgr;
50
51     ITfContext*  contextStack[2]; /* limit of 2 contexts */
52     ITfThreadMgrEventSink* ThreadMgrSink;
53 } DocumentMgr;
54
55 typedef struct tagEnumTfContext {
56     IEnumTfContexts IEnumTfContexts_iface;
57     LONG refCount;
58
59     DWORD   index;
60     DocumentMgr *docmgr;
61 } EnumTfContext;
62
63 static HRESULT EnumTfContext_Constructor(DocumentMgr* mgr, IEnumTfContexts **ppOut);
64
65 static inline DocumentMgr *impl_from_ITfDocumentMgr(ITfDocumentMgr *iface)
66 {
67     return CONTAINING_RECORD(iface, DocumentMgr, ITfDocumentMgr_iface);
68 }
69
70 static inline DocumentMgr *impl_from_ITfSource(ITfSource *iface)
71 {
72     return CONTAINING_RECORD(iface, DocumentMgr, ITfSource_iface);
73 }
74
75 static inline EnumTfContext *impl_from_IEnumTfContexts(IEnumTfContexts *iface)\
76 {
77     return CONTAINING_RECORD(iface, EnumTfContext, IEnumTfContexts_iface);
78 }
79
80 static void DocumentMgr_Destructor(DocumentMgr *This)
81 {
82     ITfThreadMgr *tm;
83     TRACE("destroying %p\n", This);
84
85     TF_GetThreadMgr(&tm);
86     ThreadMgr_OnDocumentMgrDestruction(tm, &This->ITfDocumentMgr_iface);
87
88     if (This->contextStack[0])
89         ITfContext_Release(This->contextStack[0]);
90     if (This->contextStack[1])
91         ITfContext_Release(This->contextStack[1]);
92     CompartmentMgr_Destructor(This->CompartmentMgr);
93     HeapFree(GetProcessHeap(),0,This);
94 }
95
96 static HRESULT WINAPI DocumentMgr_QueryInterface(ITfDocumentMgr *iface, REFIID iid, LPVOID *ppvOut)
97 {
98     DocumentMgr *This = impl_from_ITfDocumentMgr(iface);
99     *ppvOut = NULL;
100
101     if (IsEqualIID(iid, &IID_IUnknown) || IsEqualIID(iid, &IID_ITfDocumentMgr))
102     {
103         *ppvOut = This;
104     }
105     else if (IsEqualIID(iid, &IID_ITfSource))
106     {
107         *ppvOut = &This->ITfSource_iface;
108     }
109     else if (IsEqualIID(iid, &IID_ITfCompartmentMgr))
110     {
111         *ppvOut = This->CompartmentMgr;
112     }
113
114     if (*ppvOut)
115     {
116         IUnknown_AddRef(iface);
117         return S_OK;
118     }
119
120     WARN("unsupported interface: %s\n", debugstr_guid(iid));
121     return E_NOINTERFACE;
122 }
123
124 static ULONG WINAPI DocumentMgr_AddRef(ITfDocumentMgr *iface)
125 {
126     DocumentMgr *This = impl_from_ITfDocumentMgr(iface);
127     return InterlockedIncrement(&This->refCount);
128 }
129
130 static ULONG WINAPI DocumentMgr_Release(ITfDocumentMgr *iface)
131 {
132     DocumentMgr *This = impl_from_ITfDocumentMgr(iface);
133     ULONG ret;
134
135     ret = InterlockedDecrement(&This->refCount);
136     if (ret == 0)
137         DocumentMgr_Destructor(This);
138     return ret;
139 }
140
141 /*****************************************************
142  * ITfDocumentMgr functions
143  *****************************************************/
144 static HRESULT WINAPI DocumentMgr_CreateContext(ITfDocumentMgr *iface,
145         TfClientId tidOwner,
146         DWORD dwFlags, IUnknown *punk, ITfContext **ppic,
147         TfEditCookie *pecTextStore)
148 {
149     DocumentMgr *This = impl_from_ITfDocumentMgr(iface);
150     TRACE("(%p) 0x%x 0x%x %p %p %p\n",This,tidOwner,dwFlags,punk,ppic,pecTextStore);
151     return Context_Constructor(tidOwner, punk, iface, ppic, pecTextStore);
152 }
153
154 static HRESULT WINAPI DocumentMgr_Push(ITfDocumentMgr *iface, ITfContext *pic)
155 {
156     DocumentMgr *This = impl_from_ITfDocumentMgr(iface);
157     ITfContext *check;
158
159     TRACE("(%p) %p\n",This,pic);
160
161     if (This->contextStack[1])  /* FUll */
162         return TF_E_STACKFULL;
163
164     if (!pic || FAILED(IUnknown_QueryInterface(pic,&IID_ITfContext,(LPVOID*) &check)))
165         return E_INVALIDARG;
166
167     if (This->contextStack[0] == NULL)
168         ITfThreadMgrEventSink_OnInitDocumentMgr(This->ThreadMgrSink,iface);
169
170     This->contextStack[1] = This->contextStack[0];
171     This->contextStack[0] = check;
172
173     Context_Initialize(check, iface);
174     ITfThreadMgrEventSink_OnPushContext(This->ThreadMgrSink,check);
175
176     return S_OK;
177 }
178
179 static HRESULT WINAPI DocumentMgr_Pop(ITfDocumentMgr *iface, DWORD dwFlags)
180 {
181     DocumentMgr *This = impl_from_ITfDocumentMgr(iface);
182     TRACE("(%p) 0x%x\n",This,dwFlags);
183
184     if (dwFlags == TF_POPF_ALL)
185     {
186         if (This->contextStack[0])
187         {
188             ITfThreadMgrEventSink_OnPopContext(This->ThreadMgrSink,This->contextStack[0]);
189             ITfContext_Release(This->contextStack[0]);
190             Context_Uninitialize(This->contextStack[0]);
191         }
192         if (This->contextStack[1])
193         {
194             ITfThreadMgrEventSink_OnPopContext(This->ThreadMgrSink,This->contextStack[1]);
195             ITfContext_Release(This->contextStack[1]);
196             Context_Uninitialize(This->contextStack[1]);
197         }
198         This->contextStack[0] = This->contextStack[1] = NULL;
199         ITfThreadMgrEventSink_OnUninitDocumentMgr(This->ThreadMgrSink, iface);
200         return S_OK;
201     }
202
203     if (dwFlags)
204         return E_INVALIDARG;
205
206     if (This->contextStack[1] == NULL) /* Cannot pop last context */
207         return E_FAIL;
208
209     ITfThreadMgrEventSink_OnPopContext(This->ThreadMgrSink,This->contextStack[0]);
210     ITfContext_Release(This->contextStack[0]);
211     Context_Uninitialize(This->contextStack[0]);
212     This->contextStack[0] = This->contextStack[1];
213     This->contextStack[1] = NULL;
214
215     if (This->contextStack[0] == NULL)
216         ITfThreadMgrEventSink_OnUninitDocumentMgr(This->ThreadMgrSink, iface);
217
218     return S_OK;
219 }
220
221 static HRESULT WINAPI DocumentMgr_GetTop(ITfDocumentMgr *iface, ITfContext **ppic)
222 {
223     DocumentMgr *This = impl_from_ITfDocumentMgr(iface);
224     TRACE("(%p)\n",This);
225     if (!ppic)
226         return E_INVALIDARG;
227
228     if (This->contextStack[0])
229         ITfContext_AddRef(This->contextStack[0]);
230
231     *ppic = This->contextStack[0];
232
233     return S_OK;
234 }
235
236 static HRESULT WINAPI DocumentMgr_GetBase(ITfDocumentMgr *iface, ITfContext **ppic)
237 {
238     DocumentMgr *This = impl_from_ITfDocumentMgr(iface);
239     ITfContext *tgt;
240
241     TRACE("(%p)\n",This);
242     if (!ppic)
243         return E_INVALIDARG;
244
245     if (This->contextStack[1])
246         tgt = This->contextStack[1];
247     else
248         tgt = This->contextStack[0];
249
250     if (tgt)
251         ITfContext_AddRef(tgt);
252
253     *ppic = tgt;
254
255     return S_OK;
256 }
257
258 static HRESULT WINAPI DocumentMgr_EnumContexts(ITfDocumentMgr *iface, IEnumTfContexts **ppEnum)
259 {
260     DocumentMgr *This = impl_from_ITfDocumentMgr(iface);
261     TRACE("(%p) %p\n",This,ppEnum);
262     return EnumTfContext_Constructor(This, ppEnum);
263 }
264
265 static const ITfDocumentMgrVtbl DocumentMgr_DocumentMgrVtbl =
266 {
267     DocumentMgr_QueryInterface,
268     DocumentMgr_AddRef,
269     DocumentMgr_Release,
270
271     DocumentMgr_CreateContext,
272     DocumentMgr_Push,
273     DocumentMgr_Pop,
274     DocumentMgr_GetTop,
275     DocumentMgr_GetBase,
276     DocumentMgr_EnumContexts
277 };
278
279
280 static HRESULT WINAPI Source_QueryInterface(ITfSource *iface, REFIID iid, LPVOID *ppvOut)
281 {
282     DocumentMgr *This = impl_from_ITfSource(iface);
283     return DocumentMgr_QueryInterface(&This->ITfDocumentMgr_iface, iid, *ppvOut);
284 }
285
286 static ULONG WINAPI Source_AddRef(ITfSource *iface)
287 {
288     DocumentMgr *This = impl_from_ITfSource(iface);
289     return DocumentMgr_AddRef(&This->ITfDocumentMgr_iface);
290 }
291
292 static ULONG WINAPI Source_Release(ITfSource *iface)
293 {
294     DocumentMgr *This = impl_from_ITfSource(iface);
295     return DocumentMgr_Release(&This->ITfDocumentMgr_iface);
296 }
297
298 /*****************************************************
299  * ITfSource functions
300  *****************************************************/
301 static HRESULT WINAPI DocumentMgrSource_AdviseSink(ITfSource *iface,
302         REFIID riid, IUnknown *punk, DWORD *pdwCookie)
303 {
304     DocumentMgr *This = impl_from_ITfSource(iface);
305     FIXME("STUB:(%p)\n",This);
306     return E_NOTIMPL;
307 }
308
309 static HRESULT WINAPI DocumentMgrSource_UnadviseSink(ITfSource *iface, DWORD pdwCookie)
310 {
311     DocumentMgr *This = impl_from_ITfSource(iface);
312     FIXME("STUB:(%p)\n",This);
313     return E_NOTIMPL;
314 }
315
316 static const ITfSourceVtbl DocumentMgr_SourceVtbl =
317 {
318     Source_QueryInterface,
319     Source_AddRef,
320     Source_Release,
321
322     DocumentMgrSource_AdviseSink,
323     DocumentMgrSource_UnadviseSink,
324 };
325
326 HRESULT DocumentMgr_Constructor(ITfThreadMgrEventSink *ThreadMgrSink, ITfDocumentMgr **ppOut)
327 {
328     DocumentMgr *This;
329
330     This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(DocumentMgr));
331     if (This == NULL)
332         return E_OUTOFMEMORY;
333
334     This->ITfDocumentMgr_iface.lpVtbl = &DocumentMgr_DocumentMgrVtbl;
335     This->ITfSource_iface.lpVtbl = &DocumentMgr_SourceVtbl;
336     This->refCount = 1;
337     This->ThreadMgrSink = ThreadMgrSink;
338
339     CompartmentMgr_Constructor((IUnknown*)This, &IID_IUnknown, (IUnknown**)&This->CompartmentMgr);
340
341     TRACE("returning %p\n", This);
342     *ppOut = &This->ITfDocumentMgr_iface;
343     return S_OK;
344 }
345
346 /**************************************************
347  * IEnumTfContexts implementaion
348  **************************************************/
349 static void EnumTfContext_Destructor(EnumTfContext *This)
350 {
351     TRACE("destroying %p\n", This);
352     HeapFree(GetProcessHeap(),0,This);
353 }
354
355 static HRESULT WINAPI EnumTfContext_QueryInterface(IEnumTfContexts *iface, REFIID iid, LPVOID *ppvOut)
356 {
357     EnumTfContext *This = impl_from_IEnumTfContexts(iface);
358     *ppvOut = NULL;
359
360     if (IsEqualIID(iid, &IID_IUnknown) || IsEqualIID(iid, &IID_IEnumTfContexts))
361     {
362         *ppvOut = This;
363     }
364
365     if (*ppvOut)
366     {
367         IUnknown_AddRef(iface);
368         return S_OK;
369     }
370
371     WARN("unsupported interface: %s\n", debugstr_guid(iid));
372     return E_NOINTERFACE;
373 }
374
375 static ULONG WINAPI EnumTfContext_AddRef(IEnumTfContexts *iface)
376 {
377     EnumTfContext *This = impl_from_IEnumTfContexts(iface);
378     return InterlockedIncrement(&This->refCount);
379 }
380
381 static ULONG WINAPI EnumTfContext_Release(IEnumTfContexts *iface)
382 {
383     EnumTfContext *This = impl_from_IEnumTfContexts(iface);
384     ULONG ret;
385
386     ret = InterlockedDecrement(&This->refCount);
387     if (ret == 0)
388         EnumTfContext_Destructor(This);
389     return ret;
390 }
391
392 static HRESULT WINAPI EnumTfContext_Next(IEnumTfContexts *iface,
393     ULONG ulCount, ITfContext **rgContext, ULONG *pcFetched)
394 {
395     EnumTfContext *This = impl_from_IEnumTfContexts(iface);
396     ULONG fetched = 0;
397
398     TRACE("(%p)\n",This);
399
400     if (rgContext == NULL) return E_POINTER;
401
402     while (fetched < ulCount)
403     {
404         if (This->index > 1)
405             break;
406
407         if (!This->docmgr->contextStack[This->index])
408             break;
409
410         *rgContext = This->docmgr->contextStack[This->index];
411         ITfContext_AddRef(*rgContext);
412
413         ++This->index;
414         ++fetched;
415         ++rgContext;
416     }
417
418     if (pcFetched) *pcFetched = fetched;
419     return fetched == ulCount ? S_OK : S_FALSE;
420 }
421
422 static HRESULT WINAPI EnumTfContext_Skip( IEnumTfContexts* iface, ULONG celt)
423 {
424     EnumTfContext *This = impl_from_IEnumTfContexts(iface);
425     TRACE("(%p)\n",This);
426     This->index += celt;
427     return S_OK;
428 }
429
430 static HRESULT WINAPI EnumTfContext_Reset( IEnumTfContexts* iface)
431 {
432     EnumTfContext *This = impl_from_IEnumTfContexts(iface);
433     TRACE("(%p)\n",This);
434     This->index = 0;
435     return S_OK;
436 }
437
438 static HRESULT WINAPI EnumTfContext_Clone( IEnumTfContexts *iface,
439     IEnumTfContexts **ppenum)
440 {
441     EnumTfContext *This = impl_from_IEnumTfContexts(iface);
442     HRESULT res;
443
444     TRACE("(%p)\n",This);
445
446     if (ppenum == NULL) return E_POINTER;
447
448     res = EnumTfContext_Constructor(This->docmgr, ppenum);
449     if (SUCCEEDED(res))
450     {
451         EnumTfContext *new_This = impl_from_IEnumTfContexts(*ppenum);
452         new_This->index = This->index;
453     }
454     return res;
455 }
456
457 static const IEnumTfContextsVtbl IEnumTfContexts_Vtbl ={
458     EnumTfContext_QueryInterface,
459     EnumTfContext_AddRef,
460     EnumTfContext_Release,
461
462     EnumTfContext_Clone,
463     EnumTfContext_Next,
464     EnumTfContext_Reset,
465     EnumTfContext_Skip
466 };
467
468 static HRESULT EnumTfContext_Constructor(DocumentMgr *mgr, IEnumTfContexts **ppOut)
469 {
470     EnumTfContext *This;
471
472     This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(EnumTfContext));
473     if (This == NULL)
474         return E_OUTOFMEMORY;
475
476     This->IEnumTfContexts_iface.lpVtbl = &IEnumTfContexts_Vtbl;
477     This->refCount = 1;
478     This->docmgr = mgr;
479
480     TRACE("returning %p\n", This);
481     *ppOut = &This->IEnumTfContexts_iface;
482     return S_OK;
483 }