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