Flesh out the remaining keys in RegisterProduct.
[wine] / dlls / msi / dialog.c
1 /*
2  * Implementation of the Microsoft Installer (msi.dll)
3  *
4  * Copyright 2005 Mike McCormack for 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  */
20
21 #define COBJMACROS
22
23 #include <stdarg.h>
24
25 #include "windef.h"
26 #include "winbase.h"
27 #include "winuser.h"
28 #include "winnls.h"
29 #include "wingdi.h"
30 #include "msi.h"
31 #include "msipriv.h"
32 #include "msidefs.h"
33 #include "ocidl.h"
34 #include "olectl.h"
35 #include "richedit.h"
36
37 #include "wine/debug.h"
38 #include "wine/unicode.h"
39
40 #include "action.h"
41
42 WINE_DEFAULT_DEBUG_CHANNEL(msi);
43
44
45 struct msi_control_tag;
46 typedef struct msi_control_tag msi_control;
47 typedef UINT (*msi_handler)( msi_dialog *, msi_control *, WPARAM );
48
49 struct msi_control_tag
50 {
51     struct msi_control_tag *next;
52     HWND hwnd;
53     msi_handler handler;
54     LPWSTR property;
55     LPWSTR value;
56     IPicture *pic;
57     WCHAR name[1];
58 };
59
60 typedef struct msi_font_tag
61 {
62     struct msi_font_tag *next;
63     HFONT hfont;
64     WCHAR name[1];
65 } msi_font;
66
67 struct msi_dialog_tag
68 {
69     MSIPACKAGE *package;
70     msi_dialog_event_handler event_handler;
71     BOOL finished;
72     INT scale;
73     DWORD attributes;
74     HWND hwnd;
75     LPWSTR default_font;
76     msi_font *font_list;
77     msi_control *control_list;
78     WCHAR name[1];
79 };
80
81 typedef UINT (*msi_dialog_control_func)( msi_dialog *dialog, MSIRECORD *rec );
82 struct control_handler 
83 {
84     LPCWSTR control_type;
85     msi_dialog_control_func func;
86 };
87
88 typedef struct
89 {
90     msi_dialog* dialog;
91     msi_control *parent;
92     DWORD       attributes;
93 } radio_button_group_descr;
94
95 const WCHAR szMsiDialogClass[] = {
96     'M','s','i','D','i','a','l','o','g','C','l','o','s','e','C','l','a','s','s',0
97 };
98 const WCHAR szMsiHiddenWindow[] = {
99     'M','s','i','H','i','d','d','e','n','W','i','n','d','o','w',0 };
100 const static WCHAR szStatic[] = { 'S','t','a','t','i','c',0 };
101 const static WCHAR szButton[] = { 'B','U','T','T','O','N', 0 };
102 const static WCHAR szButtonData[] = { 'M','S','I','D','A','T','A',0 };
103 static const WCHAR szText[] = { 'T','e','x','t',0 };
104 static const WCHAR szPushButton[] = { 'P','u','s','h','B','u','t','t','o','n',0 };
105 static const WCHAR szLine[] = { 'L','i','n','e',0 };
106 static const WCHAR szBitmap[] = { 'B','i','t','m','a','p',0 };
107 static const WCHAR szCheckBox[] = { 'C','h','e','c','k','B','o','x',0 };
108 static const WCHAR szScrollableText[] = {
109     'S','c','r','o','l','l','a','b','l','e','T','e','x','t',0 };
110 static const WCHAR szComboBox[] = { 'C','o','m','b','o','B','o','x',0 };
111 static const WCHAR szEdit[] = { 'E','d','i','t',0 };
112 static const WCHAR szMaskedEdit[] = { 'M','a','s','k','e','d','E','d','i','t',0 };
113 static const WCHAR szPathEdit[] = { 'P','a','t','h','E','d','i','t',0 };
114 static const WCHAR szRadioButtonGroup[] = { 
115     'R','a','d','i','o','B','u','t','t','o','n','G','r','o','u','p',0 };
116
117 static UINT msi_dialog_checkbox_handler( msi_dialog *, msi_control *, WPARAM );
118 static void msi_dialog_checkbox_sync_state( msi_dialog *, msi_control * );
119 static UINT msi_dialog_button_handler( msi_dialog *, msi_control *, WPARAM );
120 static UINT msi_dialog_edit_handler( msi_dialog *, msi_control *, WPARAM );
121 static UINT msi_dialog_radiogroup_handler( msi_dialog *, msi_control *, WPARAM param );
122 static UINT msi_dialog_evaluate_control_conditions( msi_dialog *dialog );
123 static LRESULT WINAPI MSIRadioGroup_WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam);
124
125
126 /* dialog sequencing */
127
128 #define WM_MSI_DIALOG_CREATE  (WM_USER+0x100)
129 #define WM_MSI_DIALOG_DESTROY (WM_USER+0x101)
130
131 static DWORD uiThreadId;
132 static HWND hMsiHiddenWindow;
133 static HMODULE hRichedit;
134
135 static INT msi_dialog_scale_unit( msi_dialog *dialog, INT val )
136 {
137     return (dialog->scale * val + 5) / 10;
138 }
139
140 static msi_control *msi_dialog_find_control( msi_dialog *dialog, LPCWSTR name )
141 {
142     msi_control *control;
143
144     for( control = dialog->control_list; control; control = control->next )
145         if( !strcmpW( control->name, name ) ) /* FIXME: case sensitive? */
146             break;
147     return control;
148 }
149
150 static msi_control *msi_dialog_find_control_by_hwnd( msi_dialog *dialog, HWND hwnd )
151 {
152     msi_control *control;
153
154     for( control = dialog->control_list; control; control = control->next )
155         if( hwnd == control->hwnd )
156             break;
157     return control;
158 }
159
160 /*
161  * msi_dialog_get_style
162  *
163  * Extract the {\style} string from the front of the text to display and
164  *  update the pointer.
165  */
166 static LPWSTR msi_dialog_get_style( LPCWSTR *text )
167 {
168     LPWSTR ret = NULL;
169     LPCWSTR p = *text, q;
170     DWORD len;
171
172     if( *p++ != '{' )
173         return ret;
174     q = strchrW( p, '}' );
175     if( !q )
176         return ret;
177     *text = ++q;
178     if( *p++ != '\\' )
179         return ret;
180     len = q - p;
181     
182     ret = HeapAlloc( GetProcessHeap(), 0, len*sizeof(WCHAR) );
183     if( !ret )
184         return ret;
185     memcpy( ret, p, len*sizeof(WCHAR) );
186     ret[len-1] = 0;
187     return ret;
188 }
189
190 static UINT msi_dialog_add_font( MSIRECORD *rec, LPVOID param )
191 {
192     msi_dialog *dialog = param;
193     msi_font *font;
194     LPCWSTR face, name;
195     LOGFONTW lf;
196     INT style;
197     HDC hdc;
198
199     /* create a font and add it to the list */
200     name = MSI_RecordGetString( rec, 1 );
201     font = HeapAlloc( GetProcessHeap(), 0,
202                       sizeof *font + strlenW( name )*sizeof (WCHAR) );
203     strcpyW( font->name, name );
204     font->next = dialog->font_list;
205     dialog->font_list = font;
206
207     memset( &lf, 0, sizeof lf );
208     face = MSI_RecordGetString( rec, 2 );
209     lf.lfHeight = MSI_RecordGetInteger( rec, 3 );
210     style = MSI_RecordGetInteger( rec, 5 );
211     if( style & msidbTextStyleStyleBitsBold )
212         lf.lfWeight = FW_BOLD;
213     if( style & msidbTextStyleStyleBitsItalic )
214         lf.lfItalic = TRUE;
215     if( style & msidbTextStyleStyleBitsUnderline )
216         lf.lfUnderline = TRUE;
217     if( style & msidbTextStyleStyleBitsStrike )
218         lf.lfStrikeOut = TRUE;
219     lstrcpynW( lf.lfFaceName, face, LF_FACESIZE );
220
221     /* adjust the height */
222     hdc = GetDC( dialog->hwnd );
223     if (hdc)
224     {
225         lf.lfHeight = -MulDiv(lf.lfHeight, GetDeviceCaps(hdc, LOGPIXELSY), 72);
226         ReleaseDC( dialog->hwnd, hdc );
227     }
228
229     font->hfont = CreateFontIndirectW( &lf );
230
231     TRACE("Adding font style %s\n", debugstr_w(font->name) );
232
233     return ERROR_SUCCESS;
234 }
235
236 static msi_font *msi_dialog_find_font( msi_dialog *dialog, LPCWSTR name )
237 {
238     msi_font *font;
239
240     for( font = dialog->font_list; font; font = font->next )
241         if( !strcmpW( font->name, name ) )  /* FIXME: case sensitive? */
242             break;
243
244     return font;
245 }
246
247 static UINT msi_dialog_set_font( msi_dialog *dialog, HWND hwnd, LPCWSTR name )
248 {
249     msi_font *font;
250
251     font = msi_dialog_find_font( dialog, name );
252     if( font )
253         SendMessageW( hwnd, WM_SETFONT, (WPARAM) font->hfont, TRUE );
254     else
255         ERR("No font entry for %s\n", debugstr_w(name));
256     return ERROR_SUCCESS;
257 }
258
259 static UINT msi_dialog_build_font_list( msi_dialog *dialog )
260 {
261     static const WCHAR query[] = {
262       'S','E','L','E','C','T',' ','*',' ',
263       'F','R','O','M',' ','`','T','e','x','t','S','t','y','l','e','`',' ',0
264     };
265     UINT r;
266     MSIQUERY *view = NULL;
267
268     TRACE("dialog %p\n", dialog );
269
270     r = MSI_OpenQuery( dialog->package->db, &view, query );
271     if( r != ERROR_SUCCESS )
272         return r;
273
274     r = MSI_IterateRecords( view, NULL, msi_dialog_add_font, dialog );
275     msiobj_release( &view->hdr );
276
277     return r;
278 }
279
280 static msi_control *msi_dialog_create_window( msi_dialog *dialog,
281                 MSIRECORD *rec, LPCWSTR szCls, LPCWSTR name, LPCWSTR text,
282                 DWORD style, HWND parent )
283 {
284     DWORD x, y, width, height;
285     LPWSTR font = NULL, title = NULL;
286     msi_control *control;
287
288     style |= WS_CHILD;
289
290     control = HeapAlloc( GetProcessHeap(), 0,
291                          sizeof *control + strlenW(name)*sizeof(WCHAR) );
292     strcpyW( control->name, name );
293     control->next = dialog->control_list;
294     dialog->control_list = control;
295     control->handler = NULL;
296     control->property = NULL;
297     control->value = NULL;
298     control->pic = NULL;
299
300     x = MSI_RecordGetInteger( rec, 4 );
301     y = MSI_RecordGetInteger( rec, 5 );
302     width = MSI_RecordGetInteger( rec, 6 );
303     height = MSI_RecordGetInteger( rec, 7 );
304
305     x = msi_dialog_scale_unit( dialog, x );
306     y = msi_dialog_scale_unit( dialog, y );
307     width = msi_dialog_scale_unit( dialog, width );
308     height = msi_dialog_scale_unit( dialog, height );
309
310     if( text )
311     {
312         font = msi_dialog_get_style( &text );
313         deformat_string( dialog->package, text, &title );
314     }
315
316     control->hwnd = CreateWindowW( szCls, title, style,
317                           x, y, width, height, parent, NULL, NULL, NULL );
318
319     TRACE("Dialog %s control %s hwnd %p\n",
320            debugstr_w(dialog->name), debugstr_w(text), control->hwnd );
321
322     msi_dialog_set_font( dialog, control->hwnd,
323                          font ? font : dialog->default_font );
324
325     HeapFree( GetProcessHeap(), 0, font );
326     HeapFree( GetProcessHeap(), 0, title );
327
328     return control;
329 }
330
331 /* called from the Control Event subscription code */
332 void msi_dialog_handle_event( msi_dialog* dialog, LPCWSTR control, 
333                               LPCWSTR attribute, MSIRECORD *rec )
334 {
335     msi_control* ctrl;
336     LPCWSTR text;
337
338     ctrl = msi_dialog_find_control( dialog, control );
339     if (!ctrl)
340         return;
341     if( lstrcmpW(attribute, szText) )
342         return;
343     text = MSI_RecordGetString( rec , 1 );
344     SetWindowTextW( ctrl->hwnd, text );
345 }
346
347 static void msi_dialog_map_events(msi_dialog* dialog, LPCWSTR control)
348 {
349     static WCHAR Query[] = {
350         'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
351          '`','E','v','e','n','t','M','a','p','p','i','n','g','`',' ',
352         'W','H','E','R','E',' ',
353          '`','D','i','a','l','o','g','_','`',' ','=',' ','\'','%','s','\'',' ',
354         'A','N','D',' ',
355          '`','C','o','n','t','r','o','l','_','`',' ','=',' ','\'','%','s','\'',0
356     };
357     MSIRECORD *row;
358     LPCWSTR event, attribute;
359
360     row = MSI_QueryGetRecord( dialog->package->db, Query, dialog->name, control );
361     if (!row)
362         return;
363
364     event = MSI_RecordGetString( row, 3 );
365     attribute = MSI_RecordGetString( row, 4 );
366     ControlEvent_SubscribeToEvent( dialog->package, event, control, attribute );
367     msiobj_release( &row->hdr );
368 }
369
370 /* everything except radio buttons */
371 static msi_control *msi_dialog_add_control( msi_dialog *dialog,
372                 MSIRECORD *rec, LPCWSTR szCls, DWORD style )
373 {
374     DWORD attributes;
375     LPCWSTR text, name;
376
377     name = MSI_RecordGetString( rec, 2 );
378     attributes = MSI_RecordGetInteger( rec, 8 );
379     text = MSI_RecordGetString( rec, 10 );
380     if( attributes & 1 )
381         style |= WS_VISIBLE;
382     if( ~attributes & 2 )
383         style |= WS_DISABLED;
384
385     msi_dialog_map_events(dialog, name);
386
387     return msi_dialog_create_window( dialog, rec, szCls, name, text,
388                                      style, dialog->hwnd );
389 }
390
391 static UINT msi_dialog_text_control( msi_dialog *dialog, MSIRECORD *rec )
392 {
393     TRACE("%p %p\n", dialog, rec);
394
395     msi_dialog_add_control( dialog, rec, szStatic, 0 );
396     return ERROR_SUCCESS;
397 }
398
399 static UINT msi_dialog_button_control( msi_dialog *dialog, MSIRECORD *rec )
400 {
401     msi_control *control;
402
403     TRACE("%p %p\n", dialog, rec);
404
405     control = msi_dialog_add_control( dialog, rec, szButton, 0 );
406     control->handler = msi_dialog_button_handler;
407
408     return ERROR_SUCCESS;
409 }
410
411 static LPWSTR msi_get_checkbox_value( msi_dialog *dialog, LPCWSTR prop )
412 {
413     const static WCHAR query[] = {
414         'S','E','L','E','C','T',' ','*',' ',
415         'F','R','O','M',' ','`','C','h','e','c','k','B','o','x',' ','`',
416         'W','H','E','R','E',' ',
417         '`','P','r','o','p','e','r','t','y','`',' ','=',' ',
418         '\'','%','s','\'',0
419     };
420     MSIRECORD *rec = NULL;
421     LPCWSTR val = NULL;
422     LPWSTR ret = NULL;
423
424     /* find if there is a value associated with the checkbox */
425     rec = MSI_QueryGetRecord( dialog->package->db, query, prop );
426     if (!rec)
427         return ret;
428
429     val = MSI_RecordGetString( rec, 2 );
430     if (val)
431     {
432         deformat_string( dialog->package, val, &ret );
433         if( ret && !ret[0] )
434         {
435             HeapFree( GetProcessHeap(), 0, ret );
436             ret = NULL;
437         }
438     }
439     msiobj_release( &rec->hdr );
440     if (ret)
441         return ret;
442
443     ret = load_dynamic_property(dialog->package, prop, NULL);
444     if( ret && !ret[0] )
445     {
446         HeapFree( GetProcessHeap(), 0, ret );
447         ret = NULL;
448     }
449
450     return ret;
451 }
452
453 static UINT msi_dialog_checkbox_control( msi_dialog *dialog, MSIRECORD *rec )
454 {
455     msi_control *control;
456     LPCWSTR prop;
457
458     TRACE("%p %p\n", dialog, rec);
459
460     control = msi_dialog_add_control( dialog, rec, szButton,
461                                       BS_CHECKBOX | BS_MULTILINE );
462     control->handler = msi_dialog_checkbox_handler;
463     prop = MSI_RecordGetString( rec, 9 );
464     if( prop )
465     {
466         control->property = strdupW( prop );
467         control->value = msi_get_checkbox_value( dialog, prop );
468         TRACE("control %s value %s\n", debugstr_w(control->property),
469               debugstr_w(control->value));
470     }
471     msi_dialog_checkbox_sync_state( dialog, control );
472
473     return ERROR_SUCCESS;
474 }
475
476 static UINT msi_dialog_line_control( msi_dialog *dialog, MSIRECORD *rec )
477 {
478     TRACE("%p %p\n", dialog, rec);
479
480     msi_dialog_add_control( dialog, rec, szStatic, SS_ETCHEDHORZ | SS_SUNKEN );
481     return ERROR_SUCCESS;
482 }
483
484 struct msi_streamin_info
485 {
486     LPSTR string;
487     DWORD offset;
488     DWORD length;
489 };
490
491 static DWORD CALLBACK
492 msi_richedit_stream_in( DWORD_PTR arg, LPBYTE buffer, LONG count, LONG *pcb )
493 {
494     struct msi_streamin_info *info = (struct msi_streamin_info*) arg;
495
496     if( (count + info->offset) > info->length )
497         count = info->length - info->offset;
498     memcpy( buffer, &info->string[ info->offset ], count );
499     *pcb = count;
500     info->offset += count;
501
502     TRACE("%ld/%ld\n", info->offset, info->length);
503
504     return 0;
505 }
506
507 static UINT msi_dialog_scrolltext_control( msi_dialog *dialog, MSIRECORD *rec )
508 {
509     const static WCHAR szRichEdit20W[] = {
510         'R','i','c','h','E','d','i','t','2','0','W',0
511     };
512     struct msi_streamin_info info;
513     msi_control *control;
514     LPCWSTR text;
515     EDITSTREAM es;
516     DWORD style;
517
518     style = WS_BORDER | ES_MULTILINE | WS_VSCROLL | ES_READONLY | ES_AUTOVSCROLL;
519     control = msi_dialog_add_control( dialog, rec, szRichEdit20W, style );
520
521     text = MSI_RecordGetString( rec, 10 );
522     info.string = strdupWtoA( text );
523     info.offset = 0;
524     info.length = lstrlenA( info.string ) + 1;
525
526     es.dwCookie = (DWORD_PTR) &info;
527     es.dwError = 0;
528     es.pfnCallback = msi_richedit_stream_in;
529
530     SendMessageW( control->hwnd, EM_STREAMIN, SF_RTF, (LPARAM) &es );
531
532     HeapFree( GetProcessHeap(), 0, info.string );
533
534     return ERROR_SUCCESS;
535 }
536
537 static UINT msi_load_bitmap( MSIDATABASE *db, LPCWSTR name, IPicture **pic )
538 {
539     const static WCHAR query[] = {
540         's','e','l','e','c','t',' ','*',' ',
541         'f','r','o','m',' ','B','i','n','a','r','y',' ',
542         'w','h','e','r','e',' ',
543             '`','N','a','m','e','`',' ','=',' ','\'','%','s','\'',0
544     };
545     MSIRECORD *rec = NULL;
546     IStream *stm = NULL;
547     UINT r;
548
549     rec = MSI_QueryGetRecord( db, query, name );
550     if( !rec )
551         return ERROR_FUNCTION_FAILED;
552
553     r = MSI_RecordGetIStream( rec, 2, &stm );
554     msiobj_release( &rec->hdr );
555     if( r != ERROR_SUCCESS )
556         return r;
557
558     r = OleLoadPicture( stm, 0, TRUE, &IID_IPicture, (LPVOID*) pic );
559     IStream_Release( stm );
560     if( FAILED( r ) )
561         return ERROR_FUNCTION_FAILED;
562
563     return ERROR_SUCCESS;
564 }
565
566 static UINT msi_dialog_bitmap_control( msi_dialog *dialog, MSIRECORD *rec )
567 {
568     IPicture *pic = NULL;
569     msi_control *control;
570     OLE_HANDLE hBitmap = 0;
571     LPCWSTR text;
572     UINT r;
573
574     control = msi_dialog_add_control( dialog, rec, szStatic,
575                             SS_BITMAP | SS_LEFT | SS_CENTERIMAGE );
576     text = MSI_RecordGetString( rec, 10 );
577     r = msi_load_bitmap( dialog->package->db, text, &pic );
578     if( r == ERROR_SUCCESS )
579     {
580         r = IPicture_get_Handle( pic, &hBitmap );
581         if( SUCCEEDED( r ) )
582             SendMessageW( control->hwnd, STM_SETIMAGE, IMAGE_BITMAP, hBitmap );
583         control->pic = pic;
584     }
585     
586     return ERROR_SUCCESS;
587 }
588
589 static UINT msi_dialog_combo_control( msi_dialog *dialog, MSIRECORD *rec )
590 {
591     static const WCHAR szCombo[] = { 'C','O','M','B','O','B','O','X',0 };
592
593     msi_dialog_add_control( dialog, rec, szCombo,
594                             SS_BITMAP | SS_LEFT | SS_CENTERIMAGE );
595     return ERROR_SUCCESS;
596 }
597
598 static UINT msi_dialog_edit_control( msi_dialog *dialog, MSIRECORD *rec )
599 {
600     msi_control *control;
601     LPCWSTR prop;
602     LPWSTR val;
603
604     control = msi_dialog_add_control( dialog, rec, szEdit, WS_BORDER );
605     control->handler = msi_dialog_edit_handler;
606     prop = MSI_RecordGetString( rec, 9 );
607     if( prop )
608         control->property = strdupW( prop );
609     val = load_dynamic_property( dialog->package, control->property, NULL );
610     SetWindowTextW( control->hwnd, val );
611     HeapFree( GetProcessHeap(), 0, val );
612     return ERROR_SUCCESS;
613 }
614
615 /******************** Masked Edit ********************************************/
616
617 #define MASK_MAX_GROUPS 10
618
619 struct msi_mask_group
620 {
621     UINT len;
622     UINT ofs;
623     WCHAR type;
624     HWND hwnd;
625 };
626
627 struct msi_maskedit_info
628 {
629     msi_dialog *dialog;
630     WNDPROC oldproc;
631     HWND hwnd;
632     LPWSTR prop;
633     UINT num_chars;
634     UINT num_groups;
635     struct msi_mask_group group[MASK_MAX_GROUPS];
636 };
637
638 static void msi_mask_control_change( struct msi_maskedit_info *info )
639 {
640     LPWSTR val;
641     UINT i, n, r;
642
643     val = HeapAlloc( GetProcessHeap(), 0, (info->num_chars+1)*sizeof(WCHAR) );
644     for( i=0, n=0; i<info->num_groups; i++ )
645     {
646         if( (info->group[i].len + n) > info->num_chars )
647         {
648             ERR("can't fit control %d text into template\n",i);
649             break;
650         }
651         r = GetWindowTextW( info->group[i].hwnd, &val[n], info->group[i].len+1 );
652         if( r != info->group[i].len )
653             break;
654         n += r;
655     }
656
657     TRACE("%d/%d controls were good\n", i, info->num_groups);
658
659     if( i == info->num_groups )
660     {
661         TRACE("Set property %s to %s\n",
662               debugstr_w(info->prop), debugstr_w(val) );
663         CharUpperBuffW( val, info->num_chars );
664         MSI_SetPropertyW( info->dialog->package, info->prop, val );
665         msi_dialog_evaluate_control_conditions( info->dialog );
666     }
667     HeapFree( GetProcessHeap(), 0, val );
668 }
669
670 static LRESULT WINAPI
671 MSIMaskedEdit_WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
672 {
673     struct msi_maskedit_info *info;
674     HRESULT r;
675
676     TRACE("%p %04x %08x %08lx\n", hWnd, msg, wParam, lParam);
677
678     info = GetPropW(hWnd, szButtonData);
679
680     r = CallWindowProcW(info->oldproc, hWnd, msg, wParam, lParam);
681
682     switch( msg )
683     {
684     case WM_COMMAND:
685         if (HIWORD(wParam) == EN_CHANGE)
686             msi_mask_control_change( info );
687         break;
688     case WM_NCDESTROY:
689         HeapFree( GetProcessHeap(), 0, info->prop );
690         HeapFree( GetProcessHeap(), 0, info );
691         RemovePropW( hWnd, szButtonData );
692         break;
693     }
694
695     return r;
696 }
697
698 /* fish the various bits of the property out and put them in the control */
699 static void
700 msi_maskedit_set_text( struct msi_maskedit_info *info, LPCWSTR text )
701 {
702     LPCWSTR p;
703     UINT i;
704
705     p = text;
706     for( i = 0; i < info->num_groups; i++ )
707     {
708         if( info->group[i].len < lstrlenW( p ) )
709         {
710             LPWSTR chunk = strdupW( p );
711             chunk[ info->group[i].len ] = 0;
712             SetWindowTextW( info->group[i].hwnd, chunk );
713             HeapFree( GetProcessHeap(), 0, chunk );
714         }
715         else
716         {
717             SetWindowTextW( info->group[i].hwnd, p );
718             break;
719         }
720         p += info->group[i].len;
721     }
722 }
723
724 static struct msi_maskedit_info * msi_dialog_parse_groups( LPCWSTR mask )
725 {
726     struct msi_maskedit_info * info = NULL;
727     int i = 0, n = 0, total = 0;
728     LPCWSTR p;
729
730     TRACE("masked control, template %s\n", debugstr_w(mask));
731
732     if( !mask )
733         return info;
734
735     p = strchrW(mask, '<');
736     if( !p )
737         return info;
738
739     info = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof *info );
740     if( !info )
741         return info;
742
743     p++;
744     for( i=0; i<MASK_MAX_GROUPS; i++ )
745     {
746         /* stop at the end of the string */
747         if( p[0] == 0 || p[0] == '>' )
748             break;
749
750         /* count the number of the same identifier */
751         for( n=0; p[n] == p[0]; n++ )
752             ;
753         info->group[i].ofs = total;
754         info->group[i].type = p[0];
755         if( p[n] == '=' )
756         {
757             n++;
758             total++; /* an extra not part of the group */
759         }
760         info->group[i].len = n;
761         total += n;
762         p += n;
763     }
764
765     TRACE("%d characters in %d groups\n", total, info->num_groups );
766     if( i == MASK_MAX_GROUPS )
767         ERR("too many groups in PIDTemplate %s\n", debugstr_w(mask));
768
769     info->num_chars = total;
770     info->num_groups = i;
771
772     return info;
773 }
774
775 static void
776 msi_maskedit_create_children( struct msi_maskedit_info *info )
777 {
778     DWORD width, height, style, wx, ww;
779     LPCWSTR text, font = NULL;
780     RECT rect;
781     HWND hwnd;
782     UINT i;
783
784     style = WS_CHILD | WS_BORDER | WS_VISIBLE | WS_TABSTOP;
785
786     GetClientRect( info->hwnd, &rect );
787
788     width = rect.right - rect.left;
789     height = rect.bottom - rect.top;
790
791     if( text )
792         font = msi_dialog_get_style( &text );
793
794     for( i = 0; i < info->num_groups; i++ )
795     {
796         wx = (info->group[i].ofs * width) / info->num_chars;
797         ww = (info->group[i].len * width) / info->num_chars;
798
799         hwnd = CreateWindowW( szEdit, NULL, style, wx, 0, ww, height,
800                               info->hwnd, NULL, NULL, NULL );
801         if( !hwnd )
802         {
803             ERR("failed to create mask edit sub window\n");
804             break;
805         }
806
807         SendMessageW( hwnd, EM_LIMITTEXT, info->group[i].len, 0 );
808
809         msi_dialog_set_font( info->dialog, hwnd,
810                        font ? font : info->dialog->default_font );
811         info->group[i].hwnd = hwnd;
812     }
813 }
814
815 /* office 2003 uses "73931<````=````=````=````=`````>@@@@@" */
816 static UINT msi_dialog_maskedit_control( msi_dialog *dialog, MSIRECORD *rec )
817 {
818     const static WCHAR pidt[] = {'P','I','D','T','e','m','p','l','a','t','e',0};
819     LPWSTR mask = NULL, title = NULL, val = NULL;
820     struct msi_maskedit_info *info = NULL;
821     UINT ret = ERROR_SUCCESS;
822     msi_control *control;
823     LPCWSTR prop;
824
825     mask = load_dynamic_property( dialog->package, pidt, NULL );
826     if( !mask )
827     {
828         ERR("PIDTemplate is empty\n");
829         goto end;
830     }
831
832     info = msi_dialog_parse_groups( mask );
833     if( !info )
834     {
835         ERR("template %s is invalid\n", debugstr_w(mask));
836         goto end;
837     }
838
839     info->dialog = dialog;
840
841     control = msi_dialog_add_control( dialog, rec, szStatic, SS_OWNERDRAW|WS_GROUP );
842     if( !control )
843     {
844         ERR("Failed to create maskedit container\n");
845         ret = ERROR_FUNCTION_FAILED;
846         goto end;
847     }
848
849     info->hwnd = control->hwnd;
850
851     /* subclass the static control */
852     info->oldproc = (WNDPROC) SetWindowLongPtrW( info->hwnd, GWLP_WNDPROC,
853                                           (LONG_PTR)MSIMaskedEdit_WndProc );
854     SetPropW( control->hwnd, szButtonData, info );
855
856     prop = MSI_RecordGetString( rec, 9 );
857     if( prop )
858         info->prop = strdupW( prop );
859
860     msi_maskedit_create_children( info );
861
862     if( prop )
863     {
864         val = load_dynamic_property( dialog->package, prop, NULL );
865         if( val )
866         {
867             msi_maskedit_set_text( info, val );
868             HeapFree( GetProcessHeap(), 0, val );
869         }
870     }
871
872 end:
873     if( ret != ERROR_SUCCESS )
874         HeapFree( GetProcessHeap(), 0, info );
875     HeapFree( GetProcessHeap(), 0, title );
876     HeapFree( GetProcessHeap(), 0, mask );
877     return ret;
878 }
879
880 /******************** Path Edit ********************************************/
881
882 static UINT msi_dialog_pathedit_control( msi_dialog *dialog, MSIRECORD *rec )
883 {
884     FIXME("not implemented properly\n");
885     return msi_dialog_edit_control( dialog, rec );
886 }
887
888 /* radio buttons are a bit different from normal controls */
889 static UINT msi_dialog_create_radiobutton( MSIRECORD *rec, LPVOID param )
890 {
891     radio_button_group_descr *group = (radio_button_group_descr *)param;
892     msi_dialog *dialog = group->dialog;
893     msi_control *control;
894     LPCWSTR prop, text, name;
895     DWORD style;
896     DWORD attributes = group->attributes;
897
898     style = WS_CHILD | BS_AUTORADIOBUTTON | BS_MULTILINE;
899     name = MSI_RecordGetString( rec, 3 );
900     text = MSI_RecordGetString( rec, 8 );
901     if( attributes & 1 )
902         style |= WS_VISIBLE;
903     if( ~attributes & 2 )
904         style |= WS_DISABLED;
905
906     control = msi_dialog_create_window( dialog, rec, szButton, name, text,
907                                         style, group->parent->hwnd );
908     control->handler = msi_dialog_radiogroup_handler;
909
910     prop = MSI_RecordGetString( rec, 1 );
911     if( prop )
912         control->property = strdupW( prop );
913
914     return ERROR_SUCCESS;
915 }
916
917 static UINT msi_dialog_radiogroup_control( msi_dialog *dialog, MSIRECORD *rec )
918 {
919     static const WCHAR query[] = {
920         'S','E','L','E','C','T',' ','*',' ',
921         'F','R','O','M',' ','R','a','d','i','o','B','u','t','t','o','n',' ',
922         'W','H','E','R','E',' ',
923            '`','P','r','o','p','e','r','t','y','`',' ','=',' ','\'','%','s','\'',0};
924     UINT r;
925     LPCWSTR prop;
926     msi_control *control;
927     MSIQUERY *view = NULL;
928     radio_button_group_descr group;
929     MSIPACKAGE *package = dialog->package;
930     WNDPROC oldproc;
931
932     prop = MSI_RecordGetString( rec, 9 );
933
934     TRACE("%p %p %s\n", dialog, rec, debugstr_w( prop ));
935
936     /* Create parent group box to hold radio buttons */
937     control = msi_dialog_add_control( dialog, rec, szButton, BS_OWNERDRAW|WS_GROUP );
938     if( !control )
939         return ERROR_FUNCTION_FAILED;
940
941     oldproc = (WNDPROC) SetWindowLongPtrW( control->hwnd, GWLP_WNDPROC,
942                                            (LONG_PTR)MSIRadioGroup_WndProc );
943     SetPropW(control->hwnd, szButtonData, oldproc);
944
945     if( prop )
946         control->property = strdupW( prop );
947
948     /* query the Radio Button table for all control in this group */
949     r = MSI_OpenQuery( package->db, &view, query, prop );
950     if( r != ERROR_SUCCESS )
951     {
952         ERR("query failed for dialog %s radio group %s\n", 
953             debugstr_w(dialog->name), debugstr_w(prop));
954         return ERROR_INVALID_PARAMETER;
955     }
956
957     group.dialog = dialog;
958     group.parent = control;
959     group.attributes = MSI_RecordGetInteger( rec, 8 );
960
961     r = MSI_IterateRecords( view, 0, msi_dialog_create_radiobutton, &group );
962     msiobj_release( &view->hdr );
963
964     return r;
965 }
966
967 struct control_handler msi_dialog_handler[] =
968 {
969     { szText, msi_dialog_text_control },
970     { szPushButton, msi_dialog_button_control },
971     { szLine, msi_dialog_line_control },
972     { szBitmap, msi_dialog_bitmap_control },
973     { szCheckBox, msi_dialog_checkbox_control },
974     { szScrollableText, msi_dialog_scrolltext_control },
975     { szComboBox, msi_dialog_combo_control },
976     { szEdit, msi_dialog_edit_control },
977     { szMaskedEdit, msi_dialog_maskedit_control },
978     { szPathEdit, msi_dialog_pathedit_control },
979     { szRadioButtonGroup, msi_dialog_radiogroup_control },
980 };
981
982 #define NUM_CONTROL_TYPES (sizeof msi_dialog_handler/sizeof msi_dialog_handler[0])
983
984 static UINT msi_dialog_create_controls( MSIRECORD *rec, LPVOID param )
985 {
986     msi_dialog *dialog = param;
987     LPCWSTR control_type;
988     UINT i;
989
990     /* find and call the function that can create this type of control */
991     control_type = MSI_RecordGetString( rec, 3 );
992     for( i=0; i<NUM_CONTROL_TYPES; i++ )
993         if (!strcmpiW( msi_dialog_handler[i].control_type, control_type ))
994             break;
995     if( i != NUM_CONTROL_TYPES )
996         msi_dialog_handler[i].func( dialog, rec );
997     else
998         ERR("no handler for element type %s\n", debugstr_w(control_type));
999
1000     return ERROR_SUCCESS;
1001 }
1002
1003 static UINT msi_dialog_fill_controls( msi_dialog *dialog )
1004 {
1005     static const WCHAR query[] = {
1006         'S','E','L','E','C','T',' ','*',' ',
1007         'F','R','O','M',' ','C','o','n','t','r','o','l',' ',
1008         'W','H','E','R','E',' ',
1009            '`','D','i','a','l','o','g','_','`',' ','=',' ','\'','%','s','\'',0};
1010     UINT r;
1011     MSIQUERY *view = NULL;
1012     MSIPACKAGE *package = dialog->package;
1013
1014     TRACE("%p %s\n", dialog, debugstr_w(dialog->name) );
1015
1016     /* query the Control table for all the elements of the control */
1017     r = MSI_OpenQuery( package->db, &view, query, dialog->name );
1018     if( r != ERROR_SUCCESS )
1019     {
1020         ERR("query failed for dialog %s\n", debugstr_w(dialog->name));
1021         return ERROR_INVALID_PARAMETER;
1022     }
1023
1024     r = MSI_IterateRecords( view, 0, msi_dialog_create_controls, dialog );
1025     msiobj_release( &view->hdr );
1026
1027     return r;
1028 }
1029
1030 static UINT msi_dialog_set_control_condition( MSIRECORD *rec, LPVOID param )
1031 {
1032     static const WCHAR szHide[] = { 'H','i','d','e',0 };
1033     static const WCHAR szShow[] = { 'S','h','o','w',0 };
1034     static const WCHAR szDisable[] = { 'D','i','s','a','b','l','e',0 };
1035     static const WCHAR szEnable[] = { 'E','n','a','b','l','e',0 };
1036     msi_dialog *dialog = param;
1037     msi_control *control;
1038     LPCWSTR name, action, condition;
1039     UINT r;
1040
1041     name = MSI_RecordGetString( rec, 2 );
1042     action = MSI_RecordGetString( rec, 3 );
1043     condition = MSI_RecordGetString( rec, 4 );
1044     r = MSI_EvaluateConditionW( dialog->package, condition );
1045     control = msi_dialog_find_control( dialog, name );
1046     if( r && control )
1047     {
1048         TRACE("%s control %s\n", debugstr_w(action), debugstr_w(name));
1049
1050         /* FIXME: case sensitive? */
1051         if(!strcmpW(action, szHide))
1052             ShowWindow(control->hwnd, SW_HIDE);
1053         else if(!strcmpW(action, szShow))
1054             ShowWindow(control->hwnd, SW_SHOW);
1055         else if(!strcmpW(action, szDisable))
1056             EnableWindow(control->hwnd, FALSE);
1057         else if(!strcmpW(action, szEnable))
1058             EnableWindow(control->hwnd, TRUE);
1059         else
1060             FIXME("Unhandled action %s\n", debugstr_w(action));
1061     }
1062
1063     return ERROR_SUCCESS;
1064 }
1065
1066 static UINT msi_dialog_evaluate_control_conditions( msi_dialog *dialog )
1067 {
1068     static const WCHAR query[] = {
1069       'S','E','L','E','C','T',' ','*',' ',
1070       'F','R','O','M',' ',
1071         'C','o','n','t','r','o','l','C','o','n','d','i','t','i','o','n',' ',
1072       'W','H','E','R','E',' ',
1073         '`','D','i','a','l','o','g','_','`',' ','=',' ','\'','%','s','\'',0
1074     };
1075     UINT r;
1076     MSIQUERY *view = NULL;
1077     MSIPACKAGE *package = dialog->package;
1078
1079     TRACE("%p %s\n", dialog, debugstr_w(dialog->name) );
1080
1081     /* query the Control table for all the elements of the control */
1082     r = MSI_OpenQuery( package->db, &view, query, dialog->name );
1083     if( r != ERROR_SUCCESS )
1084     {
1085         ERR("query failed for dialog %s\n", debugstr_w(dialog->name));
1086         return ERROR_INVALID_PARAMETER;
1087     }
1088
1089     r = MSI_IterateRecords( view, 0, msi_dialog_set_control_condition, dialog );
1090     msiobj_release( &view->hdr );
1091
1092     return r;
1093 }
1094
1095 /* figure out the height of 10 point MS Sans Serif */
1096 static INT msi_dialog_get_sans_serif_height( HWND hwnd )
1097 {
1098     static const WCHAR szSansSerif[] = {
1099         'M','S',' ','S','a','n','s',' ','S','e','r','i','f',0 };
1100     LOGFONTW lf;
1101     TEXTMETRICW tm;
1102     BOOL r;
1103     LONG height = 0;
1104     HFONT hFont, hOldFont;
1105     HDC hdc;
1106
1107     hdc = GetDC( hwnd );
1108     if (hdc)
1109     {
1110         memset( &lf, 0, sizeof lf );
1111         lf.lfHeight = MulDiv(10, GetDeviceCaps(hdc, LOGPIXELSY), 72);
1112         strcpyW( lf.lfFaceName, szSansSerif );
1113         hFont = CreateFontIndirectW(&lf);
1114         if (hFont)
1115         {
1116             hOldFont = SelectObject( hdc, hFont );
1117             r = GetTextMetricsW( hdc, &tm );
1118             if (r)
1119                 height = tm.tmHeight;
1120             SelectObject( hdc, hOldFont );
1121             DeleteObject( hFont );
1122         }
1123         ReleaseDC( hwnd, hdc );
1124     }
1125     return height;
1126 }
1127
1128 /* fetch the associated record from the Dialog table */
1129 static MSIRECORD *msi_get_dialog_record( msi_dialog *dialog )
1130 {
1131     static const WCHAR query[] = {
1132         'S','E','L','E','C','T',' ','*',' ',
1133         'F','R','O','M',' ','D','i','a','l','o','g',' ',
1134         'W','H','E','R','E',' ',
1135            '`','D','i','a','l','o','g','`',' ','=',' ','\'','%','s','\'',0};
1136     MSIPACKAGE *package = dialog->package;
1137     MSIRECORD *rec = NULL;
1138
1139     TRACE("%p %s\n", dialog, debugstr_w(dialog->name) );
1140
1141     rec = MSI_QueryGetRecord( package->db, query, dialog->name );
1142     if( !rec )
1143         ERR("query failed for dialog %s\n", debugstr_w(dialog->name));
1144
1145     return rec;
1146 }
1147
1148 static void msi_dialog_adjust_dialog_size( msi_dialog *dialog, LPSIZE sz )
1149 {
1150     RECT rect;
1151     LONG style;
1152
1153     /* turn the client size into the window rectangle */
1154     rect.left = 0;
1155     rect.top = 0;
1156     rect.right = msi_dialog_scale_unit( dialog, sz->cx );
1157     rect.bottom = msi_dialog_scale_unit( dialog, sz->cy );
1158     style = GetWindowLongPtrW( dialog->hwnd, GWL_STYLE );
1159     AdjustWindowRect( &rect, style, FALSE );
1160     sz->cx = rect.right - rect.left;
1161     sz->cy = rect.bottom - rect.top;
1162 }
1163
1164 static LRESULT msi_dialog_oncreate( HWND hwnd, LPCREATESTRUCTW cs )
1165 {
1166     static const WCHAR df[] = {
1167         'D','e','f','a','u','l','t','U','I','F','o','n','t',0 };
1168     msi_dialog *dialog = (msi_dialog*) cs->lpCreateParams;
1169     MSIRECORD *rec = NULL;
1170     LPCWSTR text;
1171     LPWSTR title = NULL;
1172     SIZE size;
1173
1174     TRACE("%p %p\n", dialog, dialog->package);
1175
1176     dialog->hwnd = hwnd;
1177     SetWindowLongPtrW( hwnd, GWLP_USERDATA, (LONG_PTR) dialog );
1178
1179     rec = msi_get_dialog_record( dialog );
1180     if( !rec )
1181     {
1182         TRACE("No record found for dialog %s\n", debugstr_w(dialog->name));
1183         return -1;
1184     }
1185
1186     dialog->scale = msi_dialog_get_sans_serif_height(dialog->hwnd);
1187
1188     size.cx = MSI_RecordGetInteger( rec, 4 );
1189     size.cy = MSI_RecordGetInteger( rec, 5 );
1190     msi_dialog_adjust_dialog_size( dialog, &size );
1191
1192     dialog->attributes = MSI_RecordGetInteger( rec, 6 );
1193     text = MSI_RecordGetString( rec, 7 );
1194
1195     dialog->default_font = load_dynamic_property( dialog->package, df, NULL );
1196
1197     deformat_string( dialog->package, text, &title );
1198     SetWindowTextW( hwnd, title );
1199     SetWindowPos( hwnd, 0, 0, 0, size.cx, size.cy,
1200                   SWP_NOMOVE | SWP_NOACTIVATE | SWP_NOZORDER | SWP_NOREDRAW );
1201
1202     HeapFree( GetProcessHeap(), 0, title );
1203     msiobj_release( &rec->hdr );
1204
1205     msi_dialog_build_font_list( dialog );
1206     msi_dialog_fill_controls( dialog );
1207     msi_dialog_evaluate_control_conditions( dialog );
1208
1209     return 0;
1210 }
1211
1212 static UINT msi_dialog_send_event( msi_dialog *dialog, LPCWSTR event, LPCWSTR arg )
1213 {
1214     LPWSTR event_fmt = NULL, arg_fmt = NULL;
1215
1216     TRACE("Sending control event %s %s\n", debugstr_w(event), debugstr_w(arg));
1217
1218     deformat_string( dialog->package, event, &event_fmt );
1219     deformat_string( dialog->package, arg, &arg_fmt );
1220
1221     dialog->event_handler( dialog->package, event_fmt, arg_fmt, dialog );
1222
1223     HeapFree( GetProcessHeap(), 0, event_fmt );
1224     HeapFree( GetProcessHeap(), 0, arg_fmt );
1225
1226     return ERROR_SUCCESS;
1227 }
1228
1229 static UINT msi_dialog_set_property( msi_dialog *dialog, LPCWSTR event, LPCWSTR arg )
1230 {
1231     static const WCHAR szNullArg[] = { '{','}',0 };
1232     LPWSTR p, prop, arg_fmt = NULL;
1233     UINT len;
1234
1235     len = strlenW(event);
1236     prop = HeapAlloc( GetProcessHeap(), 0, len*sizeof(WCHAR));
1237     strcpyW( prop, &event[1] );
1238     p = strchrW( prop, ']' );
1239     if( p && p[1] == 0 )
1240     {
1241         *p = 0;
1242         if( strcmpW( szNullArg, arg ) )
1243             deformat_string( dialog->package, arg, &arg_fmt );
1244         MSI_SetPropertyW( dialog->package, prop, arg_fmt );
1245     }
1246     else
1247         ERR("Badly formatted property string - what happens?\n");
1248     HeapFree( GetProcessHeap(), 0, prop );
1249     return ERROR_SUCCESS;
1250 }
1251
1252 static UINT msi_dialog_control_event( MSIRECORD *rec, LPVOID param )
1253 {
1254     msi_dialog *dialog = param;
1255     LPCWSTR condition, event, arg;
1256     UINT r;
1257
1258     condition = MSI_RecordGetString( rec, 5 );
1259     r = MSI_EvaluateConditionW( dialog->package, condition );
1260     if( r )
1261     {
1262         event = MSI_RecordGetString( rec, 3 );
1263         arg = MSI_RecordGetString( rec, 4 );
1264         if( event[0] == '[' )
1265             msi_dialog_set_property( dialog, event, arg );
1266         else
1267             msi_dialog_send_event( dialog, event, arg );
1268     }
1269
1270     return ERROR_SUCCESS;
1271 }
1272
1273 static UINT msi_dialog_button_handler( msi_dialog *dialog,
1274                                        msi_control *control, WPARAM param )
1275 {
1276     static const WCHAR query[] = {
1277       'S','E','L','E','C','T',' ','*',' ',
1278       'F','R','O','M',' ','C','o','n','t','r','o','l','E','v','e','n','t',' ',
1279       'W','H','E','R','E',' ',
1280          '`','D','i','a','l','o','g','_','`',' ','=',' ','\'','%','s','\'',' ',
1281       'A','N','D',' ',
1282          '`','C','o','n','t','r','o','l','_','`',' ','=',' ','\'','%','s','\'',' ',
1283       'O','R','D','E','R',' ','B','Y',' ','`','O','r','d','e','r','i','n','g','`',0
1284     };
1285     MSIQUERY *view = NULL;
1286     UINT r;
1287
1288     if( HIWORD(param) != BN_CLICKED )
1289         return ERROR_SUCCESS;
1290
1291     r = MSI_OpenQuery( dialog->package->db, &view, query,
1292                        dialog->name, control->name );
1293     if( r != ERROR_SUCCESS )
1294     {
1295         ERR("query failed\n");
1296         return 0;
1297     }
1298
1299     r = MSI_IterateRecords( view, 0, msi_dialog_control_event, dialog );
1300     msiobj_release( &view->hdr );
1301
1302     return r;
1303 }
1304
1305 static UINT msi_dialog_get_checkbox_state( msi_dialog *dialog,
1306                 msi_control *control )
1307 {
1308     WCHAR state[2] = { 0 };
1309     DWORD sz = 2;
1310
1311     MSI_GetPropertyW( dialog->package, control->property, state, &sz );
1312     return state[0] ? 1 : 0;
1313 }
1314
1315 static void msi_dialog_set_checkbox_state( msi_dialog *dialog,
1316                 msi_control *control, UINT state )
1317 {
1318     static const WCHAR szState[] = { '1', 0 };
1319     LPCWSTR val;
1320
1321     /* if uncheck then the property is set to NULL */
1322     if (!state)
1323     {
1324         MSI_SetPropertyW( dialog->package, control->property, NULL );
1325         return;
1326     }
1327
1328     /* check for a custom state */
1329     if (control->value && control->value[0])
1330         val = control->value;
1331     else
1332         val = szState;
1333
1334     MSI_SetPropertyW( dialog->package, control->property, val );
1335 }
1336
1337 static void msi_dialog_checkbox_sync_state( msi_dialog *dialog,
1338                 msi_control *control )
1339 {
1340     UINT state;
1341
1342     state = msi_dialog_get_checkbox_state( dialog, control );
1343     SendMessageW( control->hwnd, BM_SETCHECK,
1344                   state ? BST_CHECKED : BST_UNCHECKED, 0 );
1345 }
1346
1347 static UINT msi_dialog_checkbox_handler( msi_dialog *dialog,
1348                 msi_control *control, WPARAM param )
1349 {
1350     UINT state;
1351
1352     if( HIWORD(param) != BN_CLICKED )
1353         return ERROR_SUCCESS;
1354
1355     TRACE("clicked checkbox %s, set %s\n", debugstr_w(control->name),
1356           debugstr_w(control->property));
1357
1358     state = msi_dialog_get_checkbox_state( dialog, control );
1359     state = state ? 0 : 1;
1360     msi_dialog_set_checkbox_state( dialog, control, state );
1361     msi_dialog_checkbox_sync_state( dialog, control );
1362
1363     return msi_dialog_button_handler( dialog, control, param );
1364 }
1365
1366 static UINT msi_dialog_edit_handler( msi_dialog *dialog,
1367                 msi_control *control, WPARAM param )
1368 {
1369     UINT sz, r;
1370     LPWSTR buf;
1371
1372     if( HIWORD(param) != EN_CHANGE )
1373         return ERROR_SUCCESS;
1374
1375     TRACE("edit %s contents changed, set %s\n", debugstr_w(control->name),
1376           debugstr_w(control->property));
1377
1378     sz = 0x20;
1379     buf = HeapAlloc( GetProcessHeap(), 0, sz*sizeof(WCHAR) );
1380     while( buf )
1381     {
1382         r = GetWindowTextW( control->hwnd, buf, sz );
1383         if( r < (sz-1) )
1384             break;
1385             sz *= 2;
1386         buf = HeapReAlloc( GetProcessHeap(), 0, buf, sz*sizeof(WCHAR) );
1387     }
1388
1389     MSI_SetPropertyW( dialog->package, control->property, buf );
1390
1391     HeapFree( GetProcessHeap(), 0, buf );
1392
1393     return ERROR_SUCCESS;
1394 }
1395
1396 static UINT msi_dialog_radiogroup_handler( msi_dialog *dialog,
1397                 msi_control *control, WPARAM param )
1398 {
1399     if( HIWORD(param) != BN_CLICKED )
1400         return ERROR_SUCCESS;
1401
1402     TRACE("clicked radio button %s, set %s\n", debugstr_w(control->name),
1403           debugstr_w(control->property));
1404
1405     MSI_SetPropertyW( dialog->package, control->property, control->name );
1406
1407     return msi_dialog_button_handler( dialog, control, param );
1408 }
1409
1410 static LRESULT msi_dialog_oncommand( msi_dialog *dialog, WPARAM param, HWND hwnd )
1411 {
1412     msi_control *control;
1413
1414     TRACE("%p %p %08x\n", dialog, hwnd, param);
1415
1416     control = msi_dialog_find_control_by_hwnd( dialog, hwnd );
1417     if( control )
1418     {
1419         if( control->handler )
1420         {
1421             control->handler( dialog, control, param );
1422             msi_dialog_evaluate_control_conditions( dialog );
1423         }
1424     }
1425     else
1426         ERR("button click from nowhere\n");
1427     return 0;
1428 }
1429
1430 static LRESULT WINAPI MSIDialog_WndProc( HWND hwnd, UINT msg,
1431                 WPARAM wParam, LPARAM lParam )
1432 {
1433     msi_dialog *dialog = (LPVOID) GetWindowLongPtrW( hwnd, GWLP_USERDATA );
1434
1435     TRACE("0x%04x\n", msg);
1436
1437     switch (msg)
1438     {
1439     case WM_CREATE:
1440         return msi_dialog_oncreate( hwnd, (LPCREATESTRUCTW)lParam );
1441
1442     case WM_COMMAND:
1443         return msi_dialog_oncommand( dialog, wParam, (HWND)lParam );
1444
1445     case WM_DESTROY:
1446         dialog->hwnd = NULL;
1447         return 0;
1448     }
1449     return DefWindowProcW(hwnd, msg, wParam, lParam);
1450 }
1451
1452 static LRESULT WINAPI MSIRadioGroup_WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
1453 {
1454     WNDPROC oldproc = (WNDPROC) GetPropW(hWnd, szButtonData);
1455
1456     TRACE("hWnd %p msg %04x wParam 0x%08x lParam 0x%08lx\n", hWnd, msg, wParam, lParam);
1457
1458     if (msg == WM_COMMAND) /* Forward notifications to dialog */
1459         SendMessageW(GetParent(hWnd), msg, wParam, lParam);
1460
1461     return CallWindowProcW(oldproc, hWnd, msg, wParam, lParam);
1462 }
1463
1464 static LRESULT WINAPI MSIHiddenWindowProc( HWND hwnd, UINT msg,
1465                 WPARAM wParam, LPARAM lParam )
1466 {
1467     msi_dialog *dialog = (msi_dialog*) lParam;
1468
1469     TRACE("%d %p\n", msg, dialog);
1470
1471     switch (msg)
1472     {
1473     case WM_MSI_DIALOG_CREATE:
1474         return msi_dialog_run_message_loop( dialog );
1475     case WM_MSI_DIALOG_DESTROY:
1476         msi_dialog_destroy( dialog );
1477         return 0;
1478     }
1479     return DefWindowProcW( hwnd, msg, wParam, lParam );
1480 }
1481
1482 /* functions that interface to other modules within MSI */
1483
1484 msi_dialog *msi_dialog_create( MSIPACKAGE* package, LPCWSTR szDialogName,
1485                                 msi_dialog_event_handler event_handler )
1486 {
1487     MSIRECORD *rec = NULL;
1488     msi_dialog *dialog;
1489
1490     TRACE("%p %s\n", package, debugstr_w(szDialogName));
1491
1492     /* allocate the structure for the dialog to use */
1493     dialog = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY,
1494                         sizeof *dialog + sizeof(WCHAR)*strlenW(szDialogName) );
1495     if( !dialog )
1496         return NULL;
1497     strcpyW( dialog->name, szDialogName );
1498     msiobj_addref( &package->hdr );
1499     dialog->package = package;
1500     dialog->event_handler = event_handler;
1501     dialog->finished = 0;
1502
1503     /* verify that the dialog exists */
1504     rec = msi_get_dialog_record( dialog );
1505     if( !rec )
1506     {
1507         HeapFree( GetProcessHeap(), 0, dialog );
1508         return NULL;
1509     }
1510     dialog->attributes = MSI_RecordGetInteger( rec, 6 );
1511     msiobj_release( &rec->hdr );
1512
1513     return dialog;
1514 }
1515
1516 static void msi_process_pending_messages(void)
1517 {
1518     MSG msg;
1519
1520     while( PeekMessageW( &msg, 0, 0, 0, PM_REMOVE ) )
1521     {
1522         TranslateMessage( &msg );
1523         DispatchMessageW( &msg );
1524     }
1525 }
1526
1527 void msi_dialog_end_dialog( msi_dialog *dialog )
1528 {
1529     TRACE("%p\n", dialog);
1530     dialog->finished = 1;
1531     PostMessageW(dialog->hwnd, WM_NULL, 0, 0);
1532 }
1533
1534 void msi_dialog_check_messages( HANDLE handle )
1535 {
1536     DWORD r;
1537
1538     /* in threads other than the UI thread, block */
1539     if( uiThreadId != GetCurrentThreadId() )
1540     {
1541         if( handle )
1542             WaitForSingleObject( handle, INFINITE );
1543         return;
1544     }
1545
1546     /* there's two choices for the UI thread */
1547     while (1)
1548     {
1549         msi_process_pending_messages();
1550
1551         if( !handle )
1552             break;
1553
1554         /*
1555          * block here until somebody creates a new dialog or
1556          * the handle we're waiting on becomes ready
1557          */
1558         r = MsgWaitForMultipleObjects( 1, &handle, 0, INFINITE, QS_ALLINPUT );
1559         if( r == WAIT_OBJECT_0 )
1560             break;
1561     }
1562 }
1563
1564 UINT msi_dialog_run_message_loop( msi_dialog *dialog )
1565 {
1566     HWND hwnd;
1567
1568     if( !(dialog->attributes & msidbDialogAttributesVisible) )
1569         return ERROR_SUCCESS;
1570
1571     if( uiThreadId != GetCurrentThreadId() )
1572         return SendMessageW( hMsiHiddenWindow, WM_MSI_DIALOG_CREATE, 0, (LPARAM) dialog );
1573
1574     /* create the dialog window, don't show it yet */
1575     hwnd = CreateWindowW( szMsiDialogClass, dialog->name, WS_OVERLAPPEDWINDOW,
1576                      CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
1577                      NULL, NULL, NULL, dialog );
1578     if( !hwnd )
1579     {
1580         ERR("Failed to create dialog %s\n", debugstr_w( dialog->name ));
1581         return ERROR_FUNCTION_FAILED;
1582     }
1583
1584     ShowWindow( hwnd, SW_SHOW );
1585     UpdateWindow( hwnd );
1586
1587     if( dialog->attributes & msidbDialogAttributesModal )
1588     {
1589         while( !dialog->finished )
1590         {
1591             MsgWaitForMultipleObjects( 0, NULL, 0, INFINITE, QS_ALLEVENTS );
1592             msi_process_pending_messages();
1593         }
1594     }
1595     else
1596         return ERROR_IO_PENDING;
1597
1598     return ERROR_SUCCESS;
1599 }
1600
1601 void msi_dialog_do_preview( msi_dialog *dialog )
1602 {
1603     TRACE("\n");
1604     dialog->attributes |= msidbDialogAttributesVisible;
1605     dialog->attributes &= ~msidbDialogAttributesModal;
1606     msi_dialog_run_message_loop( dialog );
1607 }
1608
1609 void msi_dialog_destroy( msi_dialog *dialog )
1610 {
1611     if( uiThreadId != GetCurrentThreadId() )
1612     {
1613         SendMessageW( hMsiHiddenWindow, WM_MSI_DIALOG_DESTROY, 0, (LPARAM) dialog );
1614         return;
1615     }
1616
1617     if( dialog->hwnd )
1618         ShowWindow( dialog->hwnd, SW_HIDE );
1619     
1620     /* destroy the list of controls */
1621     while( dialog->control_list )
1622     {
1623         msi_control *t = dialog->control_list;
1624         dialog->control_list = t->next;
1625         /* leave dialog->hwnd - destroying parent destroys child windows */
1626         HeapFree( GetProcessHeap(), 0, t->property );
1627         HeapFree( GetProcessHeap(), 0, t->value );
1628         if( t->pic )
1629             IPicture_Release( t->pic );
1630         HeapFree( GetProcessHeap(), 0, t );
1631     }
1632
1633     /* destroy the list of fonts */
1634     while( dialog->font_list )
1635     {
1636         msi_font *t = dialog->font_list;
1637         dialog->font_list = t->next;
1638         DeleteObject( t->hfont );
1639         HeapFree( GetProcessHeap(), 0, t );
1640     }
1641     HeapFree( GetProcessHeap(), 0, dialog->default_font );
1642
1643     if( dialog->hwnd )
1644         DestroyWindow( dialog->hwnd );
1645
1646     msiobj_release( &dialog->package->hdr );
1647     dialog->package = NULL;
1648     HeapFree( GetProcessHeap(), 0, dialog );
1649 }
1650
1651 BOOL msi_dialog_register_class( void )
1652 {
1653     WNDCLASSW cls;
1654
1655     ZeroMemory( &cls, sizeof cls );
1656     cls.lpfnWndProc   = MSIDialog_WndProc;
1657     cls.hInstance     = NULL;
1658     cls.hIcon         = LoadIconW(0, (LPWSTR)IDI_APPLICATION);
1659     cls.hCursor       = LoadCursorW(0, (LPWSTR)IDC_ARROW);
1660     cls.hbrBackground = (HBRUSH)(COLOR_WINDOW);
1661     cls.lpszMenuName  = NULL;
1662     cls.lpszClassName = szMsiDialogClass;
1663
1664     if( !RegisterClassW( &cls ) )
1665         return FALSE;
1666
1667     cls.lpfnWndProc   = MSIHiddenWindowProc;
1668     cls.lpszClassName = szMsiHiddenWindow;
1669
1670     if( !RegisterClassW( &cls ) )
1671         return FALSE;
1672
1673     uiThreadId = GetCurrentThreadId();
1674
1675     hMsiHiddenWindow = CreateWindowW( szMsiHiddenWindow, NULL, WS_OVERLAPPED,
1676                                    0, 0, 100, 100, NULL, NULL, NULL, NULL );
1677     if( !hMsiHiddenWindow )
1678         return FALSE;
1679
1680     hRichedit = LoadLibraryA("riched20");
1681
1682     return TRUE;
1683 }
1684
1685 void msi_dialog_unregister_class( void )
1686 {
1687     DestroyWindow( hMsiHiddenWindow );
1688     UnregisterClassW( szMsiDialogClass, NULL );
1689     uiThreadId = 0;
1690     FreeLibrary( hRichedit );
1691 }