#include "BrowseDialog.h"
#include <shlobj.h>
#include <string.h>

int CALLBACK BrowseCallbackProc(HWND hwnd, UINT uMsg, LPARAM lParam,
	LPARAM lpData)
{
   if (uMsg == BFFM_INITIALIZED && lpData)
      PostMessage(hwnd, BFFM_SETSELECTION, TRUE, lpData);

   return  0 ;
}

bool __fastcall BrowseFolder(const char *label, char *&file_name,
   char *prevpath)
{
	LPITEMIDLIST iidl, root;
	BROWSEINFO bi;

   SHGetSpecialFolderLocation(NULL, CSIDL_DRIVES, &root);
   bi.hwndOwner = NULL;
   bi.pidlRoot = root;
   bi.pszDisplayName = NULL;
   bi.lpszTitle = label;
   bi.ulFlags = BIF_NEWDIALOGSTYLE;
   bi.lpfn = BrowseCallbackProc;
   bi.lParam = reinterpret_cast<LPARAM>(prevpath);

   IMalloc *mal;
   SHGetMalloc(&mal);

   iidl = SHBrowseForFolder(&bi);
   mal->Free(root);

   if (!iidl)
   {
      mal->Release();
      return false;
   }

   char f_name[MAX_PATH];
   SHGetPathFromIDList(iidl, f_name);
   file_name = new char[strlen(f_name) +  1 ];
   strcpy(file_name, f_name);

   mal->Free(iidl);
   mal->Release();

   return true;
}

#ifndef BrowseDialogH
#define BrowseDialogH

bool __fastcall BrowseFolder(const char *label, char *&file_name,
   char *prevpath =  0 );

#endif