git.oblomov.eu Git - wine/blob - dlls/comdlg32/itemdlg.c

   1 /*
   2  * Common Item Dialog
   3  *
   4  * Copyright 2010,2011 David Hedberg
   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 <stdarg.h>
  22
  23 #define COBJMACROS
  24 #define NONAMELESSUNION
  25 #define NONAMELESSSTRUCT
  26
  27 #include "windef.h"
  28 #include "winbase.h"
  29 #include "winuser.h"
  30 #include "wingdi.h"
  31 #include "winreg.h"
  32 #include "shlwapi.h"
  33
  34 #include "commdlg.h"
  35 #include "cdlg.h"
  36 #include "filedlgbrowser.h"
  37
  38 #include "wine/debug.h"
  39 #include "wine/list.h"
  40
  41 #define IDC_NAV_TOOLBAR      200
  42 #define IDC_NAVBACK          201
  43 #define IDC_NAVFORWARD       202
  44
  45 #include <initguid.h>
  46 /* This seems to be another version of IID_IFileDialogCustomize. If
  47  * there is any difference I have yet to find it. */
  48 DEFINE_GUID(IID_IFileDialogCustomizeAlt, 0x8016B7B3, 0x3D49, 0x4504, 0xA0,0xAA, 0x2A,0x37,0x49,0x4E,0x60,0x6F);
  49
  50 WINE_DEFAULT_DEBUG_CHANNEL(commdlg);
  51
  52 static const WCHAR notifysink_childW[] = {'n','f','s','_','c','h','i','l','d',0};
  53 static const WCHAR floatnotifysinkW[] = {'F','l','o','a','t','N','o','t','i','f','y','S','i','n','k',0};
  54
  55 enum ITEMDLG_TYPE {
  56     ITEMDLG_TYPE_OPEN,
  57     ITEMDLG_TYPE_SAVE
  58 };
  59
  60 enum ITEMDLG_CCTRL_TYPE {
  61     IDLG_CCTRL_MENU,
  62     IDLG_CCTRL_PUSHBUTTON,
  63     IDLG_CCTRL_COMBOBOX,
  64     IDLG_CCTRL_RADIOBUTTONLIST,
  65     IDLG_CCTRL_CHECKBUTTON,
  66     IDLG_CCTRL_EDITBOX,
  67     IDLG_CCTRL_SEPARATOR,
  68     IDLG_CCTRL_TEXT
  69 };
  70
  71 typedef struct {
  72     HWND hwnd, wrapper_hwnd;
  73     UINT id, dlgid;
  74     enum ITEMDLG_CCTRL_TYPE type;
  75     CDCONTROLSTATEF cdcstate;
  76     struct list entry;
  77 } customctrl;
  78
  79 typedef struct {
  80     struct list entry;
  81     IFileDialogEvents *pfde;
  82     DWORD cookie;
  83 } events_client;
  84
  85 typedef struct FileDialogImpl {
  86     IFileDialog2 IFileDialog2_iface;
  87     union {
  88         IFileOpenDialog IFileOpenDialog_iface;
  89         IFileSaveDialog IFileSaveDialog_iface;
  90     } u;
  91     enum ITEMDLG_TYPE dlg_type;
  92     IExplorerBrowserEvents IExplorerBrowserEvents_iface;
  93     IServiceProvider       IServiceProvider_iface;
  94     ICommDlgBrowser3       ICommDlgBrowser3_iface;
  95     IOleWindow             IOleWindow_iface;
  96     IFileDialogCustomize   IFileDialogCustomize_iface;
  97     LONG ref;
  98
  99     FILEOPENDIALOGOPTIONS options;
 100     COMDLG_FILTERSPEC *filterspecs;
 101     UINT filterspec_count;
 102     UINT filetypeindex;
 103
 104     struct list events_clients;
 105     DWORD events_next_cookie;
 106
 107     IShellItemArray *psia_selection;
 108     IShellItemArray *psia_results;
 109     IShellItem *psi_defaultfolder;
 110     IShellItem *psi_setfolder;
 111     IShellItem *psi_folder;
 112
 113     HWND dlg_hwnd;
 114     IExplorerBrowser *peb;
 115     DWORD ebevents_cookie;
 116
 117     LPWSTR set_filename;
 118     LPWSTR default_ext;
 119     LPWSTR custom_title;
 120     LPWSTR custom_okbutton;
 121     LPWSTR custom_cancelbutton;
 122     LPWSTR custom_filenamelabel;
 123
 124     UINT cctrl_width, cctrl_def_height, cctrls_cols;
 125     HWND cctrls_hwnd;
 126     struct list cctrls;
 127     UINT_PTR cctrl_next_dlgid;
 128 } FileDialogImpl;
 129
 130 /**************************************************************************
 131  * Event wrappers.
 132  */
 133 static HRESULT events_OnFileOk(FileDialogImpl *This)
 134 {
 135     events_client *cursor;
 136     HRESULT hr = S_OK;
 137     TRACE("%p\n", This);
 138
 139     LIST_FOR_EACH_ENTRY(cursor, &This->events_clients, events_client, entry)
 140     {
 141         TRACE("Notifying %p\n", cursor);
 142         hr = IFileDialogEvents_OnFileOk(cursor->pfde, (IFileDialog*)&This->IFileDialog2_iface);
 143         if(FAILED(hr) && hr != E_NOTIMPL)
 144             break;
 145     }
 146
 147     if(hr == E_NOTIMPL)
 148         hr = S_OK;
 149
 150     return hr;
 151 }
 152
 153 static HRESULT events_OnFolderChanging(FileDialogImpl *This, IShellItem *folder)
 154 {
 155     events_client *cursor;
 156     HRESULT hr = S_OK;
 157     TRACE("%p (%p)\n", This, folder);
 158
 159     LIST_FOR_EACH_ENTRY(cursor, &This->events_clients, events_client, entry)
 160     {
 161         TRACE("Notifying %p\n", cursor);
 162         hr = IFileDialogEvents_OnFolderChanging(cursor->pfde, (IFileDialog*)&This->IFileDialog2_iface, folder);
 163         if(FAILED(hr) && hr != E_NOTIMPL)
 164             break;
 165     }
 166
 167     if(hr == E_NOTIMPL)
 168         hr = S_OK;
 169
 170     return hr;
 171 }
 172
 173 static void events_OnFolderChange(FileDialogImpl *This)
 174 {
 175     events_client *cursor;
 176     TRACE("%p\n", This);
 177
 178     LIST_FOR_EACH_ENTRY(cursor, &This->events_clients, events_client, entry)
 179     {
 180         TRACE("Notifying %p\n", cursor);
 181         IFileDialogEvents_OnFolderChange(cursor->pfde, (IFileDialog*)&This->IFileDialog2_iface);
 182     }
 183 }
 184
 185 static void events_OnSelectionChange(FileDialogImpl *This)
 186 {
 187     events_client *cursor;
 188     TRACE("%p\n", This);
 189
 190     LIST_FOR_EACH_ENTRY(cursor, &This->events_clients, events_client, entry)
 191     {
 192         TRACE("Notifying %p\n", cursor);
 193         IFileDialogEvents_OnSelectionChange(cursor->pfde, (IFileDialog*)&This->IFileDialog2_iface);
 194     }
 195 }
 196
 197 static inline HRESULT get_cctrl_event(IFileDialogEvents *pfde, IFileDialogControlEvents **pfdce)
 198 {
 199     return IFileDialogEvents_QueryInterface(pfde, &IID_IFileDialogControlEvents, (void**)pfdce);
 200 }
 201
 202 static HRESULT cctrl_event_OnButtonClicked(FileDialogImpl *This, DWORD ctl_id)
 203 {
 204     events_client *cursor;
 205     TRACE("%p\n", This);
 206
 207     LIST_FOR_EACH_ENTRY(cursor, &This->events_clients, events_client, entry)
 208     {
 209         IFileDialogControlEvents *pfdce;
 210         if(SUCCEEDED(get_cctrl_event(cursor->pfde, &pfdce)))
 211         {
 212             TRACE("Notifying %p\n", cursor);
 213             IFileDialogControlEvents_OnButtonClicked(pfdce, &This->IFileDialogCustomize_iface, ctl_id);
 214             IFileDialogControlEvents_Release(pfdce);
 215         }
 216     }
 217
 218     return S_OK;
 219 }
 220
 221 static HRESULT cctrl_event_OnItemSelected(FileDialogImpl *This, DWORD ctl_id, DWORD item_id)
 222 {
 223     events_client *cursor;
 224     TRACE("%p\n", This);
 225
 226     LIST_FOR_EACH_ENTRY(cursor, &This->events_clients, events_client, entry)
 227     {
 228         IFileDialogControlEvents *pfdce;
 229         if(SUCCEEDED(get_cctrl_event(cursor->pfde, &pfdce)))
 230         {
 231             TRACE("Notifying %p\n", cursor);
 232             IFileDialogControlEvents_OnItemSelected(pfdce, &This->IFileDialogCustomize_iface, ctl_id, item_id);
 233             IFileDialogControlEvents_Release(pfdce);
 234         }
 235     }
 236
 237     return S_OK;
 238 }
 239
 240 static HRESULT cctrl_event_OnCheckButtonToggled(FileDialogImpl *This, DWORD ctl_id, BOOL checked)
 241 {
 242     events_client *cursor;
 243     TRACE("%p\n", This);
 244
 245     LIST_FOR_EACH_ENTRY(cursor, &This->events_clients, events_client, entry)
 246     {
 247         IFileDialogControlEvents *pfdce;
 248         if(SUCCEEDED(get_cctrl_event(cursor->pfde, &pfdce)))
 249         {
 250             TRACE("Notifying %p\n", cursor);
 251             IFileDialogControlEvents_OnCheckButtonToggled(pfdce, &This->IFileDialogCustomize_iface, ctl_id, checked);
 252             IFileDialogControlEvents_Release(pfdce);
 253         }
 254     }
 255
 256     return S_OK;
 257 }
 258
 259 static HRESULT cctrl_event_OnControlActivating(FileDialogImpl *This,
 260                                                   DWORD ctl_id)
 261 {
 262     events_client *cursor;
 263     TRACE("%p\n", This);
 264
 265     LIST_FOR_EACH_ENTRY(cursor, &This->events_clients, events_client, entry)
 266     {
 267         IFileDialogControlEvents *pfdce;
 268         if(SUCCEEDED(get_cctrl_event(cursor->pfde, &pfdce)))
 269         {
 270             TRACE("Notifying %p\n", cursor);
 271             IFileDialogControlEvents_OnControlActivating(pfdce, &This->IFileDialogCustomize_iface, ctl_id);
 272             IFileDialogControlEvents_Release(pfdce);
 273         }
 274     }
 275
 276     return S_OK;
 277 }
 278
 279 /**************************************************************************
 280  * Helper functions.
 281  */
 282 static UINT get_file_name(FileDialogImpl *This, LPWSTR *str)
 283 {
 284     HWND hwnd_edit = GetDlgItem(This->dlg_hwnd, IDC_FILENAME);
 285     UINT len;
 286
 287     if(!hwnd_edit)
 288     {
 289         if(This->set_filename)
 290         {
 291             len = lstrlenW(This->set_filename);
 292             *str = CoTaskMemAlloc(sizeof(WCHAR)*(len+1));
 293             lstrcpyW(*str, This->set_filename);
 294             return len;
 295         }
 296         return FALSE;
 297     }
 298
 299     len = SendMessageW(hwnd_edit, WM_GETTEXTLENGTH, 0, 0);
 300     *str = CoTaskMemAlloc(sizeof(WCHAR)*(len+1));
 301     if(!*str)
 302         return FALSE;
 303
 304     SendMessageW(hwnd_edit, WM_GETTEXT, len+1, (LPARAM)*str);
 305     return len;
 306 }
 307
 308 static BOOL set_file_name(FileDialogImpl *This, LPCWSTR str)
 309 {
 310     HWND hwnd_edit = GetDlgItem(This->dlg_hwnd, IDC_FILENAME);
 311
 312     if(This->set_filename)
 313         LocalFree(This->set_filename);
 314
 315     This->set_filename = StrDupW(str);
 316
 317     return SendMessageW(hwnd_edit, WM_SETTEXT, 0, (LPARAM)str);
 318 }
 319
 320 static void fill_filename_from_selection(FileDialogImpl *This)
 321 {
 322     IShellItem *psi;
 323     LPWSTR *names;
 324     HRESULT hr;
 325     UINT item_count, valid_count;
 326     UINT len_total, i;
 327
 328     if(!This->psia_selection)
 329         return;
 330
 331     hr = IShellItemArray_GetCount(This->psia_selection, &item_count);
 332     if(FAILED(hr) || !item_count)
 333         return;
 334
 335     names = HeapAlloc(GetProcessHeap(), 0, item_count*sizeof(LPWSTR));
 336
 337     /* Get names of the selected items */
 338     valid_count = 0; len_total = 0;
 339     for(i = 0; i < item_count; i++)
 340     {
 341         hr = IShellItemArray_GetItemAt(This->psia_selection, i, &psi);
 342         if(SUCCEEDED(hr))
 343         {
 344             UINT attr;
 345
 346             hr = IShellItem_GetAttributes(psi, SFGAO_FOLDER, &attr);
 347             if(SUCCEEDED(hr) && (attr & SFGAO_FOLDER))
 348                 continue; /* FIXME: FOS_PICKFOLDERS */
 349
 350             hr = IShellItem_GetDisplayName(psi, SIGDN_PARENTRELATIVEPARSING, &names[valid_count]);
 351             if(SUCCEEDED(hr))
 352             {
 353                 len_total += lstrlenW(names[valid_count]) + 3;
 354                 valid_count++;
 355             }
 356             IShellItem_Release(psi);
 357         }
 358     }
 359
 360     if(valid_count == 1)
 361     {
 362         set_file_name(This, names[0]);
 363         CoTaskMemFree(names[0]);
 364     }
 365     else if(valid_count > 1)
 366     {
 367         LPWSTR string = HeapAlloc(GetProcessHeap(), 0, sizeof(WCHAR)*len_total);
 368         LPWSTR cur_point = string;
 369
 370         for(i = 0; i < valid_count; i++)
 371         {
 372             LPWSTR file = names[i];
 373             *cur_point++ = '\"';
 374             lstrcpyW(cur_point, file);
 375             cur_point += lstrlenW(file);
 376             *cur_point++ = '\"';
 377             *cur_point++ = ' ';
 378             CoTaskMemFree(file);
 379         }
 380         *(cur_point-1) = '\0';
 381
 382         set_file_name(This, string);
 383         HeapFree(GetProcessHeap(), 0, string);
 384     }
 385
 386     HeapFree(GetProcessHeap(), 0, names);
 387     return;
 388 }
 389
 390 static LPWSTR get_first_ext_from_spec(LPWSTR buf, LPCWSTR spec)
 391 {
 392     WCHAR *endpos, *ext;
 393
 394     lstrcpyW(buf, spec);
 395     if( (endpos = StrChrW(buf, ';')) )
 396         *endpos = '\0';
 397
 398     ext = PathFindExtensionW(buf);
 399     if(StrChrW(ext, '*'))
 400         return NULL;
 401
 402     return ext;
 403 }
 404
 405 static HRESULT on_default_action(FileDialogImpl *This)
 406 {
 407     IShellFolder *psf_parent, *psf_desktop;
 408     LPITEMIDLIST *pidla;
 409     LPITEMIDLIST current_folder;
 410     LPWSTR fn_iter, files, tmp_files;
 411     UINT file_count = 0, len, i;
 412     int open_action;
 413     HRESULT hr, ret = E_FAIL;
 414
 415     len = get_file_name(This, &tmp_files);
 416     if(len)
 417     {
 418         UINT size_used;
 419         file_count = COMDLG32_SplitFileNames(tmp_files, len, &files, &size_used);
 420     }
 421     if(!file_count) return E_FAIL;
 422
 423     hr = SHGetIDListFromObject((IUnknown*)This->psi_folder, &current_folder);
 424     if(FAILED(hr))
 425     {
 426         ERR("Failed to get pidl for current directory.\n");
 427         return hr;
 428     }
 429
 430     TRACE("Acting on %d file(s).\n", file_count);
 431
 432     pidla = HeapAlloc(GetProcessHeap(), 0, sizeof(LPITEMIDLIST) * file_count);
 433     open_action = ONOPEN_OPEN;
 434     fn_iter = files;
 435
 436     for(i = 0; i < file_count && open_action == ONOPEN_OPEN; i++)
 437     {
 438         WCHAR canon_filename[MAX_PATH];
 439         psf_parent = NULL;
 440
 441         COMDLG32_GetCanonicalPath(current_folder, fn_iter, canon_filename);
 442
 443         if( (This->options & FOS_NOVALIDATE) &&
 444             !(This->options & FOS_FILEMUSTEXIST) )
 445             open_action = ONOPEN_OPEN;
 446         else
 447             open_action = ONOPEN_BROWSE;
 448
 449         open_action = FILEDLG95_ValidatePathAction(canon_filename, &psf_parent, This->dlg_hwnd,
 450                                                    This->options & ~FOS_FILEMUSTEXIST,
 451                                                    (This->dlg_type == ITEMDLG_TYPE_SAVE),
 452                                                    open_action);
 453
 454         /* Add the proper extension */
 455         if(open_action == ONOPEN_OPEN)
 456         {
 457             static const WCHAR dotW[] = {'.',0};
 458
 459             if(This->dlg_type == ITEMDLG_TYPE_SAVE)
 460             {
 461                 WCHAR extbuf[MAX_PATH], *newext = NULL;
 462
 463                 if(This->filterspec_count)
 464                 {
 465                     newext = get_first_ext_from_spec(extbuf, This->filterspecs[This->filetypeindex].pszSpec);
 466                 }
 467                 else if(This->default_ext)
 468                 {
 469                     lstrcpyW(extbuf, dotW);
 470                     lstrcatW(extbuf, This->default_ext);
 471                     newext = extbuf;
 472                 }
 473
 474                 if(newext)
 475                 {
 476                     WCHAR *ext = PathFindExtensionW(canon_filename);
 477                     if(lstrcmpW(ext, newext))
 478                         lstrcatW(canon_filename, newext);
 479                 }
 480             }
 481             else
 482             {
 483                 if( !(This->options & FOS_NOVALIDATE) && (This->options & FOS_FILEMUSTEXIST) &&
 484                     !PathFileExistsW(canon_filename))
 485                 {
 486                     if(This->default_ext)
 487                     {
 488                         lstrcatW(canon_filename, dotW);
 489                         lstrcatW(canon_filename, This->default_ext);
 490
 491                         if(!PathFileExistsW(canon_filename))
 492                         {
 493                             FILEDLG95_OnOpenMessage(This->dlg_hwnd, 0, IDS_FILENOTEXISTING);
 494                             open_action = ONOPEN_BROWSE;
 495                         }
 496                     }
 497                     else
 498                     {
 499                         FILEDLG95_OnOpenMessage(This->dlg_hwnd, 0, IDS_FILENOTEXISTING);
 500                         open_action = ONOPEN_BROWSE;
 501                     }
 502                 }
 503             }
 504         }
 505
 506         pidla[i] = COMDLG32_SHSimpleIDListFromPathAW(canon_filename);
 507
 508         if(psf_parent && !(open_action == ONOPEN_BROWSE))
 509             IShellItem_Release(psf_parent);
 510
 511         fn_iter += (WCHAR)lstrlenW(fn_iter) + 1;
 512     }
 513
 514     HeapFree(GetProcessHeap(), 0, files);
 515     ILFree(current_folder);
 516
 517     if((This->options & FOS_PICKFOLDERS) && open_action == ONOPEN_BROWSE)
 518         open_action = ONOPEN_OPEN; /* FIXME: Multiple folders? */
 519
 520     switch(open_action)
 521     {
 522     case ONOPEN_SEARCH:
 523         FIXME("Filtering not implemented.\n");
 524         break;
 525
 526     case ONOPEN_BROWSE:
 527         hr = IExplorerBrowser_BrowseToObject(This->peb, (IUnknown*)psf_parent, SBSP_DEFBROWSER);
 528         if(FAILED(hr))
 529             ERR("Failed to browse to directory: %08x\n", hr);
 530
 531         IShellItem_Release(psf_parent);
 532         break;
 533
 534     case ONOPEN_OPEN:
 535         if(events_OnFileOk(This) != S_OK)
 536             break;
 537
 538         hr = SHGetDesktopFolder(&psf_desktop);
 539         if(SUCCEEDED(hr))
 540         {
 541             if(This->psia_results)
 542                 IShellItemArray_Release(This->psia_results);
 543
 544             hr = SHCreateShellItemArray(NULL, psf_desktop, file_count, (PCUITEMID_CHILD_ARRAY)pidla,
 545                                         &This->psia_results);
 546             if(SUCCEEDED(hr))
 547                 ret = S_OK;
 548
 549             IShellFolder_Release(psf_desktop);
 550         }
 551         break;
 552
 553     default:
 554         ERR("Failed.\n");
 555         break;
 556     }
 557
 558     /* Clean up */
 559     for(i = 0; i < file_count; i++)
 560         ILFree(pidla[i]);
 561     HeapFree(GetProcessHeap(), 0, pidla);
 562
 563     /* Success closes the dialog */
 564     return ret;
 565 }
 566
 567 /**************************************************************************
 568  * Control functions.
 569  */
 570 static inline customctrl *get_cctrl_from_dlgid(FileDialogImpl *This, DWORD dlgid)
 571 {
 572     customctrl *ctrl;
 573
 574     LIST_FOR_EACH_ENTRY(ctrl, &This->cctrls, customctrl, entry)
 575         if(ctrl->dlgid == dlgid)
 576             return ctrl;
 577
 578     ERR("Failed to find control with dialog id %d\n", dlgid);
 579     return NULL;
 580 }
 581
 582 static inline customctrl *get_cctrl(FileDialogImpl *This, DWORD ctlid)
 583 {
 584     customctrl *ctrl;
 585
 586     LIST_FOR_EACH_ENTRY(ctrl, &This->cctrls, customctrl, entry)
 587         if(ctrl->id == ctlid)
 588             return ctrl;
 589
 590     ERR("Failed to find control with control id %d\n", ctlid);
 591     return NULL;
 592 }
 593
 594 static void ctrl_resize(HWND hctrl, UINT min_width, UINT max_width, BOOL multiline)
 595 {
 596     LPWSTR text;
 597     UINT len, final_width;
 598     UINT lines, final_height;
 599     SIZE size;
 600     RECT rc;
 601     HDC hdc;
 602     WCHAR *c;
 603
 604     TRACE("\n");
 605
 606     len = SendMessageW(hctrl, WM_GETTEXTLENGTH, 0, 0);
 607     text = HeapAlloc(GetProcessHeap(), 0, sizeof(WCHAR)*(len+1));
 608     if(!text) return;
 609     SendMessageW(hctrl, WM_GETTEXT, len+1, (LPARAM)text);
 610
 611     hdc = GetDC(hctrl);
 612     GetTextExtentPoint32W(hdc, text, lstrlenW(text), &size);
 613     ReleaseDC(hctrl, hdc);
 614
 615     if(len && multiline)
 616     {
 617         /* FIXME: line-wrap */
 618         for(lines = 1, c = text; *c != '\0'; c++)
 619             if(*c == '\n') lines++;
 620
 621         final_height = size.cy*lines + 2*4;
 622     }
 623     else
 624     {
 625         GetWindowRect(hctrl, &rc);
 626         final_height = rc.bottom - rc.top;
 627     }
 628
 629     final_width = min(max(size.cx, min_width) + 4, max_width);
 630     SetWindowPos(hctrl, NULL, 0, 0, final_width, final_height,
 631                  SWP_NOZORDER | SWP_NOMOVE | SWP_NOACTIVATE);
 632
 633     HeapFree(GetProcessHeap(), 0, text);
 634 }
 635
 636 static void customctrl_resize(FileDialogImpl *This, customctrl *ctrl)
 637 {
 638     RECT rc;
 639
 640     switch(ctrl->type)
 641     {
 642     case IDLG_CCTRL_PUSHBUTTON:
 643     case IDLG_CCTRL_COMBOBOX:
 644     case IDLG_CCTRL_CHECKBUTTON:
 645     case IDLG_CCTRL_TEXT:
 646         ctrl_resize(ctrl->hwnd, 160, 160, TRUE);
 647         GetWindowRect(ctrl->hwnd, &rc);
 648         SetWindowPos(ctrl->wrapper_hwnd, NULL, 0, 0, rc.right-rc.left, rc.bottom-rc.top,
 649                      SWP_NOZORDER|SWP_NOMOVE|SWP_NOZORDER);
 650     case IDLG_CCTRL_RADIOBUTTONLIST:
 651     case IDLG_CCTRL_EDITBOX:
 652     case IDLG_CCTRL_SEPARATOR:
 653     case IDLG_CCTRL_MENU:
 654         /* Nothing */
 655         break;
 656     }
 657 }
 658
 659 static LRESULT notifysink_on_create(HWND hwnd, CREATESTRUCTW *crs)
 660 {
 661     FileDialogImpl *This = crs->lpCreateParams;
 662     TRACE("%p\n", This);
 663
 664     SetWindowLongPtrW(hwnd, GWLP_USERDATA, (LPARAM)This);
 665     return TRUE;
 666 }
 667
 668 static LRESULT notifysink_on_bn_clicked(FileDialogImpl *This, HWND hwnd, WPARAM wparam)
 669 {
 670     customctrl *ctrl = get_cctrl_from_dlgid(This, LOWORD(wparam));
 671
 672     TRACE("%p, %lx\n", This, wparam);
 673
 674     if(ctrl)
 675     {
 676         if(ctrl->type == IDLG_CCTRL_CHECKBUTTON)
 677         {
 678             BOOL checked = (SendMessageW(ctrl->hwnd, BM_GETCHECK, 0, 0) == BST_CHECKED);
 679             cctrl_event_OnCheckButtonToggled(This, ctrl->id, checked);
 680         }
 681         else
 682             cctrl_event_OnButtonClicked(This, ctrl->id);
 683     }
 684
 685     return TRUE;
 686 }
 687
 688 static LRESULT notifysink_on_cbn_selchange(FileDialogImpl *This, HWND hwnd, WPARAM wparam)
 689 {
 690     customctrl *ctrl = get_cctrl_from_dlgid(This, LOWORD(wparam));
 691     TRACE("%p, %p (%lx)\n", This, ctrl, wparam);
 692
 693     if(ctrl)
 694     {
 695         UINT index = SendMessageW(ctrl->hwnd, CB_GETCURSEL, 0, 0);
 696         UINT selid = SendMessageW(ctrl->hwnd, CB_GETITEMDATA, index, 0);
 697
 698         cctrl_event_OnItemSelected(This, ctrl->id, selid);
 699     }
 700     return TRUE;
 701 }
 702
 703 static LRESULT notifysink_on_tvn_dropdown(FileDialogImpl *This, LPARAM lparam)
 704 {
 705     NMTOOLBARW *nmtb = (NMTOOLBARW*)lparam;
 706     customctrl *ctrl = get_cctrl_from_dlgid(This, GetDlgCtrlID(nmtb->hdr.hwndFrom));
 707     POINT pt = { 0, nmtb->rcButton.bottom };
 708     TBBUTTON tbb;
 709     UINT idcmd;
 710
 711     TRACE("%p, %p (%lx)\n", This, ctrl, lparam);
 712
 713     if(ctrl)
 714     {
 715         cctrl_event_OnControlActivating(This,ctrl->id);
 716
 717         SendMessageW(ctrl->hwnd, TB_GETBUTTON, 0, (LPARAM)&tbb);
 718         ClientToScreen(ctrl->hwnd, &pt);
 719         idcmd = TrackPopupMenu((HMENU)tbb.dwData, TPM_RETURNCMD, pt.x, pt.y, 0, This->dlg_hwnd, NULL);
 720         if(idcmd)
 721             cctrl_event_OnItemSelected(This, ctrl->id, idcmd);
 722     }
 723
 724     return TBDDRET_DEFAULT;
 725 }
 726
 727 static LRESULT notifysink_on_wm_command(FileDialogImpl *This, HWND hwnd, WPARAM wparam, LPARAM lparam)
 728 {
 729     switch(HIWORD(wparam))
 730     {
 731     case BN_CLICKED:          return notifysink_on_bn_clicked(This, hwnd, wparam);
 732     case CBN_SELCHANGE:       return notifysink_on_cbn_selchange(This, hwnd, wparam);
 733     }
 734
 735     return FALSE;
 736 }
 737
 738 static LRESULT notifysink_on_wm_notify(FileDialogImpl *This, HWND hwnd, WPARAM wparam, LPARAM lparam)
 739 {
 740     NMHDR *nmhdr = (NMHDR*)lparam;
 741
 742     switch(nmhdr->code)
 743     {
 744     case TBN_DROPDOWN:        return notifysink_on_tvn_dropdown(This, lparam);
 745     }
 746
 747     return FALSE;
 748 }
 749
 750 static LRESULT CALLBACK notifysink_proc(HWND hwnd, UINT message, WPARAM wparam, LPARAM lparam)
 751 {
 752     FileDialogImpl *This = (FileDialogImpl*)GetWindowLongPtrW(hwnd, GWLP_USERDATA);
 753     HWND hwnd_child;
 754     RECT rc;
 755
 756     switch(message)
 757     {
 758     case WM_NCCREATE:         return notifysink_on_create(hwnd, (CREATESTRUCTW*)lparam);
 759     case WM_COMMAND:          return notifysink_on_wm_command(This, hwnd, wparam, lparam);
 760     case WM_NOTIFY:           return notifysink_on_wm_notify(This, hwnd, wparam, lparam);
 761     case WM_SIZE:
 762         hwnd_child = GetPropW(hwnd, notifysink_childW);
 763         GetClientRect(hwnd, &rc);
 764         SetWindowPos(hwnd_child, NULL, 0, 0, rc.right, rc.bottom, SWP_NOMOVE|SWP_NOACTIVATE|SWP_NOZORDER);
 765         return TRUE;
 766     }
 767
 768     return DefWindowProcW(hwnd, message, wparam, lparam);
 769 }
 770
 771 static HRESULT cctrl_create_new(FileDialogImpl *This, DWORD id,
 772                                 LPCWSTR text, LPCWSTR wndclass, DWORD ctrl_wsflags,
 773                                 DWORD ctrl_exflags, UINT height, customctrl **ppctrl)
 774 {
 775     HWND ns_hwnd, control_hwnd;
 776     DWORD wsflags = WS_CHILD | WS_VISIBLE | WS_CLIPSIBLINGS;
 777     customctrl *ctrl;
 778
 779     if(get_cctrl(This, id))
 780         return E_UNEXPECTED; /* Duplicate id */
 781
 782     ns_hwnd = CreateWindowExW(0, floatnotifysinkW, NULL, wsflags,
 783                               0, 0, This->cctrl_width, height, This->cctrls_hwnd,
 784                               (HMENU)This->cctrl_next_dlgid, COMDLG32_hInstance, This);
 785     control_hwnd = CreateWindowExW(ctrl_exflags, wndclass, text, wsflags | ctrl_wsflags,
 786                                    0, 0, This->cctrl_width, height, ns_hwnd,
 787                                    (HMENU)This->cctrl_next_dlgid, COMDLG32_hInstance, 0);
 788
 789     if(!ns_hwnd || !control_hwnd)
 790     {
 791         ERR("Failed to create wrapper (%p) or control (%p)\n", ns_hwnd, control_hwnd);
 792         DestroyWindow(ns_hwnd);
 793         DestroyWindow(control_hwnd);
 794
 795         return E_FAIL;
 796     }
 797
 798     SetPropW(ns_hwnd, notifysink_childW, control_hwnd);
 799
 800     ctrl = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(customctrl));
 801     if(!ctrl)
 802         return E_OUTOFMEMORY;
 803
 804     ctrl->hwnd = control_hwnd;
 805     ctrl->wrapper_hwnd = ns_hwnd;
 806     ctrl->id = id;
 807     ctrl->dlgid = This->cctrl_next_dlgid;
 808     ctrl->cdcstate = CDCS_ENABLED | CDCS_VISIBLE;
 809     list_add_tail(&This->cctrls, &ctrl->entry);
 810     if(ppctrl) *ppctrl = ctrl;
 811
 812     This->cctrl_next_dlgid++;
 813     return S_OK;
 814 }
 815
 816 /**************************************************************************
 817  * Container functions.
 818  */
 819 static UINT ctrl_container_resize(FileDialogImpl *This, UINT container_width)
 820 {
 821     UINT container_height;
 822     UINT column_width;
 823     UINT nr_of_cols;
 824     UINT max_control_height, total_height = 0;
 825     UINT cur_col_pos, cur_row_pos;
 826     customctrl *ctrl;
 827     BOOL fits_height;
 828     static const UINT col_indent = 100; /* The first column is indented 100px */
 829     static const UINT cspacing = 90;    /* Columns are spaced with 90px */
 830     static const UINT rspacing = 4;     /* Rows are spaced with 4 px. */
 831
 832     /* Given the new width of the container, this function determines the
 833      * needed height of the container and places the controls according to
 834      * the new layout. Returns the new height.
 835      */
 836
 837     TRACE("%p\n", This);
 838
 839     column_width = This->cctrl_width + cspacing;
 840     nr_of_cols = (container_width - col_indent + cspacing) / column_width;
 841
 842     /* We don't need to do anything unless the number of visible columns has changed. */
 843     if(nr_of_cols == This->cctrls_cols)
 844     {
 845         RECT rc;
 846         GetWindowRect(This->cctrls_hwnd, &rc);
 847         return rc.bottom - rc.top;
 848     }
 849
 850    This->cctrls_cols = nr_of_cols;
 851
 852     /* Get the size of the tallest control, and the total size of
 853      * all the controls to figure out the number of slots we need.
 854      */
 855     max_control_height = 0;
 856     LIST_FOR_EACH_ENTRY(ctrl, &This->cctrls, customctrl, entry)
 857     {
 858         if(ctrl->cdcstate & CDCS_VISIBLE)
 859         {
 860             RECT rc;
 861             UINT control_height;
 862             GetWindowRect(ctrl->wrapper_hwnd, &rc);
 863             control_height = rc.bottom - rc.top;
 864             max_control_height = max(max_control_height, control_height);
 865
 866             total_height +=  control_height + rspacing;
 867         }
 868     }
 869
 870     if(!total_height)
 871         return 0;
 872
 873     container_height = max(total_height / nr_of_cols, max_control_height + rspacing);
 874     TRACE("Guess: container_height: %d\n",container_height);
 875
 876     /* Incrementally increase container_height until all the controls
 877      * fit.
 878      */
 879     do {
 880         UINT columns_needed = 1;
 881         cur_row_pos = 0;
 882
 883         fits_height = TRUE;
 884         LIST_FOR_EACH_ENTRY(ctrl, &This->cctrls, customctrl, entry)
 885         {
 886             if(ctrl->cdcstate & CDCS_VISIBLE)
 887             {
 888                 RECT rc;
 889                 UINT control_height;
 890                 GetWindowRect(ctrl->wrapper_hwnd, &rc);
 891                 control_height = rc.bottom - rc.top;
 892
 893                 if(cur_row_pos + control_height > container_height)
 894                 {
 895                     if(++columns_needed > nr_of_cols)
 896                     {
 897                         container_height += 1;
 898                         fits_height = FALSE;
 899                         break;
 900                     }
 901                     cur_row_pos = 0;
 902                 }
 903
 904                 cur_row_pos += control_height + rspacing;
 905             }
 906         }
 907     } while(!fits_height);
 908
 909     TRACE("Final container height: %d\n", container_height);
 910
 911     /* Move the controls to their final destination
 912      */
 913     cur_col_pos = col_indent, cur_row_pos = 0;
 914     LIST_FOR_EACH_ENTRY(ctrl, &This->cctrls, customctrl, entry)
 915     {
 916         if(ctrl->cdcstate & CDCS_VISIBLE)
 917         {
 918             RECT rc;
 919             UINT control_height;
 920             GetWindowRect(ctrl->wrapper_hwnd, &rc);
 921             control_height = rc.bottom - rc.top;
 922
 923             if(cur_row_pos + control_height > container_height)
 924             {
 925                 cur_row_pos = 0;
 926                 cur_col_pos += This->cctrl_width + cspacing;
 927             }
 928
 929             SetWindowPos(ctrl->wrapper_hwnd, NULL, cur_col_pos, cur_row_pos, 0, 0,
 930                          SWP_NOACTIVATE | SWP_NOSIZE | SWP_NOZORDER);
 931
 932             cur_row_pos += control_height + rspacing;
 933         }
 934     }
 935
 936     /* Sanity check */
 937     if(cur_row_pos + This->cctrl_width > container_width)
 938         ERR("-- Failed to place controls properly.\n");
 939
 940     return container_height;
 941 }
 942
 943 static void ctrl_container_reparent(FileDialogImpl *This, HWND parent)
 944 {
 945     LONG wndstyle;
 946
 947     if(parent)
 948     {
 949         customctrl *ctrl;
 950         HFONT font;
 951
 952         wndstyle = GetWindowLongW(This->cctrls_hwnd, GWL_STYLE);
 953         wndstyle &= ~(WS_POPUP);
 954         wndstyle |= WS_CHILD;
 955         SetWindowLongW(This->cctrls_hwnd, GWL_STYLE, wndstyle);
 956
 957         SetParent(This->cctrls_hwnd, parent);
 958         ShowWindow(This->cctrls_hwnd, TRUE);
 959
 960         /* Set the fonts to match the dialog font. */
 961         font = (HFONT)SendMessageW(parent, WM_GETFONT, 0, 0);
 962         if(!font)
 963             ERR("Failed to get font handle from dialog.\n");
 964
 965         LIST_FOR_EACH_ENTRY(ctrl, &This->cctrls, customctrl, entry)
 966         {
 967             if(font) SendMessageW(ctrl->hwnd, WM_SETFONT, (WPARAM)font, TRUE);
 968             customctrl_resize(This, ctrl);
 969         }
 970     }
 971     else
 972     {
 973         ShowWindow(This->cctrls_hwnd, FALSE);
 974
 975         wndstyle = GetWindowLongW(This->cctrls_hwnd, GWL_STYLE);
 976         wndstyle &= ~(WS_CHILD);
 977         wndstyle |= WS_POPUP;
 978         SetWindowLongW(This->cctrls_hwnd, GWL_STYLE, wndstyle);
 979
 980         SetParent(This->cctrls_hwnd, NULL);
 981     }
 982 }
 983
 984 static LRESULT ctrl_container_on_create(HWND hwnd, CREATESTRUCTW *crs)
 985 {
 986     FileDialogImpl *This = crs->lpCreateParams;
 987     TRACE("%p\n", This);
 988
 989     SetWindowLongPtrW(hwnd, GWLP_USERDATA, (LPARAM)This);
 990     return TRUE;
 991 }
 992
 993 static LRESULT ctrl_container_on_wm_destroy(FileDialogImpl *This)
 994 {
 995     customctrl *cur1, *cur2;
 996     TRACE("%p\n", This);
 997
 998     LIST_FOR_EACH_ENTRY_SAFE(cur1, cur2, &This->cctrls, customctrl, entry)
 999     {
1000         TRACE("Freeing control %p\n", cur1);
1001         list_remove(&cur1->entry);
1002
1003         if(cur1->type == IDLG_CCTRL_MENU)
1004         {
1005             TBBUTTON tbb;
1006             SendMessageW(cur1->hwnd, TB_GETBUTTON, 0, (LPARAM)&tbb);
1007             DestroyMenu((HMENU)tbb.dwData);
1008         }
1009
1010         DestroyWindow(cur1->hwnd);
1011         HeapFree(GetProcessHeap(), 0, cur1);
1012     }
1013
1014     return TRUE;
1015 }
1016
1017 static LRESULT CALLBACK ctrl_container_wndproc(HWND hwnd, UINT umessage, WPARAM wparam, LPARAM lparam)
1018 {
1019     FileDialogImpl *This = (FileDialogImpl*)GetWindowLongPtrW(hwnd, GWLP_USERDATA);
1020
1021     switch(umessage)
1022     {
1023     case WM_NCCREATE:         return ctrl_container_on_create(hwnd, (CREATESTRUCTW*)lparam);
1024     case WM_DESTROY:          return ctrl_container_on_wm_destroy(This);
1025     default:                  return DefWindowProcW(hwnd, umessage, wparam, lparam);
1026     }
1027
1028     return FALSE;
1029 }
1030
1031 static HRESULT init_custom_controls(FileDialogImpl *This)
1032 {
1033     WNDCLASSW wc;
1034     static const WCHAR ctrl_container_classname[] =
1035         {'i','d','l','g','_','c','o','n','t','a','i','n','e','r','_','p','a','n','e',0};
1036
1037     InitCommonControlsEx(NULL);
1038
1039     This->cctrl_width = 160;      /* Controls have a fixed width */
1040     This->cctrl_def_height = 23;
1041     This->cctrls_cols = 0;
1042
1043     This->cctrl_next_dlgid = 0x2000;
1044     list_init(&This->cctrls);
1045
1046     if( !GetClassInfoW(COMDLG32_hInstance, ctrl_container_classname, &wc) )
1047     {
1048         wc.style            = CS_HREDRAW | CS_VREDRAW;
1049         wc.lpfnWndProc      = ctrl_container_wndproc;
1050         wc.cbClsExtra       = 0;
1051         wc.cbWndExtra       = 0;
1052         wc.hInstance        = COMDLG32_hInstance;
1053         wc.hIcon            = 0;
1054         wc.hCursor          = LoadCursorW(0, (LPWSTR)IDC_ARROW);
1055         wc.hbrBackground    = (HBRUSH)(COLOR_BTNFACE + 1);
1056         wc.lpszMenuName     = NULL;
1057         wc.lpszClassName    = ctrl_container_classname;
1058
1059         if(!RegisterClassW(&wc)) return E_FAIL;
1060     }
1061
1062     This->cctrls_hwnd = CreateWindowExW(0, ctrl_container_classname, NULL,
1063                                         WS_CLIPSIBLINGS | WS_CLIPCHILDREN,
1064                                         0, 0, 0, 0, NULL, 0,
1065                                         COMDLG32_hInstance, (void*)This);
1066     if(!This->cctrls_hwnd)
1067         return E_FAIL;
1068
1069     SetWindowLongW(This->cctrls_hwnd, GWL_STYLE, WS_TABSTOP);
1070
1071     /* Register class for  */
1072     if( !GetClassInfoW(COMDLG32_hInstance, floatnotifysinkW, &wc) ||
1073         wc.hInstance != COMDLG32_hInstance)
1074     {
1075         wc.style            = CS_HREDRAW | CS_VREDRAW;
1076         wc.lpfnWndProc      = notifysink_proc;
1077         wc.cbClsExtra       = 0;
1078         wc.cbWndExtra       = 0;
1079         wc.hInstance        = COMDLG32_hInstance;
1080         wc.hIcon            = 0;
1081         wc.hCursor          = LoadCursorW(0, (LPWSTR)IDC_ARROW);
1082         wc.hbrBackground    = (HBRUSH)(COLOR_BTNFACE + 1);
1083         wc.lpszMenuName     = NULL;
1084         wc.lpszClassName    = floatnotifysinkW;
1085
1086         if (!RegisterClassW(&wc))
1087             ERR("Failed to register FloatNotifySink window class.\n");
1088     }
1089
1090     return S_OK;
1091 }
1092
1093 /**************************************************************************
1094  * Window related functions.
1095  */
1096 static SIZE update_layout(FileDialogImpl *This)
1097 {
1098     HDWP hdwp;
1099     HWND hwnd;
1100     RECT dialog_rc;
1101     RECT cancel_rc, open_rc;
1102     RECT filetype_rc, filename_rc, filenamelabel_rc;
1103     RECT toolbar_rc, ebrowser_rc, customctrls_rc;
1104     int missing_width, missing_height;
1105     static const UINT vspacing = 4, hspacing = 4;
1106     SIZE ret;
1107
1108     GetClientRect(This->dlg_hwnd, &dialog_rc);
1109
1110     missing_width = max(0, 320 - dialog_rc.right);
1111     missing_height = max(0, 200 - dialog_rc.bottom);
1112
1113     if(missing_width || missing_height)
1114     {
1115         TRACE("Missing (%d, %d)\n", missing_width, missing_height);
1116         ret.cx = missing_width;
1117         ret.cy = missing_height;
1118         return ret;
1119     }
1120
1121     /****
1122      * Calculate the size of the dialog and all the parts.
1123      */
1124
1125     /* Cancel button */
1126     hwnd = GetDlgItem(This->dlg_hwnd, IDCANCEL);
1127     if(hwnd)
1128     {
1129         int cancel_width, cancel_height;
1130         GetWindowRect(hwnd, &cancel_rc);
1131         cancel_width = cancel_rc.right - cancel_rc.left;
1132         cancel_height = cancel_rc.bottom - cancel_rc.top;
1133
1134         cancel_rc.left = dialog_rc.right - cancel_width - hspacing;
1135         cancel_rc.top = dialog_rc.bottom - cancel_height - vspacing;
1136         cancel_rc.right = cancel_rc.left + cancel_width;
1137         cancel_rc.bottom = cancel_rc.top + cancel_height;
1138     }
1139
1140     /* Open/Save button */
1141     hwnd = GetDlgItem(This->dlg_hwnd, IDOK);
1142     if(hwnd)
1143     {
1144         int open_width, open_height;
1145         GetWindowRect(hwnd, &open_rc);
1146         open_width = open_rc.right - open_rc.left;
1147         open_height = open_rc.bottom - open_rc.top;
1148
1149         open_rc.left = cancel_rc.left - open_width - hspacing;
1150         open_rc.top = cancel_rc.top;
1151         open_rc.right = open_rc.left + open_width;
1152         open_rc.bottom = open_rc.top + open_height;
1153     }
1154
1155     /* The filetype combobox. */
1156     hwnd = GetDlgItem(This->dlg_hwnd, IDC_FILETYPE);
1157     if(hwnd)
1158     {
1159         int filetype_width, filetype_height;
1160         GetWindowRect(hwnd, &filetype_rc);
1161
1162         filetype_width = filetype_rc.right - filetype_rc.left;
1163         filetype_height = filetype_rc.bottom - filetype_rc.top;
1164
1165         filetype_rc.right = cancel_rc.right;
1166
1167         filetype_rc.left = filetype_rc.right - filetype_width;
1168         filetype_rc.top = cancel_rc.top - filetype_height - vspacing;
1169         filetype_rc.bottom = filetype_rc.top + filetype_height;
1170
1171         if(!This->filterspec_count)
1172             filetype_rc.left = filetype_rc.right;
1173     }
1174
1175     /* Filename label. */
1176     hwnd = GetDlgItem(This->dlg_hwnd, IDC_FILENAMESTATIC);
1177     if(hwnd)
1178     {
1179         int filetypelabel_width, filetypelabel_height;
1180         GetWindowRect(hwnd, &filenamelabel_rc);
1181
1182         filetypelabel_width = filenamelabel_rc.right - filenamelabel_rc.left;
1183         filetypelabel_height = filenamelabel_rc.bottom - filenamelabel_rc.top;
1184
1185         filenamelabel_rc.left = 160; /* FIXME */
1186         filenamelabel_rc.top = filetype_rc.top;
1187         filenamelabel_rc.right = filenamelabel_rc.left + filetypelabel_width;
1188         filenamelabel_rc.bottom = filenamelabel_rc.top + filetypelabel_height;
1189     }
1190
1191     /* Filename edit box. */
1192     hwnd = GetDlgItem(This->dlg_hwnd, IDC_FILENAME);
1193     if(hwnd)
1194     {
1195         int filename_width, filename_height;
1196         GetWindowRect(hwnd, &filename_rc);
1197
1198         filename_width = filetype_rc.left - filenamelabel_rc.right - hspacing*2;
1199         filename_height = filename_rc.bottom - filename_rc.top;
1200
1201         filename_rc.left = filenamelabel_rc.right + hspacing;
1202         filename_rc.top = filetype_rc.top;
1203         filename_rc.right = filename_rc.left + filename_width;
1204         filename_rc.bottom = filename_rc.top + filename_height;
1205     }
1206
1207     hwnd = GetDlgItem(This->dlg_hwnd, IDC_NAV_TOOLBAR);
1208     if(hwnd)
1209     {
1210         GetWindowRect(hwnd, &toolbar_rc);
1211         MapWindowPoints(NULL, This->dlg_hwnd, (POINT*)&toolbar_rc, 2);
1212     }
1213
1214     /* The custom controls */
1215     customctrls_rc.left = dialog_rc.left + vspacing;
1216     customctrls_rc.right = dialog_rc.right - vspacing;
1217     customctrls_rc.bottom = filename_rc.top - hspacing;
1218     customctrls_rc.top = customctrls_rc.bottom -
1219         ctrl_container_resize(This, customctrls_rc.right - customctrls_rc.left);
1220
1221     /* The ExplorerBrowser control. */
1222     ebrowser_rc.left = dialog_rc.left + vspacing;
1223     ebrowser_rc.top = toolbar_rc.bottom + vspacing;
1224     ebrowser_rc.right = dialog_rc.right - hspacing;
1225     ebrowser_rc.bottom = customctrls_rc.top - hspacing;
1226
1227     /****
1228      * Move everything to the right place.
1229      */
1230
1231     /* FIXME: The Save Dialog uses a slightly different layout. */
1232     hdwp = BeginDeferWindowPos(7);
1233
1234     if(hdwp && This->peb)
1235         IExplorerBrowser_SetRect(This->peb, &hdwp, ebrowser_rc);
1236
1237     if(hdwp && This->cctrls_hwnd)
1238         DeferWindowPos(hdwp, This->cctrls_hwnd, NULL,
1239                        customctrls_rc.left, customctrls_rc.top,
1240                        customctrls_rc.right - customctrls_rc.left, customctrls_rc.bottom - customctrls_rc.top,
1241                        SWP_NOZORDER | SWP_NOACTIVATE);
1242
1243     /* The default controls */
1244     if(hdwp && (hwnd = GetDlgItem(This->dlg_hwnd, IDC_FILETYPE)) )
1245         DeferWindowPos(hdwp, hwnd, NULL, filetype_rc.left, filetype_rc.top, 0, 0,
1246                        SWP_NOZORDER | SWP_NOSIZE | SWP_NOACTIVATE);
1247
1248     if(hdwp && (hwnd = GetDlgItem(This->dlg_hwnd, IDC_FILENAME)) )
1249         DeferWindowPos(hdwp, hwnd, NULL, filename_rc.left, filename_rc.top,
1250                        filename_rc.right - filename_rc.left, filename_rc.bottom - filename_rc.top,
1251                        SWP_NOZORDER | SWP_NOACTIVATE);
1252
1253     if(hdwp && (hwnd = GetDlgItem(This->dlg_hwnd, IDC_FILENAMESTATIC)) )
1254         DeferWindowPos(hdwp, hwnd, NULL, filenamelabel_rc.left, filenamelabel_rc.top, 0, 0,
1255                        SWP_NOZORDER | SWP_NOSIZE | SWP_NOACTIVATE);
1256
1257     if(hdwp && (hwnd = GetDlgItem(This->dlg_hwnd, IDOK)) )
1258         DeferWindowPos(hdwp, hwnd, NULL, open_rc.left, open_rc.top, 0, 0,
1259                        SWP_NOZORDER | SWP_NOSIZE | SWP_NOACTIVATE);
1260
1261     if(hdwp && (hwnd = GetDlgItem(This->dlg_hwnd, IDCANCEL)) )
1262         DeferWindowPos(hdwp, hwnd, NULL, cancel_rc.left, cancel_rc.top, 0, 0,
1263                        SWP_NOZORDER | SWP_NOSIZE | SWP_NOACTIVATE);
1264
1265     if(hdwp)
1266         EndDeferWindowPos(hdwp);
1267     else
1268         ERR("Failed to position dialog controls.\n");
1269
1270     ret.cx = 0; ret.cy = 0;
1271     return ret;
1272 }
1273
1274 static HRESULT init_explorerbrowser(FileDialogImpl *This)
1275 {
1276     IShellItem *psi_folder;
1277     FOLDERSETTINGS fos;
1278     RECT rc = {0};
1279     HRESULT hr;
1280
1281     /* Create ExplorerBrowser instance */
1282     OleInitialize(NULL);
1283
1284     hr = CoCreateInstance(&CLSID_ExplorerBrowser, NULL, CLSCTX_INPROC_SERVER,
1285                           &IID_IExplorerBrowser, (void**)&This->peb);
1286     if(FAILED(hr))
1287     {
1288         ERR("Failed to instantiate ExplorerBrowser control.\n");
1289         return hr;
1290     }
1291
1292     IExplorerBrowser_SetOptions(This->peb, EBO_SHOWFRAMES);
1293
1294     hr = IExplorerBrowser_Initialize(This->peb, This->dlg_hwnd, &rc, NULL);
1295     if(FAILED(hr))
1296     {
1297         ERR("Failed to initialize the ExplorerBrowser control.\n");
1298         IExplorerBrowser_Release(This->peb);
1299         This->peb = NULL;
1300         return hr;
1301     }
1302     hr = IExplorerBrowser_Advise(This->peb, &This->IExplorerBrowserEvents_iface, &This->ebevents_cookie);
1303     if(FAILED(hr))
1304         ERR("Advise (ExplorerBrowser) failed.\n");
1305
1306     /* Get previous options? */
1307     fos.ViewMode = fos.fFlags = 0;
1308     if(!(This->options & FOS_ALLOWMULTISELECT))
1309         fos.fFlags |= FWF_SINGLESEL;
1310
1311     IExplorerBrowser_SetFolderSettings(This->peb, &fos);
1312
1313     hr = IUnknown_SetSite((IUnknown*)This->peb, (IUnknown*)This);
1314     if(FAILED(hr))
1315         ERR("SetSite (ExplorerBrowser) failed.\n");
1316
1317     /* Browse somewhere */
1318     psi_folder = This->psi_setfolder ? This->psi_setfolder : This->psi_defaultfolder;
1319     IExplorerBrowser_BrowseToObject(This->peb, (IUnknown*)psi_folder, SBSP_DEFBROWSER);
1320
1321     return S_OK;
1322 }
1323
1324 static void init_toolbar(FileDialogImpl *This, HWND hwnd)
1325 {
1326     HWND htoolbar;
1327     TBADDBITMAP tbab;
1328     TBBUTTON button[2];
1329
1330     htoolbar = CreateWindowExW(0, TOOLBARCLASSNAMEW, NULL, TBSTYLE_FLAT | WS_CHILD | WS_VISIBLE,
1331                                0, 0, 0, 0,
1332                                hwnd, (HMENU)IDC_NAV_TOOLBAR, NULL, NULL);
1333
1334     tbab.hInst = HINST_COMMCTRL;
1335     tbab.nID = IDB_HIST_LARGE_COLOR;
1336     SendMessageW(htoolbar, TB_ADDBITMAP, 0, (LPARAM)&tbab);
1337
1338     button[0].iBitmap = HIST_BACK;
1339     button[0].idCommand = IDC_NAVBACK;
1340     button[0].fsState = TBSTATE_ENABLED;
1341     button[0].fsStyle = BTNS_BUTTON;
1342     button[0].dwData = 0;
1343     button[0].iString = 0;
1344
1345     button[1].iBitmap = HIST_FORWARD;
1346     button[1].idCommand = IDC_NAVFORWARD;
1347     button[1].fsState = TBSTATE_ENABLED;
1348     button[1].fsStyle = BTNS_BUTTON;
1349     button[1].dwData = 0;
1350     button[1].iString = 0;
1351
1352     SendMessageW(htoolbar, TB_ADDBUTTONSW, 2, (LPARAM)&button);
1353     SendMessageW(htoolbar, TB_SETBUTTONSIZE, 0, MAKELPARAM(24,24));
1354     SendMessageW(htoolbar, TB_AUTOSIZE, 0, 0);
1355 }
1356
1357 static void update_control_text(FileDialogImpl *This)
1358 {
1359     HWND hitem;
1360     if(This->custom_title)
1361         SetWindowTextW(This->dlg_hwnd, This->custom_title);
1362
1363     if(This->custom_okbutton &&
1364        (hitem = GetDlgItem(This->dlg_hwnd, IDOK)))
1365     {
1366         SetWindowTextW(hitem, This->custom_okbutton);
1367         ctrl_resize(hitem, 50, 250, FALSE);
1368     }
1369
1370     if(This->custom_cancelbutton &&
1371        (hitem = GetDlgItem(This->dlg_hwnd, IDCANCEL)))
1372     {
1373         SetWindowTextW(hitem, This->custom_cancelbutton);
1374         ctrl_resize(hitem, 50, 250, FALSE);
1375     }
1376
1377     if(This->custom_filenamelabel &&
1378        (hitem = GetDlgItem(This->dlg_hwnd, IDC_FILENAMESTATIC)))
1379     {
1380         SetWindowTextW(hitem, This->custom_filenamelabel);
1381         ctrl_resize(hitem, 50, 250, FALSE);
1382     }
1383 }
1384
1385 static LRESULT on_wm_initdialog(HWND hwnd, LPARAM lParam)
1386 {
1387     FileDialogImpl *This = (FileDialogImpl*)lParam;
1388     HWND hitem;
1389
1390     TRACE("(%p, %p)\n", This, hwnd);
1391
1392     SetWindowLongPtrW(hwnd, GWLP_USERDATA, (LPARAM)This);
1393     This->dlg_hwnd = hwnd;
1394
1395     hitem = GetDlgItem(This->dlg_hwnd, pshHelp);
1396     if(hitem) ShowWindow(hitem, SW_HIDE);
1397
1398     hitem = GetDlgItem(This->dlg_hwnd, IDC_FILETYPESTATIC);
1399     if(hitem) ShowWindow(hitem, SW_HIDE);
1400
1401     /* Fill filetypes combobox, or hide it. */
1402     hitem = GetDlgItem(This->dlg_hwnd, IDC_FILETYPE);
1403     if(This->filterspec_count)
1404     {
1405         UINT i;
1406         for(i = 0; i < This->filterspec_count; i++)
1407             SendMessageW(hitem, CB_ADDSTRING, 0, (LPARAM)This->filterspecs[i].pszName);
1408
1409         SendMessageW(hitem, CB_SETCURSEL, This->filetypeindex, 0);
1410     }
1411     else
1412         ShowWindow(hitem, SW_HIDE);
1413
1414     if(This->set_filename &&
1415        (hitem = GetDlgItem(This->dlg_hwnd, IDC_FILENAME)) )
1416         SendMessageW(hitem, WM_SETTEXT, 0, (LPARAM)This->set_filename);
1417
1418     ctrl_container_reparent(This, This->dlg_hwnd);
1419     init_explorerbrowser(This);
1420     init_toolbar(This, hwnd);
1421     update_control_text(This);
1422     update_layout(This);
1423
1424     return TRUE;
1425 }
1426
1427 static LRESULT on_wm_size(FileDialogImpl *This)
1428 {
1429     update_layout(This);
1430     return FALSE;
1431 }
1432
1433 static LRESULT on_wm_getminmaxinfo(FileDialogImpl *This, LPARAM lparam)
1434 {
1435     MINMAXINFO *mmi = (MINMAXINFO*)lparam;
1436     TRACE("%p (%p)\n", This, mmi);
1437
1438     /* FIXME */
1439     mmi->ptMinTrackSize.x = 640;
1440     mmi->ptMinTrackSize.y = 480;
1441
1442     return FALSE;
1443 }
1444
1445 static LRESULT on_wm_destroy(FileDialogImpl *This)
1446 {
1447     TRACE("%p\n", This);
1448
1449     if(This->peb)
1450     {
1451         IExplorerBrowser_Destroy(This->peb);
1452         IExplorerBrowser_Release(This->peb);
1453         This->peb = NULL;
1454     }
1455
1456     ctrl_container_reparent(This, NULL);
1457     This->dlg_hwnd = NULL;
1458
1459     return TRUE;
1460 }
1461
1462 static LRESULT on_idok(FileDialogImpl *This)
1463 {
1464     TRACE("%p\n", This);
1465
1466     if(SUCCEEDED(on_default_action(This)))
1467         EndDialog(This->dlg_hwnd, S_OK);
1468
1469     return FALSE;
1470 }
1471
1472 static LRESULT on_idcancel(FileDialogImpl *This)
1473 {
1474     TRACE("%p\n", This);
1475
1476     EndDialog(This->dlg_hwnd, HRESULT_FROM_WIN32(ERROR_CANCELLED));
1477
1478     return FALSE;
1479 }
1480
1481 static LRESULT on_browse_back(FileDialogImpl *This)
1482 {
1483     TRACE("%p\n", This);
1484     IExplorerBrowser_BrowseToIDList(This->peb, NULL, SBSP_NAVIGATEBACK);
1485     return FALSE;
1486 }
1487
1488 static LRESULT on_browse_forward(FileDialogImpl *This)
1489 {
1490     TRACE("%p\n", This);
1491     IExplorerBrowser_BrowseToIDList(This->peb, NULL, SBSP_NAVIGATEFORWARD);
1492     return FALSE;
1493 }
1494
1495 static LRESULT on_command_filetype(FileDialogImpl *This, WPARAM wparam, LPARAM lparam)
1496 {
1497     if(HIWORD(wparam) == CBN_SELCHANGE)
1498     {
1499         IShellView *psv;
1500         HRESULT hr;
1501         LPWSTR filename;
1502         UINT prev_index = This->filetypeindex;
1503
1504         This->filetypeindex = SendMessageW((HWND)lparam, CB_GETCURSEL, 0, 0);
1505         TRACE("File type selection changed to %d.\n", This->filetypeindex);
1506
1507         if(prev_index == This->filetypeindex)
1508             return FALSE;
1509
1510         hr = IExplorerBrowser_GetCurrentView(This->peb, &IID_IShellView, (void**)&psv);
1511         if(SUCCEEDED(hr))
1512         {
1513             IShellView_Refresh(psv);
1514             IShellView_Release(psv);
1515         }
1516
1517         if(This->dlg_type == ITEMDLG_TYPE_SAVE && get_file_name(This, &filename))
1518         {
1519             WCHAR buf[MAX_PATH], extbuf[MAX_PATH], *ext;
1520
1521             ext = get_first_ext_from_spec(extbuf, This->filterspecs[This->filetypeindex].pszSpec);
1522             if(ext)
1523             {
1524                 lstrcpyW(buf, filename);
1525
1526                 if(PathMatchSpecW(buf, This->filterspecs[prev_index].pszSpec))
1527                     PathRemoveExtensionW(buf);
1528
1529                 lstrcatW(buf, ext);
1530                 set_file_name(This, buf);
1531             }
1532             CoTaskMemFree(filename);
1533         }
1534     }
1535
1536     return FALSE;
1537 }
1538
1539 static LRESULT on_wm_command(FileDialogImpl *This, WPARAM wparam, LPARAM lparam)
1540 {
1541     switch(LOWORD(wparam))
1542     {
1543     case IDOK:                return on_idok(This);
1544     case IDCANCEL:            return on_idcancel(This);
1545     case IDC_NAVBACK:         return on_browse_back(This);
1546     case IDC_NAVFORWARD:      return on_browse_forward(This);
1547     case IDC_FILETYPE:        return on_command_filetype(This, wparam, lparam);
1548     default:                  TRACE("Unknown command.\n");
1549     }
1550     return FALSE;
1551 }
1552
1553 static LRESULT CALLBACK itemdlg_dlgproc(HWND hwnd, UINT umessage, WPARAM wparam, LPARAM lparam)
1554 {
1555     FileDialogImpl *This = (FileDialogImpl*)GetWindowLongPtrW(hwnd, GWLP_USERDATA);
1556
1557     switch(umessage)
1558     {
1559     case WM_INITDIALOG:       return on_wm_initdialog(hwnd, lparam);
1560     case WM_COMMAND:          return on_wm_command(This, wparam, lparam);
1561     case WM_SIZE:             return on_wm_size(This);
1562     case WM_GETMINMAXINFO:    return on_wm_getminmaxinfo(This, lparam);
1563     case WM_DESTROY:          return on_wm_destroy(This);
1564     }
1565
1566     return FALSE;
1567 }
1568
1569 static HRESULT create_dialog(FileDialogImpl *This, HWND parent)
1570 {
1571     INT_PTR res;
1572
1573     SetLastError(0);
1574     res = DialogBoxParamW(COMDLG32_hInstance,
1575                           MAKEINTRESOURCEW(NEWFILEOPENV3ORD),
1576                           parent, itemdlg_dlgproc, (LPARAM)This);
1577     This->dlg_hwnd = NULL;
1578     if(res == -1)
1579     {
1580         ERR("Failed to show dialog (LastError: %d)\n", GetLastError());
1581         return E_FAIL;
1582     }
1583
1584     TRACE("Returning 0x%08x\n", (HRESULT)res);
1585     return (HRESULT)res;
1586 }
1587
1588 /**************************************************************************
1589  * IFileDialog implementation
1590  */
1591 static inline FileDialogImpl *impl_from_IFileDialog2(IFileDialog2 *iface)
1592 {
1593     return CONTAINING_RECORD(iface, FileDialogImpl, IFileDialog2_iface);
1594 }
1595
1596 static HRESULT WINAPI IFileDialog2_fnQueryInterface(IFileDialog2 *iface,
1597                                                     REFIID riid,
1598                                                     void **ppvObject)
1599 {
1600     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1601     TRACE("%p (%s, %p)\n", This, debugstr_guid(riid), ppvObject);
1602
1603     *ppvObject = NULL;
1604     if(IsEqualGUID(riid, &IID_IUnknown) ||
1605        IsEqualGUID(riid, &IID_IFileDialog) ||
1606        IsEqualGUID(riid, &IID_IFileDialog2))
1607     {
1608         *ppvObject = iface;
1609     }
1610     else if(IsEqualGUID(riid, &IID_IFileOpenDialog) && This->dlg_type == ITEMDLG_TYPE_OPEN)
1611     {
1612         *ppvObject = &This->u.IFileOpenDialog_iface;
1613     }
1614     else if(IsEqualGUID(riid, &IID_IFileSaveDialog) && This->dlg_type == ITEMDLG_TYPE_SAVE)
1615     {
1616         *ppvObject = &This->u.IFileSaveDialog_iface;
1617     }
1618     else if(IsEqualGUID(riid, &IID_IExplorerBrowserEvents))
1619     {
1620         *ppvObject = &This->IExplorerBrowserEvents_iface;
1621     }
1622     else if(IsEqualGUID(riid, &IID_IServiceProvider))
1623     {
1624         *ppvObject = &This->IServiceProvider_iface;
1625     }
1626     else if(IsEqualGUID(&IID_ICommDlgBrowser3, riid) ||
1627             IsEqualGUID(&IID_ICommDlgBrowser2, riid) ||
1628             IsEqualGUID(&IID_ICommDlgBrowser, riid))
1629     {
1630         *ppvObject = &This->ICommDlgBrowser3_iface;
1631     }
1632     else if(IsEqualGUID(&IID_IOleWindow, riid))
1633     {
1634         *ppvObject = &This->IOleWindow_iface;
1635     }
1636     else if(IsEqualGUID(riid, &IID_IFileDialogCustomize) ||
1637             IsEqualGUID(riid, &IID_IFileDialogCustomizeAlt))
1638     {
1639         *ppvObject = &This->IFileDialogCustomize_iface;
1640     }
1641     else
1642         FIXME("Unknown interface requested: %s.\n", debugstr_guid(riid));
1643
1644     if(*ppvObject)
1645     {
1646         IUnknown_AddRef((IUnknown*)*ppvObject);
1647         return S_OK;
1648     }
1649
1650     return E_NOINTERFACE;
1651 }
1652
1653 static ULONG WINAPI IFileDialog2_fnAddRef(IFileDialog2 *iface)
1654 {
1655     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1656     LONG ref = InterlockedIncrement(&This->ref);
1657     TRACE("%p - ref %d\n", This, ref);
1658
1659     return ref;
1660 }
1661
1662 static ULONG WINAPI IFileDialog2_fnRelease(IFileDialog2 *iface)
1663 {
1664     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1665     LONG ref = InterlockedDecrement(&This->ref);
1666     TRACE("%p - ref %d\n", This, ref);
1667
1668     if(!ref)
1669     {
1670         UINT i;
1671         for(i = 0; i < This->filterspec_count; i++)
1672         {
1673             LocalFree((void*)This->filterspecs[i].pszName);
1674             LocalFree((void*)This->filterspecs[i].pszSpec);
1675         }
1676         HeapFree(GetProcessHeap(), 0, This->filterspecs);
1677
1678         DestroyWindow(This->cctrls_hwnd);
1679
1680         if(This->psi_defaultfolder) IShellItem_Release(This->psi_defaultfolder);
1681         if(This->psi_setfolder)     IShellItem_Release(This->psi_setfolder);
1682         if(This->psi_folder)        IShellItem_Release(This->psi_folder);
1683         if(This->psia_selection)    IShellItemArray_Release(This->psia_selection);
1684         if(This->psia_results)      IShellItemArray_Release(This->psia_results);
1685
1686         LocalFree(This->set_filename);
1687         LocalFree(This->default_ext);
1688         LocalFree(This->custom_title);
1689         LocalFree(This->custom_okbutton);
1690         LocalFree(This->custom_cancelbutton);
1691         LocalFree(This->custom_filenamelabel);
1692
1693         HeapFree(GetProcessHeap(), 0, This);
1694     }
1695
1696     return ref;
1697 }
1698
1699 static HRESULT WINAPI IFileDialog2_fnShow(IFileDialog2 *iface, HWND hwndOwner)
1700 {
1701     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1702     TRACE("%p (%p)\n", iface, hwndOwner);
1703
1704     return create_dialog(This, hwndOwner);
1705 }
1706
1707 static HRESULT WINAPI IFileDialog2_fnSetFileTypes(IFileDialog2 *iface, UINT cFileTypes,
1708                                                   const COMDLG_FILTERSPEC *rgFilterSpec)
1709 {
1710     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1711     UINT i;
1712     TRACE("%p (%d, %p)\n", This, cFileTypes, rgFilterSpec);
1713
1714     if(This->filterspecs)
1715         return E_UNEXPECTED;
1716
1717     if(!rgFilterSpec)
1718         return E_INVALIDARG;
1719
1720     if(!cFileTypes)
1721         return S_OK;
1722
1723     This->filterspecs = HeapAlloc(GetProcessHeap(), 0, sizeof(COMDLG_FILTERSPEC)*cFileTypes);
1724     for(i = 0; i < cFileTypes; i++)
1725     {
1726         This->filterspecs[i].pszName = StrDupW(rgFilterSpec[i].pszName);
1727         This->filterspecs[i].pszSpec = StrDupW(rgFilterSpec[i].pszSpec);
1728     }
1729     This->filterspec_count = cFileTypes;
1730
1731     return S_OK;
1732 }
1733
1734 static HRESULT WINAPI IFileDialog2_fnSetFileTypeIndex(IFileDialog2 *iface, UINT iFileType)
1735 {
1736     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1737     TRACE("%p (%d)\n", This, iFileType);
1738
1739     if(!This->filterspecs)
1740         return E_FAIL;
1741
1742     if(iFileType >= This->filterspec_count)
1743         This->filetypeindex = This->filterspec_count - 1;
1744     else
1745         This->filetypeindex = iFileType;
1746
1747     return S_OK;
1748 }
1749
1750 static HRESULT WINAPI IFileDialog2_fnGetFileTypeIndex(IFileDialog2 *iface, UINT *piFileType)
1751 {
1752     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1753     TRACE("%p (%p)\n", This, piFileType);
1754
1755     if(!piFileType)
1756         return E_INVALIDARG;
1757
1758     *piFileType = This->filetypeindex;
1759
1760     return S_OK;
1761 }
1762
1763 static HRESULT WINAPI IFileDialog2_fnAdvise(IFileDialog2 *iface, IFileDialogEvents *pfde, DWORD *pdwCookie)
1764 {
1765     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1766     events_client *client;
1767     TRACE("%p (%p, %p)\n", This, pfde, pdwCookie);
1768
1769     if(!pfde || !pdwCookie)
1770         return E_INVALIDARG;
1771
1772     client = HeapAlloc(GetProcessHeap(), 0, sizeof(events_client));
1773     client->pfde = pfde;
1774     client->cookie = ++This->events_next_cookie;
1775
1776     IFileDialogEvents_AddRef(pfde);
1777     *pdwCookie = client->cookie;
1778
1779     list_add_tail(&This->events_clients, &client->entry);
1780
1781     return S_OK;
1782 }
1783
1784 static HRESULT WINAPI IFileDialog2_fnUnadvise(IFileDialog2 *iface, DWORD dwCookie)
1785 {
1786     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1787     events_client *client, *found = NULL;
1788     TRACE("%p (%d)\n", This, dwCookie);
1789
1790     LIST_FOR_EACH_ENTRY(client, &This->events_clients, events_client, entry)
1791     {
1792         if(client->cookie == dwCookie)
1793         {
1794             found = client;
1795             break;
1796         }
1797     }
1798
1799     if(found)
1800     {
1801         list_remove(&found->entry);
1802         IFileDialogEvents_Release(found->pfde);
1803         HeapFree(GetProcessHeap(), 0, found);
1804         return S_OK;
1805     }
1806
1807     return E_INVALIDARG;
1808 }
1809
1810 static HRESULT WINAPI IFileDialog2_fnSetOptions(IFileDialog2 *iface, FILEOPENDIALOGOPTIONS fos)
1811 {
1812     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1813     TRACE("%p (0x%x)\n", This, fos);
1814
1815     This->options = fos;
1816
1817     return S_OK;
1818 }
1819
1820 static HRESULT WINAPI IFileDialog2_fnGetOptions(IFileDialog2 *iface, FILEOPENDIALOGOPTIONS *pfos)
1821 {
1822     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1823     TRACE("%p (%p)\n", This, pfos);
1824
1825     if(!pfos)
1826         return E_INVALIDARG;
1827
1828     *pfos = This->options;
1829
1830     return S_OK;
1831 }
1832
1833 static HRESULT WINAPI IFileDialog2_fnSetDefaultFolder(IFileDialog2 *iface, IShellItem *psi)
1834 {
1835     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1836     TRACE("%p (%p)\n", This, psi);
1837     if(This->psi_defaultfolder)
1838         IShellItem_Release(This->psi_defaultfolder);
1839
1840     This->psi_defaultfolder = psi;
1841
1842     if(This->psi_defaultfolder)
1843         IShellItem_AddRef(This->psi_defaultfolder);
1844
1845     return S_OK;
1846 }
1847
1848 static HRESULT WINAPI IFileDialog2_fnSetFolder(IFileDialog2 *iface, IShellItem *psi)
1849 {
1850     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1851     TRACE("%p (%p)\n", This, psi);
1852     if(This->psi_setfolder)
1853         IShellItem_Release(This->psi_setfolder);
1854
1855     This->psi_setfolder = psi;
1856
1857     if(This->psi_setfolder)
1858         IShellItem_AddRef(This->psi_setfolder);
1859
1860     return S_OK;
1861 }
1862
1863 static HRESULT WINAPI IFileDialog2_fnGetFolder(IFileDialog2 *iface, IShellItem **ppsi)
1864 {
1865     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1866     TRACE("%p (%p)\n", This, ppsi);
1867     if(!ppsi)
1868         return E_INVALIDARG;
1869
1870     /* FIXME:
1871        If the dialog is shown, return the current(ly selected) folder. */
1872
1873     *ppsi = NULL;
1874     if(This->psi_folder)
1875         *ppsi = This->psi_folder;
1876     else if(This->psi_setfolder)
1877         *ppsi = This->psi_setfolder;
1878     else if(This->psi_defaultfolder)
1879         *ppsi = This->psi_defaultfolder;
1880
1881     if(*ppsi)
1882     {
1883         IShellItem_AddRef(*ppsi);
1884         return S_OK;
1885     }
1886
1887     return E_FAIL;
1888 }
1889
1890 static HRESULT WINAPI IFileDialog2_fnGetCurrentSelection(IFileDialog2 *iface, IShellItem **ppsi)
1891 {
1892     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1893     HRESULT hr;
1894     TRACE("%p (%p)\n", This, ppsi);
1895
1896     if(!ppsi)
1897         return E_INVALIDARG;
1898
1899     if(This->psia_selection)
1900     {
1901         /* FIXME: Check filename edit box */
1902         hr = IShellItemArray_GetItemAt(This->psia_selection, 0, ppsi);
1903         return hr;
1904     }
1905
1906     return E_FAIL;
1907 }
1908
1909 static HRESULT WINAPI IFileDialog2_fnSetFileName(IFileDialog2 *iface, LPCWSTR pszName)
1910 {
1911     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1912     TRACE("%p (%s)\n", iface, debugstr_w(pszName));
1913
1914     set_file_name(This, pszName);
1915
1916     return S_OK;
1917 }
1918
1919 static HRESULT WINAPI IFileDialog2_fnGetFileName(IFileDialog2 *iface, LPWSTR *pszName)
1920 {
1921     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1922     TRACE("%p (%p)\n", iface, pszName);
1923
1924     if(!pszName)
1925         return E_INVALIDARG;
1926
1927     *pszName = NULL;
1928     if(get_file_name(This, pszName))
1929         return S_OK;
1930     else
1931         return E_FAIL;
1932 }
1933
1934 static HRESULT WINAPI IFileDialog2_fnSetTitle(IFileDialog2 *iface, LPCWSTR pszTitle)
1935 {
1936     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1937     TRACE("%p (%s)\n", This, debugstr_w(pszTitle));
1938
1939     LocalFree(This->custom_title);
1940     This->custom_title = StrDupW(pszTitle);
1941     update_control_text(This);
1942
1943     return S_OK;
1944 }
1945
1946 static HRESULT WINAPI IFileDialog2_fnSetOkButtonLabel(IFileDialog2 *iface, LPCWSTR pszText)
1947 {
1948     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1949     TRACE("%p (%s)\n", This, debugstr_w(pszText));
1950
1951     LocalFree(This->custom_okbutton);
1952     This->custom_okbutton = StrDupW(pszText);
1953     update_control_text(This);
1954     update_layout(This);
1955
1956     return S_OK;
1957 }
1958
1959 static HRESULT WINAPI IFileDialog2_fnSetFileNameLabel(IFileDialog2 *iface, LPCWSTR pszLabel)
1960 {
1961     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1962     TRACE("%p (%s)\n", This, debugstr_w(pszLabel));
1963
1964     LocalFree(This->custom_filenamelabel);
1965     This->custom_filenamelabel = StrDupW(pszLabel);
1966     update_control_text(This);
1967     update_layout(This);
1968
1969     return S_OK;
1970 }
1971
1972 static HRESULT WINAPI IFileDialog2_fnGetResult(IFileDialog2 *iface, IShellItem **ppsi)
1973 {
1974     FileDialogImpl *This = impl_from_IFileDialog2(iface);
1975     HRESULT hr;
1976     TRACE("%p (%p)\n", This, ppsi);
1977
1978     if(!ppsi)
1979         return E_INVALIDARG;
1980
1981     if(This->psia_results)
1982     {
1983         UINT item_count;
1984         hr = IShellItemArray_GetCount(This->psia_results, &item_count);
1985         if(SUCCEEDED(hr))
1986         {
1987             if(item_count != 1)
1988                 return E_FAIL;
1989
1990             /* Adds a reference. */
1991             hr = IShellItemArray_GetItemAt(This->psia_results, 0, ppsi);
1992         }
1993
1994         return hr;
1995     }
1996
1997     return E_UNEXPECTED;
1998 }
1999
2000 static HRESULT WINAPI IFileDialog2_fnAddPlace(IFileDialog2 *iface, IShellItem *psi, FDAP fdap)
2001 {
2002     FileDialogImpl *This = impl_from_IFileDialog2(iface);
2003     FIXME("stub - %p (%p, %d)\n", This, psi, fdap);
2004     return E_NOTIMPL;
2005 }
2006
2007 static HRESULT WINAPI IFileDialog2_fnSetDefaultExtension(IFileDialog2 *iface, LPCWSTR pszDefaultExtension)
2008 {
2009     FileDialogImpl *This = impl_from_IFileDialog2(iface);
2010     TRACE("%p (%s)\n", This, debugstr_w(pszDefaultExtension));
2011
2012     LocalFree(This->default_ext);
2013     This->default_ext = StrDupW(pszDefaultExtension);
2014
2015     return S_OK;
2016 }
2017
2018 static HRESULT WINAPI IFileDialog2_fnClose(IFileDialog2 *iface, HRESULT hr)
2019 {
2020     FileDialogImpl *This = impl_from_IFileDialog2(iface);
2021     TRACE("%p (0x%08x)\n", This, hr);
2022
2023     if(This->dlg_hwnd)
2024         EndDialog(This->dlg_hwnd, hr);
2025
2026     return S_OK;
2027 }
2028
2029 static HRESULT WINAPI IFileDialog2_fnSetClientGuid(IFileDialog2 *iface, REFGUID guid)
2030 {
2031     FileDialogImpl *This = impl_from_IFileDialog2(iface);
2032     FIXME("stub - %p (%s)\n", This, debugstr_guid(guid));
2033     return E_NOTIMPL;
2034 }
2035
2036 static HRESULT WINAPI IFileDialog2_fnClearClientData(IFileDialog2 *iface)
2037 {
2038     FileDialogImpl *This = impl_from_IFileDialog2(iface);
2039     FIXME("stub - %p\n", This);
2040     return E_NOTIMPL;
2041 }
2042
2043 static HRESULT WINAPI IFileDialog2_fnSetFilter(IFileDialog2 *iface, IShellItemFilter *pFilter)
2044 {
2045     FileDialogImpl *This = impl_from_IFileDialog2(iface);
2046     FIXME("stub - %p (%p)\n", This, pFilter);
2047     return E_NOTIMPL;
2048 }
2049
2050 static HRESULT WINAPI IFileDialog2_fnSetCancelButtonLabel(IFileDialog2 *iface, LPCWSTR pszLabel)
2051 {
2052     FileDialogImpl *This = impl_from_IFileDialog2(iface);
2053     TRACE("%p (%s)\n", This, debugstr_w(pszLabel));
2054
2055     LocalFree(This->custom_cancelbutton);
2056     This->custom_cancelbutton = StrDupW(pszLabel);
2057     update_control_text(This);
2058     update_layout(This);
2059
2060     return S_OK;
2061 }
2062
2063 static HRESULT WINAPI IFileDialog2_fnSetNavigationRoot(IFileDialog2 *iface, IShellItem *psi)
2064 {
2065     FileDialogImpl *This = impl_from_IFileDialog2(iface);
2066     FIXME("stub - %p (%p)\n", This, psi);
2067     return E_NOTIMPL;
2068 }
2069
2070 static const IFileDialog2Vtbl vt_IFileDialog2 = {
2071     IFileDialog2_fnQueryInterface,
2072     IFileDialog2_fnAddRef,
2073     IFileDialog2_fnRelease,
2074     IFileDialog2_fnShow,
2075     IFileDialog2_fnSetFileTypes,
2076     IFileDialog2_fnSetFileTypeIndex,
2077     IFileDialog2_fnGetFileTypeIndex,
2078     IFileDialog2_fnAdvise,
2079     IFileDialog2_fnUnadvise,
2080     IFileDialog2_fnSetOptions,
2081     IFileDialog2_fnGetOptions,
2082     IFileDialog2_fnSetDefaultFolder,
2083     IFileDialog2_fnSetFolder,
2084     IFileDialog2_fnGetFolder,
2085     IFileDialog2_fnGetCurrentSelection,
2086     IFileDialog2_fnSetFileName,
2087     IFileDialog2_fnGetFileName,
2088     IFileDialog2_fnSetTitle,
2089     IFileDialog2_fnSetOkButtonLabel,
2090     IFileDialog2_fnSetFileNameLabel,
2091     IFileDialog2_fnGetResult,
2092     IFileDialog2_fnAddPlace,
2093     IFileDialog2_fnSetDefaultExtension,
2094     IFileDialog2_fnClose,
2095     IFileDialog2_fnSetClientGuid,
2096     IFileDialog2_fnClearClientData,
2097     IFileDialog2_fnSetFilter,
2098     IFileDialog2_fnSetCancelButtonLabel,
2099     IFileDialog2_fnSetNavigationRoot
2100 };
2101
2102 /**************************************************************************
2103  * IFileOpenDialog
2104  */
2105 static inline FileDialogImpl *impl_from_IFileOpenDialog(IFileOpenDialog *iface)
2106 {
2107     return CONTAINING_RECORD(iface, FileDialogImpl, u.IFileOpenDialog_iface);
2108 }
2109
2110 static HRESULT WINAPI IFileOpenDialog_fnQueryInterface(IFileOpenDialog *iface,
2111                                                        REFIID riid, void **ppvObject)
2112 {
2113     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2114     return IFileDialog2_QueryInterface(&This->IFileDialog2_iface, riid, ppvObject);
2115 }
2116
2117 static ULONG WINAPI IFileOpenDialog_fnAddRef(IFileOpenDialog *iface)
2118 {
2119     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2120     return IFileDialog2_AddRef(&This->IFileDialog2_iface);
2121 }
2122
2123 static ULONG WINAPI IFileOpenDialog_fnRelease(IFileOpenDialog *iface)
2124 {
2125     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2126     return IFileDialog2_Release(&This->IFileDialog2_iface);
2127 }
2128
2129 static HRESULT WINAPI IFileOpenDialog_fnShow(IFileOpenDialog *iface, HWND hwndOwner)
2130 {
2131     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2132     return IFileDialog2_Show(&This->IFileDialog2_iface, hwndOwner);
2133 }
2134
2135 static HRESULT WINAPI IFileOpenDialog_fnSetFileTypes(IFileOpenDialog *iface, UINT cFileTypes,
2136                                                      const COMDLG_FILTERSPEC *rgFilterSpec)
2137 {
2138     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2139     return IFileDialog2_SetFileTypes(&This->IFileDialog2_iface, cFileTypes, rgFilterSpec);
2140 }
2141
2142 static HRESULT WINAPI IFileOpenDialog_fnSetFileTypeIndex(IFileOpenDialog *iface, UINT iFileType)
2143 {
2144     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2145     return IFileDialog2_SetFileTypeIndex(&This->IFileDialog2_iface, iFileType);
2146 }
2147
2148 static HRESULT WINAPI IFileOpenDialog_fnGetFileTypeIndex(IFileOpenDialog *iface, UINT *piFileType)
2149 {
2150     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2151     return IFileDialog2_GetFileTypeIndex(&This->IFileDialog2_iface, piFileType);
2152 }
2153
2154 static HRESULT WINAPI IFileOpenDialog_fnAdvise(IFileOpenDialog *iface, IFileDialogEvents *pfde,
2155                                                DWORD *pdwCookie)
2156 {
2157     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2158     return IFileDialog2_Advise(&This->IFileDialog2_iface, pfde, pdwCookie);
2159 }
2160
2161 static HRESULT WINAPI IFileOpenDialog_fnUnadvise(IFileOpenDialog *iface, DWORD dwCookie)
2162 {
2163     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2164     return IFileDialog2_Unadvise(&This->IFileDialog2_iface, dwCookie);
2165 }
2166
2167 static HRESULT WINAPI IFileOpenDialog_fnSetOptions(IFileOpenDialog *iface, FILEOPENDIALOGOPTIONS fos)
2168 {
2169     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2170     return IFileDialog2_SetOptions(&This->IFileDialog2_iface, fos);
2171 }
2172
2173 static HRESULT WINAPI IFileOpenDialog_fnGetOptions(IFileOpenDialog *iface, FILEOPENDIALOGOPTIONS *pfos)
2174 {
2175     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2176     return IFileDialog2_GetOptions(&This->IFileDialog2_iface, pfos);
2177 }
2178
2179 static HRESULT WINAPI IFileOpenDialog_fnSetDefaultFolder(IFileOpenDialog *iface, IShellItem *psi)
2180 {
2181     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2182     return IFileDialog2_SetDefaultFolder(&This->IFileDialog2_iface, psi);
2183 }
2184
2185 static HRESULT WINAPI IFileOpenDialog_fnSetFolder(IFileOpenDialog *iface, IShellItem *psi)
2186 {
2187     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2188     return IFileDialog2_SetFolder(&This->IFileDialog2_iface, psi);
2189 }
2190
2191 static HRESULT WINAPI IFileOpenDialog_fnGetFolder(IFileOpenDialog *iface, IShellItem **ppsi)
2192 {
2193     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2194     return IFileDialog2_GetFolder(&This->IFileDialog2_iface, ppsi);
2195 }
2196
2197 static HRESULT WINAPI IFileOpenDialog_fnGetCurrentSelection(IFileOpenDialog *iface, IShellItem **ppsi)
2198 {
2199     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2200     return IFileDialog2_GetCurrentSelection(&This->IFileDialog2_iface, ppsi);
2201 }
2202
2203 static HRESULT WINAPI IFileOpenDialog_fnSetFileName(IFileOpenDialog *iface, LPCWSTR pszName)
2204 {
2205     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2206     return IFileDialog2_SetFileName(&This->IFileDialog2_iface, pszName);
2207 }
2208
2209 static HRESULT WINAPI IFileOpenDialog_fnGetFileName(IFileOpenDialog *iface, LPWSTR *pszName)
2210 {
2211     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2212     return IFileDialog2_GetFileName(&This->IFileDialog2_iface, pszName);
2213 }
2214
2215 static HRESULT WINAPI IFileOpenDialog_fnSetTitle(IFileOpenDialog *iface, LPCWSTR pszTitle)
2216 {
2217     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2218     return IFileDialog2_SetTitle(&This->IFileDialog2_iface, pszTitle);
2219 }
2220
2221 static HRESULT WINAPI IFileOpenDialog_fnSetOkButtonLabel(IFileOpenDialog *iface, LPCWSTR pszText)
2222 {
2223     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2224     return IFileDialog2_SetOkButtonLabel(&This->IFileDialog2_iface, pszText);
2225 }
2226
2227 static HRESULT WINAPI IFileOpenDialog_fnSetFileNameLabel(IFileOpenDialog *iface, LPCWSTR pszLabel)
2228 {
2229     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2230     return IFileDialog2_SetFileNameLabel(&This->IFileDialog2_iface, pszLabel);
2231 }
2232
2233 static HRESULT WINAPI IFileOpenDialog_fnGetResult(IFileOpenDialog *iface, IShellItem **ppsi)
2234 {
2235     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2236     return IFileDialog2_GetResult(&This->IFileDialog2_iface, ppsi);
2237 }
2238
2239 static HRESULT WINAPI IFileOpenDialog_fnAddPlace(IFileOpenDialog *iface, IShellItem *psi, FDAP fdap)
2240 {
2241     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2242     return IFileDialog2_AddPlace(&This->IFileDialog2_iface, psi, fdap);
2243 }
2244
2245 static HRESULT WINAPI IFileOpenDialog_fnSetDefaultExtension(IFileOpenDialog *iface,
2246                                                             LPCWSTR pszDefaultExtension)
2247 {
2248     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2249     return IFileDialog2_SetDefaultExtension(&This->IFileDialog2_iface, pszDefaultExtension);
2250 }
2251
2252 static HRESULT WINAPI IFileOpenDialog_fnClose(IFileOpenDialog *iface, HRESULT hr)
2253 {
2254     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2255     return IFileDialog2_Close(&This->IFileDialog2_iface, hr);
2256 }
2257
2258 static HRESULT WINAPI IFileOpenDialog_fnSetClientGuid(IFileOpenDialog *iface, REFGUID guid)
2259 {
2260     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2261     return IFileDialog2_SetClientGuid(&This->IFileDialog2_iface, guid);
2262 }
2263
2264 static HRESULT WINAPI IFileOpenDialog_fnClearClientData(IFileOpenDialog *iface)
2265 {
2266     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2267     return IFileDialog2_ClearClientData(&This->IFileDialog2_iface);
2268 }
2269
2270 static HRESULT WINAPI IFileOpenDialog_fnSetFilter(IFileOpenDialog *iface, IShellItemFilter *pFilter)
2271 {
2272     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2273     return IFileDialog2_SetFilter(&This->IFileDialog2_iface, pFilter);
2274 }
2275
2276 static HRESULT WINAPI IFileOpenDialog_fnGetResults(IFileOpenDialog *iface, IShellItemArray **ppenum)
2277 {
2278     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2279     TRACE("%p (%p)\n", This, ppenum);
2280
2281     *ppenum = This->psia_results;
2282
2283     if(*ppenum)
2284     {
2285         IShellItemArray_AddRef(*ppenum);
2286         return S_OK;
2287     }
2288
2289     return E_FAIL;
2290 }
2291
2292 static HRESULT WINAPI IFileOpenDialog_fnGetSelectedItems(IFileOpenDialog *iface, IShellItemArray **ppsai)
2293 {
2294     FileDialogImpl *This = impl_from_IFileOpenDialog(iface);
2295     TRACE("%p (%p)\n", This, ppsai);
2296
2297     if(This->psia_selection)
2298     {
2299         *ppsai = This->psia_selection;
2300         IShellItemArray_AddRef(*ppsai);
2301         return S_OK;
2302     }
2303
2304     return E_FAIL;
2305 }
2306
2307 static const IFileOpenDialogVtbl vt_IFileOpenDialog = {
2308     IFileOpenDialog_fnQueryInterface,
2309     IFileOpenDialog_fnAddRef,
2310     IFileOpenDialog_fnRelease,
2311     IFileOpenDialog_fnShow,
2312     IFileOpenDialog_fnSetFileTypes,
2313     IFileOpenDialog_fnSetFileTypeIndex,
2314     IFileOpenDialog_fnGetFileTypeIndex,
2315     IFileOpenDialog_fnAdvise,
2316     IFileOpenDialog_fnUnadvise,
2317     IFileOpenDialog_fnSetOptions,
2318     IFileOpenDialog_fnGetOptions,
2319     IFileOpenDialog_fnSetDefaultFolder,
2320     IFileOpenDialog_fnSetFolder,
2321     IFileOpenDialog_fnGetFolder,
2322     IFileOpenDialog_fnGetCurrentSelection,
2323     IFileOpenDialog_fnSetFileName,
2324     IFileOpenDialog_fnGetFileName,
2325     IFileOpenDialog_fnSetTitle,
2326     IFileOpenDialog_fnSetOkButtonLabel,
2327     IFileOpenDialog_fnSetFileNameLabel,
2328     IFileOpenDialog_fnGetResult,
2329     IFileOpenDialog_fnAddPlace,
2330     IFileOpenDialog_fnSetDefaultExtension,
2331     IFileOpenDialog_fnClose,
2332     IFileOpenDialog_fnSetClientGuid,
2333     IFileOpenDialog_fnClearClientData,
2334     IFileOpenDialog_fnSetFilter,
2335     IFileOpenDialog_fnGetResults,
2336     IFileOpenDialog_fnGetSelectedItems
2337 };
2338
2339 /**************************************************************************
2340  * IFileSaveDialog
2341  */
2342 static inline FileDialogImpl *impl_from_IFileSaveDialog(IFileSaveDialog *iface)
2343 {
2344     return CONTAINING_RECORD(iface, FileDialogImpl, u.IFileSaveDialog_iface);
2345 }
2346
2347 static HRESULT WINAPI IFileSaveDialog_fnQueryInterface(IFileSaveDialog *iface,
2348                                                        REFIID riid,
2349                                                        void **ppvObject)
2350 {
2351     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2352     return IFileDialog2_QueryInterface(&This->IFileDialog2_iface, riid, ppvObject);
2353 }
2354
2355 static ULONG WINAPI IFileSaveDialog_fnAddRef(IFileSaveDialog *iface)
2356 {
2357     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2358     return IFileDialog2_AddRef(&This->IFileDialog2_iface);
2359 }
2360
2361 static ULONG WINAPI IFileSaveDialog_fnRelease(IFileSaveDialog *iface)
2362 {
2363     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2364     return IFileDialog2_Release(&This->IFileDialog2_iface);
2365 }
2366
2367 static HRESULT WINAPI IFileSaveDialog_fnShow(IFileSaveDialog *iface, HWND hwndOwner)
2368 {
2369     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2370     return IFileDialog2_Show(&This->IFileDialog2_iface, hwndOwner);
2371 }
2372
2373 static HRESULT WINAPI IFileSaveDialog_fnSetFileTypes(IFileSaveDialog *iface, UINT cFileTypes,
2374                                                      const COMDLG_FILTERSPEC *rgFilterSpec)
2375 {
2376     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2377     return IFileDialog2_SetFileTypes(&This->IFileDialog2_iface, cFileTypes, rgFilterSpec);
2378 }
2379
2380 static HRESULT WINAPI IFileSaveDialog_fnSetFileTypeIndex(IFileSaveDialog *iface, UINT iFileType)
2381 {
2382     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2383     return IFileDialog2_SetFileTypeIndex(&This->IFileDialog2_iface, iFileType);
2384 }
2385
2386 static HRESULT WINAPI IFileSaveDialog_fnGetFileTypeIndex(IFileSaveDialog *iface, UINT *piFileType)
2387 {
2388     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2389     return IFileDialog2_GetFileTypeIndex(&This->IFileDialog2_iface, piFileType);
2390 }
2391
2392 static HRESULT WINAPI IFileSaveDialog_fnAdvise(IFileSaveDialog *iface, IFileDialogEvents *pfde,
2393                                                DWORD *pdwCookie)
2394 {
2395     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2396     return IFileDialog2_Advise(&This->IFileDialog2_iface, pfde, pdwCookie);
2397 }
2398
2399 static HRESULT WINAPI IFileSaveDialog_fnUnadvise(IFileSaveDialog *iface, DWORD dwCookie)
2400 {
2401     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2402     return IFileDialog2_Unadvise(&This->IFileDialog2_iface, dwCookie);
2403 }
2404
2405 static HRESULT WINAPI IFileSaveDialog_fnSetOptions(IFileSaveDialog *iface, FILEOPENDIALOGOPTIONS fos)
2406 {
2407     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2408     return IFileDialog2_SetOptions(&This->IFileDialog2_iface, fos);
2409 }
2410
2411 static HRESULT WINAPI IFileSaveDialog_fnGetOptions(IFileSaveDialog *iface, FILEOPENDIALOGOPTIONS *pfos)
2412 {
2413     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2414     return IFileDialog2_GetOptions(&This->IFileDialog2_iface, pfos);
2415 }
2416
2417 static HRESULT WINAPI IFileSaveDialog_fnSetDefaultFolder(IFileSaveDialog *iface, IShellItem *psi)
2418 {
2419     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2420     return IFileDialog2_SetDefaultFolder(&This->IFileDialog2_iface, psi);
2421 }
2422
2423 static HRESULT WINAPI IFileSaveDialog_fnSetFolder(IFileSaveDialog *iface, IShellItem *psi)
2424 {
2425     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2426     return IFileDialog2_SetFolder(&This->IFileDialog2_iface, psi);
2427 }
2428
2429 static HRESULT WINAPI IFileSaveDialog_fnGetFolder(IFileSaveDialog *iface, IShellItem **ppsi)
2430 {
2431     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2432     return IFileDialog2_GetFolder(&This->IFileDialog2_iface, ppsi);
2433 }
2434
2435 static HRESULT WINAPI IFileSaveDialog_fnGetCurrentSelection(IFileSaveDialog *iface, IShellItem **ppsi)
2436 {
2437     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2438     return IFileDialog2_GetCurrentSelection(&This->IFileDialog2_iface, ppsi);
2439 }
2440
2441 static HRESULT WINAPI IFileSaveDialog_fnSetFileName(IFileSaveDialog *iface, LPCWSTR pszName)
2442 {
2443     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2444     return IFileDialog2_SetFileName(&This->IFileDialog2_iface, pszName);
2445 }
2446
2447 static HRESULT WINAPI IFileSaveDialog_fnGetFileName(IFileSaveDialog *iface, LPWSTR *pszName)
2448 {
2449     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2450     return IFileDialog2_GetFileName(&This->IFileDialog2_iface, pszName);
2451 }
2452
2453 static HRESULT WINAPI IFileSaveDialog_fnSetTitle(IFileSaveDialog *iface, LPCWSTR pszTitle)
2454 {
2455     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2456     return IFileDialog2_SetTitle(&This->IFileDialog2_iface, pszTitle);
2457 }
2458
2459 static HRESULT WINAPI IFileSaveDialog_fnSetOkButtonLabel(IFileSaveDialog *iface, LPCWSTR pszText)
2460 {
2461     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2462     return IFileDialog2_SetOkButtonLabel(&This->IFileDialog2_iface, pszText);
2463 }
2464
2465 static HRESULT WINAPI IFileSaveDialog_fnSetFileNameLabel(IFileSaveDialog *iface, LPCWSTR pszLabel)
2466 {
2467     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2468     return IFileDialog2_SetFileNameLabel(&This->IFileDialog2_iface, pszLabel);
2469 }
2470
2471 static HRESULT WINAPI IFileSaveDialog_fnGetResult(IFileSaveDialog *iface, IShellItem **ppsi)
2472 {
2473     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2474     return IFileDialog2_GetResult(&This->IFileDialog2_iface, ppsi);
2475 }
2476
2477 static HRESULT WINAPI IFileSaveDialog_fnAddPlace(IFileSaveDialog *iface, IShellItem *psi, FDAP fdap)
2478 {
2479     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2480     return IFileDialog2_AddPlace(&This->IFileDialog2_iface, psi, fdap);
2481 }
2482
2483 static HRESULT WINAPI IFileSaveDialog_fnSetDefaultExtension(IFileSaveDialog *iface,
2484                                                             LPCWSTR pszDefaultExtension)
2485 {
2486     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2487     return IFileDialog2_SetDefaultExtension(&This->IFileDialog2_iface, pszDefaultExtension);
2488 }
2489
2490 static HRESULT WINAPI IFileSaveDialog_fnClose(IFileSaveDialog *iface, HRESULT hr)
2491 {
2492     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2493     return IFileDialog2_Close(&This->IFileDialog2_iface, hr);
2494 }
2495
2496 static HRESULT WINAPI IFileSaveDialog_fnSetClientGuid(IFileSaveDialog *iface, REFGUID guid)
2497 {
2498     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2499     return IFileDialog2_SetClientGuid(&This->IFileDialog2_iface, guid);
2500 }
2501
2502 static HRESULT WINAPI IFileSaveDialog_fnClearClientData(IFileSaveDialog *iface)
2503 {
2504     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2505     return IFileDialog2_ClearClientData(&This->IFileDialog2_iface);
2506 }
2507
2508 static HRESULT WINAPI IFileSaveDialog_fnSetFilter(IFileSaveDialog *iface, IShellItemFilter *pFilter)
2509 {
2510     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2511     return IFileDialog2_SetFilter(&This->IFileDialog2_iface, pFilter);
2512 }
2513
2514 static HRESULT WINAPI IFileSaveDialog_fnSetSaveAsItem(IFileSaveDialog* iface, IShellItem *psi)
2515 {
2516     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2517     FIXME("stub - %p (%p)\n", This, psi);
2518     return E_NOTIMPL;
2519 }
2520
2521 static HRESULT WINAPI IFileSaveDialog_fnSetProperties(IFileSaveDialog* iface, IPropertyStore *pStore)
2522 {
2523     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2524     FIXME("stub - %p (%p)\n", This, pStore);
2525     return E_NOTIMPL;
2526 }
2527
2528 static HRESULT WINAPI IFileSaveDialog_fnSetCollectedProperties(IFileSaveDialog* iface,
2529                                                                IPropertyDescriptionList *pList,
2530                                                                BOOL fAppendDefault)
2531 {
2532     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2533     FIXME("stub - %p (%p, %d)\n", This, pList, fAppendDefault);
2534     return E_NOTIMPL;
2535 }
2536
2537 static HRESULT WINAPI IFileSaveDialog_fnGetProperties(IFileSaveDialog* iface, IPropertyStore **ppStore)
2538 {
2539     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2540     FIXME("stub - %p (%p)\n", This, ppStore);
2541     return E_NOTIMPL;
2542 }
2543
2544 static HRESULT WINAPI IFileSaveDialog_fnApplyProperties(IFileSaveDialog* iface,
2545                                                         IShellItem *psi,
2546                                                         IPropertyStore *pStore,
2547                                                         HWND hwnd,
2548                                                         IFileOperationProgressSink *pSink)
2549 {
2550     FileDialogImpl *This = impl_from_IFileSaveDialog(iface);
2551     FIXME("%p (%p, %p, %p, %p)\n", This, psi, pStore, hwnd, pSink);
2552     return E_NOTIMPL;
2553 }
2554
2555 static const IFileSaveDialogVtbl vt_IFileSaveDialog = {
2556     IFileSaveDialog_fnQueryInterface,
2557     IFileSaveDialog_fnAddRef,
2558     IFileSaveDialog_fnRelease,
2559     IFileSaveDialog_fnShow,
2560     IFileSaveDialog_fnSetFileTypes,
2561     IFileSaveDialog_fnSetFileTypeIndex,
2562     IFileSaveDialog_fnGetFileTypeIndex,
2563     IFileSaveDialog_fnAdvise,
2564     IFileSaveDialog_fnUnadvise,
2565     IFileSaveDialog_fnSetOptions,
2566     IFileSaveDialog_fnGetOptions,
2567     IFileSaveDialog_fnSetDefaultFolder,
2568     IFileSaveDialog_fnSetFolder,
2569     IFileSaveDialog_fnGetFolder,
2570     IFileSaveDialog_fnGetCurrentSelection,
2571     IFileSaveDialog_fnSetFileName,
2572     IFileSaveDialog_fnGetFileName,
2573     IFileSaveDialog_fnSetTitle,
2574     IFileSaveDialog_fnSetOkButtonLabel,
2575     IFileSaveDialog_fnSetFileNameLabel,
2576     IFileSaveDialog_fnGetResult,
2577     IFileSaveDialog_fnAddPlace,
2578     IFileSaveDialog_fnSetDefaultExtension,
2579     IFileSaveDialog_fnClose,
2580     IFileSaveDialog_fnSetClientGuid,
2581     IFileSaveDialog_fnClearClientData,
2582     IFileSaveDialog_fnSetFilter,
2583     IFileSaveDialog_fnSetSaveAsItem,
2584     IFileSaveDialog_fnSetProperties,
2585     IFileSaveDialog_fnSetCollectedProperties,
2586     IFileSaveDialog_fnGetProperties,
2587     IFileSaveDialog_fnApplyProperties
2588 };
2589
2590 /**************************************************************************
2591  * IExplorerBrowserEvents implementation
2592  */
2593 static inline FileDialogImpl *impl_from_IExplorerBrowserEvents(IExplorerBrowserEvents *iface)
2594 {
2595     return CONTAINING_RECORD(iface, FileDialogImpl, IExplorerBrowserEvents_iface);
2596 }
2597
2598 static HRESULT WINAPI IExplorerBrowserEvents_fnQueryInterface(IExplorerBrowserEvents *iface,
2599                                                               REFIID riid, void **ppvObject)
2600 {
2601     FileDialogImpl *This = impl_from_IExplorerBrowserEvents(iface);
2602     TRACE("%p (%s, %p)\n", This, debugstr_guid(riid), ppvObject);
2603
2604     return IFileDialog2_QueryInterface(&This->IFileDialog2_iface, riid, ppvObject);
2605 }
2606
2607 static ULONG WINAPI IExplorerBrowserEvents_fnAddRef(IExplorerBrowserEvents *iface)
2608 {
2609     FileDialogImpl *This = impl_from_IExplorerBrowserEvents(iface);
2610     TRACE("%p\n", This);
2611     return IFileDialog2_AddRef(&This->IFileDialog2_iface);
2612 }
2613
2614 static ULONG WINAPI IExplorerBrowserEvents_fnRelease(IExplorerBrowserEvents *iface)
2615 {
2616     FileDialogImpl *This = impl_from_IExplorerBrowserEvents(iface);
2617     TRACE("%p\n", This);
2618     return IFileDialog2_Release(&This->IFileDialog2_iface);
2619 }
2620
2621 static HRESULT WINAPI IExplorerBrowserEvents_fnOnNavigationPending(IExplorerBrowserEvents *iface,
2622                                                                    PCIDLIST_ABSOLUTE pidlFolder)
2623 {
2624     FileDialogImpl *This = impl_from_IExplorerBrowserEvents(iface);
2625     IShellItem *psi;
2626     HRESULT hr;
2627     TRACE("%p (%p)\n", This, pidlFolder);
2628
2629     hr = SHCreateItemFromIDList(pidlFolder, &IID_IShellItem, (void**)&psi);
2630     if(SUCCEEDED(hr))
2631     {
2632         hr = events_OnFolderChanging(This, psi);
2633         IShellItem_Release(psi);
2634
2635         /* The ExplorerBrowser treats S_FALSE as S_OK, we don't. */
2636         if(hr == S_FALSE)
2637             hr = E_FAIL;
2638
2639         return hr;
2640     }
2641     else
2642         ERR("Failed to convert pidl (%p) to a shellitem.\n", pidlFolder);
2643
2644     return S_OK;
2645 }
2646
2647 static HRESULT WINAPI IExplorerBrowserEvents_fnOnViewCreated(IExplorerBrowserEvents *iface,
2648                                                              IShellView *psv)
2649 {
2650     FileDialogImpl *This = impl_from_IExplorerBrowserEvents(iface);
2651     TRACE("%p (%p)\n", This, psv);
2652     return S_OK;
2653 }
2654
2655 static HRESULT WINAPI IExplorerBrowserEvents_fnOnNavigationComplete(IExplorerBrowserEvents *iface,
2656                                                                     PCIDLIST_ABSOLUTE pidlFolder)
2657 {
2658     FileDialogImpl *This = impl_from_IExplorerBrowserEvents(iface);
2659     HRESULT hr;
2660     TRACE("%p (%p)\n", This, pidlFolder);
2661
2662     if(This->psi_folder)
2663         IShellItem_Release(This->psi_folder);
2664
2665     hr = SHCreateItemFromIDList(pidlFolder, &IID_IShellItem, (void**)&This->psi_folder);
2666     if(FAILED(hr))
2667     {
2668         ERR("Failed to get the current folder.\n");
2669         This->psi_folder = NULL;
2670     }
2671
2672     events_OnFolderChange(This);
2673
2674     return S_OK;
2675 }
2676
2677 static HRESULT WINAPI IExplorerBrowserEvents_fnOnNavigationFailed(IExplorerBrowserEvents *iface,
2678                                                                   PCIDLIST_ABSOLUTE pidlFolder)
2679 {
2680     FileDialogImpl *This = impl_from_IExplorerBrowserEvents(iface);
2681     TRACE("%p (%p)\n", This, pidlFolder);
2682     return S_OK;
2683 }
2684
2685 static const IExplorerBrowserEventsVtbl vt_IExplorerBrowserEvents = {
2686     IExplorerBrowserEvents_fnQueryInterface,
2687     IExplorerBrowserEvents_fnAddRef,
2688     IExplorerBrowserEvents_fnRelease,
2689     IExplorerBrowserEvents_fnOnNavigationPending,
2690     IExplorerBrowserEvents_fnOnViewCreated,
2691     IExplorerBrowserEvents_fnOnNavigationComplete,
2692     IExplorerBrowserEvents_fnOnNavigationFailed
2693 };
2694
2695 /**************************************************************************
2696  * IServiceProvider implementation
2697  */
2698 static inline FileDialogImpl *impl_from_IServiceProvider(IServiceProvider *iface)
2699 {
2700     return CONTAINING_RECORD(iface, FileDialogImpl, IServiceProvider_iface);
2701 }
2702
2703 static HRESULT WINAPI IServiceProvider_fnQueryInterface(IServiceProvider *iface,
2704                                                         REFIID riid, void **ppvObject)
2705 {
2706     FileDialogImpl *This = impl_from_IServiceProvider(iface);
2707     TRACE("%p\n", This);
2708     return IFileDialog2_QueryInterface(&This->IFileDialog2_iface, riid, ppvObject);
2709 }
2710
2711 static ULONG WINAPI IServiceProvider_fnAddRef(IServiceProvider *iface)
2712 {
2713     FileDialogImpl *This = impl_from_IServiceProvider(iface);
2714     TRACE("%p\n", This);
2715     return IFileDialog2_AddRef(&This->IFileDialog2_iface);
2716 }
2717
2718 static ULONG WINAPI IServiceProvider_fnRelease(IServiceProvider *iface)
2719 {
2720     FileDialogImpl *This = impl_from_IServiceProvider(iface);
2721     TRACE("%p\n", This);
2722     return IFileDialog2_Release(&This->IFileDialog2_iface);
2723 }
2724
2725 static HRESULT WINAPI IServiceProvider_fnQueryService(IServiceProvider *iface,
2726                                                       REFGUID guidService,
2727                                                       REFIID riid, void **ppv)
2728 {
2729     FileDialogImpl *This = impl_from_IServiceProvider(iface);
2730     HRESULT hr = E_FAIL;
2731     TRACE("%p (%s, %s, %p)\n", This, debugstr_guid(guidService), debugstr_guid(riid), ppv);
2732
2733     *ppv = NULL;
2734     if(IsEqualGUID(guidService, &SID_STopLevelBrowser) && This->peb)
2735         hr = IExplorerBrowser_QueryInterface(This->peb, riid, ppv);
2736     else if(IsEqualGUID(guidService, &SID_SExplorerBrowserFrame))
2737         hr = IFileDialog2_QueryInterface(&This->IFileDialog2_iface, riid, ppv);
2738     else
2739         FIXME("Interface %s requested from unknown service %s\n",
2740               debugstr_guid(riid), debugstr_guid(guidService));
2741
2742     return hr;
2743 }
2744
2745 static const IServiceProviderVtbl vt_IServiceProvider = {
2746     IServiceProvider_fnQueryInterface,
2747     IServiceProvider_fnAddRef,
2748     IServiceProvider_fnRelease,
2749     IServiceProvider_fnQueryService
2750 };
2751
2752 /**************************************************************************
2753  * ICommDlgBrowser3 implementation
2754  */
2755 static inline FileDialogImpl *impl_from_ICommDlgBrowser3(ICommDlgBrowser3 *iface)
2756 {
2757     return CONTAINING_RECORD(iface, FileDialogImpl, ICommDlgBrowser3_iface);
2758 }
2759
2760 static HRESULT WINAPI ICommDlgBrowser3_fnQueryInterface(ICommDlgBrowser3 *iface,
2761                                                         REFIID riid, void **ppvObject)
2762 {
2763     FileDialogImpl *This = impl_from_ICommDlgBrowser3(iface);
2764     TRACE("%p\n", This);
2765     return IFileDialog2_QueryInterface(&This->IFileDialog2_iface, riid, ppvObject);
2766 }
2767
2768 static ULONG WINAPI ICommDlgBrowser3_fnAddRef(ICommDlgBrowser3 *iface)
2769 {
2770     FileDialogImpl *This = impl_from_ICommDlgBrowser3(iface);
2771     TRACE("%p\n", This);
2772     return IFileDialog2_AddRef(&This->IFileDialog2_iface);
2773 }
2774
2775 static ULONG WINAPI ICommDlgBrowser3_fnRelease(ICommDlgBrowser3 *iface)
2776 {
2777     FileDialogImpl *This = impl_from_ICommDlgBrowser3(iface);
2778     TRACE("%p\n", This);
2779     return IFileDialog2_Release(&This->IFileDialog2_iface);
2780 }
2781
2782 static HRESULT WINAPI ICommDlgBrowser3_fnOnDefaultCommand(ICommDlgBrowser3 *iface,
2783                                                           IShellView *shv)
2784 {
2785     FileDialogImpl *This = impl_from_ICommDlgBrowser3(iface);
2786     HRESULT hr;
2787     TRACE("%p (%p)\n", This, shv);
2788
2789     hr = on_default_action(This);
2790
2791     if(SUCCEEDED(hr))
2792         EndDialog(This->dlg_hwnd, S_OK);
2793
2794     return S_OK;
2795 }
2796
2797 static HRESULT WINAPI ICommDlgBrowser3_fnOnStateChange(ICommDlgBrowser3 *iface,
2798                                                        IShellView *shv, ULONG uChange )
2799 {
2800     FileDialogImpl *This = impl_from_ICommDlgBrowser3(iface);
2801     IDataObject *new_selection;
2802     HRESULT hr;
2803     TRACE("%p (%p, %x)\n", This, shv, uChange);
2804
2805     switch(uChange)
2806     {
2807     case CDBOSC_SELCHANGE:
2808         if(This->psia_selection)
2809         {
2810             IShellItemArray_Release(This->psia_selection);
2811             This->psia_selection = NULL;
2812         }
2813
2814         hr = IShellView_GetItemObject(shv, SVGIO_SELECTION, &IID_IDataObject, (void**)&new_selection);
2815         if(SUCCEEDED(hr))
2816         {
2817             hr = SHCreateShellItemArrayFromDataObject(new_selection, &IID_IShellItemArray,
2818                                                       (void**)&This->psia_selection);
2819             if(SUCCEEDED(hr))
2820             {
2821                 fill_filename_from_selection(This);
2822                 events_OnSelectionChange(This);
2823             }
2824
2825             IDataObject_Release(new_selection);
2826         }
2827         break;
2828     default:
2829         TRACE("Unhandled state change\n");
2830     }
2831     return S_OK;
2832 }
2833
2834 static HRESULT WINAPI ICommDlgBrowser3_fnIncludeObject(ICommDlgBrowser3 *iface,
2835                                                        IShellView *shv, LPCITEMIDLIST pidl)
2836 {
2837     FileDialogImpl *This = impl_from_ICommDlgBrowser3(iface);
2838     IShellItem *psi;
2839     LPWSTR filename;
2840     LPITEMIDLIST parent_pidl;
2841     HRESULT hr;
2842     ULONG attr;
2843     TRACE("%p (%p, %p)\n", This, shv, pidl);
2844
2845     if(!This->filterspec_count)
2846         return S_OK;
2847
2848     hr = SHGetIDListFromObject((IUnknown*)shv, &parent_pidl);
2849     if(SUCCEEDED(hr))
2850     {
2851         LPITEMIDLIST full_pidl = ILCombine(parent_pidl, pidl);
2852         hr = SHCreateItemFromIDList(full_pidl, &IID_IShellItem, (void**)&psi);
2853         ILFree(parent_pidl);
2854         ILFree(full_pidl);
2855     }
2856     if(FAILED(hr))
2857     {
2858         ERR("Failed to get shellitem (%08x).\n", hr);
2859         return S_OK;
2860     }
2861
2862     hr = IShellItem_GetAttributes(psi, SFGAO_FOLDER|SFGAO_LINK, &attr);
2863     if(FAILED(hr) || (attr & (SFGAO_FOLDER | SFGAO_LINK)))
2864     {
2865         IShellItem_Release(psi);
2866         return S_OK;
2867     }
2868
2869     hr = S_OK;
2870     if(SUCCEEDED(IShellItem_GetDisplayName(psi, SIGDN_PARENTRELATIVEPARSING, &filename)))
2871     {
2872         if(!PathMatchSpecW(filename, This->filterspecs[This->filetypeindex].pszSpec))
2873             hr = S_FALSE;
2874         CoTaskMemFree(filename);
2875     }
2876
2877     IShellItem_Release(psi);
2878     return hr;
2879 }
2880
2881 static HRESULT WINAPI ICommDlgBrowser3_fnNotify(ICommDlgBrowser3 *iface,
2882                                                 IShellView *ppshv, DWORD dwNotifyType)
2883 {
2884     FileDialogImpl *This = impl_from_ICommDlgBrowser3(iface);
2885     FIXME("Stub: %p (%p, 0x%x)\n", This, ppshv, dwNotifyType);
2886     return E_NOTIMPL;
2887 }
2888
2889 static HRESULT WINAPI ICommDlgBrowser3_fnGetDefaultMenuText(ICommDlgBrowser3 *iface,
2890                                                             IShellView *pshv,
2891                                                             LPWSTR pszText, int cchMax)
2892 {
2893     FileDialogImpl *This = impl_from_ICommDlgBrowser3(iface);
2894     FIXME("Stub: %p (%p, %p, %d)\n", This, pshv, pszText, cchMax);
2895     return E_NOTIMPL;
2896 }
2897
2898 static HRESULT WINAPI ICommDlgBrowser3_fnGetViewFlags(ICommDlgBrowser3 *iface, DWORD *pdwFlags)
2899 {
2900     FileDialogImpl *This = impl_from_ICommDlgBrowser3(iface);
2901     FIXME("Stub: %p (%p)\n", This, pdwFlags);
2902     return E_NOTIMPL;
2903 }
2904
2905 static HRESULT WINAPI ICommDlgBrowser3_fnOnColumnClicked(ICommDlgBrowser3 *iface,
2906                                                          IShellView *pshv, int iColumn)
2907 {
2908     FileDialogImpl *This = impl_from_ICommDlgBrowser3(iface);
2909     FIXME("Stub: %p (%p, %d)\n", This, pshv, iColumn);
2910     return E_NOTIMPL;
2911 }
2912
2913 static HRESULT WINAPI ICommDlgBrowser3_fnGetCurrentFilter(ICommDlgBrowser3 *iface,
2914                                                           LPWSTR pszFileSpec, int cchFileSpec)
2915 {
2916     FileDialogImpl *This = impl_from_ICommDlgBrowser3(iface);
2917     FIXME("Stub: %p (%p, %d)\n", This, pszFileSpec, cchFileSpec);
2918     return E_NOTIMPL;
2919 }
2920
2921 static HRESULT WINAPI ICommDlgBrowser3_fnOnPreviewCreated(ICommDlgBrowser3 *iface,
2922                                                           IShellView *pshv)
2923 {
2924     FileDialogImpl *This = impl_from_ICommDlgBrowser3(iface);
2925     FIXME("Stub: %p (%p)\n", This, pshv);
2926     return E_NOTIMPL;
2927 }
2928
2929 static const ICommDlgBrowser3Vtbl vt_ICommDlgBrowser3 = {
2930     ICommDlgBrowser3_fnQueryInterface,
2931     ICommDlgBrowser3_fnAddRef,
2932     ICommDlgBrowser3_fnRelease,
2933     ICommDlgBrowser3_fnOnDefaultCommand,
2934     ICommDlgBrowser3_fnOnStateChange,
2935     ICommDlgBrowser3_fnIncludeObject,
2936     ICommDlgBrowser3_fnNotify,
2937     ICommDlgBrowser3_fnGetDefaultMenuText,
2938     ICommDlgBrowser3_fnGetViewFlags,
2939     ICommDlgBrowser3_fnOnColumnClicked,
2940     ICommDlgBrowser3_fnGetCurrentFilter,
2941     ICommDlgBrowser3_fnOnPreviewCreated
2942 };
2943
2944 /**************************************************************************
2945  * IOleWindow implementation
2946  */
2947 static inline FileDialogImpl *impl_from_IOleWindow(IOleWindow *iface)
2948 {
2949     return CONTAINING_RECORD(iface, FileDialogImpl, IOleWindow_iface);
2950 }
2951
2952 static HRESULT WINAPI IOleWindow_fnQueryInterface(IOleWindow *iface, REFIID riid, void **ppvObject)
2953 {
2954     FileDialogImpl *This = impl_from_IOleWindow(iface);
2955     return IFileDialog2_QueryInterface(&This->IFileDialog2_iface, riid, ppvObject);
2956 }
2957
2958 static ULONG WINAPI IOleWindow_fnAddRef(IOleWindow *iface)
2959 {
2960     FileDialogImpl *This = impl_from_IOleWindow(iface);
2961     return IFileDialog2_AddRef(&This->IFileDialog2_iface);
2962 }
2963
2964 static ULONG WINAPI IOleWindow_fnRelease(IOleWindow *iface)
2965 {
2966     FileDialogImpl *This = impl_from_IOleWindow(iface);
2967     return IFileDialog2_Release(&This->IFileDialog2_iface);
2968 }
2969
2970 static HRESULT WINAPI IOleWindow_fnContextSensitiveHelp(IOleWindow *iface, BOOL fEnterMOde)
2971 {
2972     FileDialogImpl *This = impl_from_IOleWindow(iface);
2973     FIXME("Stub: %p (%d)\n", This, fEnterMOde);
2974     return E_NOTIMPL;
2975 }
2976
2977 static HRESULT WINAPI IOleWindow_fnGetWindow(IOleWindow *iface, HWND *phwnd)
2978 {
2979     FileDialogImpl *This = impl_from_IOleWindow(iface);
2980     TRACE("%p (%p)\n", This, phwnd);
2981     *phwnd = This->dlg_hwnd;
2982     return S_OK;
2983 }
2984
2985 static const IOleWindowVtbl vt_IOleWindow = {
2986     IOleWindow_fnQueryInterface,
2987     IOleWindow_fnAddRef,
2988     IOleWindow_fnRelease,
2989     IOleWindow_fnGetWindow,
2990     IOleWindow_fnContextSensitiveHelp
2991 };
2992
2993 /**************************************************************************
2994  * IFileDialogCustomize implementation
2995  */
2996 static inline FileDialogImpl *impl_from_IFileDialogCustomize(IFileDialogCustomize *iface)
2997 {
2998     return CONTAINING_RECORD(iface, FileDialogImpl, IFileDialogCustomize_iface);
2999 }
3000
3001 static HRESULT WINAPI IFileDialogCustomize_fnQueryInterface(IFileDialogCustomize *iface,
3002                                                             REFIID riid, void **ppvObject)
3003 {
3004     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3005     return IFileDialog2_QueryInterface(&This->IFileDialog2_iface, riid, ppvObject);
3006 }
3007
3008 static ULONG WINAPI IFileDialogCustomize_fnAddRef(IFileDialogCustomize *iface)
3009 {
3010     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3011     return IFileDialog2_AddRef(&This->IFileDialog2_iface);
3012 }
3013
3014 static ULONG WINAPI IFileDialogCustomize_fnRelease(IFileDialogCustomize *iface)
3015 {
3016     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3017     return IFileDialog2_Release(&This->IFileDialog2_iface);
3018 }
3019
3020 static HRESULT WINAPI IFileDialogCustomize_fnEnableOpenDropDown(IFileDialogCustomize *iface,
3021                                                                 DWORD dwIDCtl)
3022 {
3023     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3024     FIXME("stub - %p (%d)\n", This, dwIDCtl);
3025     return E_NOTIMPL;
3026 }
3027
3028 static HRESULT WINAPI IFileDialogCustomize_fnAddMenu(IFileDialogCustomize *iface,
3029                                                      DWORD dwIDCtl,
3030                                                      LPCWSTR pszLabel)
3031 {
3032     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3033     customctrl *ctrl;
3034     TBBUTTON tbb;
3035     HRESULT hr;
3036     TRACE("%p (%d, %p)\n", This, dwIDCtl, pszLabel);
3037
3038     hr = cctrl_create_new(This, dwIDCtl, NULL, TOOLBARCLASSNAMEW,
3039                           TBSTYLE_FLAT | CCS_NODIVIDER, 0,
3040                           This->cctrl_def_height, &ctrl);
3041     if(SUCCEEDED(hr))
3042     {
3043         ctrl->type = IDLG_CCTRL_MENU;
3044
3045         /* Add the actual button with a popup menu. */
3046         tbb.iBitmap = I_IMAGENONE;
3047         tbb.dwData = (DWORD_PTR)CreatePopupMenu();
3048         tbb.iString = (DWORD_PTR)pszLabel;
3049         tbb.fsState = TBSTATE_ENABLED;
3050         tbb.fsStyle = BTNS_WHOLEDROPDOWN;
3051         tbb.idCommand = 1;
3052
3053         SendMessageW(ctrl->hwnd, TB_ADDBUTTONSW, 1, (LPARAM)&tbb);
3054     }
3055
3056     return hr;
3057 }
3058
3059 static HRESULT WINAPI IFileDialogCustomize_fnAddPushButton(IFileDialogCustomize *iface,
3060                                                            DWORD dwIDCtl,
3061                                                            LPCWSTR pszLabel)
3062 {
3063     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3064     customctrl *ctrl;
3065     HRESULT hr;
3066     TRACE("%p (%d, %p)\n", This, dwIDCtl, pszLabel);
3067
3068     hr = cctrl_create_new(This, dwIDCtl, pszLabel, WC_BUTTONW, BS_MULTILINE, 0,
3069                           This->cctrl_def_height, &ctrl);
3070     if(SUCCEEDED(hr))
3071         ctrl->type = IDLG_CCTRL_PUSHBUTTON;
3072
3073     return hr;
3074 }
3075
3076 static HRESULT WINAPI IFileDialogCustomize_fnAddComboBox(IFileDialogCustomize *iface,
3077                                                          DWORD dwIDCtl)
3078 {
3079     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3080     customctrl *ctrl;
3081     HRESULT hr;
3082     TRACE("%p (%d)\n", This, dwIDCtl);
3083
3084     hr =  cctrl_create_new(This, dwIDCtl, NULL, WC_COMBOBOXW, CBS_DROPDOWNLIST, 0,
3085                            This->cctrl_def_height, &ctrl);
3086     if(SUCCEEDED(hr))
3087         ctrl->type = IDLG_CCTRL_COMBOBOX;
3088
3089     return hr;
3090 }
3091
3092 static HRESULT WINAPI IFileDialogCustomize_fnAddRadioButtonList(IFileDialogCustomize *iface,
3093                                                                 DWORD dwIDCtl)
3094 {
3095     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3096     FIXME("stub - %p (%d)\n", This, dwIDCtl);
3097     return E_NOTIMPL;
3098 }
3099
3100 static HRESULT WINAPI IFileDialogCustomize_fnAddCheckButton(IFileDialogCustomize *iface,
3101                                                             DWORD dwIDCtl,
3102                                                             LPCWSTR pszLabel,
3103                                                             BOOL bChecked)
3104 {
3105     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3106     customctrl *ctrl;
3107     HRESULT hr;
3108     TRACE("%p (%d, %p, %d)\n", This, dwIDCtl, pszLabel, bChecked);
3109
3110     hr = cctrl_create_new(This, dwIDCtl, pszLabel, WC_BUTTONW, BS_AUTOCHECKBOX, 0,
3111                           This->cctrl_def_height, &ctrl);
3112     if(SUCCEEDED(hr))
3113     {
3114         ctrl->type = IDLG_CCTRL_CHECKBUTTON;
3115         SendMessageW(ctrl->hwnd, BM_SETCHECK, bChecked ? BST_CHECKED : BST_UNCHECKED, 0);
3116     }
3117
3118     return hr;
3119 }
3120
3121 static HRESULT WINAPI IFileDialogCustomize_fnAddEditBox(IFileDialogCustomize *iface,
3122                                                         DWORD dwIDCtl,
3123                                                         LPCWSTR pszText)
3124 {
3125     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3126     customctrl *ctrl;
3127     HRESULT hr;
3128     TRACE("%p (%d, %p)\n", This, dwIDCtl, pszText);
3129
3130     hr = cctrl_create_new(This, dwIDCtl, pszText, WC_EDITW, ES_AUTOHSCROLL, WS_EX_CLIENTEDGE,
3131                           This->cctrl_def_height, &ctrl);
3132     if(SUCCEEDED(hr))
3133         ctrl->type = IDLG_CCTRL_EDITBOX;
3134
3135     return hr;
3136 }
3137
3138 static HRESULT WINAPI IFileDialogCustomize_fnAddSeparator(IFileDialogCustomize *iface,
3139                                                           DWORD dwIDCtl)
3140 {
3141     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3142     customctrl *ctrl;
3143     HRESULT hr;
3144     TRACE("%p (%d)\n", This, dwIDCtl);
3145
3146     hr = cctrl_create_new(This, dwIDCtl, NULL, WC_STATICW, SS_ETCHEDHORZ, 0,
3147                           GetSystemMetrics(SM_CYEDGE), &ctrl);
3148     if(SUCCEEDED(hr))
3149         ctrl->type = IDLG_CCTRL_SEPARATOR;
3150
3151     return hr;
3152 }
3153
3154 static HRESULT WINAPI IFileDialogCustomize_fnAddText(IFileDialogCustomize *iface,
3155                                                      DWORD dwIDCtl,
3156                                                      LPCWSTR pszText)
3157 {
3158     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3159     customctrl *ctrl;
3160     HRESULT hr;
3161     TRACE("%p (%d, %p)\n", This, dwIDCtl, pszText);
3162
3163     hr = cctrl_create_new(This, dwIDCtl, pszText, WC_STATICW, 0, 0,
3164                           This->cctrl_def_height, &ctrl);
3165     if(SUCCEEDED(hr))
3166         ctrl->type = IDLG_CCTRL_TEXT;
3167
3168     return hr;
3169 }
3170
3171 static HRESULT WINAPI IFileDialogCustomize_fnSetControlLabel(IFileDialogCustomize *iface,
3172                                                              DWORD dwIDCtl,
3173                                                              LPCWSTR pszLabel)
3174 {
3175     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3176     customctrl *ctrl = get_cctrl(This, dwIDCtl);
3177     TRACE("%p (%d, %p)\n", This, dwIDCtl, pszLabel);
3178
3179     if(!ctrl) return E_INVALIDARG;
3180
3181     switch(ctrl->type)
3182     {
3183     case IDLG_CCTRL_MENU:
3184     case IDLG_CCTRL_PUSHBUTTON:
3185     case IDLG_CCTRL_CHECKBUTTON:
3186     case IDLG_CCTRL_TEXT:
3187         SendMessageW(ctrl->hwnd, WM_SETTEXT, 0, (LPARAM)pszLabel);
3188         break;
3189     default:
3190         break;
3191     }
3192
3193     return S_OK;
3194 }
3195
3196 static HRESULT WINAPI IFileDialogCustomize_fnGetControlState(IFileDialogCustomize *iface,
3197                                                              DWORD dwIDCtl,
3198                                                              CDCONTROLSTATEF *pdwState)
3199 {
3200     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3201     customctrl *ctrl = get_cctrl(This, dwIDCtl);
3202     TRACE("%p (%d, %p)\n", This, dwIDCtl, pdwState);
3203
3204     if(!ctrl) return E_NOTIMPL;
3205
3206     *pdwState = ctrl->cdcstate;
3207     return S_OK;
3208 }
3209
3210 static HRESULT WINAPI IFileDialogCustomize_fnSetControlState(IFileDialogCustomize *iface,
3211                                                              DWORD dwIDCtl,
3212                                                              CDCONTROLSTATEF dwState)
3213 {
3214     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3215     customctrl *ctrl = get_cctrl(This,dwIDCtl);
3216     TRACE("%p (%d, %x)\n", This, dwIDCtl, dwState);
3217
3218     if(ctrl)
3219     {
3220         LONG wndstyle = GetWindowLongW(ctrl->hwnd, GWL_STYLE);
3221
3222         if(dwState & CDCS_ENABLED)
3223             wndstyle &= ~(WS_DISABLED);
3224         else
3225             wndstyle |= WS_DISABLED;
3226
3227         if(dwState & CDCS_VISIBLE)
3228             wndstyle |= WS_VISIBLE;
3229         else
3230             wndstyle &= ~(WS_VISIBLE);
3231
3232         SetWindowLongW(ctrl->hwnd, GWL_STYLE, wndstyle);
3233
3234         /* We save the state separately since at least one application
3235          * relies on being able to hide a control. */
3236         ctrl->cdcstate = dwState;
3237     }
3238
3239     return S_OK;
3240 }
3241
3242 static HRESULT WINAPI IFileDialogCustomize_fnGetEditBoxText(IFileDialogCustomize *iface,
3243                                                             DWORD dwIDCtl,
3244                                                             WCHAR **ppszText)
3245 {
3246     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3247     customctrl *ctrl = get_cctrl(This, dwIDCtl);
3248     WCHAR len, *text;
3249     TRACE("%p (%d, %p)\n", This, dwIDCtl, ppszText);
3250
3251     if(!ctrl || !(len = SendMessageW(ctrl->hwnd, WM_GETTEXTLENGTH, 0, 0)))
3252         return E_FAIL;
3253
3254     text = CoTaskMemAlloc(sizeof(WCHAR)*(len+1));
3255     if(!text) return E_FAIL;
3256
3257     SendMessageW(ctrl->hwnd, WM_GETTEXT, len+1, (LPARAM)text);
3258     *ppszText = text;
3259     return S_OK;
3260 }
3261
3262 static HRESULT WINAPI IFileDialogCustomize_fnSetEditBoxText(IFileDialogCustomize *iface,
3263                                                             DWORD dwIDCtl,
3264                                                             LPCWSTR pszText)
3265 {
3266     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3267     customctrl *ctrl = get_cctrl(This, dwIDCtl);
3268     TRACE("%p (%d, %s)\n", This, dwIDCtl, debugstr_w(pszText));
3269
3270     if(!ctrl || ctrl->type != IDLG_CCTRL_EDITBOX)
3271         return E_FAIL;
3272
3273     SendMessageW(ctrl->hwnd, WM_SETTEXT, 0, (LPARAM)pszText);
3274     return S_OK;
3275 }
3276
3277 static HRESULT WINAPI IFileDialogCustomize_fnGetCheckButtonState(IFileDialogCustomize *iface,
3278                                                                  DWORD dwIDCtl,
3279                                                                  BOOL *pbChecked)
3280 {
3281     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3282     customctrl *ctrl = get_cctrl(This, dwIDCtl);
3283     TRACE("%p (%d, %p)\n", This, dwIDCtl, pbChecked);
3284
3285     if(ctrl)
3286         *pbChecked = (SendMessageW(ctrl->hwnd, BM_GETCHECK, 0, 0) == BST_CHECKED);
3287
3288     return S_OK;
3289 }
3290
3291 static HRESULT WINAPI IFileDialogCustomize_fnSetCheckButtonState(IFileDialogCustomize *iface,
3292                                                                  DWORD dwIDCtl,
3293                                                                  BOOL bChecked)
3294 {
3295     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3296     customctrl *ctrl = get_cctrl(This, dwIDCtl);
3297     TRACE("%p (%d, %d)\n", This, dwIDCtl, bChecked);
3298
3299     if(ctrl)
3300         SendMessageW(ctrl->hwnd, BM_SETCHECK, bChecked ? BST_CHECKED:BST_UNCHECKED, 0);
3301
3302     return S_OK;
3303 }
3304
3305 static UINT get_combobox_index_from_id(HWND cb_hwnd, DWORD dwIDItem)
3306 {
3307     UINT count = SendMessageW(cb_hwnd, CB_GETCOUNT, 0, 0);
3308     UINT i;
3309     if(!count || (count == CB_ERR))
3310         return -1;
3311
3312     for(i = 0; i < count; i++)
3313         if(SendMessageW(cb_hwnd, CB_GETITEMDATA, i, 0) == dwIDItem)
3314             return i;
3315
3316     TRACE("Item with id %d not found in combobox %p (item count: %d)\n", dwIDItem, cb_hwnd, count);
3317     return -1;
3318 }
3319
3320 static HRESULT WINAPI IFileDialogCustomize_fnAddControlItem(IFileDialogCustomize *iface,
3321                                                             DWORD dwIDCtl,
3322                                                             DWORD dwIDItem,
3323                                                             LPCWSTR pszLabel)
3324 {
3325     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3326     customctrl *ctrl = get_cctrl(This, dwIDCtl);
3327     TRACE("%p (%d, %d, %s)\n", This, dwIDCtl, dwIDItem, debugstr_w(pszLabel));
3328
3329     if(!ctrl) return E_FAIL;
3330
3331     switch(ctrl->type)
3332     {
3333     case IDLG_CCTRL_COMBOBOX:
3334     {
3335         UINT index;
3336
3337         if(get_combobox_index_from_id(ctrl->hwnd, dwIDItem) != -1)
3338             return E_INVALIDARG;
3339
3340         index = SendMessageW(ctrl->hwnd, CB_ADDSTRING, 0, (LPARAM)pszLabel);
3341         SendMessageW(ctrl->hwnd, CB_SETITEMDATA, index, dwIDItem);
3342
3343         return S_OK;
3344     }
3345     case IDLG_CCTRL_MENU:
3346     {
3347         TBBUTTON tbb;
3348         SendMessageW(ctrl->hwnd, TB_GETBUTTON, 0, (LPARAM)&tbb);
3349
3350         if(GetMenuState((HMENU)tbb.dwData, dwIDItem, MF_BYCOMMAND) != -1)
3351             return E_INVALIDARG;
3352
3353         AppendMenuW((HMENU)tbb.dwData, MF_STRING, dwIDItem, pszLabel);
3354         return S_OK;
3355     }
3356     default:
3357         break;
3358     }
3359
3360     return E_NOINTERFACE; /* win7 */
3361 }
3362
3363 static HRESULT WINAPI IFileDialogCustomize_fnRemoveControlItem(IFileDialogCustomize *iface,
3364                                                                DWORD dwIDCtl,
3365                                                                DWORD dwIDItem)
3366 {
3367     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3368     customctrl *ctrl = get_cctrl(This, dwIDCtl);
3369     TRACE("%p (%d, %d)\n", This, dwIDCtl, dwIDItem);
3370
3371     if(!ctrl) return E_FAIL;
3372
3373     switch(ctrl->type)
3374     {
3375     case IDLG_CCTRL_COMBOBOX:
3376     {
3377         UINT i, count = SendMessageW(ctrl->hwnd, CB_GETCOUNT, 0, 0);
3378         if(!count || (count == CB_ERR))
3379             return E_FAIL;
3380
3381         for(i = 0; i < count; i++)
3382             if(SendMessageW(ctrl->hwnd, CB_GETITEMDATA, 0, 0) == dwIDItem)
3383             {
3384                 if(SendMessageW(ctrl->hwnd, CB_DELETESTRING, i, 0) == CB_ERR)
3385                     return E_FAIL;
3386                 return S_OK;
3387             }
3388
3389         return E_UNEXPECTED;
3390     }
3391     case IDLG_CCTRL_MENU:
3392     {
3393         TBBUTTON tbb;
3394         HMENU hmenu;
3395         SendMessageW(ctrl->hwnd, TB_GETBUTTON, 0, (LPARAM)&tbb);
3396         hmenu = (HMENU)tbb.dwData;
3397
3398         if(!hmenu || !DeleteMenu(hmenu, dwIDItem, MF_BYCOMMAND))
3399             return E_UNEXPECTED;
3400
3401         return S_OK;
3402     }
3403     default:
3404         break;
3405     }
3406
3407     return E_FAIL;
3408 }
3409
3410 static HRESULT WINAPI IFileDialogCustomize_fnRemoveAllControlItems(IFileDialogCustomize *iface,
3411                                                                    DWORD dwIDCtl)
3412 {
3413     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3414     TRACE("%p (%d)\n", This, dwIDCtl);
3415
3416     /* Not implemented by native */
3417     return E_NOTIMPL;
3418 }
3419
3420 static HRESULT WINAPI IFileDialogCustomize_fnGetControlItemState(IFileDialogCustomize *iface,
3421                                                                  DWORD dwIDCtl,
3422                                                                  DWORD dwIDItem,
3423                                                                  CDCONTROLSTATEF *pdwState)
3424 {
3425     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3426     FIXME("stub - %p\n", This);
3427     return E_NOTIMPL;
3428 }
3429
3430 static HRESULT WINAPI IFileDialogCustomize_fnSetControlItemState(IFileDialogCustomize *iface,
3431                                                                  DWORD dwIDCtl,
3432                                                                  DWORD dwIDItem,
3433                                                                  CDCONTROLSTATEF dwState)
3434 {
3435     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3436     FIXME("stub - %p\n", This);
3437     return E_NOTIMPL;
3438 }
3439
3440 static HRESULT WINAPI IFileDialogCustomize_fnGetSelectedControlItem(IFileDialogCustomize *iface,
3441                                                                     DWORD dwIDCtl,
3442                                                                     DWORD *pdwIDItem)
3443 {
3444     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3445     customctrl *ctrl = get_cctrl(This, dwIDCtl);
3446     TRACE("%p (%d, %p)\n", This, dwIDCtl, pdwIDItem);
3447
3448     if(!ctrl) return E_FAIL;
3449
3450     switch(ctrl->type)
3451     {
3452     case IDLG_CCTRL_COMBOBOX:
3453     {
3454         UINT index = SendMessageW(ctrl->hwnd, CB_GETCURSEL, 0, 0);
3455         if(index == CB_ERR)
3456             return E_FAIL;
3457
3458         *pdwIDItem = SendMessageW(ctrl->hwnd, CB_GETITEMDATA, index, 0);
3459         return S_OK;
3460     }
3461     default:
3462         FIXME("Unsupported control type %d\n", ctrl->type);
3463     }
3464
3465     return E_NOTIMPL;
3466 }
3467
3468 static HRESULT WINAPI IFileDialogCustomize_fnSetSelectedControlItem(IFileDialogCustomize *iface,
3469                                                                     DWORD dwIDCtl,
3470                                                                     DWORD dwIDItem)
3471 {
3472     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3473     customctrl *ctrl = get_cctrl(This, dwIDCtl);
3474     TRACE("%p (%d, %d)\n", This, dwIDCtl, dwIDItem);
3475
3476     if(!ctrl) return E_INVALIDARG;
3477
3478     switch(ctrl->type)
3479     {
3480     case IDLG_CCTRL_COMBOBOX:
3481     {
3482         UINT index = get_combobox_index_from_id(ctrl->hwnd, dwIDItem);
3483
3484         if(index == -1)
3485             return E_INVALIDARG;
3486
3487         if(SendMessageW(ctrl->hwnd, CB_SETCURSEL, index, 0) == CB_ERR)
3488             return E_FAIL;
3489
3490         return S_OK;
3491     }
3492     default:
3493         FIXME("Unsupported control type %d\n", ctrl->type);
3494     }
3495
3496     return E_INVALIDARG;
3497 }
3498
3499 static HRESULT WINAPI IFileDialogCustomize_fnStartVisualGroup(IFileDialogCustomize *iface,
3500                                                               DWORD dwIDCtl,
3501                                                               LPCWSTR pszLabel)
3502 {
3503     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3504     FIXME("stub - %p (%d, %s)\n", This, dwIDCtl, debugstr_w(pszLabel));
3505     return E_NOTIMPL;
3506 }
3507
3508 static HRESULT WINAPI IFileDialogCustomize_fnEndVisualGroup(IFileDialogCustomize *iface)
3509 {
3510     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3511     FIXME("stub - %p\n", This);
3512     return E_NOTIMPL;
3513 }
3514
3515 static HRESULT WINAPI IFileDialogCustomize_fnMakeProminent(IFileDialogCustomize *iface,
3516                                                            DWORD dwIDCtl)
3517 {
3518     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3519     FIXME("stub - %p (%d)\n", This, dwIDCtl);
3520     return E_NOTIMPL;
3521 }
3522
3523 static HRESULT WINAPI IFileDialogCustomize_fnSetControlItemText(IFileDialogCustomize *iface,
3524                                                                 DWORD dwIDCtl,
3525                                                                 DWORD dwIDItem,
3526                                                                 LPCWSTR pszLabel)
3527 {
3528     FileDialogImpl *This = impl_from_IFileDialogCustomize(iface);
3529     FIXME("stub - %p (%d, %d, %p)\n", This, dwIDCtl, dwIDItem, debugstr_w(pszLabel));
3530     return E_NOTIMPL;
3531 }
3532
3533 static const IFileDialogCustomizeVtbl vt_IFileDialogCustomize = {
3534     IFileDialogCustomize_fnQueryInterface,
3535     IFileDialogCustomize_fnAddRef,
3536     IFileDialogCustomize_fnRelease,
3537     IFileDialogCustomize_fnEnableOpenDropDown,
3538     IFileDialogCustomize_fnAddMenu,
3539     IFileDialogCustomize_fnAddPushButton,
3540     IFileDialogCustomize_fnAddComboBox,
3541     IFileDialogCustomize_fnAddRadioButtonList,
3542     IFileDialogCustomize_fnAddCheckButton,
3543     IFileDialogCustomize_fnAddEditBox,
3544     IFileDialogCustomize_fnAddSeparator,
3545     IFileDialogCustomize_fnAddText,
3546     IFileDialogCustomize_fnSetControlLabel,
3547     IFileDialogCustomize_fnGetControlState,
3548     IFileDialogCustomize_fnSetControlState,
3549     IFileDialogCustomize_fnGetEditBoxText,
3550     IFileDialogCustomize_fnSetEditBoxText,
3551     IFileDialogCustomize_fnGetCheckButtonState,
3552     IFileDialogCustomize_fnSetCheckButtonState,
3553     IFileDialogCustomize_fnAddControlItem,
3554     IFileDialogCustomize_fnRemoveControlItem,
3555     IFileDialogCustomize_fnRemoveAllControlItems,
3556     IFileDialogCustomize_fnGetControlItemState,
3557     IFileDialogCustomize_fnSetControlItemState,
3558     IFileDialogCustomize_fnGetSelectedControlItem,
3559     IFileDialogCustomize_fnSetSelectedControlItem,
3560     IFileDialogCustomize_fnStartVisualGroup,
3561     IFileDialogCustomize_fnEndVisualGroup,
3562     IFileDialogCustomize_fnMakeProminent,
3563     IFileDialogCustomize_fnSetControlItemText
3564 };
3565
3566 static HRESULT FileDialog_constructor(IUnknown *pUnkOuter, REFIID riid, void **ppv, enum ITEMDLG_TYPE type)
3567 {
3568     FileDialogImpl *fdimpl;
3569     HRESULT hr;
3570     IShellFolder *psf;
3571     TRACE("%p, %s, %p\n", pUnkOuter, debugstr_guid(riid), ppv);
3572
3573     if(!ppv)
3574         return E_POINTER;
3575     if(pUnkOuter)
3576         return CLASS_E_NOAGGREGATION;
3577
3578     fdimpl = HeapAlloc(GetProcessHeap(), 0, sizeof(FileDialogImpl));
3579     if(!fdimpl)
3580         return E_OUTOFMEMORY;
3581
3582     fdimpl->ref = 1;
3583     fdimpl->IFileDialog2_iface.lpVtbl = &vt_IFileDialog2;
3584     fdimpl->IExplorerBrowserEvents_iface.lpVtbl = &vt_IExplorerBrowserEvents;
3585     fdimpl->IServiceProvider_iface.lpVtbl = &vt_IServiceProvider;
3586     fdimpl->ICommDlgBrowser3_iface.lpVtbl = &vt_ICommDlgBrowser3;
3587     fdimpl->IOleWindow_iface.lpVtbl = &vt_IOleWindow;
3588     fdimpl->IFileDialogCustomize_iface.lpVtbl = &vt_IFileDialogCustomize;
3589
3590     if(type == ITEMDLG_TYPE_OPEN)
3591     {
3592         fdimpl->dlg_type = ITEMDLG_TYPE_OPEN;
3593         fdimpl->u.IFileOpenDialog_iface.lpVtbl = &vt_IFileOpenDialog;
3594         fdimpl->options = FOS_PATHMUSTEXIST | FOS_FILEMUSTEXIST | FOS_NOCHANGEDIR;
3595         fdimpl->custom_title = fdimpl->custom_okbutton = NULL;
3596     }
3597     else
3598     {
3599         WCHAR buf[16];
3600         fdimpl->dlg_type = ITEMDLG_TYPE_SAVE;
3601         fdimpl->u.IFileSaveDialog_iface.lpVtbl = &vt_IFileSaveDialog;
3602         fdimpl->options = FOS_OVERWRITEPROMPT | FOS_NOREADONLYRETURN | FOS_PATHMUSTEXIST | FOS_NOCHANGEDIR;
3603
3604         LoadStringW(COMDLG32_hInstance, IDS_SAVE, buf, sizeof(buf)/sizeof(WCHAR));
3605         fdimpl->custom_title = StrDupW(buf);
3606         fdimpl->custom_okbutton = StrDupW(buf);
3607     }
3608
3609     fdimpl->filterspecs = NULL;
3610     fdimpl->filterspec_count = 0;
3611     fdimpl->filetypeindex = 0;
3612
3613     fdimpl->psia_selection = fdimpl->psia_results = NULL;
3614     fdimpl->psi_setfolder = fdimpl->psi_folder = NULL;
3615
3616     list_init(&fdimpl->events_clients);
3617     fdimpl->events_next_cookie = 0;
3618
3619     fdimpl->dlg_hwnd = NULL;
3620     fdimpl->peb = NULL;
3621
3622     fdimpl->set_filename = NULL;
3623     fdimpl->default_ext = NULL;
3624     fdimpl->custom_cancelbutton = fdimpl->custom_filenamelabel = NULL;
3625
3626     /* FIXME: The default folder setting should be restored for the
3627      * application if it was previously set. */
3628     SHGetDesktopFolder(&psf);
3629     SHGetItemFromObject((IUnknown*)psf, &IID_IShellItem, (void**)&fdimpl->psi_defaultfolder);
3630     IShellFolder_Release(psf);
3631
3632     hr = init_custom_controls(fdimpl);
3633     if(FAILED(hr))
3634     {
3635         ERR("Failed to initialize custom controls (0x%08x).\n", hr);
3636         IUnknown_Release((IUnknown*)fdimpl);
3637         return E_FAIL;
3638     }
3639
3640     hr = IUnknown_QueryInterface((IUnknown*)fdimpl, riid, ppv);
3641     IUnknown_Release((IUnknown*)fdimpl);
3642     return hr;
3643 }
3644
3645 HRESULT FileOpenDialog_Constructor(IUnknown *pUnkOuter, REFIID riid, void **ppv)
3646 {
3647     return FileDialog_constructor(pUnkOuter, riid, ppv, ITEMDLG_TYPE_OPEN);
3648 }
3649
3650 HRESULT FileSaveDialog_Constructor(IUnknown *pUnkOuter, REFIID riid, void **ppv)
3651 {
3652     return FileDialog_constructor(pUnkOuter, riid, ppv, ITEMDLG_TYPE_SAVE);
3653 }